wishdev-rio 0.4.3.1

data/lib/rio/stream/base.rb

@@ -0,0 +1,54 @@
+#--
+# =============================================================================== 
+# 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/state'
+module RIO
+  module Stream
+
+    class Base < State::Base
+      def iostate(*args) self end
+      def stream?() true end
+      def recno() nil end
+
+      # The copy code requires a directory test
+      # Streams are never directories
+      def dir?() false end
+
+    end
+
+  end
+end
+

data/lib/rio/stream/duplex.rb

@@ -0,0 +1,79 @@
+#--
+# =============================================================================== 
+# 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/stream'
+require 'rio/stream/open'
+
+module RIO
+  module Stream
+    module Duplex
+      module Ops
+        module Output
+          def wclose()
+            #p "wclose #{self}"
+            ioh.close_write
+            return self.close.softreset if ioh.closed?
+            self
+          end
+        end
+      end
+      class Open < RIO::Stream::Open
+        def output() super.extend(Ops::Output) end
+        def inout() super.extend(Ops::Output) end
+      end
+    end
+  end
+end
+__END__
+module RIO
+  module Stream
+    module Duplex
+
+      class Input < RIO::Stream::Input
+        include Ops::Input
+      end
+
+      class Output < RIO::Stream::Output
+        include Ops::Output
+      end
+      
+      class InOut < RIO::Stream::InOut
+        include Ops::Output
+        include Ops::Input
+      end
+    end
+  end
+end

data/lib/rio/stream/open.rb

@@ -0,0 +1,202 @@
+#--
+# =============================================================================== 
+# 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/ioh'
+require 'rio/stream/base'
+require 'rio/ops/stream'
+require 'rio/ops/path'
+require 'rio/cp'
+require 'rio/piper/cp'
+require 'rio/util'
+
+
+module RIO
+  module Stream
+    class Open < Base
+      include Ops::Stream::Status
+      include Ops::Path::URI
+      include Cp::Open::Output
+      include Cp::Open::Input
+      include Piper::Cp::Input
+
+      def check?() true end
+
+      def open(m=nil,*args)
+        #p callstr('open',m,*args)
+        case
+        when open? then open_(*args)
+        when m.nil? then open_(*args)
+        else mode(m).open_(*args)
+        end
+      end
+
+      protected
+
+      def open_(*args)
+        #p callstr('open_',args.inspect)+" mode='#{mode?}' (#{mode?.class}) ioh=#{self.ioh} open?=#{open?}"
+        self.ioh = self.rl.open(mode?,*args) unless open?
+        self
+      end
+
+
+    end
+  end
+end
+
+
+module RIO
+  module Stream
+    class Open < Base
+
+      public
+
+      #def size() self.contents.size end
+      #def empty?() self.eof? end
+
+      def iostate(sym)
+        if mode? && mode?.allows_both?
+          open_.inout()
+        else
+          implicit_state(sym)
+        end
+      end
+
+      protected
+
+      OUTPUT_SYMS = Util::build_sym_hash([:print,:printf,:puts,:putc,:write,
+                                          :print!,:printf!,:puts!,:putc!,:write!,
+                                          :put_,:putrec,:putrec!,:<,:<< ])
+
+      def when_missing(sym,*args) 
+        #p callstr('when_missing',sym,*args)+" mode?(#{mode?.class})=#{mode?}"
+        nobj = iostate(sym)
+        return nobj unless nobj.nil?
+        gofigure(sym,*args)
+      end
+
+      def stream_state(cl)
+        #p callstr('stream_state',cl)
+        #p "LOOP: retry:#{cx['retrystate']} => #{cl}" 
+        return nil if cx['retrystate'] == cl
+        cx['retrystate'] = cl
+
+        become(cl).add_rec_methods.add_extensions.add_filters.setup
+      end
+      def output() stream_state('Stream::Output') end
+      def input()  stream_state('Stream::Input')  end
+      def inout()  stream_state('Stream::InOut')  end
+
+      private
+
+      def sym_state(sym,im,om)
+       if OUTPUT_SYMS[sym.to_s] or RIO::Ext::OUTPUT_SYMS[sym.to_s]
+         om ||= (sym.to_s == '<<' ? 'a' : 'w')
+         mode_(om).open_.output()
+        else
+         im ||= 'r'
+         mode_(im).open_.input()
+       end
+      end
+
+      def implicit_state(sym,*args)
+        #p callstr('implicit_state',sym,*args) + " mode='#{mode?}' im='#{inputmode?}' om='#{outputmode?}'"
+        case
+        when (inputmode?.nil? and outputmode?.nil?)
+          sym_state(sym,nil,nil)
+        when outputmode?.nil?
+          im = inputmode?
+          if im.allows_both?
+            mode_(im).open_.inout()
+          else
+            sym_state(sym,im,nil)
+          end
+        when inputmode?.nil?
+          om = outputmode?
+          if om.allows_both?
+            mode_(om).open_.inout()
+          else
+            sym_state(sym,nil,om)
+          end
+        else
+          sym_state(sym,inputmode?,outputmode?)
+        end
+      end
+
+
+    end
+
+    class Close < State::Base
+      include Ops::Stream::Status
+
+      def check?() true end
+      def close() 
+        #p callstr('close')+" mode='#{mode?}' ioh=#{self.ioh} open?=#{open?}"
+        return self unless self.open? 
+        self.close_
+        cx['retrystate'] = nil
+        self
+      end
+
+      def reopen(*args) self.close.softreset.open(*args) end
+
+      protected
+
+      def base_state() 'Stream::Reset'  end
+      def when_missing(sym,*args) 
+        self.close_.retryreset()
+      end
+
+      def close_() 
+        #p callstr('close_')+" mode='#{mode?}' ioh=#{self.ioh} open?=#{open?}"
+        return self unless self.open? 
+        #clear_selection
+        self.ioh.close 
+        self.ioh = nil
+        self.rl.close
+        self
+      end
+
+      private
+
+      def clear_selection()
+        cx.delete('stream_sel')
+        cx.delete('stream_nosel')
+        self
+      end
+
+    end
+  end
+end

