sloth-reel 0.1.0

data/lib/reel/websocket.rb

@@ -0,0 +1,132 @@
+require 'forwardable'
+require 'websocket'
+
+module Reel
+  class WebSocket
+    extend Forwardable
+    include ConnectionMixin
+    include RequestMixin
+
+    attr_reader :socket, :url
+    def_delegators :@socket, :addr, :peeraddr
+
+    READ_SIZE  =  0x4000
+    @@sockets  =  []
+    @@conns  =  {}
+    @@actions  =  Hash.new { |h,k| h[k] = {} }
+    @@thread  =  nil
+
+    def initialize(info, connection)
+      @opened  =  false
+      @socket  =  connection.hijack_socket
+      @request_info  =  info
+      @url  =  @request_info.url
+
+      @handshake  =  ::WebSocket::Handshake::Server.new
+      @frame  =  ::WebSocket::Frame::Incoming::Server.new(:version => @handshake.version)
+
+      line  =  "#{@request_info.method} / HTTP/#{@request_info.version}\r\n"
+      @handshake  <<  line
+      @request_info.headers.each do |k, v|
+        line  =  "#{k}: #{v}\r\n"
+        @handshake  <<  line
+      end
+      @handshake  <<  "\r\n"
+
+      if not @handshake.finished? or not @handshake.valid?
+        @socket.close
+        return
+      end
+
+      lines  =  @handshake.to_s
+      @socket.write  lines
+
+      @@sockets.push( @socket )
+      @@conns[@socket]  =  self
+
+      @@thread  ||=  Thread.start do
+        loop do
+          readables,  =  ::IO.select( @@sockets, nil, nil, 1 )
+          while  sock  =  readables&.shift
+            if sock.eof?
+              @@conns[sock]&.close
+              break
+            else
+              payload  =  sock.readpartial( READ_SIZE )    rescue  nil
+              @@conns[sock]&.parse( payload )    if  payload
+            end
+          end
+        end
+      end
+
+    end
+
+    def parse( payload )
+      @frame  <<  payload
+      messages  =  []
+      while  frame = @frame.next
+        if (frame.type == :close)
+          close
+          break
+        else
+          messages.push  frame.to_s
+        end
+      end
+
+      conns  =  @@conns.values.map do |conn|
+        conn    if  conn.url == @url
+      end.compact
+      messages.each do |mesg|
+        @@actions[@url][:onmessage]&.call( mesg, self, conns )    rescue  nil
+      end
+    end
+
+    def send( mesg, type: :text )
+      frame  =  ::WebSocket::Frame::Outgoing::Server.new( :version => @handshake.version, :data => mesg, :type => type )
+      begin
+        @socket.write  frame.to_s
+        @socket.flush
+      rescue
+        close
+        raise  Reel::SocketError
+      end
+    end
+    alias  write  send
+    alias  :<<    send
+
+    def close
+      return    if  @socket.closed?
+      @@actions[@url][:onclose]&.call( self )    rescue  nil
+      @@conns.delete( @socket )
+      @@sockets.delete( @socket )
+      @@sockets.compact!
+      @socket.close    rescue  nil
+    end
+
+    def closed?
+      @socket.closed?
+    end
+
+    def connections
+      @@conns.values
+    end
+
+    def on_open( &block )
+      @@actions[@url][:onopen]  =  block
+      if  not @opened
+        @opened  =  true
+        block.call( self )    rescue  nil
+      end
+    end
+
+    def on_message( &block )
+      @@actions[@url][:onmessage]  =  block
+    end
+
+    def on_close( &block )
+      @@actions[@url][:onclose]  =  block
+    end
+
+  end
+end
+

data/lib/sloth/reel.rb

@@ -0,0 +1,13 @@
+require "sloth/reel/version"
+require "celluloid/current"
+require "celluloid/autostart"
+require "celluloid/fsm"
+require "reel/rack"
+require "sinatra/base"
+require "sloth/reel/sinatra"
+
+module Sloth
+  module Reel
+    class Error < StandardError; end
+  end
+end

data/lib/sloth/reel/sinatra.rb

@@ -0,0 +1,18 @@
+require "sinatra/base"
+
+module Sloth
+  module Reel
+    module Sinatra
+      def websocket?
+        !!env['websocket']
+      end
+
+      def websocket
+        env['websocket']
+      end
+    end
+  end
+end
+
+defined?( ::Sinatra ) && Sinatra::Request.send( :include, ::Sloth::Reel::Sinatra )
+

data/lib/sloth/reel/version.rb

@@ -0,0 +1,5 @@
+module Sloth
+  module Reel
+    VERSION = "0.1.0"
+  end
+end

data/sample/rack_server_hello_1.rb

