ruby_ex 0.1.1

data/src/commands/datas.rb

@@ -0,0 +1,16 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: datas.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Datas
+
+    autoloaded_module(__FILE__, false)
+
+  end # module Datas
+
+end # module Commands

data/src/commands/datas/composite.rb

@@ -0,0 +1,31 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: composite.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Datas
+
+    class Composite < Data
+
+      def initialize ( *cmd )
+        @commands = cmd.flatten
+      end
+
+      def each ( &block )
+        @commands.each(&block)
+      end
+
+      def waitpid
+        each { |x| x.waitpid }
+      end
+
+
+    end # class Composite
+
+  end # module Datas
+
+end # module Commands

data/src/commands/datas/data.rb

@@ -0,0 +1,65 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: data.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Datas
+
+    #
+    # A CommandData instance contains all information returned by a Command#run:
+    #
+    #   * It's exit status:
+    #     - Exit number.
+    #     - Signals
+    #     - ...
+    #
+    #   * It's outputs:
+    #     - What it writes on stdout.
+    #     - stderr.
+    class Data
+
+      attr_accessor :pid
+      attr_accessor :input
+      attr_accessor :output
+      attr_accessor :error
+      attr_accessor :status
+
+
+      def waitpid
+        return @status unless @status.nil?
+        Process.waitpid(@pid)
+        @status = $?
+      end
+
+
+      def | ( rhs ) # FIXME CHECK ME
+        raise unless rhs.is_a? Command
+        rhs < output
+      end
+
+
+      def method_missing ( meth, *a, &b )
+        if output.respond_to?(meth)
+          output.send(meth, *a, &b)
+        else
+          super
+        end
+      end
+
+
+      def display ( out=STDOUT, err=STDERR )
+        waitpid
+        out.print @output.read
+        err.print @error.read
+      end
+
+
+    end # class Data
+
+  end # module Datas
+
+end # module Commands

data/src/commands/datas/factory.rb

@@ -0,0 +1,69 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: factory.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Datas
+
+    class Factory < Data
+
+      attr_accessor :command_data_class
+
+      def initialize
+        @command_data_class = Temp
+      end
+
+      def new_command_data ( *a, &b )
+        data = @command_data_class.new(*a, &b)
+        instance_variables.each do |var|
+          next if var =~ /command_data_class/
+          val = instance_variable_get(var)
+          data.send(var[1..-1] + '=', val) unless val.nil?
+        end
+        data
+      end
+
+      alias :create :new_command_data
+
+    end # class Factory
+
+
+
+
+    test_section __FILE__ do
+
+      class FactoryTest < Test::Unit::TestCase
+
+        def setup
+          assert_nothing_raised { @fa = Factory.new }
+        end
+
+        def test_0_initialize
+        end
+
+        def test_new_command_data
+          assert_nothing_raised do
+            assert_kind_of(Data, @fa.create)
+            assert_kind_of(Data, @fa.new_command_data)
+          end
+        end
+
+        def test_simple
+          assert_nothing_raised do
+            @fa.pid = 42
+            @data = @fa.create
+          end
+          assert_equal(42, @data.pid)
+        end
+
+      end # class FactoryTest
+
+    end
+
+  end # module Datas
+
+end # module Commands

data/src/commands/datas/temp.rb

@@ -0,0 +1,26 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: temp.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Datas
+
+    class Temp < Data
+
+      def initialize ( *a, &b )
+        super
+        @output = TempPath.new('cmd-output')
+        @output.open_mode = 'w'
+        @error = TempPath.new('cmd-error')
+        @error.open_mode = 'w'
+      end
+
+    end # class Temp
+
+  end # module Datas
+
+end # module Commands

data/src/commands/factory.rb

