wishdev-rio 0.4.3.1

data/lib/rio/ext/yaml/tie.rb

@@ -0,0 +1,149 @@
+#--
+# =============================================================================== 
+# 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 'yaml'
+require 'forwardable'
+require 'delegate'
+
+module RIO
+  module Ext
+    module YAML #:nodoc: all
+      module Tie
+        class Doc
+          extend Forwardable
+          def initialize(fn)
+            @filename = fn
+            @io = nil
+            @doc = nil
+            @root = nil
+          end
+          def open()
+            @io = ::File.new('database.yml',"r")
+            @doc = ::YAML.load(@io)
+            @root = RIO::Ext::YAML::Tie::Root.new(@doc)
+            self
+          end
+
+          def close()
+            if @root.dirty?
+              @io.close
+              ::File.open('database.yml',"w") do |ios|
+                ::YAML.dump(@doc,ios)
+              end
+            end
+          end
+
+          def_instance_delegators(:@root,:inspect)
+          def method_missing(sym,*args)
+            @root.__send__(sym,*args)
+          end
+          def self.new_node(doc,cont)
+          end
+        end
+      end
+    end
+  end
+end
+module RIO
+  module Ext
+    module YAML #:nodoc: all
+      module Tie
+        class Base
+          attr :doc,:cont
+          def initialize(doc,cont=nil)
+            @doc = doc
+            @cont = cont
+          end
+        end
+        class Root < Base
+          extend Forwardable
+          attr_accessor :dirty
+          def initialize(doc)
+            @doc = doc
+            @root = Tie::Hash.new(@doc,self)
+            @dirty = false
+          end
+          def dirty=(val)
+            @dirty = val
+          end
+          def dirty?()
+            @dirty
+          end
+          def_instance_delegators(:@root,:inspect)
+          def method_missing(sym,*args)
+            @root.__send__(sym,*args)
+          end
+        end
+      end
+    end
+  end
+end
+
+module RIO
+  module Ext
+    module YAML #:nodoc: all
+      module Tie
+        module Node
+          def dirty=(val)
+            cont.dirty=(val)
+          end
+        end
+        class Hash < DelegateClass(::Hash)
+          include Tie::Node
+          attr_reader :doc
+          attr_reader :cont
+          def initialize(doc,cont=nil)
+            @doc = doc
+            @cont = cont
+            super(@doc)
+          end
+          def [](key)
+            val = super
+            case val
+            when ::Hash then Tie::Hash.new(val,self)
+            else val
+            end
+          end
+          def []=(key,val)
+            self.dirty = true
+            super
+          end
+        end
+      end
+    end
+  end
+end
+__END__

data/lib/rio/ext/yaml.rb

