class IRBAppEnvPrompt - irb_app_env_prompt-0.1.0 Documentation

Public Class Methods

configure(app, env) click to toggle source

# File lib/irb_app_env_prompt.rb, line 4
def self.configure(app, env)
  return unless defined? IRB

  new.call(app, env)
end

Public Instance Methods

call(app, env) click to toggle source

# File lib/irb_app_env_prompt.rb, line 10
def call(app, env)
  ansi_red = 31
  ansi_green = 32
  ansi_yellow = 33

  env_description = {
    "development" => "dev",
    "staging" => "stag",
    "production" => "prod",
  }.fetch(env) { "unknown" }

  env_color = {
    "dev" => ansi_green,
    "stag" => ansi_yellow,
    "prod" => ansi_red,
  }.fetch(env_description) { ansi_red }

  env_prompt = "#{app}(#{paint(env_color, env_description)})"

  IRB.conf[:PROMPT][prompt_sym] = {
    PROMPT_I: "#{env_prompt}> ",
    PROMPT_N: "#{env_prompt}> ",
    PROMPT_S: "#{env_prompt}%l ",
    PROMPT_C: "#{env_prompt}* ",
    RETURN: "=> %s\n",
  }
  IRB.conf[:PROMPT_MODE] = prompt_sym
end

Private Instance Methods

paint(color, text) click to toggle source

# File lib/irb_app_env_prompt.rb, line 41
def paint(color, text)
  "\033[#{color}m#{text}\033[0m"
end

prompt_sym() click to toggle source

# File lib/irb_app_env_prompt.rb, line 45
def prompt_sym
  :APP_ENV
end