ruby_ex 0.1.1

data/src/sendmail.rb

@@ -0,0 +1,214 @@
+#!/usr/bin/env ruby
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2004, 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: header 98 2004-09-29 12:07:43Z ertai $
+
+require 'pathname'
+require 'ostruct'
+require 'optparse'
+
+
+class Sendmail
+
+  MAIL_FILE = Pathname.new(',mail')
+
+  def initialize
+    require 'ruby_ex'
+    require 'yaml/chop_header'
+  end
+
+  def parse_mail_options ( *args )
+    email = (defined? EMAIL)? EMAIL : ENV['EMAIL']
+    full_name = (defined? FULLNAME)? FULLNAME : ENV['FULLNAME']
+    full_email = (defined? FULL_EMAIL)? FULL_EMAIL : "#{full_name}  <#{email}>"
+    options = OpenStruct.new(
+      :from_name => full_email,
+      :from => email,
+      :to => [],
+      :server => ENV['SMTPSERVER'] || 'localhost:25',
+      :body => STDIN
+    )
+    string_args, hash_args = args.partition { |x| x.is_a?(String) }
+    hash_args.each do |hash|
+      raise ArgumentError, "Bad argument #{hash}" unless hash.is_a?(Hash)
+      hash.each do |k,v|
+	options.send("#{k}=", v)
+      end
+    end
+    OptionParser.new do |opts|
+      opts.separator ''
+      opts.on('-b', '--body FILE', 'Choose a file for the mail body') do |aFile|
+	options.body = File.open(aFile)
+      end
+      opts.on('-t', '--mail-to NAME', 'Choose a recipient') do |aString|
+	options.to << aString
+      end
+      opts.on('-s', '--server NAME:PORT', 'Choose a mail server') do |aString|
+	options.server, options.port = aString.split(/:/)
+        options.port ||= 25
+      end
+      opts.on('-S', '--subject NAME', 'Choose your mail subject') do |aString|
+	options.subject = aString.sub(/\.?$/, '.')
+      end
+      opts.on('--[no-]sign', 'Sign the message with gpg') do |signed|
+	options.signed = signed
+      end
+      opts.on('--passphrase FILE', 'the passphrase file') do |aFile|
+	options.pass = Pathname.new(aFile)
+      end
+      opts.on('-a', '--[no-]ask', 'Ask a confirmation before sending') do |ask|
+        options.ask = ask
+      end
+      opts.on('--comment STRING', 'Choose a comment for GPG') do |aComment|
+        options.comment = aComment
+      end
+      opts.on('-m', '--[no-]mime', 'Choose the mime protocole') do |mime|
+        options.mime = mime
+      end
+      opts.on_tail('-h', '--help', 'Show this message') do
+	puts opts
+	exit
+      end
+    end.parse!(string_args)
+    check_mail_options(options)
+    options
+  end
+
+
+  def check_mail_options ( options )
+    raise ArgumentError, 'No recipents' if options.to.empty?
+    raise ArgumentError, 'No mail server' if options.server.nil?
+    raise ArgumentError, 'No mail subject' if options.subject.nil?
+  end
+
+  #
+  # Mail.
+  #
+  def sendmail ( *args )
+
+    body, opts = nil, nil
+    if MAIL_FILE.exist?
+      STDERR.puts "Using the cache file #{MAIL_FILE}"
+      MAIL_FILE.open do |mail_file|
+        opts = YAML::chop_header(mail_file)
+        body = mail_file.read
+      end
+    else
+      opts = parse_mail_options(*args)
+      body = (opts.body.nil?)? '' : opts.body.read
+    end
+    server, port = opts.server.split(/:/)
+    port ||= 25
+    STDERR.puts "Smtp Server: #{server}"
+
+    #
+    # Ask the user
+    #
+    if opts.ask
+      require 'ask'
+      question = %Q[
+        |Send a mail, with this subject: #{opts.subject}
+        |  to #{opts.to.join(', ')}
+        |  from #{opts.from_name}
+        |  #{(opts.signed.nil?)? 'not signed !' : 'signed by ' + opts.from}
+        |Are you sure?].head_cut!
+      raise 'Aborting' if ask(question, :y) == :n
+    end
+
+    require 'net/smtp'
+    Net::SMTP.start(server, port) do |smtp|
+      smtp.open_message_stream(opts.from, opts.to) do |f|
+        f.print %Q[From: #{opts.from_name}
+                  |Subject: #{opts.subject}
+                  |To: #{opts.to.join(', ')}\n].head_cut!
+        if opts.signed
+          TempPath.new do |mail_body|
+            mail_body.open('w') do |out|
+              if opts.mime
+                out.puts '
+                  |Content-Type: text/plain; charset=ISO-8859-1; format=flowed
+                  |Content-Transfer-Encoding: 8bit
+                  |
+                  |'.head_cut!
+              end
+              out.print body
+            end
+            f.print sign_wrap(opts, mail_body)
+          end
+        else
+          f.print "\n" + body
+        end
+      end
+    end
+  end
+
+
+  def new_id ( n )
+    ([0] * n).map!{ |x| "%X" % rand(15) }.join
+  end
+  protected :new_id
+
+
+  def sign_wrap ( opts, body )
+    id = "===============#{new_id 10}==" if opts.mime
+
+    TempPath.new do |tmp_name|
+      tmp_name.open('w') do |tmp|
+        if opts.mime
+          tmp.print %Q[
+            |MIME-Version: 1.0
+            |Content-Type: multipart/signed; micalg=pgp-sha1;
+            | protocol="application/pgp-signature";
+            | boundary="#{id}"
+            |Content-Transfer-Encoding: 8bit
+            |
+            |This is an OpenPGP/MIME signed message (RFC 2440 and 3156)
+            |--#{id}
+            |].head_cut!
+
+          tmp.print body.read
+
+          tmp.print %Q[
+            |
+            |--#{id}
+            |Content-Type: application/pgp-signature; name="signature.asc"
+            |Content-Description: OpenPGP digital signature
+            |Content-Disposition: attachment; filename="signature.asc"
+            |
+            |].head_cut!
+        end
+    
+        TempPath.new do |tmp_out|
+          pid = fork do
+            cmd = %w[ gpg --status-fd 2 --sign --textmode --armor
+                          --digest-algo sha1 --no-use-agent -u ]
+            cmd << "<#{opts.from}>"
+            cmd << ((opts.mime)? '--detach-sign' : '--clearsign')
+            cmd += [ '--output', tmp_out ]
+            if opts.pass
+              cmd += [ '--no-tty', '--batch',
+                       '--passphrase-fd', opts.pass.open.to_i.to_s ]
+            end
+            cmd += [ '--comment', "'#{opts.comment}'" ] if opts.comment
+            STDIN.reopen(body.open)
+            exec(*cmd)
+          end
+          Process.waitpid pid
+
+          tmp.print tmp_out.read
+          tmp.puts "\n--#{id}--" if opts.mime
+        end
+      end
+
+      return tmp_name.read
+    end
+  end
+
+end # class Sendmail
+
+
+if $0 == __FILE__
+  $: << Pathname.new(__FILE__).dirname.to_s
+  Sendmail.new.sendmail(*ARGV)
+end

data/src/service_manager.rb

@@ -0,0 +1,122 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: service_manager.rb 266 2005-06-01 14:27:18Z ertai $
+
+require 'ruby_ex'
+require 'drb/drb'
+
+# A service must implement at least:
+#   - start_service(uri, *args, &block)
+#   - stop_service
+#   - server
+#   - thread
+#   - uri
+module ServiceManager
+
+  def start_service(service, uri=nil, *args, &block)
+    @services = {} unless defined? @services
+    service.start_service(uri, *args, &block)
+    @services[service.uri] = service
+  end
+
+  def stop_service(uri)
+    if defined? @services
+      if service = @services.delete(uri)
+        service.stop_service
+      end
+    end
+  end
+
+  def stop_services
+    if defined? @services
+      @services.keys.each do |uri|
+        stop_service(uri)
+      end
+    end
+  end
+
+  def count_services
+    if defined? @services
+      @services.size
+    else
+      0
+    end
+  end
+
+  def list_services
+    if defined? @services
+      @services.values
+    else
+      []
+    end
+  end
+
+  def join_services
+    if defined? @services
+      @services.each { |uri, service| service.thread.join }
+    end
+  end
+
+  def each_services(&block)
+    if defined? @services
+      @services.each { |uri, service| block[uri, service] }
+    end
+  end
+
+  def service_server(uri)
+    @services[uri].server if defined? @services
+  end
+
+end # module ServiceManager
+
+
+test_section __FILE__ do
+
+
+require 'drb'
+
+
+class ServiceManagerTest < Test::Unit::TestCase
+
+  class Foo
+    include ServiceManager
+
+  end
+
+  module MyService
+
+    module_function
+    def start_service(uri); end
+    def stop_service; end
+    def uri; 'toto'; end
+
+  end
+
+  def test_simple
+    f = Foo.new
+    f.start_service(DRb)
+    assert_equal(1, f.count_services)
+    f.start_service(DRb)
+    assert_equal(2, f.count_services)
+    f.start_service(MyService)
+    [ DRb, DRb, MyService ].each do |s|
+      assert(f.list_services.include?(s))
+    end
+    assert_equal(3, f.count_services)
+    f.stop_service(MyService.uri)
+    assert_equal(2, f.count_services)
+    f.start_service(MyService)
+    [ DRb, DRb, MyService ].each do |s|
+      assert(f.list_services.include?(s))
+    end
+    f.stop_services
+    assert_equal([], f.list_services)
+  end
+
+end # class ServiceManagerTest
+
+end
+

data/src/shuffle.rb

@@ -0,0 +1,30 @@
+# Author::    Nicolas Pouillard  <ertai@lrde.epita.fr>.
+# Copyright:: Copyright (c) 2005 Nicolas Pouillard. All rights reserved.
+# License::   GNU General Public License (GPL).
+# Revision::  $Id: shuffle.rb 220 2005-05-09 12:28:26Z ertai $
+
+class Array
+
+  def shuffle! ( generator=nil )
+    n = size
+    n.times do |i|
+      swap!(i, n.choose(generator))
+    end
+    self
+  end
+
+  def shuffle ( generator=nil )
+    dup.shuffle!(generator)
+  end
+
+  def swap! ( i, j )
+    self[i], self[j] = self[j], self[i]
+  end
+
+  def swap ( i, j )
+    dup.swap!(i, j)
+  end
+
+end # class Array
+
+

data/src/spring.rb

@@ -0,0 +1,134 @@
+# Copyright: Copyright (c) 2004  Nicolas Despres. All rights reserved.
+# Author: Nicolas Despres  <polrop@lrde.epita.fr>.
+# License: Gnu General Public License.
+
+# $LastChangedBy: ertai $
+# $Id: spring.rb 266 2005-06-01 14:27:18Z ertai $
+
+
+require 'ruby_ex'
+require 'thread'
+require 'observer'
+
+
+class Spring
+
+  include Observable
+
+  GET = :get
+  SIGN_UP = :sign_up
+
+  def initialize(*items)
+    @q = *items
+    @mutex = Mutex.new
+    @collecter = []
+  end
+
+  def get(collecter_id)
+    @mutex.synchronize do
+      if @collecter.include?(collecter_id)
+        item = @q.shift
+        changed(true)
+        notify_observers(self.object_id, GET, collecter_id)
+        return item
+      end
+    end
+  end
+
+  def size
+    @mutex.synchronize { return @q.size }
+  end
+  alias length size
+
+  def empty?
+    @mutex.synchronize { return @q.empty? }
+  end
+
+  def sign_up(collecter_id)
+    @mutex.synchronize do
+      unless @collecter.include?(collecter_id)
+        @collecter << collecter_id
+        changed(true)
+        notify_observers(self.object_id, SIGN_UP, collecter_id)
+        return nil
+      end
+    end
+  end
+
+  def num_collecter
+    @mutex.synchronize { @collecter.length }
+  end
+
+end # class Spring
+
+
+#
+# Unit test suite
+#
+test_section __FILE__ do
+
+
+class SpringTest < Test::Unit::TestCase
+
+  class Obs
+
+    attr_reader :updated, :msg_seq
+
+    def initialize(s_id, msg_seq, t_id, spring_test_instance)
+      @s_id = s_id
+      @msg_seq = msg_seq
+      @t_id = t_id
+      @updated = false
+      @spring_test_instance = spring_test_instance
+    end
+
+    def update(s_id, msg, t_id)
+      @updated = true
+      @spring_test_instance.test_update(@s_id,
+                                        @msg_seq,
+                                        @t_id,
+                                        s_id,
+                                        msg,
+                                        t_id)
+    end
+
+  end
+
+  #
+  # tests
+  #
+  def test_simple
+    s = Spring.new('hello', 'world')
+    q = Queue.new
+    thread = Thread.new do
+      s.add_observer(q.pop)
+      assert_equal(2, s.size)
+      assert_equal(2, s.length)
+      assert_nil(s.get(thread.object_id))
+      s.sign_up(thread.object_id)
+      assert_equal(1, s.num_collecter)
+      assert_equal('hello', s.get(thread.object_id))
+      assert_equal('world', s.get(thread.object_id))
+    end
+    o = Obs.new(s.object_id,
+                [Spring::SIGN_UP, Spring::GET, Spring::GET],
+                thread.object_id,
+                self)
+    q.push(o)
+    thread.join
+    assert_equal(true, o.updated, 'not visited at all')
+    assert_equal([], o.msg_seq)
+  end
+
+  # tools
+
+  def test_update(r_s_id, r_msg, r_t_id, s_id, msg, t_id)
+    assert_equal(r_t_id, t_id, 'bad collecter id')
+    assert_equal(r_s_id, s_id, 'bad spring id')
+    assert_equal(r_msg.shift, msg, 'bad message')
+  end
+
+end # class SpringTest
+
+
+end