nostr-gem/spec/support/echo_server.rb
2023-01-12 17:21:17 +07:00

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