faye-tls1-websocket 0.8.0

data/examples/app.rb

@@ -0,0 +1,53 @@
+require 'faye/websocket'
+require 'rack'
+
+static = Rack::File.new(File.dirname(__FILE__))
+
+App = lambda do |env|
+  if Faye::WebSocket.websocket?(env)
+    ws = Faye::WebSocket.new(env, ['irc', 'xmpp'], :ping => 5)
+    p [:open, ws.url, ws.version, ws.protocol]
+
+    ws.onmessage = lambda do |event|
+      ws.send(event.data)
+    end
+
+    ws.onclose = lambda do |event|
+      p [:close, event.code, event.reason]
+      ws = nil
+    end
+
+    ws.rack_response
+
+  elsif Faye::EventSource.eventsource?(env)
+    es   = Faye::EventSource.new(env)
+    time = es.last_event_id.to_i
+
+    p [:open, es.url, es.last_event_id]
+
+    loop = EM.add_periodic_timer(2) do
+      time += 1
+      es.send("Time: #{time}")
+      EM.add_timer(1) do
+        es.send('Update!!', :event => 'update', :id => time) if es
+      end
+    end
+
+    es.send("Welcome!\n\nThis is an EventSource server.")
+
+    es.onclose = lambda do |event|
+      EM.cancel_timer(loop)
+      p [:close, es.url]
+      es = nil
+    end
+
+    es.rack_response
+
+  else
+    static.call(env)
+  end
+end
+
+def App.log(message)
+end
+

data/examples/autobahn_client.rb

@@ -0,0 +1,51 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'faye/websocket'
+require 'cgi'
+require 'progressbar'
+
+EM.run {
+  host  = 'ws://localhost:9001'
+  ruby  = RUBY_PLATFORM =~ /java/ ? 'JRuby' : 'MRI'
+  agent = "#{ruby} #{RUBY_VERSION}"
+  cases = 0
+  skip  = []
+
+  socket   = Faye::WebSocket::Client.new("#{host}/getCaseCount")
+  progress = nil
+
+  socket.onmessage = lambda do |event|
+    puts "Total cases to run: #{event.data}"
+    cases = event.data.to_i
+    progress = ProgressBar.new('Autobahn', cases)
+  end
+
+  socket.onclose = lambda do |event|
+    run_case = lambda do |n|
+      progress.inc
+
+      if n > cases
+        socket = Faye::WebSocket::Client.new("#{host}/updateReports?agent=#{CGI.escape agent}")
+        progress.finish
+        socket.onclose = lambda { |e| EM.stop }
+
+      elsif skip.include?(n)
+        EM.next_tick { run_case.call(n+1) }
+
+      else
+        socket = Faye::WebSocket::Client.new("#{host}/runCase?case=#{n}&agent=#{CGI.escape agent}")
+
+        socket.onmessage = lambda do |event|
+          socket.send(event.data)
+        end
+
+        socket.on :close do |event|
+          run_case.call(n + 1)
+        end
+      end
+    end
+
+    run_case.call(1)
+  end
+}
+

data/examples/client.rb

@@ -0,0 +1,32 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'faye/websocket'
+require 'eventmachine'
+
+port   = ARGV[0] || 7000
+secure = ARGV[1] == 'ssl'
+
+EM.run {
+  scheme  = secure ? 'wss' : 'ws'
+  url     = "#{scheme}://localhost:#{port}/"
+  headers = {'Origin' => 'http://faye.jcoglan.com'}
+  ws      = Faye::WebSocket::Client.new(url, nil, :headers => headers)
+
+  puts "Connecting to #{ws.url}"
+
+  ws.onopen = lambda do |event|
+    p [:open]
+    ws.send("Hello, WebSocket!")
+  end
+
+  ws.onmessage = lambda do |event|
+    p [:message, event.data]
+    # ws.close 1002, 'Going away'
+  end
+
+  ws.onclose = lambda do |event|
+    p [:close, event.code, event.reason]
+    EM.stop
+  end
+}
+

data/examples/config.ru

