Improve the break command support

lib/irb/cmd/break.rb

@@ -7,8 +7,12 @@ module IRB
 
   module ExtendCommand
     class Break < Debug
-      def execute(*args)
-        super(['break', *args].join(' '))
+      def self.transform_args(args)
+        args&.dump
+      end
+
+      def execute(args = nil)
+        super(pre_cmds: "break #{args}")
       end
     end
   end

lib/irb/cmd/continue.rb

@@ -8,7 +8,7 @@ module IRB
   module ExtendCommand
     class Continue < Debug
       def execute(*args)
-        super(['continue', *args].join(' '))
+        super(do_cmds: ["continue", *args].join(" "))
       end
     end
   end

lib/irb/cmd/debug.rb

@@ -11,7 +11,7 @@ class Debug < Nop
       ].map { |file| /\A#{Regexp.escape(file)}:\d+:in `irb'\z/ }
       IRB_DIR = File.expand_path('..', __dir__)
 
-      def execute(debug_command = nil)
+      def execute(pre_cmds: nil, do_cmds: nil)
         unless binding_irb?
           puts "`debug` command is only available when IRB is started with binding.irb"
           return
@@ -26,8 +26,8 @@ def execute(debug_command = nil)
         end
 
         command = nil
-        if debug_command
-          command = ['irb', nil, debug_command]
+        if pre_cmds || do_cmds
+          command = ['irb', pre_cmds, do_cmds]
         end
 
         # To make debugger commands like `next` or `continue` work without asking

lib/irb/cmd/finish.rb

@@ -8,7 +8,7 @@ module IRB
   module ExtendCommand
     class Finish < Debug
       def execute(*args)
-        super(['finish', *args].join(' '))
+        super(do_cmds: ["finish", *args].join(" "))
       end
     end
   end

lib/irb/cmd/next.rb

@@ -8,7 +8,7 @@ module IRB
   module ExtendCommand
     class Next < Debug
       def execute(*args)
-        super(['next', *args].join(' '))
+        super(do_cmds: ["next", *args].join(" "))
       end
     end
   end

lib/irb/context.rb

@@ -545,10 +545,16 @@ def local_variables # :nodoc:
       workspace.binding.local_variables
     end
 
-    # Return a command name if it's aliased from the argument and it's not an identifier.
+    # Return true if it's aliased from the argument and it's not an identifier.
     def symbol_alias?(command)
       return nil if command.match?(/\A\w+\z/)
       command_aliases.key?(command.to_sym)
     end
+
+    # Return true if the command supports transforming args
+    def transform_args?(command)
+      command = command_aliases.fetch(command.to_sym, command)
+      ExtendCommandBundle.load_command(command)&.respond_to?(:transform_args)
+    end
   end
 end

lib/irb/ruby-lex.rb

@@ -65,9 +65,9 @@ def set_input(io, p = nil, context:, &block)
             false
           end
         else
-          # Accept any single-line input starting with a non-identifier alias (ex: @, $)
+          # Accept any single-line input for symbol aliases or commands that transform args
           command = code.split(/\s/, 2).first
-          if context.symbol_alias?(command)
+          if context.symbol_alias?(command) || context.transform_args?(command)
             next true
           end