wishdev-rio 0.4.3.1

data/lib/rio/ftp/ftpfile.rb

@@ -0,0 +1,20 @@
+
+require 'tempfile'
+require 'delegate'
+module RIO
+  module FTP
+    class FTPFile < DelegateClass( ::Tempfile )
+      def initialize(remote_path, netftp)
+        @remote_path = remote_path
+        @netftp = netftp
+        @ftpfile = ::Tempfile.new('ftpfile')
+        super(@ftpfile)
+      end
+      def close()
+        super
+        @netftp.put(path(),@remote_path)
+      end
+      
+    end
+  end
+end

data/lib/rio/grande.rb

@@ -0,0 +1,97 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005,2006,2007 Christopher Kleckner
+# All rights reserved
+#
+# This file is part of the Rio library for ruby.
+#
+# Rio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# Rio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rio; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# =============================================================================== 
+#++
+#
+# To create the documentation for Rio run the command
+#  ruby build_doc.rb
+# from the distribution directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Doc::EXAMPLES
+# * RIO::Rio
+#
+
+
+require 'rio/match'
+module RIO
+  module Grande #:nodoc: all
+    include Match::Common
+    def [](*args)
+      #p "#{callstr('[]',*args)} ss_type=#{cx['ss_type']} stream_iter=#{stream_iter?}"
+      ss_args = cx['ss_args'] = args
+      rtn = nil
+      if ss_args and (ss_type = ss_type?(_ss_keys()))
+        rtn = self.__send__(ss_type,*(ss_args)).to_a
+      else
+        rtn = to_a()
+      end
+      _ss_returns_first? ? _ss_return_first(rtn) : rtn
+    end
+
+    private
+
+    def _ss_return_first(ary)
+      _ss_clear_single_return()
+      ary[0] 
+    end
+
+    SINGLE_RETURN_KEYS = %[line record row]
+
+    def _ss_clear_single_return
+      SINGLE_RETURN_KEYS.each do |key|
+        cx.delete(key)
+      end
+    end
+
+    def _ss_returns_first?
+      cx['line'] || cx['record'] || cx['row']
+    end
+
+    def fixnumss(*args)
+
+      #p args[0].class,ss_type?(_ss_keys())
+      ss_args = cx['ss_args'] = args
+      if ss_args.length == 1 and ss_args[0].kind_of?(Fixnum) and !cx.has_key?('dirlines')
+        ans = nil
+        ss_type = ss_type?(_ss_keys())
+
+        if ss_args and ss_type
+          ans = self.__send__(ss_type,*(ss_args)).to_a
+        else
+          ans = to_a()
+        end
+        
+        return (ans.empty? ? nil : ans[0])
+      else
+        if ss_args and (ss_type = ss_type?(_ss_keys()))
+          return self.__send__(ss_type,*(ss_args)).to_a
+        else
+          return to_a()
+        end
+      end
+    end
+
+  end
+end

data/lib/rio/handle.rb

@@ -0,0 +1,100 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005,2006,2007 Christopher Kleckner
+# All rights reserved
+#
+# This file is part of the Rio library for ruby.
+#
+# Rio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# Rio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rio; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# =============================================================================== 
+#++
+#
+# To create the documentation for Rio run the command
+#  ruby build_doc.rb
+# from the distribution directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Doc::EXAMPLES
+# * RIO::Rio
+#
+
+
+require 'rio/base'
+
+module RIO
+  class Handle < Base #:nodoc: all
+    attr_accessor :target
+    def initialize(st=nil) 
+      @target = st 
+    end
+    def initialize_copy(*args)
+      #p callstr('initialize_copy',*args)
+      super
+      @target = @target.clone
+    end
+    def method_missing(sym,*args,&block)
+      #  p callstr('method_missing',*args)
+      @target.__send__(sym,*args,&block)
+    end
+    def t_instance_of?(*args) @target.instance_of?(*args) end
+    def t_kind_of?(*args) @target.kind_of?(*args) end
+    def t_class(*args) @target.class(*args) end
+    def to_s() @target.to_s() end
+    def split(*args,&block) @target.split(*args,&block) end
+    def callstr(func,*args)
+      self.class.to_s+'['+self.to_s+']'+'.'+func.to_s+'('+args.join(',')+')'
+    end
+  end
+end
+
+if $0 == __FILE__
+  eval DATA.read, nil, $0, __LINE__+4
+end
+
+__END__
+
+require 'test/unit'
+
+class TC_Handle < Test::Unit::TestCase
+  def test_handle
+    assert_nothing_raised { 
+      h = Handle.new()
+    }
+    h = Handle.new()
+    assert_nil(h.v)
+    object = String.new("Zippy")
+    h = FwdHandle.new(object,$stdout)
+    s = "Hello World\n"
+
+    rtn = h.fwd_rtn(:write,s)
+    assert_equal(rtn,s.length)
+
+    rtn = h.fwd_rtn_obj(:write,s)
+    assert_same(rtn,object)
+
+    rtn = h.fwd_rtn_new(:write,s)
+    assert_equal(object.class,rtn.class)
+    assert_not_same(object,rtn)
+    assert_equal(rtn,s.length.to_s)
+
+    rtn = h.write(s)
+    p rtn
+#    assert_equal(sdocdir,docdir.to_s)
+#    assert_instance_of(RIO::Rio,doc,"catpath does not return an FS")
+  end
+end