@@ -0,0 +1,14 @@
+# Run using your favourite server:
+#
+#     thin start -R examples/config.ru -p 7000
+#     rainbows -c examples/rainbows.conf -E production examples/config.ru -p 7000
+
+require 'rubygems'
+require 'bundler/setup'
+require File.expand_path('../app', __FILE__)
+
+Faye::WebSocket.load_adapter('thin')
+Faye::WebSocket.load_adapter('rainbows')
+
+run App
+

data/examples/haproxy.conf

@@ -0,0 +1,21 @@
+defaults
+   mode http
+   timeout client  5s
+   timeout connect 5s
+   timeout server  5s
+
+frontend all 0.0.0.0:3000
+   mode http
+   timeout client 120s
+
+   option forwardfor
+   option http-server-close
+   option http-pretend-keepalive
+
+   default_backend sockets
+
+backend sockets
+   balance uri depth 2
+   timeout server  120s
+   server socket1 127.0.0.1:7000
+

data/examples/rainbows.conf

@@ -0,0 +1,3 @@
+Rainbows! do
+  use :EventMachine
+end

data/examples/server.rb

@@ -0,0 +1,53 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'rack/content_length'
+require 'rack/chunked'
+
+port   = ARGV[0] || 7000
+secure = ARGV[1] == 'ssl'
+engine = ARGV[2] || 'thin'
+spec   = File.expand_path('../../spec', __FILE__)
+
+require File.expand_path('../app', __FILE__)
+Faye::WebSocket.load_adapter(engine)
+
+case engine
+
+when 'goliath'
+  class WebSocketServer < Goliath::API
+    def response(env)
+      App.call(env)
+    end
+  end
+
+when 'puma'
+  events = Puma::Events.new($stdout, $stderr)
+  binder = Puma::Binder.new(events)
+  binder.parse(["tcp://0.0.0.0:#{port}"], App)
+  server = Puma::Server.new(App, events)
+  server.binder = binder
+  server.run.join
+
+when 'rainbows'
+  rackup = Unicorn::Configurator::RACKUP
+  rackup[:port] = port
+  rackup[:set_listener] = true
+  options = rackup[:options]
+  options[:config_file] = File.expand_path('../rainbows.conf', __FILE__)
+  Rainbows::HttpServer.new(App, options).start.join
+
+when 'thin'
+  EM.run {
+    thin = Rack::Handler.get('thin')
+    thin.run(App, :Port => port) do |server|
+      if secure
+        server.ssl_options = {
+          :private_key_file => spec + '/server.key',
+          :cert_chain_file  => spec + '/server.crt'
+        }
+        server.ssl = true
+      end
+    end
+  }
+end
+

data/examples/sse.html

@@ -0,0 +1,39 @@
+<!doctype html>
+<html>
+  <head>
+    <meta http-equiv="Content-type" content="text/html; charset=utf-8">
+    <title>EventSource test</title>
+  </head>
+  <body>
+
+    <h1>EventSource test</h1>
+    <ul></ul>
+
+    <script type="text/javascript">
+      var logger = document.getElementsByTagName('ul')[0],
+          socket = new EventSource('/');
+
+      var log = function(text) {
+        logger.innerHTML += '<li>' + text + '</li>';
+      };
+
+      socket.onopen = function() {
+        log('OPEN');
+      };
+
+      socket.onmessage = function(event) {
+        log('MESSAGE: ' + event.data);
+      };
+
+      socket.addEventListener('update', function(event) {
+        log('UPDATE(' + event.lastEventId + '): ' + event.data);
+      });
+
+      socket.onerror = function(event) {
+        log('ERROR: ' + event.message);
+      };
+    </script>
+
+  </body>
+</html>
+

data/examples/ws.html

