Merge pull request #80 from k0kubun/byebug-stop

Stop Byebug after continue

lib/pry-byebug.rb

@@ -2,3 +2,4 @@
 require 'pry-byebug/base'
 require 'pry-byebug/pry_ext'
 require 'pry-byebug/commands'
+require 'pry-byebug/control_d_handler'

lib/pry-byebug/commands.rb

@@ -6,3 +6,4 @@
 require 'pry-byebug/commands/down'
 require 'pry-byebug/commands/frame'
 require 'pry-byebug/commands/breakpoint'
+require 'pry-byebug/commands/exit_all'

lib/pry-byebug/commands/continue.rb

@@ -30,6 +30,8 @@ def process
       breakpoints.add_file(current_file, args.first.to_i) if args.first
 
       breakout_navigation :continue
+    ensure
+      Byebug.stop if Byebug.stoppable?
     end
   end
 end

lib/pry-byebug/commands/exit_all.rb

@@ -0,0 +1,14 @@
+module PryByebug
+  #
+  # Exit pry REPL with Byebug.stop
+  #
+  class ExitAllCommand < Pry::Command::ExitAll
+    def process
+      super
+    ensure
+      Byebug.stop if Byebug.stoppable?
+    end
+  end
+end
+
+Pry::Commands.add_command(PryByebug::ExitAllCommand)

lib/pry-byebug/control_d_handler.rb

@@ -0,0 +1,7 @@
+original_handler = Pry.config.control_d_handler
+
+Pry.config.control_d_handler = proc do |eval_string, _pry_|
+  Byebug.stop if Byebug.stoppable?
+
+  original_handler.call(eval_string, _pry_)
+end