data/lib/rio/if/basic.rb

@@ -0,0 +1,64 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005,2006,2007 Christopher Kleckner
+# All rights reserved
+#
+# This file is part of the Rio library for ruby.
+#
+# Rio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# Rio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rio; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# =============================================================================== 
+#++
+#
+# To create the documentation for Rio run the command
+#  ruby build_doc.rb
+# from the distribution directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Doc::EXAMPLES
+# * RIO::Rio
+#
+
+
+module RIO
+  class Rio
+
+    #def to_rl() target.to_rl end
+    
+    # Returns the length of the Rio's String representation
+    # 
+    # To get the size of the underlying file system object use RIO::IF::Test#size
+    def length() target.length end
+
+    # Equality - calls to_s on _other_ and compares its return value 
+    # with the value returned by Rio#to_s
+    def ==(other) target == other end
+
+    # Equality (for case statements) same as Rio#==
+    def ===(other) target === other end
+
+    # Rios are hashed based on their String representation
+    def hash() target.hash end
+
+    # Returns true if their String representations are eql?
+    def eql?(other) target.eql?(other) end
+
+    # Match - invokes _other_.=~, passing the value returned by Rio#to_str
+    def =~(other) target =~ other end
+      
+  end
+end

data/lib/rio/if/csv.rb

@@ -0,0 +1,76 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005,2006,2007 Christopher Kleckner
+# All rights reserved
+#
+# This file is part of the Rio library for ruby.
+#
+# Rio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# Rio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rio; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# =============================================================================== 
+#++
+#
+# To create the documentation for Rio run the command
+#  ruby build_doc.rb
+# from the distribution directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Doc::EXAMPLES
+# * RIO::Rio
+#
+
+
+module RIO
+  module IF
+    module CSV
+      #def file() target.file end
+      #def dir() target.dir end
+      
+      # Puts a Rio in CSV mode and sets the field and record separators.
+      # In csv mode selecting with IF::GrandeStream#records will cause each line
+      # read to be parsed into a line with the CSV standard library.
+      # Specifying using IF::GrandeStream#lines to select will return unparsed strings
+      # as normal.
+      #
+      #  # copy a csv file, changing the field separator
+      #  rio('afile.csv').csv > rio('afile_semicolons.csv').csv(';')
+      #
+      # CSV mode also adds two methods #columns and #skipcolumns which 
+      # allows selecting columns by column index using Fixnums or Ranges
+      # in a way similar to how lines are selected.
+      #
+      #  # iterate through every line but only get the first three columns
+      #  rio("afile.csv").csv.columns(0..2) { |array_of_fields| ... }
+      #
+      #  # iterate through every line but skip the columns 2 and 3 through 5
+      #  rio("afile.csv").csv.skipcolumns(2,3..5) { |array_of_fields| ... }
+      #
+      #  # an array containg all but the first line returning columns 5,6 and 7
+      #  rio("afile.csv").csv.columns(5..7).skiplines[0]
+      #
+      # See RIO::Doc::INTRO for complete documentation on csv mode.
+      def csv(field_separator=',',record_separator=nil,&block) 
+        target.csv(field_separator,record_separator,&block); 
+        self 
+      end
+      # Select columns from a CSV file. See #csv and RIO::Doc::INTRO.
+      def columns(*ranges,&block) target.columns(*ranges,&block); self end
+      # Reject columns from a CSV file. See #csv and RIO::Doc::INTRO.
+      def skipcolumns(*ranges,&block) target.skipcolumns(*ranges,&block); self end
+    end
+  end
+end

data/lib/rio/if/dir.rb

