em-websocket-proxy 0.1.1

data/README.rdoc

@@ -0,0 +1,20 @@
+= EventMachine Websocket Proxy
+
+Don't want to reinvent the wheel to use a websocket? Don't. Use your existing tcp servers, with this simple proxy.
+
+== Getting Started
+
+em-websocket-proxy -p 8080 -r localhost -q 80
+
+== License
+
+(The MIT License)
+
+Copyright © 2010 Matt Colyer
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the ‘Software’), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED ‘AS IS’, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 
+

data/Rakefile

@@ -0,0 +1,145 @@
+require 'rubygems'
+require 'rake'
+require 'date'
+
+#############################################################################
+#
+# Helper functions
+#
+#############################################################################
+
+def name
+  @name ||= Dir['*.gemspec'].first.split('.').first
+end
+
+def version
+  line = File.read("lib/#{name}.rb")[/^\s*VERSION\s*=\s*.*/]
+  line.match(/.*VERSION\s*=\s*['"](.*)['"]/)[1]
+end
+
+def date
+  Date.today.to_s
+end
+
+def rubyforge_project
+  name
+end
+
+def gemspec_file
+  "#{name}.gemspec"
+end
+
+def gem_file
+  "#{name}-#{version}.gem"
+end
+
+def replace_header(head, header_name)
+  head.sub!(/(\.#{header_name}\s*= ').*'/) { "#{$1}#{send(header_name)}'"}
+end
+
+#############################################################################
+#
+# Standard tasks
+#
+#############################################################################
+
+task :default => :test
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+desc "Generate RCov test coverage and open in your browser"
+task :coverage do
+  require 'rcov'
+  sh "rm -fr coverage"
+  sh "rcov test/test_*.rb"
+  sh "open coverage/index.html"
+end
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "#{name} #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+desc "Open an irb session preloaded with this library"
+task :console do
+  sh "irb -rubygems -r ./lib/#{name}.rb"
+end
+
+#############################################################################
+#
+# Custom tasks (add your own tasks here)
+#
+#############################################################################
+
+
+
+#############################################################################
+#
+# Packaging tasks
+#
+#############################################################################
+
+task :release => :build do
+  unless `git branch` =~ /^\* master$/
+    puts "You must be on the master branch to release!"
+    exit!
+  end
+  sh "git commit --allow-empty -a -m 'Release #{version}'"
+  sh "git tag v#{version}"
+  sh "git push --tags origin master"
+  sh "gem push pkg/#{name}-#{version}.gem"
+end
+
+task :build => :gemspec do
+  sh "mkdir -p pkg"
+  sh "gem build #{gemspec_file}"
+  sh "mv #{gem_file} pkg"
+end
+
+task :gemspec => :validate do
+  # read spec file and split out manifest section
+  spec = File.read(gemspec_file)
+  head, manifest, tail = spec.split("  # = MANIFEST =\n")
+
+  # replace name version and date
+  replace_header(head, :name)
+  replace_header(head, :version)
+  replace_header(head, :date)
+  #comment this out if your rubyforge_project has a different name
+  replace_header(head, :rubyforge_project)
+
+  # determine file list from git ls-files
+  files = `git ls-files`.
+    split("\n").
+    sort.
+    reject { |file| file =~ /^\./ }.
+    reject { |file| file =~ /^(rdoc|pkg)/ }.
+    map { |file| "    #{file}" }.
+    join("\n")
+
+  # piece file back together and write
+  manifest = "  s.files = %w[\n#{files}\n  ]\n"
+  spec = [head, manifest, tail].join("  # = MANIFEST =\n")
+  File.open(gemspec_file, 'w') { |io| io.write(spec) }
+  puts "Updated #{gemspec_file}"
+end
+
+task :validate do
+  libfiles = Dir['lib/*'] - ["lib/#{name}.rb", "lib/#{name}"]
+  unless libfiles.empty?
+    puts "Directory `lib` should only contain a `#{name}.rb` file and `#{name}` dir."
+    exit!
+  end
+  unless Dir['VERSION*'].empty?
+    puts "A `VERSION` file at root level violates Gem best practices."
+    exit!
+  end
+end

data/bin/em-websocket-proxy

@@ -0,0 +1,77 @@
+#!/usr/bin/env ruby
+require 'optparse'
+
+require 'rubygems'
+require 'eventmachine'
+require 'em-websocket'
+
+options = {}
+OptionParser.new do |opts|
+  opts.banner = "Usage: em-websocket-proxy [options]"
+
+  opts.on("-p", "--port [PORT]", "Port to listen for Websocket connections") do |port|
+    options[:port] = port
+  end
+
+  opts.on("-r", "--remote-host [HOST]", "Host to proxy to") do |host|
+    options[:remote_host] = host
+  end
+
+  opts.on("-q", "--remote-port [PORT]", "Port to proxy to") do |port|
+    options[:remote_port] = port
+  end
+end.parse!
+
+required_opts = [:port, :remote_host, :remote_port]
+required_opts.each do |opt|
+  unless options.has_key? opt
+    puts "Required option --#{opt.to_s.gsub("_", "-")} missing. Use -h for details."
+    exit(-1)
+  end
+end
+
+EventMachine.run {
+
+  class Server < EventMachine::Connection
+      def initialize(input, output, server_close, client_close)
+        @input = input
+        @output = output
+        @server_close = server_close
+        @client_close = client_close
+
+        @input_sid = @input.subscribe { |msg| send_data msg }
+        @client_close_sid = @client_close.subscribe { |msg| close_connection }
+      end
+
+      def receive_data(data)
+        @output.push(data)
+      end
+
+      def unbind
+        @server_close.push("exit")
+        @input.unsubscribe(@input_sid)
+        @client_close.unsubscribe(@client_close_sid)
+      end
+  end
+
+  EventMachine::WebSocket.start(:host => "0.0.0.0", :port => options[:port]) do |ws|
+    ws.onopen {
+      output = EM::Channel.new
+      input = EM::Channel.new
+      server_close = EM::Channel.new
+      client_close = EM::Channel.new
+
+      output_sid = output.subscribe { |msg| ws.send msg }
+      server_close_sid = server_close.subscribe { |msg| ws.close_connection }
+      EventMachine::connect options[:remote_host], options[:remote_port], Server, input, output, server_close, client_close
+
+      ws.onmessage { |msg| input.push(msg)}
+
+      ws.onclose {
+        output.unsubscribe(output_sid)
+        server_close.unsubscribe(server_close_sid)
+        client_close.push("exit")
+      }
+    }
+  end
+}

data/em-websocket-proxy.gemspec

@@ -0,0 +1,74 @@
+## This is the rakegem gemspec template. Make sure you read and understand
+## all of the comments. Some sections require modification, and others can
+## be deleted if you don't need them. Once you understand the contents of
+## this file, feel free to delete any comments that begin with two hash marks.
+## You can find comprehensive Gem::Specification documentation, at
+## http://docs.rubygems.org/read/chapter/20
+Gem::Specification.new do |s|
+  s.specification_version = 2 if s.respond_to? :specification_version=
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.rubygems_version = '1.3.5'
+
+  ## Leave these as is they will be modified for you by the rake gemspec task.
+  ## If your rubyforge_project name is different, then edit it and comment out
+  ## the sub! line in the Rakefile
+  s.name              = 'em-websocket-proxy'
+  s.version           = '0.1.1'
+  s.date              = '2010-07-08'
+  s.rubyforge_project = 'em-websocket-proxy'
+
+  ## Make sure your summary is short. The description may be as long
+  ## as you like.
+  s.summary     = "A proxy for using websockets with unmodified servers."
+  s.description = "A lightweight proxy, for using websockets with unmodified TCP servers."
+
+  ## List the primary authors. If there are a bunch of authors, it's probably
+  ## better to set the email to an email list or something. If you don't have
+  ## a custom homepage, consider using your GitHub URL or the like.
+  s.authors  = ["Matt Colyer"]
+  s.email    = 'matt@nospam.colyer.name'
+  s.homepage = 'http://matt.colyer.name'
+
+  ## This gets added to the $LOAD_PATH so that 'lib/NAME.rb' can be required as
+  ## require 'NAME.rb' or'/lib/NAME/file.rb' can be as require 'NAME/file.rb'
+  s.require_paths = %w[lib]
+
+  ## This sections is only necessary if you have C extensions.
+  #s.require_paths << 'ext'
+  #s.extensions = %w[ext/extconf.rb]
+
+  ## If your gem includes any executables, list them here.
+  s.executables = ["em-websocket-proxy"]
+  s.default_executable = 'em-websocket-proxy'
+
+  ## Specify any RDoc options here. You'll want to add your README and
+  ## LICENSE files to the extra_rdoc_files list.
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.extra_rdoc_files = %w[README.rdoc]
+
+  ## List your runtime dependencies here. Runtime dependencies are those
+  ## that are needed for an end user to actually USE your code.
+  s.add_dependency('eventmachine', [">= 0.12.10"])
+  s.add_dependency('em-websocket', [">= 0.1.2"])
+
+  ## List your development dependencies here. Development dependencies are
+  ## those that are only needed during development
+  #s.add_development_dependency('DEVDEPNAME', [">= 1.1.0", "< 2.0.0"])
+
+  ## Leave this section as-is. It will be automatically generated from the
+  ## contents of your Git repository via the gemspec task. DO NOT REMOVE
+  ## THE MANIFEST COMMENTS, they are used as delimiters by the task.
+  # = MANIFEST =
+  s.files = %w[
+    README.rdoc
+    Rakefile
+    bin/em-websocket-proxy
+    em-websocket-proxy.gemspec
+    lib/em-websocket-proxy.rb
+  ]
+  # = MANIFEST =
+
+  ## Test files will be grabbed from the file list. Make sure the path glob
+  ## matches what you actually use.
+  #s.test_files = s.files.select { |path| path =~ /^test\/test_.*\.rb/ }
+end

data/lib/em-websocket-proxy.rb

@@ -0,0 +1,3 @@
+module EmWebsocketProxy
+  VERSION = '0.1.1'
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification 
+name: em-websocket-proxy
+version: !ruby/object:Gem::Version 
+  version: 0.1.1
+platform: ruby
+authors: 
+- Matt Colyer
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-07-08 00:00:00 -07:00
+default_executable: em-websocket-proxy
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.12.10
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: em-websocket
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.1.2
+    version: 
+description: A lightweight proxy, for using websockets with unmodified TCP servers.
+email: matt@nospam.colyer.name
+executables: 
+- em-websocket-proxy
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- README.rdoc
+- Rakefile
+- bin/em-websocket-proxy
+- em-websocket-proxy.gemspec
+- lib/em-websocket-proxy.rb
+has_rdoc: true
+homepage: http://matt.colyer.name
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: em-websocket-proxy
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 2
+summary: A proxy for using websockets with unmodified servers.
+test_files: []
+