44 lines
805 B
Ruby
44 lines
805 B
Ruby
# frozen_string_literal: true
|
|
|
|
require 'puma'
|
|
require 'puma/binder'
|
|
require 'puma/events'
|
|
|
|
class EchoServer
|
|
def call(env)
|
|
@socket = Faye::WebSocket.new(env, ['echo'])
|
|
|
|
@socket.onmessage = lambda do |event|
|
|
@socket.send(event.data)
|
|
end
|
|
|
|
@socket.rack_response
|
|
end
|
|
|
|
def send(message)
|
|
@socket.send(message)
|
|
end
|
|
|
|
def close(code, reason)
|
|
@socket.close(code, reason)
|
|
end
|
|
|
|
def log(*args); end
|
|
|
|
def listen(port)
|
|
events = Puma::Events.new(StringIO.new, StringIO.new)
|
|
binder = Puma::Binder.new(events)
|
|
binder.parse(["tcp://0.0.0.0:#{port}"], self)
|
|
@server = Puma::Server.new(self, events)
|
|
@server.binder = binder
|
|
@server.run
|
|
end
|
|
|
|
def stop
|
|
case @server
|
|
when Puma::Server then @server.stop(true)
|
|
else @server.stop
|
|
end
|
|
end
|
|
end
|