rio 0.3.1

data/lib/rio/context/stream.rb

@@ -0,0 +1,170 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005, 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
+#  rake rdoc
+# from the distribution directory. Then point your browser at the 'doc/rdoc' directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Rio
+#
+# <b>Rio is pre-alpha software. 
+# The documented interface and behavior is subject to change without notice.</b>
+
+
+require 'rio/context/cxx.rb'
+require 'rio/iomode'
+
+module RIO
+  module Cx
+    module Methods
+      def stream_iter?() cx.has_key?('stream_itertype') && !cx['nostreamenum'] end
+      
+      private
+
+      def _set_bytes(nb)
+        nb = 1 if nb.nil? or nb < 1
+        cx['bytes_n'] = nb
+      end
+      def _set_bytes_(nb)
+        nb = 1 if nb.nil? or nb < 1
+        cx.set_('bytes_n',nb)
+      end
+
+      def _set_sstype(ss,explicit=true)
+        case
+        when explicit then cx['ss_type'] = ss
+        else cx.set_('ss_type',ss)
+        end
+      end
+      def _set_rectype(rt,explicit=true)
+        case
+        when explicit then cx['stream_rectype'] = rt
+        else cx.set_('stream_rectype',rt)
+        end
+      end
+      def _set_itertype(rt,explicit=true)
+        case
+        when explicit then cx['stream_itertype'] = rt
+        else cx.set_('stream_itertype',rt)
+        end
+      end
+      def _set_selargs(sa,sel=true,&block)
+        key = (sel ? 'stream_sel' : 'stream_nosel')
+        cx[key] = sa
+        each(&block) if block_given?
+        self
+      end
+
+      def _lines(args,sel=true,explicit=true,&block)   
+        #p callstr('lines',*args)
+        _set_itertype('lines',explicit)
+        _set_rectype('lines',explicit)
+        _set_selargs(args,sel,&block)
+      end
+
+      def _records(args,sel=true,explicit=true,&block) 
+        _set_itertype('records',explicit)
+        _set_selargs(args,sel,&block)
+      end
+
+      def _rows(args,sel=true,explicit=true,&block) 
+        _set_itertype('rows',explicit)
+        _set_selargs(args,sel,&block)
+      end
+
+      public
+
+      def lines(*args,&block)   
+        #p callstr('lines',*args)
+        _set_sstype('lines')
+        _lines(args,&block)
+      end
+      def lines_(*args,&block) 
+        #p callstr('lines_',*args)
+        _set_sstype('lines',false)
+        _lines(args,true,false,&block)
+      end
+
+      def records(*args,&block) 
+        _set_sstype('records')
+        _records(args,&block)
+      end
+      def records_(*args,&block) 
+        _set_sstype('records',false)
+        _records(args,true,false,&block)
+      end
+
+      def rows(*args,&block) 
+        _set_sstype('rows')
+        _rows(args,&block)
+      end
+      def rows_(*args,&block) 
+        _set_sstype('rows',false)
+        _rows(args,true,false,&block)
+      end
+
+
+      def nolines(*args,&block) 
+        #p callstr('nolines',*args)
+        self.lines() unless args.empty? or cx.has_key?('stream_sel')
+        _set_sstype('nolines')
+        _lines(args,false,&block)
+      end
+      def norecords(*args,&block) 
+        self.records() unless args.empty? or cx.has_key?('stream_sel')
+        _set_sstype('norecords')
+        _records(args,false,&block)
+      end
+      def norows(*args,&block) 
+        self.rows() unless args.empty? or cx.has_key?('stream_sel')
+        _set_sstype('norows')
+        _rows(args,false,&block)
+      end
+
+
+
+      def bytes(nb=1,*args,&block)
+        _set_sstype('bytes')
+        _set_bytes(nb)
+        _set_itertype('records')
+        _set_rectype('bytes')
+        each(&block) if block_given?
+        self
+      end
+      def bytes_(nb=1,*args,&block)
+        _set_bytes_(nb)
+        _set_itertype('records',false)
+        _set_rectype('bytes',false)
+        each(&block) if block_given?
+        self
+      end
+
+
+
+    end
+  end
+end

data/lib/rio/cp.rb

