Avoid using arbitrary puts to print output

lib/irb.rb

@@ -1422,6 +1422,14 @@ def inspect
 
     private
 
+    def puts(...)
+      @context.io.puts(...)
+    end
+
+    def print(...)
+      @context.io.print(...)
+    end
+
     def generate_prompt(opens, continue, line_offset)
       ltype = @scanner.ltype_from_open_tokens(opens)
       indent = @scanner.calc_indent_level(opens)

lib/irb/command/base.rb

@@ -36,7 +36,7 @@ def help_message(help_message = nil)
         def execute(irb_context, arg)
           new(irb_context).execute(arg)
         rescue CommandArgumentError => e
-          puts e.message
+          irb_context.io.puts e.message
         end
 
         private
@@ -55,6 +55,16 @@ def initialize(irb_context)
       def execute(arg)
         #nop
       end
+
+      private
+
+      def puts(...)
+        irb_context.io.puts(...)
+      end
+
+      def print(...)
+        irb_context.io.print(...)
+      end
     end
 
     Nop = Base

lib/irb/input-method.rb

@@ -48,6 +48,14 @@ def prompting?
       false
     end
 
+    def puts
+      fail NotImplementedError
+    end
+
+    def print
+      fail NotImplementedError
+    end
+
     # For debug message
     def inspect
       'Abstract InputMethod'
@@ -73,6 +81,14 @@ def gets
       @line[@line_no += 1] = line
     end
 
+    def puts(...)
+      @stdout.puts(...)
+    end
+
+    def print(...)
+      @stdout.print(...)
+    end
+
     # Whether the end of this input method has been reached, returns +true+ if
     # there is no more data to read.
     #
@@ -155,6 +171,14 @@ def gets
       @io.gets
     end
 
+    def puts(...)
+      ::Kernel.puts(...)
+    end
+
+    def print(...)
+      ::Kernel.print(...)
+    end
+
     # The external encoding for standard input.
     def encoding
       @external_encoding
@@ -223,6 +247,14 @@ def gets
       end
     end
 
+    def puts(...)
+      @stdout.puts(...)
+    end
+
+    def print(...)
+      @stdout.format(...)
+    end
+
     # Whether the end of this input method has been reached, returns +true+
     # if there is no more data to read.
     #
@@ -474,6 +506,14 @@ def gets
       end
     end
 
+    def puts(...)
+      @stdout.puts(...)
+    end
+
+    def print(...)
+      @stdout.print(...)
+    end
+
     # Whether the end of this input method has been reached, returns +true+
     # if there is no more data to read.
     #

test/irb/helper.rb

@@ -32,6 +32,14 @@ def gets
         @list[@line_no]&.tap {@line_no += 1}
       end
 
+      def puts(...)
+        Kernel.puts(...)
+      end
+
+      def print(...)
+        Kernel.print(...)
+      end
+
       def eof?
         @line_no >= @list.size
       end