@@ -0,0 +1,157 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005,2006,2007 Christopher Kleckner
+# All rights reserved
+#
+# This file is part of the Rio library for ruby.
+#
+# Rio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# Rio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Rio; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+# =============================================================================== 
+#++
+#
+# To create the documentation for Rio run the command
+#  ruby build_doc.rb
+# from the distribution directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Doc::EXAMPLES
+# * RIO::Rio
+#
+
+
+module RIO
+  module IF
+    module Dir
+      # Calls ::Dir#chdir.
+      #
+      # Changes the current working directory of the process to the directory specified by the Rio. 
+      # Raises a SystemCallError (probably Errno::ENOENT) if the target directory does not exist or 
+      # if the Rio does not reference a directory.
+      #
+      # If a block is given changes to the directory specified by the rio for the length of the block
+      # and changes back outside the block
+      #
+      # Returns the Rio
+      #
+      #  rio('/home').chdir  # change the current working directory to /home
+      #  # the working directory here is /home
+      #  rio('/tmp/data/mydata').delete!.mkpath.chdir {
+      #     # the working directory here is /tmp/data/mydata
+      #  }
+      #  # the working directory here is /home
+      #
+      def chdir(&block) target.chdir(&block);self end
+
+
+
+      # Calls Find#find
+      #
+      # Uses ::Find#find to find all entries recursively for a Rio that 
+      # specifies a directory. Note that there are other ways to recurse through
+      # a directory structure using a Rio. See IF::Grande#each and IF::GrandeEntry#all.
+      #
+      # Calls the block passing a Rio for each entry found. The Rio inherits
+      # file attrubutes from the directory Rio.
+      #
+      # Returns itself
+      #
+      #  rio('adir').find { |entrio| puts "#{entrio}: #{entrio.file?}" }
+      #
+      #  rio('adir').chomp.find do |entrio|
+      #    next unless entrio.file?
+      #    lines = entrio[0..10]  # lines are chomped because 'chomp' was inherited
+      #  end
+      #
+      def find(*args,&block) target.find_entries(*args,&block); self end
+
+
+      # Calls ::Dir#glob
+      #
+      # Returns the filenames found by expanding the pattern given in string, 
+      # either as an array or as parameters to the block. In both cases the filenames
+      # are expressed as a Rio.
+      # Note that this pattern is not a regexp (it�s closer to a shell glob). 
+      # See File::fnmatch for details of file name matching and the meaning of the flags parameter.
+      #
+      #
+      def glob(string,*args,&block) target.glob(string,*args,&block) end
+
+
+      # Calls ::Dir#rmdir
+      #
+      # Deletes the directory referenced by the Rio. 
+      # Raises a subclass of SystemCallError if the directory isn�t empty.
+      # Returns the Rio. If the directory does not exist, just returns the Rio.
+      #
+      # See also #rmtree, IF::Grande#delete, IF::Grande#delete!
+      #
+      #    rio('adir').rmdir # remove the empty directory 'adir'
+      #
+      def rmdir() target.rmdir(); self end
+
+
+      # Calls FileUtils#rmtree
+      #
+      # Removes a directory Rio recursively. Returns the Rio. 
+      # If the directory does not exist, simply returns the Rio
+      #
+      # If called with a block, behaves as if <tt>rmtree.each(&block)</tt> had been called
+      # 
+      # See also IF::Grande#delete!
+      #
+      #  rio('adir').rmtree # removes the directory 'adir' recursively
+      #
+      #  # delete the directory 'adir', recreate it and then change to the new directory
+      #  rio('adir/asubdir').rmtree.mkpath.chdir {
+      #    ...
+      #  }
+      #
+      # 
+      def rmtree() target.rmtree(); self end
+
+      # Calls FileUtils#mkpath
+      #
+      # Makes a new directory named by the Rio and any directories in its path that do not exist.
+      # 
+      # Returns the Rio. If the directory already exists, just returns the Rio.
+      #
+      #  rio('adir/a/b').mkpath
+      def mkpath(&block) target.mkpath(&block); self end
+
+      
+      # Calls FileUtils#mkdir
+      #
+      # Makes a new directory named by the Rio with permissions specified by the optional parameter. 
+      # The permissions may be modified by the value of File::umask
+      #
+      # Returns the Rio. If the directory already exists, just returns the Rio.
+      #
+      #  rio('adir').mkdir
+      def mkdir(*args,&block) target.mkdir(*args,&block); self end
+
+
+
+    end
+  end
+end
+
+module RIO
+  class Rio
+    include RIO::IF::Dir
+  end
+end