lib/pry-state/hook_action.rb in pry-state-0.1.8 vs lib/pry-state/hook_action.rb in pry-state-0.1.9

@@ -4,40 +4,50 @@
   IGNORABLE_LOCAL_VARS = [:__, :_, :_ex_, :_pry_, :_out_, :_in_, :_dir_, :_file_]
 
   CONTROLLER_INSTANCE_VARIABLES = [:@_request, :@_response, :@_routes]
   RSPEC_INSTANCE_VARIABLES = [:@__inspect_output, :@__memoized]
   IGNORABLE_INSTANCE_VARS = CONTROLLER_INSTANCE_VARIABLES + RSPEC_INSTANCE_VARIABLES
+  IGNORABLE_GLOBAL_VARS = [:$@]
 
   def initialize binding, pry
     @binding, @pry = binding, pry
-
   end
 
   def act
     # when using guard, locals :e, :lib, :pry_state_prev get printed.
     # this 'if' cuts them off.
     if @binding.eval("self.class") == Object
       return
     end
 
-    (binding.eval('instance_variables').sort - IGNORABLE_INSTANCE_VARS).each do |var|
-      eval_and_print var, var_color: 'green'
+    if ENV['SHOW_GLOBAL_VARIABLES']
+      (binding.eval('global_variables').sort - IGNORABLE_GLOBAL_VARS).each do |var|
+        eval_and_print var, var_color: 'white', value_colore: 'yellow'
+      end
     end
 
-    (binding.eval('local_variables').sort - IGNORABLE_LOCAL_VARS).each do |var|
-      eval_and_print var, var_color: 'cyan'
+    unless ENV['HIDE_INSTANCE_VARIABLES']
+      (binding.eval('instance_variables').sort - IGNORABLE_INSTANCE_VARS).each do |var|
+        eval_and_print var, var_color: 'green'
+      end
     end
+
+    unless ENV['HIDE_LOCAL_VARIABLES']
+      (binding.eval('local_variables').sort - IGNORABLE_LOCAL_VARS).each do |var|
+        eval_and_print var, var_color: 'cyan'
+      end
+    end
   end
 
   private
   attr_reader :binding, :pry
 
   def eval_and_print var, var_color: 'green', value_color: 'white'
     value = binding.eval(var.to_s)
     if value_changed? var, value
       var_color = "bright_#{var_color}"; value_color = 'bright_yellow'
     end
-    Printer.trunc_and_print var, value, var_color, value_color
+    PryState::Printer.trunc_and_print var, value, var_color, value_color
     stick_value! var, value # to refer the value in next
   end
 
   def value_changed? var, value
     prev_state[var] and prev_state[var] != value