Add command support to CheckBreakpoint

Author: st0012

lib/debug/breakpoint.rb

@@ -317,8 +317,9 @@ def description
   end
 
   class CheckBreakpoint < Breakpoint
-    def initialize expr, path
+    def initialize expr, path, command
       @expr = expr.freeze
+      @command = command
       @key = [:check, @expr].freeze
       @path = path
 
@@ -338,7 +339,10 @@ def setup
     end
 
     def to_s
-      "#{generate_label("Check")} #{@expr}"
+      s = "#{generate_label("Check")} #{@expr}"
+      s << " pre: #{@command[1]}" if defined?(@command) && @command && @command[1]
+      s << " do: #{@command[2]}"  if defined?(@command) && @command && @command[2]
+      s
     end
   end
 

lib/debug/session.rb

@@ -1324,7 +1324,7 @@ def repl_add_breakpoint arg
         @tc << [:breakpoint, :method, $1, $2, $3, cond, cmd, path]
         return :noretry
       when nil
-        add_check_breakpoint cond, path
+        add_check_breakpoint cond, path, cmd
       else
         @ui.puts "Unknown breakpoint format: #{arg}"
         @ui.puts
@@ -1356,8 +1356,8 @@ def add_catch_breakpoint pat
       add_bp bp
     end
 
-    def add_check_breakpoint expr, path
-      bp = CheckBreakpoint.new(expr, path)
+    def add_check_breakpoint expr, path, command
+      bp = CheckBreakpoint.new(expr, path, command)
       add_bp bp
     end
 

test/debug/break_test.rb

@@ -460,6 +460,17 @@ def test_break_command_executes_do_option_and_continues_with_method_bp
         type 'c'
       end
     end
+
+    def test_break_command_executes_do_option_and_continues_with_check_bp
+      debug_code(program) do
+        type 'break if: s.is_a?(String) do: p "foobar"'
+        assert_line_text(/BP - Check  s\.is_a\?\(String\) do: p "foobar"/)
+        type 'break 9'
+        type 'c'
+        assert_line_text(/foobar/)
+        type 'c'
+      end
+    end
   end
 
   #