@@ -0,0 +1,164 @@
+#--
+# =============================================================================== 
+# 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 'yaml'
+
+module RIO
+  module Ext
+    module YAML #:nodoc: all
+      module Cx
+        def yaml(&block) 
+          cxx('yaml',true,&block) 
+        end
+        def yamldoc(&block) 
+          #@yamldoc = YAML::Tie::Doc.new(self.fspath).open
+          become('Ext::YAML::Doc::Existing')
+        end
+        def yaml?() cxx?('yaml') end 
+        def yaml_(fs=',',rs=nil) 
+          cxx_('yaml',true) 
+        end
+        protected :yaml_
+        def objects(*args) records(*args) end
+        def documents(*args) rows(*args) end
+        def object(*args) record(*args) end
+        def document(*args) row(*args) end
+        def skipobjects(*args) skiprecords(*args) end
+        def skipdocuments(*args) skiprows(*args) end
+      end
+    end
+    module YAML
+      module Input
+          
+        def cpto_(arg)
+          #p callstr('cpto_',arg.inspect)
+          
+          case arg
+          when ::Array,::String then super
+          when ::IO,::StringIO then cpto_io_(arg)
+          else super
+          end
+        end
+        def apto_(arg)
+          case arg
+          when ::Array,::String then super
+          when ::IO,::StringIO then cpto_io_(arg)
+          else super
+          end
+        end
+        def cpto_io_(ioarg)
+          recs = self.to_a
+          if recs.size == 1
+            YAML.dump(recs[0],ioarg)
+          else
+            YAML.dump(recs,ioarg)
+          end
+        end
+        def cpto_string_(string)
+          string << ::YAML.dump_stream(self.contents)
+        end
+        def get_(arg=nil)
+          case cx['stream_itertype']
+          when 'lines' then super
+          when 'records' then ::YAML.load(self.ioh)
+          when 'rows' then ::YAML.dump(::YAML.load(self.ioh))
+          else ::YAML.load(self.ioh)
+          end
+        end
+        def each_rec_(&block)
+          case cx['stream_itertype']
+          when 'lines' then super
+          when 'records' then ::YAML.load_documents(self.ioh,&block)
+          when 'rows' then ::YAML.load_documents(self.ioh) { |obj| yield obj.to_yaml }
+          else ::YAML.load_documents(self.ioh,&block)
+          end
+          self
+        end
+        def load()
+          getrec()
+        end
+        def getobj()
+          getrec()
+        end
+
+      end
+    end
+
+    module YAML
+      module Output
+        protected
+
+        def cpfrom_(arg)
+          #p "yaml#cpfrom_(#{arg})"
+          case arg
+          #when ::Array then cpfrom_array_(arg)
+          when Rio,::IO,::StringIO then super
+          else self.put_(arg)
+          end
+          self
+        end
+#        def cpfrom_array_(array)
+#          array.each { |el|
+#            self.put_(el)
+#          }
+#          self
+#        end
+#        def cpfrom_rio_(ario)
+#          p "yaml#cpfrom_rio_(#{ario})"
+#          ioh.puts(::YAML.dump_stream(*ario[]))
+#          self
+#        end
+        def put_(obj)
+          #p callstr('put_',obj)
+          ioh.puts(obj.to_yaml)
+        end
+
+        public
+
+        def dump(obj)
+          putrec!(obj)
+        end
+        def putobj(obj)
+          putrec(obj)
+        end
+        def putobj!(obj)
+          putrec!(obj)
+        end
+      end
+    end
+  end
+end
+__END__

data/lib/rio/ext/zipfile/fs.rb

@@ -0,0 +1,116 @@
+#--
+# =============================================================================== 
+# 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/fs/base'
+require 'zip/zip'
+require 'zip/zipfilesystem'
+require 'rio/scheme/path'
+require 'rio/ext/zipfile/wrap'
+
+module RIO
+  module ZipFile
+    module RootDir
+      class FS
+        attr_reader :dir
+        def initialize(zipfile)
+          @zipfile = zipfile
+          @dir = @zipfile.dir
+          @file = @zipfile.file
+          @test = @zipfile.file
+          #p "InFile: #{@file.class}"
+        end
+        def commit(&block)
+          yield if block_given?
+          @zipfile.commit
+        end
+        include RIO::FS::Str
+        include RIO::FS::Dir
+        include RIO::FS::Test
+        def mkdir(*args)
+          commit{super}
+        end
+        def rmdir(*args)
+          commit{@dir.rmdir(*args)}
+        end
+      end
+    end
+    module InFile
+      class FS
+        attr_reader :file,:dir
+        def initialize(zipfile)
+          @zipfile = zipfile
+          @file = RIO::ZipFile::Wrap::File.new(@zipfile.file)
+          @dir = RIO::ZipFile::Wrap::Dir.new(@zipfile.dir)
+          @test = @file
+          #p "InFile: #{@file.class}"
+        end
+        include RIO::FS::File
+        include RIO::FS::Dir
+        include RIO::FS::Test
+        include RIO::FS::Str
+      end
+    end
+  end
+end
+
+__END__
+
+      
+      class CentralDir < RIO::FS::Native
+        def initialize(zipfilepath)
+          @zipfilepath = zipfilepath
+          @zipfile = Zip::ZipFile.new(@zipfilepath)
+          super
+        end
+        def mkdir(path)
+          @zipfile.mkdir(path)
+        end
+        def rmdir(path)
+          @zipfile.remove(path)
+        end
+        def file()
+          self
+        end
+        def open(zipfilepath)
+          @zipfilepath = zipfilepath
+          @zipfile = Zip::ZipFile.new(@zipfilepath)
+          RIO::ZipFile::CentralDir.new(@zipfile)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/rio/ext/zipfile/rl.rb

