eventmachine-eventmachine 0.12.3

data/lib/protocols/stomp.rb

@@ -0,0 +1,130 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 15 November 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) 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; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+# 
+# 
+
+
+
+module EventMachine
+	module Protocols
+
+		# Implements Stomp (http://docs.codehaus.org/display/STOMP/Protocol).
+		#
+		module Stomp
+			include LineText2
+
+			class Message
+				attr_accessor :command, :header, :body
+				def initialize
+					@header = {}
+					@state = :precommand
+					@content_length = nil
+				end
+				def consume_line line
+					if @state == :precommand
+						unless line =~ /\A\s*\Z/
+							@command = line
+							@state = :headers
+						end
+					elsif @state == :headers
+						if line == ""
+							if @content_length
+								yield( [:sized_text, @content_length+1] )
+							else
+								@state = :body
+								yield( [:unsized_text] )
+							end
+						elsif line =~ /\A([^:]+):(.+)\Z/
+							k = $1.dup.strip
+							v = $2.dup.strip
+							@header[k] = v
+							if k == "content-length"
+								@content_length = v.to_i
+							end
+						else
+							# This is a protocol error. How to signal it?
+						end
+					elsif @state == :body
+						@body = line
+						yield( [:dispatch] )
+					end
+				end
+			end
+
+			def send_frame verb, headers={}, body=""
+				ary = [verb, "\n"]
+				headers.each {|k,v| ary << "#{k}:#{v}\n" }
+				ary << "content-length: #{body.to_s.length}\n"
+				ary << "content-type: text/plain; charset=UTF-8\n"
+				ary << "\n"
+				ary << body.to_s
+				ary << "\0"
+				send_data ary.join
+			end
+
+			def receive_line line
+				@stomp_initialized || init_message_reader
+				@stomp_message.consume_line(line) {|outcome|
+					if outcome.first == :sized_text
+						set_text_mode outcome[1]
+					elsif outcome.first == :unsized_text
+						set_delimiter "\0"
+					elsif outcome.first == :dispatch
+						receive_msg(@stomp_message) if respond_to?(:receive_msg)
+						init_message_reader
+					end
+				}
+			end
+			def receive_binary_data data
+				@stomp_message.body = data[0..-2]
+				receive_msg(@stomp_message) if respond_to?(:receive_msg)
+				init_message_reader
+			end
+
+			def init_message_reader
+				@stomp_initialized = true
+				set_delimiter "\n"
+				set_line_mode
+				@stomp_message = Message.new
+			end
+
+
+			# STOMP verbs
+			def connect parms={}
+				send_frame "CONNECT", parms
+			end
+			def send destination, body, parms={}
+				send_frame "SEND", parms.merge( :destination=>destination ), body.to_s
+			end
+			def subscribe dest, ack=false
+				send_frame "SUBSCRIBE", {:destination=>dest, :ack=>(ack ? "client" : "auto")}
+			end
+			def ack msgid
+				send_frame "ACK", { :'message-id'=> msgid }
+			end
+
+		end
+	end
+end
+

data/lib/protocols/tcptest.rb

@@ -0,0 +1,57 @@
+# $Id$
+#
+# Author:: Francis Cianfrocca (gmail: blackhedd)
+# Homepage::  http://rubyeventmachine.com
+# Date:: 16 July 2006
+# 
+# See EventMachine and EventMachine::Connection for documentation and
+# usage examples.
+#
+#----------------------------------------------------------------------------
+#
+# Copyright (C) 2006-07 by Francis Cianfrocca. All Rights Reserved.
+# Gmail: blackhedd
+# 
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of either: 1) 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; or 2) Ruby's License.
+# 
+# See the file COPYING for complete licensing information.
+#
+#---------------------------------------------------------------------------
+#
+#
+# 
+
+
+
+module EventMachine
+module Protocols
+
+class TcpConnectTester < Connection
+  include EventMachine::Deferrable
+
+  def self.test( host, port )
+    EventMachine.connect( host, port, self )
+  end
+
+  def post_init
+    @start_time = Time.now
+  end
+
+  def connection_completed
+    @completed = true
+    set_deferred_status :succeeded, (Time.now - @start_time)
+    close_connection
+  end
+
+  def unbind
+    set_deferred_status :failed, (Time.now - @start_time)  unless @completed
+  end
+end
+
+
+end
+end
+

