ruby_ex 0.1.1

data/src/thread_mutex.rb

@@ -0,0 +1,11 @@
+require 'ruby_ex'
+class Thread
+
+  alias :raise_without_exception_call :raise
+
+  def raise ( exc, *a, &b )
+    exc = exc.exception if exc.respond_to? :exception
+    raise_without_exception_call(exc, *a, &b)
+  end
+
+end

data/src/timeout_ex.rb

@@ -0,0 +1,79 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: timeout_ex.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'thread'
+
+
+module TimeoutEx
+
+  class Error < Interrupt; end
+
+  module_function
+  def timeout(sec, exception=Error, mutex=nil, &block)
+    if sec > 0
+      thread = Thread.new(mutex) do |mutex|
+        Thread.current[:timeout_mutex] = mutex
+        block[]
+      end
+      sleep(sec)
+      if thread.alive?
+        unless mutex.nil?
+          mutex.synchronize { thread.kill if thread.alive? }
+        else
+          thread.kill
+        end
+        raise(exception, 'execution expired')
+      end
+    else
+      block[]
+    end
+    true
+  end
+
+end # module TimeoutEx
+
+
+test_section __FILE__ do
+
+
+class TimeoutExTest < Test::Unit::TestCase
+
+  def test_timeout
+    run = false
+    assert_raises(TimeoutEx::Error) do
+      TimeoutEx.timeout(0.1) do
+        sleep(0.5)
+        run = true
+      end
+    end
+    assert(! run)
+  end
+
+  def test_mutex
+    run = false
+    assert_raises(TimeoutEx::Error) do
+      TimeoutEx.timeout(0.1, TimeoutEx::Error, Mutex.new) do
+        Thread.current[:timeout_mutex].synchronize do
+          sleep(0.5)
+          run = true
+        end
+      end
+    end
+    assert(run)
+  end
+
+  def test_no_timeout
+    assert(TimeoutEx.timeout(0.1) { nil })
+  end
+
+end # class TimeoutExTest
+
+
+end
+

data/src/trace.rb

@@ -0,0 +1,26 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: polrop $
+# $Id: trace.rb 90 2005-01-09 22:09:32Z polrop $
+
+
+module Kernel
+
+  def trace_on
+    set_trace_func proc { |event, file, line, id, binding, classname|
+      printf("%8s %s:%-2d %10s %8s\n",
+             event,
+             file,
+             line,
+             id,
+             classname)
+    }
+  end
+
+  def trace_off
+    set_trace_func nil
+  end
+
+end # module Kernel

data/src/uri/druby.rb

@@ -0,0 +1,78 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: druby.rb 258 2005-06-01 00:22:51Z ertai $
+
+
+require 'uri_ex'
+
+
+module URI
+
+  class DRuby < Generic
+
+    SCHEME = 'druby'
+    DEFAULT_HOST = 'localhost'
+    DEFAULT_PORT = 0
+    DEFAULT_QUERY = ''
+    COMPONENT = [ :scheme, :host, :port ].freeze
+
+    def self.build(args)
+      if args.is_a?(Array)
+        args << DEFAULT_HOST if args.size < 1
+        args << DEFAULT_PORT if args.size < 2
+      elsif args.is_a?(Hash)
+        args[:host] = DEFAULT_HOST if args[:host].nil?
+        args[:port] = DEFAULT_PORT if args[:port].nil?
+      end
+      return super(Util::make_components_hash(self, args))
+    end
+
+    #FIXME: find a way to always print the port even if it is the default one.
+
+  end # class DRuby
+
+  @@schemes[DRuby::SCHEME.upcase] = DRuby
+
+end # module URI
+
+
+test_section __FILE__ do
+
+
+module URI
+
+  class DRubyTest < ::Test::Unit::TestCase
+
+    def test_classic
+      uri = URI.parse('druby://www.foo.com:42000')
+      assert(uri.is_a?(DRuby))
+      assert_equal('druby', uri.scheme)
+      assert_nil(uri.user)
+      assert_nil(uri.password)
+      assert_nil(uri.userinfo)
+      assert_equal('www.foo.com', uri.host)
+      assert_equal(42000, uri.port)
+      assert_equal('druby://www.foo.com:42000', uri.to_s)
+    end
+
+    def test_build
+      assert_equal('druby://localhost', DRuby.build([]).to_s)
+      assert_equal('druby://www.foo.com:52000',
+                   DRuby.build(['www.foo.com', 52000 ]).to_s)
+    end
+
+    def test_query
+      uri = DRuby.build(['www.foo.com', 52000 ])
+      uri.query = 'toto=tata&foo=bar'
+      assert_equal('druby://www.foo.com:52000?toto=tata&foo=bar', uri.to_s)
+    end
+
+  end # class DRubyTest
+
+end # module URI
+
+
+end

