97 lines
2.2 KiB
Crystal
97 lines
2.2 KiB
Crystal
|
require "option_parser"
|
||
|
|
require "cbor"
|
||
|
|
require "../src/ipc.cr"
|
||
|
|
require "../src/cbor.cr"
|
||
|
|
require "./prints.cr"
|
||
|
|
|
||
|
|
require "./message"
|
||
|
|
|
||
|
|
class CLI
|
||
|
|
class_property service_name = "pong"
|
||
|
|
class_property verbosity = 1
|
||
|
|
class_property timer = 30_000
|
||
|
|
class_property no_response = false
|
||
|
|
end
|
||
|
|
|
||
|
|
OptionParser.parse do |parser|
|
||
|
|
parser.on "-s service_name", "--service-name service_name", "URI" do |optsn|
|
||
|
|
CLI.service_name = optsn
|
||
|
|
end
|
||
|
|
|
||
|
|
parser.on "-n", "--no-response", "Do not provide any response back." do
|
||
|
|
CLI.no_response = true
|
||
|
|
end
|
||
|
|
|
||
|
|
parser.on "-t timer", "--timer ms", "Timer in ms. Default: 30 000" do |optsn|
|
||
|
|
CLI.timer = optsn.to_i
|
||
|
|
end
|
||
|
|
|
||
|
|
|
||
|
|
parser.on "-v verbosity", "--verbosity verbosity", "Verbosity (0 = nothing is printed, 1 = only events, 2 = events and messages). Default: 1" do |optsn|
|
||
|
|
CLI.verbosity = optsn.to_i
|
||
|
|
end
|
||
|
|
|
||
|
|
parser.on "-h", "--help", "Show this help" do
|
||
|
|
puts parser
|
||
|
|
exit 0
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def main
|
||
|
|
service = IPC::Server.new CLI.service_name
|
||
|
|
service.base_timer = CLI.timer # default: 30 seconds
|
||
|
|
service.timer = CLI.timer # default: 30 seconds
|
||
|
|
|
||
|
|
service.loop do |event|
|
||
|
|
# service.pp
|
||
|
|
case event
|
||
|
|
when IPC::Event::Timer
|
||
|
|
info "IPC::Event::Timer"
|
||
|
|
when IPC::Event::Connection
|
||
|
|
info "IPC::Event::Connection, client: #{event.fd}"
|
||
|
|
when IPC::Event::Disconnection
|
||
|
|
info "IPC::Event::Disconnection, client: #{event.fd}"
|
||
|
|
when IPC::Event::MessageSent
|
||
|
|
begin
|
||
|
|
info "IPC::Event::MessageSent, client: #{event.fd}"
|
||
|
|
rescue e
|
||
|
|
important "#{e.message}"
|
||
|
|
service.remove_fd event.fd
|
||
|
|
end
|
||
|
|
|
||
|
|
when IPC::Event::MessageReceived
|
||
|
|
begin
|
||
|
|
info "IPC::Event::MessageReceived, client: #{event.fd}"
|
||
|
|
|
||
|
|
request = Context.requests.parse_ipc_cbor event.message
|
||
|
|
|
||
|
|
if request.nil?
|
||
|
|
raise "unknown request type"
|
||
|
|
end
|
||
|
|
|
||
|
|
info "<< #{request.class.name}"
|
||
|
|
|
||
|
|
response = begin
|
||
|
|
request.handle
|
||
|
|
rescue e
|
||
|
|
important "#{request.class.name} generic error #{e}"
|
||
|
|
::Error.new "unknown error"
|
||
|
|
end
|
||
|
|
|
||
|
|
unless CLI.no_response
|
||
|
|
info ">> #{response.class.name}"
|
||
|
|
service.send event.fd, response.not_nil!
|
||
|
|
end
|
||
|
|
|
||
|
|
rescue e
|
||
|
|
important "#{e.message}"
|
||
|
|
service.remove_fd event.fd
|
||
|
|
end
|
||
|
|
else
|
||
|
|
important "Exception: message #{event}"
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
main
|