@@ -0,0 +1,305 @@
+#--
+# =============================================================================== 
+# Copyright (c) 2005, 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
+#  rake rdoc
+# from the distribution directory. Then point your browser at the 'doc/rdoc' directory.
+#
+# Suggested Reading
+# * RIO::Doc::SYNOPSIS
+# * RIO::Doc::INTRO
+# * RIO::Doc::HOWTO
+# * RIO::Rio
+#
+# <b>Rio is pre-alpha software. 
+# The documented interface and behavior is subject to change without notice.</b>
+
+
+require 'rio/exception/copy'
+class String #:nodoc: all
+  def clear()
+    self[0..-1] = ''
+    self
+  end
+end
+module RIO
+  module Cp #:nodoc: all
+    module Util
+      module InOut
+        def cpclose(*args,&block)
+          if args.empty?
+            oldcoc,self.cx['closeoncopy'] = self.cx['closeoncopy'],false
+            rtn = yield
+            rtn.cx['closeoncopy'] = oldcoc
+            rtn.copyclose
+          else
+            if (ario = args[0]).kind_of?(Rio)
+              oldcoc,ario.cx['closeoncopy'] = ario.cx['closeoncopy'],false
+              rtn = yield
+              ario.cx['closeoncopy'] = oldcoc
+              ario.copyclose
+              rtn
+            else
+              yield
+            end
+          end
+        end
+      end
+      module Input
+        include InOut
+        protected
+
+        def cpto_obj_(obj)
+          self.each do |el|
+            obj << el
+          end
+        end
+
+      end
+      module Output
+        include InOut
+
+
+        protected
+
+        def cpfrom_obj_(obj)
+          obj.each do |el|
+            self << el
+          end
+        end
+        def cpfrom_array_(ary)
+          ary.inject(self) { |anio,el| anio << el }
+        end
+      end
+    end
+  end
+end
+
+module RIO
+  module Cp
+    module Stream
+      module Input
+        include Util::Input
+        def >>(arg)
+          cpclose(arg) {
+            case arg
+            when ::Array,::String,::IO then cpto_obj_(arg)
+            else _cpto_rio(arg,:<<)
+            end
+            self
+          }
+        end
+        def >(arg)
+          cpclose(arg) {
+            case arg 
+            when ::Array,::String then cpto_obj_(arg.clear)
+            when ::IO then cpto_obj_(arg)
+            else _cpto_rio(arg,:<)
+            end
+            self
+          }
+        end
+        private
+        def _cpto_rio(arg,sym)
+          ario = ensure_rio(arg)
+          ario = ario.join(self.filename) if ario.dir?
+          ario.cpclose {
+            ario = ario.iostate(sym)
+            self.each do |el|
+              ario << el
+            end
+            ario
+          }
+        end
+      end
+      module Output
+        include Util::Output
+        def <<(arg) cpclose { _cpfrom(arg) } end
+        def <(arg) cpclose { _cpfrom(arg) } end
+
+        private
+
+        def _cpfrom(arg)
+            case arg
+            when ::Array then cpfrom_array_(arg)
+            when ::IO then cpfrom_obj_(arg)
+            when ::String then self.put_(arg)
+            else _cpfrom_rio(arg)
+            end
+            self
+        end
+        def _cpfrom_rio(arg)
+          ensure_rio(arg).each do |el|
+            self << el
+          end
+        end
+      end
+    end
+  end
+end
+module RIO
+  module Cp
+    module File
+      module Output
+        include Util::Output
+        def <(arg) cpclose { self.iostate(:<) < arg } end
+        def <<(arg) cpclose { self.iostate(:<<) << arg } end
+      end
+      module Input
+        include Util::Input
+        def >(arg)  
+          spcp(arg) || cpclose(arg) { self.iostate(:>) > arg  } 
+        end
+        def >>(arg) 
+          spcp(arg) || cpclose(arg) { self.iostate(:>>) >> arg } 
+        end
+        def copy_as_file?(arg)
+          arg.kind_of?(Rio) and arg.scheme == 'ftp'
+        end
+        def spcp(arg)
+          if arg.kind_of?(Rio) and arg.scheme == 'ftp'
+            arg < new_rio(rl.path)
+            self
+          else
+            nil
+          end
+        end
+        alias :copy :>
+      end
+    end
+  end
+end
+module RIO
+  module Cp
+    module Open
+      module Output
+        include Util::Output
+        def <(arg) cpclose { self.iostate(:<) < arg } end
+        def <<(arg) cpclose { self.iostate(:<<) << arg } end
+      end
+      module Input
+        include Util::Input
+        def >(arg) cpclose(arg) { self.iostate(:>) > arg } end
+        def >>(arg) cpclose(arg) { self.iostate(:>>) >> arg } end
+        alias :copy :>
+      end
+    end
+  end
+end
+module RIO
+  module Cp
+    module Dir
+      module Output
+        include Util::Output
+        def <<(arg)  _cpfrom(arg); self  end
+        def <(arg)  _cpfrom(arg); self end
+
+        private
+
+        def _cpfrom(arg)
+          case arg
+          when ::Array then cpfrom_array_(arg)
+          else _cpfrom_rio(ensure_rio(arg))
+          end
+        end
+        def _cpfrom_rio(ario)
+          #p callstr('_cpfrom_rio',ario)
+          dest = self.join(ario.filename)
+          if ario.dir?
+            dest.mkdir
+            ario.nostreamenum.each do |el|
+              dest < el
+            end
+          else
+            dest < ario
+          end
+        end
+      end
+      module Input
+        include Util::Input
+        def >>(arg)
+          case arg
+          when ::Array then cpto_obj_(arg)
+          else _cpto_rio(ensure_rio(arg))
+          end
+          self
+        end
+        def >(arg)
+          case arg
+          when ::Array then cpto_obj_(arg.clear)
+          else _cpto_rio(ensure_rio(arg))
+          end
+          self
+        end
+        alias :copy :>
+
+        private
+
+        def _cpto_rio(ario)
+          ario = ario.join(self.filename) if ario.exist?
+          nostreamenum.cpto_obj_(ario.mkdir)
+        end
+      end
+    end
+  end
+end
+module RIO
+  module Cp
+    module NonExisting
+      module Output
+        include Util::Output
+        def <(arg)  
+          if _switch_direction?(arg)
+            arg > self
+            self
+          else 
+            _cpsrc(arg) < arg
+          end
+        end
+        def <<(arg)  
+          if _switch_direction?(arg)
+            arg >> self
+            self
+          else 
+            _cpsrc(arg) << arg
+          end
+        end
+
+        private
+
+        def _cpsrc(arg)
+          if arg.kind_of?(::Array) and !arg.empty? and arg[0].kind_of?(Rio)
+            self.mkdir.reset
+          else
+            self.nfile
+          end
+        end
+        def _switch_direction?(arg)
+          arg.kind_of?(Rio) and arg.dir? and !arg.stream_iter?
+        end
+      end
+    end
+  end
+end
+
+__END__