Merge pull request joeyAghion#5 from jonallured/support-separate-colo…

…r-var

Support separate color var

README.md

@@ -12,6 +12,18 @@ By default, the prompt is green in development, red in production, and yellow el
 
     ConsoleColor::COLORS['production'] = "\e[35m"  # pink
 
+Note that one can set the `CONSOLE_COLOR_ENV` var to override the Rails
+environment:
+
+```
+$ CONSOLE_COLOR_ENV=staging RAILS_ENV=production rails console
+app_name:staging>Rails.env
+=> "production"
+```
+
+This way you can set your staging Rails env to production to mimic that
+configuration but have your prompt reflect that you're using on the staging
+server.
 
 ## Installation
 

lib/console_color/railtie.rb

@@ -4,7 +4,12 @@ module IRBSetup
       def setup(*)
         super
 
-        prompt = "\001#{ConsoleColor::COLORS[Rails.env]}\002#{Rails.application.class.parent_name.downcase}:#{Rails.env}"
+        app_name = Rails.application.class.parent_name.downcase
+        environment = ENV.fetch('CONSOLE_COLOR_ENV', Rails.env)
+        color = ConsoleColor::COLORS[environment]
+
+        prompt = "\001#{color}\002#{app_name}:#{environment}"
+
         IRB.conf[:PROMPT][:RAILS_APP] = {
           PROMPT_I: "#{prompt}>\e[0m ",
           PROMPT_N: "#{prompt}>\e[0m ",
@@ -13,6 +18,7 @@ def setup(*)
           RETURN: "=> %s\n",
           AUTO_INDENT: true
         }
+
         IRB.conf[:PROMPT_MODE] = :RAILS_APP
       end
     end