data/src/uri/file.rb

@@ -0,0 +1,63 @@
+# Copyright: Copyright (c) 2005  Nicolas Pouillard. All rights reserved.
+# Author: Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: file.rb 258 2005-06-01 00:22:51Z ertai $
+
+require 'uri_ex'
+
+module URI
+
+  class File < Generic
+
+    COMPONENT = [
+      :scheme,
+      :path
+    ].freeze
+    TAR = 'tar'.to_cmd.freeze
+    GZIP = 'gzip'.to_cmd.freeze
+
+
+    def initialize ( *args )
+      super
+      unless @host.nil? or @host.empty?
+        raise ArgumentError, 
+          "You cannot neither setup a host (#{@host}), nor a relative path"
+      end
+    end
+
+    def self.build ( args )
+      tmp = Util::make_components_hash(self, args)
+      return super(tmp)
+    end
+
+    def checkout
+      p = self.pathname
+      raise CheckoutError, to_s unless p.exist?
+      p
+    end
+
+    def commit ( aPath )
+      aPath.cp_r(self.pathname)
+    end
+
+    def save
+      p = self.pathname
+      raise SaveError, to_s unless p.exist?
+      if p.directory?
+        out = TempPath.new('save', "#{p.basename}.tar.gz")
+        cmd = TAR['czf', out, p]
+      else
+        out = TempPath.new('save', "#{p.basename}.gz")
+        cmd = GZIP['-c', p] > out
+      end
+      cmd.run(self.runner)
+      out
+    end
+
+  end # class File
+
+  @@schemes['FILE'] = File
+
+end # module URI

data/src/uri/ftp_ex.rb

@@ -0,0 +1,36 @@
+# Copyright: Copyright (c) 2005  Nicolas Pouillard. All rights reserved.
+# Author: Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: ftp_ex.rb 258 2005-06-01 00:22:51Z ertai $
+
+require 'uri_ex'
+require 'uri/ftp'
+require 'net/ftp'
+
+module URI
+
+  class FTP
+
+    def checkout
+      tmp = TempPath.new('checkout', pathname.basename.to_s)
+
+      begin
+        Net::FTP.open(host) do |ftp|
+          ftp.login
+          ftp.passive = true
+          ftp.getbinaryfile(path, tmp)
+        end
+
+      rescue Exception => ex
+        tmp.unlink if tmp.exist?
+        raise CheckoutError, "Cannot checkout #{to_s} (error: #{ex.short_pp})"
+      end
+
+      tmp
+    end
+
+  end # class FTP
+
+end # module URI

data/src/uri/http_ex.rb

@@ -0,0 +1,41 @@
+# Copyright: Copyright (c) 2005  Nicolas Pouillard. All rights reserved.
+# Author: Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: http_ex.rb 258 2005-06-01 00:22:51Z ertai $
+
+require 'uri_ex'
+require 'uri/http'
+require 'net/http'
+
+module URI
+
+  class HTTP
+
+    def checkout
+      tmp = TempPath.new('checkout', pathname.basename.to_s)
+
+      begin
+        uri = URI.parse(ENV['http_proxy'] || '')
+        proxy_auth = [uri.host, uri.port, uri.user, uri.password]
+
+        Net::HTTP::Proxy(*proxy_auth).start(host) do |http|
+          tmp.open('w') do |out|
+            http.get(path) do |str|
+              out.print str
+            end
+          end
+        end
+
+      rescue Exception => ex
+        tmp.unlink if tmp.exist?
+        raise CheckoutError, "Cannot checkout #{to_s} (error: #{ex.short_pp})"
+      end
+
+      tmp
+  end
+
+  end # class HTTP
+
+end # module URI

data/src/uri/pgsql.rb