@@ -0,0 +1,251 @@
+#--
+# =============================================================================== 
+# 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 'uri'
+require 'rio/uri/file'
+require 'rio/ext/zipfile/fs'
+
+module RIO
+  module ZipFile #:nodoc: all
+    module RootDir
+      RESET_STATE = 'ZipFile::State::Reset'
+      #RESET_STATE = RIO::RL::PathBase::RESET_STATE
+      
+      require 'rio/rl/base'
+      class RL < RIO::RL::Base 
+        RIOSCHEME = 'zipfile'
+        RIOPATH = RIO::RL::CHMAP.invert[RIOSCHEME].to_s.freeze
+        attr_reader :zipfilepath,:infilepath
+        def initialize(zipfilepath,infilepath=nil)
+          @zipfilepath = zipfilepath
+          @uri = URI(@zipfilepath)
+          if infilepath
+            @infilepath = infilepath
+            @uri.query = @infilepath
+          end
+          @zipfile = ::Zip::ZipFile.new(@zipfilepath,::Zip::ZipFile::CREATE)
+          super()
+        end
+        def file_rl
+          RIO::File::RL.new(infilepath,{:fs => RIO::ZipFile::InFile::FS.new(@zipfile)})
+        end
+        def openfs_()
+          RIO::ZipFile::RootDir::FS.new(@zipfile)
+        end
+        def open()
+          IOH::Dir.new(ZipFile::Wrap::Stream::Root.new(@zipfile))
+        end
+        def close()
+          p "CLOSING ZIPFILE RL"
+        end
+        def path()
+          @infilepath
+        end
+        def fspath()
+          @infilepath
+        end
+        def base(*args) 
+          #p "base: #{args.inspect}"
+          '' 
+        end
+        def path_no_slash() self.path.to_s.sub(/\/$/,'') end
+        #def path() nil end
+        def scheme() self.class.const_get(:RIOSCHEME) end
+        def opaque()
+          @uri.to_s
+        end
+        include RIO::RL::PathMethods
+        def urlpath() 
+          return '' unless fspath
+          RIO::RL.fs2url(fspath) 
+        end
+        def urlpath=(arg) @infilepath = RIO::RL.url2fs(arg) end
+
+        # USAGE EXAMPLES
+        # zroot = rio('f.zip').zipfile
+        #
+        # rio(zroot,'tdir').mkdir
+        # rio(zroot,'tdir').mkdir
+        # rio(zroot).each {}
+        # rio(:zipfile,'t.zip','tdir').mkdir
+        # rio('zipfile:t.zip','tdir').mkdir
+        # rio('zipfile:t.zip?tdir').mkdir
+        # rio(?z,'t.zip?tdir').mkdir
+        # rio(?z,'t.zip','tdir').mkdir
+        # rio('zipfile:file:///tmp/t.zip?tdir').mkdir
+        SPLIT_RE = %r|(.+)(?:\?(.+))?$|.freeze
+        def self.splitrl(s)
+          sub,opq,whole = split_riorl(s)
+          if opq.nil? or opq.empty?
+            []
+          elsif bm = SPLIT_RE.match(opq)
+            zpath = bm[1] unless bm[1].nil? or bm[1].empty?
+            ipath = bm[2] unless bm[2].nil? or bm[2].empty?
+            [zpath,ipath]
+          else
+            []
+          end
+        end
+      end
+    end
+  end
+end
+
+
+__END__
+    module Dir
+      require 'rio/rl/path'
+      RESET_STATE = RIO::RL::PathBase::RESET_STATE
+      require 'tmpdir'
+      class RL < RIO::RL::PathBase 
+        RIOSCHEME = 'tempdir'
+        DFLT_PREFIX = Temp::RL::DFLT_PREFIX
+        DFLT_TMPDIR = Temp::RL::DFLT_TMPDIR
+        attr_reader :prefix,:tmpdir,:tmprl
+        def initialize(file_prefix=DFLT_PREFIX,temp_dir=DFLT_TMPDIR)
+          #puts "initialize(#{file_prefix.inspect},#{temp_dir.inspect})"
+          @prefix = file_prefix || DFLT_PREFIX
+          @tmpdir = temp_dir || DFLT_TMPDIR
+          require 'rio/tempdir'
+          @td = ::Tempdir.new( @prefix.to_s, @tmpdir.to_s)
+          super(@td.to_s)
+        end
+        def dir_rl() 
+          RIO::Dir::RL.new(self.uri, {:fs => self.fs})
+          #self 
+        end
+        SPLIT_RE = Temp::RL::SPLIT_RE
+        def self.splitrl(s)
+          Temp::RL.splitrl(s)
+        end
+      end
+    end
+    module File
+      require 'rio/rl/path'
+      RESET_STATE = 'Temp::Stream::Open'
+      class RL < RIO::RL::PathBase 
+        RIOSCHEME = 'tempfile'
+        DFLT_PREFIX = Temp::RL::DFLT_PREFIX
+        DFLT_TMPDIR = Temp::RL::DFLT_TMPDIR
+        attr_reader :prefix,:tmpdir,:tmprl
+        def initialize(file_prefix=DFLT_PREFIX,temp_dir=DFLT_TMPDIR)
+          #puts "initialize(#{file_prefix.inspect},#{temp_dir.inspect})"
+          @prefix = file_prefix || DFLT_PREFIX
+          @tmpdir = temp_dir || DFLT_TMPDIR
+          require 'tempfile'
+          @tf = ::Tempfile.new( @prefix.to_s, @tmpdir.to_s)
+          super(@tf.path)
+        end
+        def file_rl() 
+        RIO::File::RL.new(self.uri,{:fs => self.fs})
+          #self 
+        end
+        def open(mode='ignored')
+          #p callstr('open',mode)
+          @tf
+        end
+        def close 
+          super
+          @tf = nil
+        end
+        SPLIT_RE = Temp::RL::SPLIT_RE
+        def self.splitrl(s)
+          Temp::RL.splitrl(s)
+        end
+      end
+    end
+    require 'rio/state'
+    class Reset < State::Base
+      def initialize(*args)
+        super
+        #p args
+        @tempobj = nil
+      end
+      #def self.default_cx
+      #  Cx::Vars.new( { 'closeoneof' => false, 'closeoncopy' => false } )
+      #end
+
+      def check?() true end
+      def mkdir(prefix=rl.prefix,tmpdir=rl.tmpdir)
+        self.rl = RIO::Temp::Dir::RL.new(prefix, tmpdir)
+        become 'Dir::Existing'
+      end
+#      def mkdir()
+#        dir()
+#      end
+      def chdir(&block)
+        self.mkdir.chdir(&block)
+      end
+      def file(prefix=rl.prefix,tmpdir=rl.tmpdir)
+        self.rl = RIO::Temp::File::RL.new(prefix, tmpdir)
+        become 'Temp::Stream::Open'
+      end
+      def scheme() rl.scheme() end
+      def host() rl.host() end
+      def opaque() rl.opaque() end
+      def to_s() rl.url() end
+      def exist?() false end
+      def file?() false end
+      def dir?() false end
+      def open?() false end
+      def closed?() true end
+      def when_missing(sym,*args)
+        #p @rl.scheme
+        if @tempobj.nil?
+          file()
+        else
+          gofigure(sym,*args)
+        end
+      end
+    end
+    require 'rio/stream/open'
+    module Stream
+      class Open < RIO::Stream::Open
+        def iostate(sym)
+          mode_('w+').open_.inout()
+        end
+#        def inout() stream_state('Temp::Stream::InOut') end
+      end
+#      require 'rio/stream'
+#      class InOut < RIO::Stream::InOut
+#        def base_state() 'Temp::Stream::Close' end
+#      end
+#      class Close < RIO::Stream::Close
+#        def base_state() 'Temp::Reset' end
+#      end
+
+    end
+  end
+end