io.rb in polyphony-0.17

- old
+ new

@@ -1,19 +1,64 @@
 # frozen_string_literal: true
 
 class ::IO
-  def read_watcher
-    @read_watcher ||= EV::IO.new(self, :r)
-  end
+  class << self
+    alias_method :orig_binread, :binread
+    def binread(name, length = nil, offset = nil)
+      File.open(name, 'rb:ASCII-8BIT') do |f|
+        f.seek(offset) if offset
+        length ? f.read(length) : f.read
+      end
+    end
 
-  def write_watcher
-    @write_watcher ||= EV::IO.new(self, :w)
-  end
+    alias_method :orig_binwrite, :binwrite
+    def binwrite(name, string, offset = nil)
+      File.open(name, 'wb:ASCII-8BIT') do |f|
+        f.seek(offset) if offset
+        f.write(string)
+      end
+    end
 
-  def stop_watchers
-    @read_watcher&.stop
-    @write_watcher&.stop
+    EMPTY_HASH = {}
+
+    alias_method :orig_foreach, :foreach
+    def foreach(name, sep = $/, limit = nil, getline_args = EMPTY_HASH, &block)
+      sep, limit = $/, sep if sep.is_a?(Integer)
+      File.open(name, 'r') do |f|
+        f.each_line(sep, limit, getline_args, &block)
+      end
+    end
+
+    alias_method :orig_read, :read
+    def read(name, length = nil, offset = nil, opt = EMPTY_HASH)
+      File.open(name, opt[:mode] || 'r') do |f|
+        f.seek(offset) if offset
+        length ? f.read(length) : f.read
+      end
+    end
+
+    alias_method :orig_readlines, :readlines
+    def readlines(name, sep = $/, limit = nil, getline_args = EMPTY_HASH)
+      File.open(name, 'r') do |f|
+        f.readlines(sep, limit, getline_args)
+      end
+    end
+
+    alias_method :orig_write, :write
+    def write(name, string, offset = nil, opt = EMPTY_HASH)
+      File.open(name, opt[:mode] || 'w') do |f|
+        f.seek(offset) if offset
+        f.write(string)
+      end
+    end
+
+    alias_method :orig_popen, :popen
+    def popen(*args)
+      Open3.popen2(*args) do |i, o, t|
+        yield o
+      end
+    end
   end
 
   # def each(sep = $/, limit = nil, chomp: nil)
   #   sep, limit = $/, sep if sep.is_a?(Integer)
   # end
@@ -32,26 +77,61 @@
   # end
 
   # def getc
   # end
 
-  # def gets(sep = $/, limit = nil, chomp: nil)
-  #   sep, limit = $/, sep if sep.is_a?(Integer)
-  # end
+  alias_method :orig_gets, :gets
+  def gets(sep = $/, limit = nil, chomp: nil)
+    sep, limit = $/, sep if sep.is_a?(Integer)
+    sep_size = sep.bytesize
 
+    @gets_buffer ||= +''
+
+    loop do
+      idx = @gets_buffer.index(sep)
+      return @gets_buffer.slice!(0, idx + sep_size) if idx
+
+      data = readpartial(8192)
+      if data
+        @gets_buffer << data
+      else
+        return nil if @gets_buffer.empty?
+        line, @gets_buffer = @gets_buffer.freeze, +''
+        return line
+      end
+    end
+    # orig_gets(sep, limit, chomp: chomp)
+  end
+
   # def print(*args)
   # end
 
   # def printf(format, *args)
   # end
 
   # def putc(obj)
   # end
 
-  # def puts(*args)
-  # end
+  alias_method :orig_puts, :puts
+  def puts(*args)
+    if args.empty?
+      write "\n"
+      return
+    end
 
+    s = args.each_with_object(+'') do |a, str|
+      if a.is_a?(Array)
+        a.each { |a2| str << a2.to_s << "\n" }
+      else
+        a = a.to_s
+        str << a
+        str << "\n" unless a =~ /\n$/
+      end
+    end
+    write s
+  end
+
   # def readbyte
   # end
 
   # def readchar
   # end
@@ -59,6 +139,6 @@
   # def readline(sep = $/, limit = nil, chomp: nil)
   # end
 
   # def readlines(sep = $/, limit = nil, chomp: nil)
   # end
-end
\ No newline at end of file
+end