data/lib/rio/stream.rb

@@ -0,0 +1,181 @@
+#--
+# =============================================================================== 
+# 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/stream/base'
+require 'rio/ops/path'
+require 'rio/ops/stream'
+require 'rio/ops/stream/input'
+require 'rio/ops/stream/output'
+require 'rio/ext'
+
+require 'rio/filter/gzip'
+#require 'rio/filter/yaml'
+#require 'rio/filter/chomp'
+#require 'rio/filter/strip'
+require 'rio/filter/closeoneof'
+
+module RIO
+
+  module Stream #:nodoc: all
+    class Reset < Base
+      # Mixin the appropriate ops
+      #include Ops::Path::Str
+      include Ops::Stream::Reset
+
+      def check?() true end
+      def when_missing(sym,*args) retryreset() end
+    end
+
+    class IOBase < Base
+      # Mixin the appropriate ops
+      include Ops::Path::Str
+      include Ops::Stream::Status
+      include Ops::Stream::Manip
+      
+      def check?() open? end
+      def when_missing(sym,*args) 
+        #p callstr('when_missing',sym,*args)
+        retryreset() 
+      end
+      def base_state() 'Stream::Close' end
+      def reset()
+        self.close.softreset()
+      end
+      def setup
+        ioh.sync = sync? if cx.has_key?('sync')
+        self
+      end
+      def add_filter(mod)
+        unless ioh.kind_of?(mod)
+          ioh.extend(mod)
+        end
+      end
+#      def open_(*args)
+#        self
+#      end
+      def rectype_mod
+        case cx['stream_rectype']
+        when 'lines' then RIO::RecType::Lines
+        when 'bytes' then RIO::RecType::Bytes
+        else RIO::RecType::Lines
+        end
+      end
+      def self.copier(src,dst)
+        RIO::Copy::Strategy::Stream.instance.copier(src,dst)
+      end
+    end
+    module Ini
+      def setup
+        super
+        @recno = -1
+        @get_selrej,@get_rangetops = create_selrej()
+        self
+      end
+    end
+    class Input < IOBase
+      include Ops::Stream::Input
+      include Ini
+      include Filters
+
+
+      def add_extensions()
+        #p callstr('add_extensions')
+        Ext::Input.add_extensions(self)
+      end
+      def add_filters
+        #p "IN stream#add_filters"
+        add_filter(Filter::GZipRead) if gzip?
+        #add_filter(Filter::YAML) if yaml?
+        add_line_filters()
+#        add_filter(Filter::FasterCSV) if csv?
+#        if closeoneof?
+#          add_filter(Filter::CloseOnEOF)
+#          ioh.oncloseproc = proc { self.on_closeoneof }
+#        end
+        self
+      end
+      def add_rec_methods()
+        self.extend(rectype_mod.module_eval('Input'))
+      end
+    end 
+
+    class Output < IOBase
+      include Ops::Stream::Output
+      
+      def add_rec_methods()
+        self.extend(rectype_mod.module_eval('Output'))
+      end
+      def add_extensions()
+        Ext::Output.add_extensions(self)
+      end
+      def add_filters
+        add_filter(Filter::GZipWrite) if gzip?
+        #add_filter(Filter::FasterCSV) if csv?
+        #add_filter(Filter::YAML) if yaml?
+        self
+      end
+
+    end 
+
+    class InOut < IOBase
+      include Ops::Stream::Input
+      include Ops::Stream::Output
+      include Ini
+      include Filters
+      
+      def add_rec_methods()
+        self.extend(rectype_mod.module_eval('Input'))
+        self.extend(rectype_mod.module_eval('Output'))
+      end
+      def add_extensions()
+        Ext::Input.add_extensions(self)
+        Ext::Output.add_extensions(self)
+      end
+
+      def add_filters
+        add_line_filters()
+
+ #       if closeoneof?
+ #         add_filter(Filter::CloseOnEOF)
+ #         ioh.oncloseproc = proc { self.on_closeoneof }
+ #       end
+        self
+      end
+
+    end 
+  end
+
+end # module RIO