@@ -0,0 +1,44 @@
+<!doctype html>
+<html>
+  <head>
+    <meta http-equiv="Content-type" content="text/html; charset=utf-8">
+    <title>WebSocket test</title>
+  </head>
+  <body>
+
+    <h1>WebSocket test</h1>
+    <ul></ul>
+
+    <script type="text/javascript">
+      var logger = document.getElementsByTagName('ul')[0],
+          Socket = window.MozWebSocket || window.WebSocket,
+          protos = ['foo', 'bar', 'xmpp'],
+          socket = new Socket('ws://' + location.hostname + ':' + location.port + '/', protos),
+          index  = 0;
+
+      var log = function(text) {
+        logger.innerHTML += '<li>' + text + '</li>';
+      };
+
+      socket.addEventListener('open', function() {
+        log('OPEN: ' + socket.protocol);
+        socket.send('Hello, world');
+      });
+
+      socket.onerror = function(event) {
+        log('ERROR: ' + event.message);
+      };
+
+      socket.onmessage = function(event) {
+        log('MESSAGE: ' + event.data);
+        setTimeout(function() { socket.send(++index + ' ' + event.data) }, 2000);
+      };
+
+      socket.onclose = function(event) {
+        log('CLOSE: ' + event.code + ', ' + event.reason);
+      };
+    </script>
+
+  </body>
+</html>
+

data/lib/faye/adapters/goliath.rb

@@ -0,0 +1,47 @@
+class Goliath::Connection
+  attr_accessor :socket_stream
+  alias :goliath_receive_data :receive_data
+
+  def receive_data(data)
+    if @serving == :websocket
+      socket_stream.receive(data) if socket_stream
+    else
+      goliath_receive_data(data)
+      socket_stream.receive(@parser.upgrade_data) if socket_stream
+      @serving = :websocket if @api.websocket?
+    end
+  end
+
+  def unbind
+    super
+  ensure
+    socket_stream.fail if socket_stream
+  end
+end
+
+class Goliath::API
+  include Faye::WebSocket::Adapter
+end
+
+class Goliath::Request
+  alias :goliath_process :process
+
+  def process
+    env['em.connection'] = conn
+    goliath_process
+  end
+end
+
+class Goliath::Response
+  alias :goliath_head :head
+  alias :goliath_headers_output :headers_output
+
+  def head
+    (status == 101) ? '' : goliath_head
+  end
+
+  def headers_output
+    (status == 101) ? '' : goliath_headers_output
+  end
+end
+

data/lib/faye/adapters/rainbows.rb

@@ -0,0 +1,31 @@
+# WebSocket extensions for Rainbows
+# Based on code from the Cramp project
+# http://github.com/lifo/cramp
+
+# Copyright (c) 2009-2011 Pratik Naik
+#
+# 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.
+
+module Faye
+  class WebSocket
+    autoload :RainbowsClient, File.expand_path('../rainbows_client', __FILE__)
+  end
+end
+
+Rainbows::O[:em_client_class] = "Faye::WebSocket::RainbowsClient"

data/lib/faye/adapters/rainbows_client.rb

@@ -0,0 +1,69 @@
+# WebSocket extensions for Rainbows
+# Based on code from the Cramp project
+# http://github.com/lifo/cramp
+
+# Copyright (c) 2009-2011 Pratik Naik
+#
+# 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.
+
+module Faye
+  class WebSocket
+
+    class RainbowsClient < Rainbows::EventMachine::Client
+      include Faye::WebSocket::Adapter
+      attr_accessor :socket_stream
+
+      def receive_data(data)
+        return super unless @state == :websocket
+        socket_stream.receive(data) if socket_stream
+      end
+
+      def app_call(*args)
+        @env['em.connection'] = self
+        if args.first == NULL_IO and @hp.content_length == 0 and websocket?
+          prepare_request_body
+        else
+          super
+        end
+      end
+
+      def on_read(data)
+        if @state == :body and websocket? and @hp.body_eof?
+          @state = :websocket
+          @input.rewind
+          app_call StringIO.new(@buf)
+        else
+          super
+        end
+      end
+
+      def unbind
+        super
+      ensure
+        socket_stream.fail if socket_stream
+      end
+
+      def write_headers(status, headers, *args)
+        super unless socket_connection? and status == 101
+      end
+    end
+
+  end
+end
+