Enable pry-like stop on binding.pry

lib/byebug/processors/pry_processor.rb

@@ -15,7 +15,11 @@ def self.start
       Byebug.start
       Setting[:autolist] = false
       Context.processor = self
-      Byebug.current_context.step_out(3, true)
+      if PryByebug.binding_behavior == :pry
+        Byebug.current_context.step_out(0, true)
+      else
+        Byebug.current_context.step_out(3, true)
+      end
     end
 
     #
@@ -90,6 +94,10 @@ def n_hits(breakpoint)
     # Resume an existing Pry REPL at the paused point.
     #
     def resume_pry
+      if PryByebug.binding_behavior == :pry
+        Byebug::UpCommand.new(self, "up 3").execute
+      end
+
       new_binding = frame._binding
 
       run do

lib/pry-byebug/base.rb

@@ -22,4 +22,14 @@ def check_file_context(target, e = nil)
 
   # Reference to currently running pry-remote server. Used by the processor.
   attr_accessor :current_remote_server
+
+  def binding_behavior
+    @binding_behavior ||= :byebug
+  end
+  module_function :binding_behavior
+
+  def binding_behavior=(style)
+    @binding_behavior = style
+  end
+  module_function :binding_behavior=
 end

lib/pry-byebug/pry_ext.rb

@@ -4,6 +4,7 @@ class << Pry
   alias_method :start_without_pry_byebug, :start
 
   def start_with_pry_byebug(target = TOPLEVEL_BINDING, options = {})
+    Pry.initial_session_setup
     if target.is_a?(Binding) && PryByebug.file_context?(target)
       Byebug::PryProcessor.start
     else

test/base_test.rb

@@ -16,3 +16,25 @@ def test_other_file_context
     end
   end
 end
+
+#
+# Tests binding.pry stops at that line with PryByebug.break_behavior = :pry
+#
+
+class TestStopsOnBinding < MiniTest::Spec
+  def setup
+    super
+    PryByebug.binding_behavior = :pry
+    @output = StringIO.new
+    @input = InputTester.new('')
+    redirect_pry_io(@input, @output) { load test_file('last_line') }
+  end
+
+  def test_stops_on_binding
+    assert_match(/\=> \s*2:/, @output.string)
+  end
+
+  def teardown
+    PryByebug.binding_behavior = :byebug
+  end
+end

test/examples/last_line.rb

@@ -0,0 +1,2 @@
+require 'pry'
+binding.pry