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 = [
+      :<<
+    ].freeze
+
+    READER_DELEGATE = [
+      :eof,
+      :eof?
+    ].freeze
+
     alias_method :sync_close?, :sync_close
     alias_method :closed?, :closed
 
@@ -33,53 +44,26 @@ 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(...)
-    end
-
-    # Gets a string from `reader`.
-    def gets(...)
-      check_open
-
-      @reader.gets(...)
-    end
-
-    def puts(...)
-      check_open
-
-      @writer.puts(...)
-    end
-
-    def print(...)
-      check_open
-
-      @writer.print(...)
+    def method_missing(name, *args, &block) # rubocop:disable Style/MissingRespondToMissing
+      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
 
-    def printf(...)
-      check_open
-
-      @writer.print(...)
+    def write_methods?(name)
+      name.include?("put") || name.include?("print") || name.include?("write") ||
+        WRITER_DELEGATE.include?(name)
     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`.
@@ -105,14 +89,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