@@ -0,0 +1,74 @@
+require 'sloth/reel'
+
+class WebApp
+  def call( env )
+    r  =  nil
+    case env['REQUEST_METHOD']
+    when 'GET'
+      case  env['PATH_INFO']
+      when  '/'
+        get_index
+      when  '/heavy'
+        get_heavy
+      end
+
+    when 'POST'
+      case  env['PATH_INFO']
+      when  '/'
+        post_index
+      end
+
+    else
+      not_found
+
+    end
+  end
+
+  def not_found
+    [
+      404,
+      {'Content-Type' => 'text/html'},
+      ["<html><body><h1> 404: Not found: #{env['REQUEST_METHOD']} #{env['PATH_INFO']} </h1> </body></html>"]
+    ]
+  end
+
+  def get_index
+    [
+      200,
+      {'Content-Type' => 'text/html'},
+      ['<html><body><h1> GET </h1><a href="/"> Hello (GET) </a> <form method="POST"><input type="submit" value="Howdy (POST)" /></form></body></html>']
+    ]
+  end
+
+  def post_index
+    [
+      200,
+      {'Content-Type' => 'text/html'},
+      ['<html><body><h1> POST </h1><a href="/"> Hello (GET) </a> <form method="POST"><input type="submit" value="Howdy (POST)" /></form></body></html>']
+    ]
+  end
+
+  def get_heavy
+    sec  =  10
+    tm1  =  Time.now.to_s
+    Kernel.sleep  sec
+    tm2  =  Time.now.to_s
+
+    [
+      200,
+      {'Content-Type' => 'text/html'},
+      ["<html><body><p> #{sec}: interval <br/> #{tm1}: sleep <br/> #{tm2}: wakeup </p></body></html>"]
+    ]
+  end
+end
+
+ENV['RACK_ENV'] ||= 'production'
+options  =  {Host: "0.0.0.0", Port: 3000, max_connection: 16}
+Reel::Rack::Server.new( WebApp.new, options )
+
+Signal.trap(:INT) do
+  exit
+end
+
+sleep
+

data/sample/server_http_hello_1.rb

@@ -0,0 +1,84 @@
+require 'sloth/reel'
+
+class WebServer < Reel::Server::HTTP
+  include Celluloid::Internals::Logger
+
+  def initialize( options )
+    host  =  options[:Host] || "0.0.0.0"
+    port  =  options[:Port] || 3000
+    opts  =  {}
+    opts[:max_connection]  =  options[:max_connection] || 16
+
+    info  "Reel::Server starting on #{host}:#{port}"
+    super( host, port, opts, &method(:on_connection) )
+  end
+
+  def on_connection( connection )
+    while request = connection.request
+      route_request  connection, request
+    end
+  end
+
+  def route_request( connection, request )
+    case  request.method
+    when  "GET"
+      case  request.url
+      when  "/"
+        get_index( connection )
+      when  "/heavy"
+        get_heavy( connection )
+      end
+
+    when  "POST"
+      case  request.url
+      when  "/"
+        post_index( connection )
+      end
+
+    else
+      not_found( connection )
+
+    end
+  end
+
+  def not_found( connection )
+    info "404 Not Found: #{request.method} #{request.path}"
+    connection.respond :not_found, "Not found"
+  end
+
+  def get_index( connection )
+    info "200 OK: GET /"
+    connection.respond :ok, <<-HTML
+      <html><body><h1> GET </h1><a href="/"> Hello (GET) </a> <form method="POST"><input type="submit" value="Howdy (POST)" /></form></body></html>
+    HTML
+  end
+
+  def post_index( connection )
+    info "200 OK: POST /"
+    connection.respond :ok, <<-HTML
+      <html><body><h1> POST </h1><a href="/"> Hello (GET) </a> <form method="POST"><input type="submit" value="Howdy (POST)" /></form></body></html>
+    HTML
+  end
+
+  def get_heavy( connection )
+    info "200 OK: POST /"
+
+    sec  =  10
+    tm1  =  Time.now.to_s
+    Kernel.sleep  sec
+    tm2  =  Time.now.to_s
+
+    connection.respond :ok, <<-HTML
+      <html><body><p> #{sec}: interval <br/> #{tm1}: sleep <br/> #{tm2}: wakeup </p></body></html>
+    HTML
+  end
+end
+
+options  =  {Host: "0.0.0.0", Port: 3000, max_connection: 16}
+WebServer.new( options )
+
+Signal.trap(:INT) do
+  exit
+end
+
+sleep

data/sample/server_http_ws_chat_1.rb

