feat: yield Process::Pipe when block.arity == 1

Author: initdc

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rb-process (0.2.0)
+    rb-process (0.2.1)
 
 GEM
   remote: https://rubygems.org/

lib/rb/process.rb

@@ -5,7 +5,19 @@
 require_relative "process/version"
 
 module Process
-  def self.run(*args, output: STDOUT, error: STDERR, **options)
+  class Pipe
+    attr_reader :input
+    attr_reader :output
+    attr_reader :error
+
+    def initialize(input, output, error)
+      @input = input
+      @output = output
+      @error = error
+    end
+  end
+
+  def self.run(*args, output: STDOUT, error: STDERR, **options, &block)
     output_strio = StringIO.new
     error_strio = StringIO.new
 
@@ -34,14 +46,23 @@ def self.run(*args, output: STDOUT, error: STDERR, **options)
 
     if block_given?
       begin
-        yield in_w, out_r, err_r
+        case block.arity
+        when 1
+          yield Pipe.new(in_w, out_r, err_r)
+        when 2
+          yield in_w, out_r
+        when 3
+          yield in_w, out_r, err_r
+        else
+          raise ArgumentError.new("block must take 1 to 3 arguments")
+        end
       rescue StandardError => e
         Process.detach(pid)
         raise e
       end
     end
 
-    in_w.close
+    in_w.close unless in_w.closed?
     out_w.close
     err_w.close
 

lib/rb/process/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Process
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

spec/rb/process_spec.rb

@@ -45,6 +45,8 @@
   end
 
   it "answer with cmd with ruby style" do
+    expect(Process.run("bash") { |pipe| pipe.input.puts "uname" }).to eq "Linux\n"
+    expect(Process.run("bash") { |i, _o| i.puts "uname" }).to eq "Linux\n"
     expect(Process.run("bash") { |i, _o, _e| i.puts "uname" }).to eq "Linux\n"
   end