data/tasks/cpp.rake

@@ -0,0 +1,77 @@
+# EventMachine C++ Rakefile Stab Case
+# TODO : track header files as a build dependency...
+# TODO : cross platform support
+# TODO : configure style functionality
+namespace :cpp do
+
+  require 'rake/clean'
+
+  # *nix only atm...
+  module Cpp
+    class <<self
+      def cpp; "g++"; end
+      def archive; "ar"; end
+      def compile file, output, includes=nil, flags=nil
+        sh %{#{cpp} #{file} #{includes} #{flags} -c -o #{output}}
+      end
+      def link file, output, libs=nil, flags=nil
+        sh %{#{cpp} #{file} #{libs} #{flags} -o #{output}}
+      end
+      def static output, files
+        sh %{#{archive} cr #{output} #{files}}
+      end
+    end
+  end
+
+  module EmConfig
+    Path = ENV['EVENTMACHINE_SOURCE'] || 'ext'
+    Sources = FileList["#{Path}/*.cpp"]
+    Sources.delete_if { |s| /ruby/ =~ s }
+    Compiled = Sources.sub(%r{^#{Path}/(.*)\.cpp}, "#{Path}/\\1.o")
+
+    Flags = "-O2 -pipe -fno-common -DOS_UNIX -DWITHOUT_SSL"
+    Includes = ""
+    Libs = ''
+  end
+  CLEAN.include(EmConfig::Compiled)
+
+  rule %r{^#{EmConfig::Path}/.*\.o$} => [proc { |targ| 
+    targ.sub(%r{^#{EmConfig::Path}/(.*)\.o$}, "#{EmConfig::Path}/\\1.cpp")
+    }] do |t|
+    Cpp.compile t.source, t.name, EmConfig::Includes, EmConfig::Flags
+  end
+
+  file "#{EmConfig::Path}/libeventmachine.a" => EmConfig::Compiled do |t|
+    Cpp.static t.name, EmConfig::Compiled
+  end
+  CLEAN.include("#{EmConfig::Path}/libeventmachine.a")
+
+  module AppConfig
+    Appname = 'echo_em'
+    Sources = FileList['*.cpp']
+    Compiled = Sources.sub(%r{^(.*)\.cpp}, '\\1.o')
+
+    Flags = ["", EmConfig::Flags].join(' ')
+    Includes = ["-I. -I#{EmConfig::Path}", EmConfig::Includes].join(' ')
+    Libs = ["-L#{EmConfig::Path} -leventmachine", EmConfig::Libs].join(' ')
+  end
+  CLEAN.include(AppConfig::Compiled)
+  CLEAN.include(AppConfig::Appname)
+
+  rule %r{^.*\.o$} => [proc { |targ| 
+    targ.sub(%r{^(.*)\.o$}, '\\1.cpp')
+    }] do |t|
+    Cpp.compile t.source, t.name, AppConfig::Includes, AppConfig::Flags
+  end
+
+  file AppConfig::Appname => ["#{EmConfig::Path}/libeventmachine.a", AppConfig::Compiled] do |t|
+    Cpp.link AppConfig::Compiled, t.name, AppConfig::Libs, AppConfig::Flags
+  end
+
+  task :build => AppConfig::Appname
+
+  task :run => AppConfig::Appname do
+    sh "./#{AppConfig::Appname}"
+  end
+
+end

data/tasks/project.rake

@@ -0,0 +1,78 @@
+require 'rake/gempackagetask'
+require 'rake/rdoctask'
+require 'rake/clean'
+
+# monkey bitchin' for windows stuffs...
+module FileUtils
+  # If any of these methods ever clobber, try removing them.
+  # Hopefully they'll do something semantically similar.
+  abort "Err: #{__FILE__}:#{__LINE__} monkey patch windows? clobbers!" unless instance_methods.grep(/windows\?/).empty?
+  abort "Err: #{__FILE__}:#{__LINE__} monkey patch sudo clobbers!" unless instance_methods.grep(/sudo/).empty?
+  abort "Err: #{__FILE__}:#{__LINE__} monkey patch gem_cmd clobbers!" unless instance_methods.grep(/gem_cmd/).empty?
+  def windows?; RUBY_PLATFORM =~ /mswin|mingw/; end
+  def sudo(cmd)
+    if windows? || (require 'etc'; Etc.getpwuid.uid == 0)
+      sh cmd
+    else
+      sh "sudo #{cmd}"
+    end
+  end
+  def gem_cmd(action, name, *args)
+    rb = Gem.ruby rescue nil
+    rb ||= (require 'rbconfig'; File.join(Config::CONFIG['bindir'], Config::CONFIG['ruby_install_name']))
+    sudo "#{rb} -r rubygems -e 'require %{rubygems/gem_runner}; Gem::GemRunner.new.run(%w{#{action} #{name} #{args.join(' ')}})'"
+  end
+end
+
+
+# Setup our packaging tasks, we're just jacking the builtins.
+Rake::GemPackageTask.new(Spec) do |pkg|
+  pkg.need_tar, pkg.need_tar_gz, pkg.need_zip = true, true, true if Package
+  pkg.gem_spec = Spec
+end
+Rake::Task[:clean].enhance [:clobber_package]
+
+# Only generate rdoc if the spec says so, again, jack the builtins.
+if Spec.has_rdoc
+  Rake::RDocTask.new do |rd|
+    rd.title = Spec.name
+    rd.rdoc_dir = 'rdoc'
+    rd.main = "docs/README" if test ?e, "docs/README"
+    rd.rdoc_files.include("lib/**/*.rb", *Spec.extra_rdoc_files)
+  end
+  Rake::Task[:clean].enhance [:clobber_rdoc]
+
+  desc 'Generate and open documentation'
+  task :docs => :rdoc do
+    case RUBY_PLATFORM
+    when /darwin/       ; sh 'open rdoc/index.html'
+    when /mswin|mingw/  ; sh 'start rdoc\index.html'
+    else 
+      sh 'firefox rdoc/index.html'
+    end
+  end
+end
+
+if Spec.default_executable
+  desc "Run #{Spec.default_executable}"
+  task :run do ruby File.join(Spec.bindir, Spec.default_executable) end
+end
+
+require 'rubygems'
+
+desc 'Install gem (and sudo if required)'
+task :install => :package do 
+  gem_cmd(:install, "pkg/#{Spec.name}-#{Spec.version}.gem")
+end
+
+desc 'Uninstall gem (and sudo if required)'
+task :uninstall do
+  gem_cmd(:uninstall, "#{Spec.name}", "-v=#{Spec.version}")
+end
+
+# Find an scm's store directory, if we do, make a task to commit to it only
+# after running all the tests (successfully).
+if scm = %w(git svn bzr hg).find { |d| File.directory? ".#{d}" }
+  desc "Run tests then commit to #{scm}"
+  task :commit => :test do sh "#{scm} commit" end
+end

data/tasks/tests.rake

@@ -0,0 +1,192 @@
+# This is used by several rake tasks, that parameterize the
+# behavior so we can use the same tests to test both the
+# extension and non-extension versions.
+def run_tests t, libr = :cascade, test_files="test_*.rb"
+  require 'test/unit/testsuite'
+  require 'test/unit/ui/console/testrunner'
+  require 'tests/testem'
+  
+  base_dir = File.expand_path(File.dirname(__FILE__) + '/../') + '/'
+  
+  runner = Test::Unit::UI::Console::TestRunner
+  
+  $eventmachine_library = libr
+  EmTestRunner.run(test_files)
+  
+  suite = Test::Unit::TestSuite.new($name)
+
+  ObjectSpace.each_object(Class) do |testcase|
+    suite << testcase.suite if testcase < Test::Unit::TestCase
+  end
+
+  runner.run(suite)
+end
+
+desc "Run tests for #{Spec.name}."
+task :test do |t|
+  # run_tests t
+  # Rake +/ friends leave threads, etc, less stable test runs.
+  ruby "-Ilib -Iext -Ijava tests/testem.rb #{'-v' if ENV['VERBOSE']}"
+end
+
+namespace :test do
+  desc "Run tests for #{Spec.name}."
+  task :partial do |t|
+    lib = RUBY_PLATFORM =~ /java/ ? :java : :extension
+    run_tests t, lib, [
+      "test_basic.rb",
+      "test_epoll.rb",
+      "test_errors.rb",
+      "test_eventables.rb",
+      "test_exc.rb",
+      "test_futures.rb",
+      "test_hc.rb",
+      "test_httpclient2.rb",
+      "test_httpclient.rb",
+      "test_kb.rb",
+      "test_ltp2.rb",
+      "test_ltp.rb",
+      "test_next_tick.rb",
+      "test_processes.rb",
+      "test_pure.rb",
+      "test_running.rb",
+      "test_sasl.rb",
+      #"test_send_file.rb",
+      "test_servers.rb",
+      "test_smtpclient.rb",
+      "test_smtpserver.rb",
+      "test_spawn.rb",
+      "test_timers.rb",
+      "test_ud.rb",
+    ].map { |tf| "tests/#{tf}" }
+  end
+  
+  desc "Run java tests for #$name."
+  task :testjava do |t|
+    run_tests t, :java
+  end
+
+  desc "Run pure-ruby tests for #$name."
+  task :testpr do |t|
+    run_tests t, :pure_ruby
+  end
+
+  desc "Run extension tests for #$name."
+  task :testext do |t|
+    run_tests t, :extension
+  end
+
+  desc "PROVISIONAL: run tests for user-defined events"
+  task :ud do |t|
+    run_tests t, :extension, "test_ud.rb"
+  end
+
+  desc "PROVISIONAL: run tests for line/text protocol handler"
+  task :ltp do |t|
+    run_tests t, :extension, "test_ltp*.rb"
+  end
+
+  desc "PROVISIONAL: run tests for header/content protocol handler"
+  task :hc do |t|
+    run_tests t, :extension, "test_hc.rb"
+  end
+
+  desc "PROVISIONAL: run tests for exceptions"
+  task :exc do |t|
+    run_tests t, :extension, "test_exc.rb"
+  end
+
+  desc "Test protocol handlers"
+  task :protocols => [ :hc, :ltp ]
+
+
+  desc "Test HTTP client"
+  task :httpclient do |t|
+    run_tests t, :extension, "test_httpclient.rb"
+  end
+
+  desc "Test HTTP client2"
+  task :httpclient2 do |t|
+    run_tests t, :extension, "test_httpclient2.rb"
+  end
+
+  desc "Test futures"
+  task :futures do |t|
+    run_tests t, :extension, "test_future*.rb"
+  end
+
+  desc "Test Timers"
+  task :timers do |t|
+    run_tests t, :extension, "test_timer*.rb"
+  end
+
+  desc "Test Next Tick"
+  task :next_tick do |t|
+    run_tests t, :extension, "test_next_tick*.rb"
+  end
+
+  desc "Test Epoll"
+  task :epoll do |t|
+    run_tests t, :extension, "test_epoll*.rb"
+  end
+
+  desc "Test Servers"
+  task :servers do |t|
+    run_tests t, :extension, "test_servers*.rb"
+  end
+
+  desc "Test Basic"
+  task :basic do |t|
+    run_tests t, :extension, "test_basic*.rb"
+  end
+
+  desc "Test Send File"
+  task :send_file do |t|
+    run_tests t, :extension, "test_send_file*.rb"
+  end
+
+  desc "Test Running"
+  task :running do |t|
+    run_tests t, :extension, "test_running*.rb"
+  end
+
+  desc "Test Keyboard Events"
+  task :keyboard do |t|
+    run_tests t, :extension, "test_kb*.rb"
+  end
+
+  desc "Test Spawn"
+  task :spawn do |t|
+    run_tests t, :extension, "test_spawn*.rb"
+  end
+
+  desc "Test SMTP"
+  task :smtp do |t|
+    run_tests t, :extension, "test_smtp*.rb"
+  end
+
+  desc "Test Errors"
+  task :errors do |t|
+    run_tests t, :extension, "test_errors*.rb"
+  end
+
+  desc "Test Pure Ruby"
+  task :pure do |t|
+    run_tests t, :extension, "test_pure*.rb"
+  end
+
+  desc "Test Processes"
+  task :processes do |t|
+    run_tests t, :extension, "test_process*.rb"
+  end
+
+  desc "Test SASL"
+  task :sasl do |t|
+    run_tests t, :java, "test_sasl*.rb"
+  end
+  
+  desc "Test Attach"
+  task :attach do |t|
+    run_tests t, :extension, "test_attach*.rb"
+  end
+end