@@ -0,0 +1,136 @@
+# Copyright:: Copyright (c) 2005  Nicolas Pouillard. All rights reserved.
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# License::   Gnu General Public License.
+# Revision::  $Id: pgsql.rb 256 2005-06-01 00:11:05Z ertai $
+
+require 'uri_ex'
+
+module URI
+
+  class PgSQL < Generic
+
+    SCHEME = 'pgsql'.freeze
+    DEFAULT_HOST = 'localhost'.freeze
+    DEFAULT_PORT = 5432
+    DEFAULT_QUERY = ''.freeze
+    PG_DUMP = 'pg_dump'.to_cmd.freeze
+
+    COMPONENT = [
+      :scheme,
+      :userinfo,
+      :host,
+      :path,
+      :query
+    ].freeze
+
+
+    def self.build ( args )
+      tmp = Util::make_components_hash(self, args)
+      return super(tmp)
+    end
+
+
+    def mk_connection_opts
+      opts = []
+      opts << '-h' << @host
+      opts << '-p' << @port if @port != DEFAULT_PORT
+      opts << '-U' << @user if @user
+      opts
+    end
+
+
+    def mk_custom_opts
+      opts = []
+      return opts if @query.nil?
+      @query.split(/,/).map do |x|
+        k, v = x.split(/=/)
+        if k.size == 1
+          opts << "-#{k}"
+        else
+          opts << "--#{k}"
+        end
+        opts << v unless v.nil?
+      end
+      opts
+    end
+
+
+    def mk_pg_dump_opts ( out )
+      opts = %w[ -F c -Z 9 -C -f ]
+      opts << out
+      base, table = pathname.split
+      base, table = base.to_s, table.to_s
+      if base =~ /[\.\/]/
+        opts << table
+      else
+        if base =~ /\//
+          raise ArgumentError, "Use database/table not #{pathname}"
+        end
+        opts << '-t' << table << base
+      end
+      opts
+    end
+
+
+    def save
+      out = TempPath.new('dump', pathname.basename.to_s)
+      cmd = PG_DUMP + mk_connection_opts + mk_pg_dump_opts(out) + mk_custom_opts
+      cmd.run(self.runner)
+      out
+    end
+
+  end # class PgSQL
+
+  @@schemes[PgSQL::SCHEME.upcase] = PgSQL
+
+  test_section __FILE__ do
+    class PgSQLTest < Test::Unit::TestCase
+
+      def setup
+        assert_nothing_raised { @foo = URI.parse('pgsql://foo/qux/bar') }
+        assert_nothing_raised { @foo_at_bar = URI.parse('pgsql://foo@bar/baz/ben') }
+        assert_nothing_raised { @foo_at_bar_42 = URI.parse('pgsql://foo@bar:42/bing') }
+        assert_nothing_raised do
+          @cmplx = URI.parse('pgsql://foo@bar/qux/bob?a=b,c,d=e,f,ghi,jkl=mno')
+        end
+      end
+
+      def test_0_parse_basic
+      end
+
+      def test_with_query
+        ref = %w[-h bar -U foo -F c -Z 9 -C -f out bob -a b -c -d e -f --ghi --jkl mno]
+        assert_equal(ref, @cmplx.mk_connection_opts +
+                          @cmplx.mk_pg_dump_opts('out') +
+                          @cmplx.mk_custom_opts)
+      end
+
+      def test_mk_connection_opts
+        assert_equal(['-h', 'foo'], @foo.mk_connection_opts)
+        assert_equal(["-h", "bar", "-U", "foo"], @foo_at_bar.mk_connection_opts)
+        assert_equal(%w[-h bar -p]+[42, '-U', 'foo'], @foo_at_bar_42.mk_connection_opts)
+        assert_equal(["-h", "bar", "-U", "foo"], @cmplx.mk_connection_opts)
+      end
+
+      def test_mk_custom_opts
+        assert_equal([], @foo.mk_custom_opts)
+        assert_equal([], @foo_at_bar.mk_custom_opts)
+        assert_equal([], @foo_at_bar_42.mk_custom_opts)
+        assert_equal(%w[-a b -c -d e -f --ghi --jkl mno], @cmplx.mk_custom_opts)
+      end
+
+      def test_mk_pg_dump_opts
+        base = %w[-F c -Z 9 -C -f out]
+        assert_equal(base + ['bar'], @foo.mk_pg_dump_opts('out'))
+        assert_equal(base + ['ben'], @foo_at_bar.mk_pg_dump_opts('out'))
+        assert_equal(base + ['bing'], @foo_at_bar_42.mk_pg_dump_opts('out'))
+        assert_equal(base + ['bob'], @cmplx.mk_pg_dump_opts('out'))
+      end
+
+      def test_save
+        # FIXME
+      end
+    end # class PgSQLTest
+  end
+
+end # module URI