Contents

Faye::WebSocket.load_adapter('thin')

module Grenache
  class WebsocketServer
    def initialize(port, &block)
      @port = port
      @callback = block
    end

    def start_server
      @server = Thin::Server.start('0.0.0.0', @port, method(:app), {signals: false})
    end

    def app(env)
      ws = Faye::WebSocket.new(env)
      ws.on :message, -> (ev) do
        req = Message.parse(ev.data)
        res = @callback.call(req)
        ws.send(Message.response_to(req, res).to_json)
      end
      ws.rack_response
    end

    def send(payload)
      @server.send(payload)
    end
  end

  class WebsocketClient
    def initialize(uri, &cb)
      @uri = uri.sub("tcp://","ws://")
      @uri.prepend("ws://") unless @uri.start_with?("ws://")
      @callback = cb
    end

    def send payload
      ws_connect unless @connected
      @ws.send(payload)
    end

    private
    def ws_connect
      @ws = Faye::WebSocket::Client.new(@uri)
      @ws.on(:open, method(:on_open))
      @ws.on(:message, method(:on_message))
      @ws.on(:close, method(:on_close))
    end

    def on_open(ev)
      @connected = true
    end

    def on_message(ev)
      msg = Message.parse(ev.data)
      disconnect
      @callback.call([nil,msg]) if @callback
    end

    def disconnect
      @ws.close
      @ws = nil
    end

    def on_close(ev)
      @connected = false
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
grenache-ruby-ws-0.1.8	lib/grenache/websocket.rb
grenache-ruby-ws-0.1.7	lib/grenache/websocket.rb
grenache-ruby-ws-0.1.6	lib/grenache/websocket.rb
grenache-ruby-ws-0.1.5	lib/grenache/websocket.rb
grenache-ruby-ws-0.1.4	lib/grenache/websocket.rb
grenache-ruby-ws-0.1.3	lib/grenache/websocket.rb