www/board/agenda/daemon/wss.rb

#!/usr/bin/env ruby
$:.unshift File.realpath(File.expand_path('../' * 5 + 'lib', __FILE__))
require 'websocket-eventmachine-server'
require 'listen'
require 'ostruct'
require 'optparse'
require 'yaml'
require 'rbconfig'
require 'json'

require_relative './session'
require_relative './channel'

########################################################################
#                         Parse argument list                          #
########################################################################

options = OpenStruct.new
options.host = '0.0.0.0'
options.port = 34234
options.privkey = Dir['/etc/letsencrypt/live/*/privkey.pem'].
  sort_by {|file| File.mtime file}.last
options.chain = Dir['/etc/letsencrypt/live/*/fullchain.pem'].
  sort_by {|file| File.mtime file}.last
options.kill = false
options.timeout = 900

opt_parser = OptionParser.new do |opts|
  opts.banner = "Usage: #{File.basename(__FILE__)} [options]"

  opts.on "-h", "--host HOST", 'Host to listen on' do |host|
    options.host = host
  end

  opts.on "-p", "--port PORT", 'Port to listen on' do |port|
    options.port = port
  end

  opts.on "-k", "--key KEY", 'Private key' do |key|
    options.privkey = key
  end

  opts.on "-c", "--chain CHAIN", 'Certificate Chain' do |chain|
    options.chain = chain
  end

  opts.on "--kill [SIGNAL]", 'Kill existing process' do |signal|
    options.kill = signal || 'INT'
  end

  opts.on '-t', "--timeout [900]", 'inactivity timeout' do |timeout|
    options.timeout = timeout.to_i
  end
end

opt_parser.parse!(ARGV)

########################################################################
#                  Verify/enforce socket availability                  #
########################################################################

begin
  test_socket = TCPSocket.new('localhost', options.port)
  test_socket.close
  if options.kill
    `lsof -Fp -i :#{options.port} -sTCP:LISTEN`.scan(/^p(\d+)/).each do |(pid)|
      Process.kill 'INT', pid.to_i
    end
  else
    STDERR.puts 'socket in use'
    exit 1
  end
rescue Errno::ECONNREFUSED
end
exit 0 if options.kill

########################################################################
#    Restart when source file changes or when inactive for an hour     #
########################################################################

def restart_process
  puts 'restarting'
  exec RbConfig.ruby, File.expand_path(__FILE__), *ARGV
end

listener = Listen.to(__dir__) do |modified, added, removed|
  restart_process
end
listener.start

active = Time.now

# restart once an hour when inactive
Thread.new do
  loop do
    sleep 90
    restart_process if Time.now - active >= 3600
  end
end

########################################################################
#                  Close all open connection on exit                   #
########################################################################

at_exit do
  Channel.close_all
end

########################################################################
#                        Start WebSocket server                        #
########################################################################

server_options = {host: options.host, port: options.port}

if options.privkey and options.chain
  server_options.merge! secure: true, 
    tls_options: {
      private_key_file: options.privkey,
      cert_chain_file: options.chain
    }
end

EM.run do
  WebSocket::EventMachine::Server.start(server_options) do |ws|
    ws.onclose do 
      Channel.delete ws
    end

    ws.onmessage do |msg|
      # extract headers
      headers = msg.slice!(/\A(\w+:\s*.*\r?\n)+\s*(\n|\Z)/).to_s
      headers = YAML.safe_load(headers) || {} rescue {}

      if headers['session']
        session = Session[headers['session']]
        if session
          restart_process if headers['restart']
          Channel.add ws, session[:id]
          ws.send JSON.dump(session.merge type: 'login')
        else
          msg = ''
          ws.send JSON.dump(type: 'unauthorized', session: headers['session'])
          EM.add_timer(1) {ws.close}
        end
      end

      # forward message
      unless msg.empty?
        if headers['private']
          Channel.post_private headers['private'], msg
        else
          Channel.post_all msg
        end
      end

      # reset activity timer
      active = Time.now
    end
  end
end