@@ -0,0 +1,105 @@
+require 'sloth/reel'
+
+class WebServer < Reel::Server::HTTP
+  include Celluloid::Internals::Logger
+
+  def initialize( options )
+    host  =  options[:Host] || "0.0.0.0"
+    port  =  options[:Port] || 3000
+    opts  =  {}
+    opts[:max_connection]  =  options[:max_connection] || 16
+
+    info  "Reel::Server starting on #{host}:#{port}"
+    super( host, port, opts, &method(:on_connection) )
+  end
+
+  def on_connection(connection)
+    while request = connection.request
+      if request.websocket?
+        connection.detach
+        route_websocket( request )
+      else
+        route_request( connection, request )
+      end
+    end
+  end
+
+  def route_websocket( request )
+    case  request.url
+    when  "/chat"
+      route_chat( request )
+    end
+  end
+
+  def route_chat(request)
+    if request.websocket?
+      ws  =  request.websocket
+      ws  <<  Time.now.to_s
+      ws.on_message do |mesg, sender, conns|
+        conns.each do |conn|
+          conn  <<  mesg
+        end
+      end
+    end
+  end
+
+  def route_request(connection, request)
+    case  request.url
+    when  "/"
+      render_index(connection)
+    else
+      info "404 Not Found: #{request.path}"
+      connection.respond :not_found, "Not found"
+    end
+  end
+
+  def render_index(connection)
+    info "200 OK: /"
+    connection.respond :ok, <<-HTML
+      <!doctype html>
+      <html>
+        <body>
+          <h1>WebSocket Chat</h1>
+          <form id="form"> <input type="text" id="input" value=""></input> </form>
+          <div id="msgs"></div>
+        </body>
+
+        <script type="text/javascript">
+          window.onload = function(){
+            (function(){
+              var show = function(el){
+                return function(str){ el.innerHTML = str + '<br />' + el.innerHTML }
+              }(document.getElementById('msgs'))
+
+              ws = new WebSocket('ws://' + window.location.host + '/chat')
+              ws.onopen = function() { show('[opened]') }
+              ws.onclose = function() { show('[closed]') }
+              ws.onmessage = function(mesg) { show(mesg.data) }
+
+              var sender = function(fm) {
+                var input = document.getElementById('input')
+                input.onclick = function(){
+                  input.value = ""
+                }
+                fm.onsubmit = function(){
+                  ws.send( input.value )
+                  input.value = ""
+                  return false
+                }
+              }(document.getElementById('form'))
+            })()
+          }
+        </script>
+      </html>
+    HTML
+  end
+end
+
+options  =  {Host: "0.0.0.0", Port: 3000, max_connection: 16}
+WebServer.new( options )
+
+Signal.trap(:INT) do
+  exit
+end
+
+sleep

data/sample/server_http_ws_time_1.rb

@@ -0,0 +1,94 @@
+require 'sloth/reel'
+
+class WebServer < Reel::Server::HTTP
+  include Celluloid::Internals::Logger
+
+  def initialize( options )
+    host  =  options[:Host] || "0.0.0.0"
+    port  =  options[:Port] || 3000
+    opts  =  {}
+    opts[:max_connection]  =  options[:max_connection] || 16
+
+    info  "Reel::Server starting on #{host}:#{port}"
+    super( host, port, opts, &method(:on_connection) )
+  end
+
+  def on_connection(connection)
+    while request = connection.request
+      if request.websocket?
+        connection.detach
+        route_websocket( request )
+      else
+        route_request( connection, request )
+      end
+    end
+  end
+
+  def route_websocket( request )
+    case  request.url
+    when  "/timeinfo"
+      route_timeinfo( request )
+    end
+  end
+
+  def route_timeinfo(request)
+    if request.websocket?
+      Thread.start do
+        begin
+          websocket  =  request.websocket
+          while not websocket.closed?
+            str  =  Time.now.to_s
+            websocket.write  str
+            sleep  1
+          end
+        rescue  Reel::SocketError
+        end
+      end
+    end
+  end
+
+  def route_request(connection, request)
+    case  request.url
+    when  "/"
+      render_index(connection)
+    else
+      info "404 Not Found: #{request.path}"
+      connection.respond :not_found, "Not found"
+    end
+  end
+
+  def render_index(connection)
+    info "200 OK: /"
+    connection.respond :ok, <<-HTML
+      <!doctype html>
+      <html lang="en">
+      <head>
+        <meta charset="utf-8">
+        <title>Reel WebSockets Time Server Example</title>
+      </head>
+      <script>
+        var ws = new WebSocket('ws://' + window.location.host + '/timeinfo');
+        ws.onmessage = function(mesg){
+          document.getElementById('current-time').innerHTML = mesg.data;
+        }
+      </script>
+      <body>
+        <div id="content">
+          <h1>Reel WebSockets Time Server Example</h1>
+          <h2>The time is now: <span id="current-time">...</span></h2>
+        </div>
+      </body>
+      </html>
+    HTML
+  end
+end
+
+options  =  {Host: "0.0.0.0", Port: 3000, max_connection: 16}
+WebServer.new( options )
+
+Signal.trap(:INT) do
+  exit
+end
+
+sleep
+