Make method_size_cutoff configurable

lib/pry/commands/whereami.rb

@@ -11,12 +11,6 @@ def initialize(*)
         @method_code = nil
       end
 
-      class << self
-        attr_accessor :method_size_cutoff
-      end
-
-      @method_size_cutoff = 30
-
       match 'whereami'
       description 'Show code surrounding the current context.'
       group 'Context'
@@ -138,7 +132,11 @@ def handle_internal_binding
       end
 
       def small_method?
-        @method.source_range.count < self.class.method_size_cutoff
+        @method.source_range.count < cutoff_size
+      end
+
+      def cutoff_size
+        pry_instance.config.method_size_cutoff || 30
       end
 
       def default_code

lib/pry/config.rb

@@ -34,6 +34,9 @@ class Config
     #   exceptions
     attribute :default_window_size
 
+    # @return [Integer] The cutoff size of lines of method to show
+    attribute :method_size_cutoff
+
     # @return [Pry::Hooks]
     attribute :hooks
 
@@ -180,6 +183,7 @@ def initialize
         system: Pry::SystemCommandHandler.method(:default),
         color: Pry::Helpers::BaseHelpers.use_ansi_codes?,
         default_window_size: 5,
+        method_size_cutoff: 30,
         editor: Pry::Editor.default,
         rc_file: default_rc_file,
         should_load_rc: true,

spec/commands/whereami_spec.rb

@@ -148,17 +148,17 @@ def blimey!
   it 'should show entire method when -m option used' do
     old_size = Pry.config.default_window_size
     Pry.config.default_window_size = 1
-    old_cutoff = Pry::Command::Whereami.method_size_cutoff
-    Pry::Command::Whereami.method_size_cutoff = 1
+    old_cutoff = Pry.config.method_size_cutoff
+    Pry.config.method_size_cutoff = 1
     class Cor
       def blimey!
         @foo = 1
         @bar = 2
         pry_eval(binding, 'whereami -m')
       end
     end
-    Pry::Command::Whereami.method_size_cutoff = old_cutoff
     Pry.config.default_window_size = old_size
+    Pry.config.method_size_cutoff = old_cutoff
     result = Cor.new.blimey!
     Object.remove_const(:Cor)
     expect(result).to match(/def blimey/)

spec/config_spec.rb

@@ -16,6 +16,7 @@
   specify { expect(subject.system).to be_a(Method) }
   specify { expect(subject.color).to be(true).or be(false) }
   specify { expect(subject.default_window_size).to be_a(Numeric) }
+  specify { expect(subject.method_size_cutoff).to be_a(Numeric) }
   specify { expect(subject.editor).to be_a(String) }
   specify { expect(subject.should_load_rc).to be(true).or be(false) }
   specify { expect(subject.should_load_local_rc).to be(true).or be(false) }