@@ -0,0 +1,67 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: factory.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  class Factory < Command
+
+    def initialize
+      super(nil)
+      @command_class = Command
+    end
+
+    def new_command ( *a, &b )
+      cmd = @command_class.new(*a, &b)
+      instance_variables.each do |var|
+        next if var =~ /^@command(_class)?$/
+        cmd.send(var[1..-1] + '=', instance_variable_get(var))
+      end
+      cmd
+    end
+
+    alias :create :new_command
+
+  end # class Factory
+
+
+
+  test_section __FILE__ do
+
+    class FactoryTest < Test::Unit::TestCase
+
+      def setup
+        assert_nothing_raised { @fa = Factory.new }
+      end
+
+      def test_0_initialize
+      end
+
+      def test_new_command
+        assert_nothing_raised do
+          assert_kind_of(Command, @fa.create('wc'))
+          assert_kind_of(Command, @fa.new_command('ls'))
+        end
+      end
+
+      def test_simple
+        assert_nothing_raised do
+          @fa << 42
+          @fa.input = 'inp'
+          @fa = @fa > 'out'
+          @cmd = @fa.create('foo')
+        end
+        assert_equal('foo', @cmd.command)
+        assert_equal(42, @cmd.args.first)
+        assert_equal('inp', @cmd.input)
+        assert_equal('out', @cmd.output)
+      end
+
+    end # class FactoryTest
+
+  end
+
+end # module Commands

data/src/commands/helpers.rb

@@ -0,0 +1,81 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: helpers.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Helpers
+
+    class ::Object
+
+      def to_io_for_commands
+        to_io
+      end
+
+    end
+
+    class ::String
+
+      #
+      # Make a command from a String.
+      #
+      # Split your command when no quotations are used.
+      #
+      # Do not touch the String and give as one argument.
+      def to_cmd
+        if self =~ /['"`]/
+          Command.new(self)
+        else
+          Command.new(*split(/\s+/))
+        end
+      end
+
+      def to_io_for_commands
+        tmp = TempPath.new
+        tmp.open('w') { |f| f.write self }
+        tmp.to_io_for_commands
+      end
+
+    end # class ::String
+
+
+
+    class ::Array
+
+      def to_cmd
+        Command.new(*self)
+      end
+
+    end # class ::Array
+
+  end # module Helpers
+
+
+
+  test_section __FILE__ do
+
+    class HelpersTest < Test::Unit::TestCase
+
+      def test_string_to_cmd
+        assert_equal(['foo', 'bar'], "foo bar".to_cmd.to_a)
+
+        assert_equal(['foo', 'b/ar', 'qu44x32'],
+                     "foo b/ar \t \n  qu44x32".to_cmd.to_a)
+
+        s = "foo 'b/ar \t' \n  qu44x32"
+        assert_equal([s], s.to_cmd.to_a)
+      end
+
+      def test_array_to_cmd
+        a = ['foo', "b/ar \t", "\n  qu44x32"]
+        assert_equal(a, a.to_cmd.to_a)
+      end
+
+    end # class HelpersTest
+
+  end
+
+end # module Commands

data/src/commands/pipe.rb

@@ -0,0 +1,66 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: pipe.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  class Pipe < Command
+
+    def initialize ( *cmds )
+      @cmds = cmds.map { |x| x.dup }
+    end
+
+    def run ( *a )
+      pids = []
+      ([nil] + @cmds).zip(@cmds + [nil]).each do |cmd1, cmd2|
+        next if cmd1.nil? or cmd2.nil?
+        rd, wr = IO.pipe
+        cmd1.output = wr
+        cmd2.input = rd
+        pids << Kernel.fork do
+          rd.close
+          cmd1.exec
+        end
+        wr.close
+      end
+      pids << Kernel.fork do
+        @cmds.last.exec
+      end
+      pids.each { |pid| Process.waitpid pid }
+    end
+
+    def input= ( arg )
+      @cmds.first.input = arg
+    end
+
+    def input
+      @cmds.first.input
+    end
+
+    def output= ( arg )
+      @cmds.last.output = arg
+    end
+
+    def output
+      @cmds.last.output
+    end
+
+    def error= ( arg )
+      @cmds.last.error = arg
+    end
+
+    def error
+      @cmds.last.error
+    end
+
+    def to_sh
+      strs = @cmds.map { |cmd| "(#{cmd.to_sh})" }
+      "(#{strs.join(' | ')})#{sh_args}"
+    end
+
+  end # class Pipe
+
+end # module Commands

data/src/commands/runners.rb

@@ -0,0 +1,16 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: runners.rb 255 2005-06-01 00:08:46Z ertai $
+
+require 'commands'
+
+module Commands
+
+  module Runners
+
+    autoloaded_module(__FILE__, false)
+
+  end # module Runners
+
+end # module Commands