lib/rash/shell.rb in rash-shell-0.1.1 vs lib/rash/shell.rb in rash-shell-0.2.0

@@ -1,14 +1,16 @@
+# frozen_string_literal: true
+
 require 'rash/shell/pry'
 require 'rash/shell/version'
 
 module Rash
   # :no-doc:
   module Shell
     module_function
 
     def environment
-      ENV['RACK_ENV'] || 'development'
+      ENV['APP_ENV'] || ENV['RAILS_ENV'] || ENV['RACK_ENV'] || 'development'
     end
 
     def environment_color
       environment == 'production' ? "\e[38;5;124m" : "\e[38;5;107m"
     end