feat: send method to writer or reader by define method_missing

Author: initdc

lib/rb/io/multi_writer.rb

@@ -15,6 +15,7 @@ class MultiWriter
     # If `#sync_close?` is `true`, closing this `IO` will close all of the underlying
     # IOs.
     attr_accessor :sync_close
+    attr_accessor :writers
     attr_reader :closed
 
     @closed = false

lib/rb/io/stapled.rb

@@ -15,6 +15,8 @@ class IO
   class Stapled
     # If `#sync_close?` is `true`, closing this `IO` will close the underlying `IO`s.
     attr_accessor :sync_close
+    attr_reader :reader
+    attr_reader :writer
 
     # Returns `true` if this `IO` is closed.
     #
@@ -23,6 +25,15 @@ class Stapled
 
     @closed = false
 
+    WRITER_DELEGATE = %w[
+      <<
+    ].freeze
+
+    READER_DELEGATE = %w[
+      eof
+      eof?
+    ].freeze
+
     alias_method :sync_close?, :sync_close
     alias_method :closed?, :closed
 
@@ -33,53 +44,30 @@ def initialize(reader, writer, sync_close: false)
       @sync_close = sync_close
     end
 
-    def read(...)
-      check_open
-
-      @reader.read(...)
-    end
-
-    def readlines(...)
-      check_open
-
-      @reader.readlines(...)
-    end
-
-    def each_line(...)
-      check_open
-
-      @reader.each_line(...)
+    def method_missing(name, *args, &block)
+      if write_methods?(name.to_s)
+        check_open
+        @writer.send(name, *args, &block)
+      elsif read_methods?(name.to_s)
+        check_open
+        @reader.send(name, *args, &block)
+      else
+        super
+      end
     end
 
-    # Gets a string from `reader`.
-    def gets(...)
-      check_open
-
-      @reader.gets(...)
+    def respond_to_missing?(name, include_private = false)
+      super || write_methods?(name.to_s) || read_methods?(name.to_s)
     end
 
-    def puts(...)
-      check_open
-
-      @writer.puts(...)
+    def write_methods?(name)
+      name.include?("put") || name.include?("prin") ||
+        name.include?("write") || WRITER_DELEGATE.include?(name)
     end
 
-    def print(...)
-      check_open
-
-      @writer.print(...)
-    end
-
-    def printf(...)
-      check_open
-
-      @writer.print(...)
-    end
-
-    def write(...)
-      check_open
-
-      @writer.write(...)
+    def read_methods?(name)
+      name.include?("get") || name.include?("read") ||
+        name.include?("each") || READER_DELEGATE.include?(name)
     end
 
     # Flushes `writer`.
@@ -91,6 +79,14 @@ def flush
       self
     end
 
+    def close_write
+      @writer.close
+    end
+
+    def close_read
+      @reader.close
+    end
+
     # Closes this `IO`.
     #
     # If `sync_close?` is `true`, it will also close the underlying `IO`s.
@@ -105,14 +101,6 @@ def close
       end
     end
 
-    def close_write
-      @writer.close
-    end
-
-    def close_read
-      @reader.close
-    end
-
     protected def check_open
       raise IOError.new("Closed stream") if closed?
     end

spec/rb/process_spec.rb

@@ -43,4 +43,29 @@
     expect(tempfile.readlines).to eq ["Linux\n"]
     tempfile.delete
   end
+
+  # https://devdocs.io/ruby~3.4/io#class-IO-label-Reading
+  it "responds to methods" do
+    Process.run("bash", out: File.open(File::NULL, "r+")) do |pipe|
+      pipe.write_nonblock("echo")
+      pipe << " "
+      pipe.write("hello ")
+      pipe.print("w")
+      pipe.printf("%s", "o")
+      pipe.putc "r"
+      pipe.write("l")
+      pipe.puts "d"
+      pipe.close_write
+
+      pipe.getbyte
+      pipe.getc
+      pipe.readbyte
+      pipe.readchar
+      pipe.readpartial(1)
+      pipe.readline
+      pipe.readlines
+      pipe.gets
+      pipe.read
+    end
+  end
 end