trabajo-afectivo/script/websocket-server.rb

$LOAD_PATH << './lib'
require 'rubygems'
require 'eventmachine'
require 'em-websocket'
require 'json'
require 'fileutils'
require 'web_socket'
require 'optparse'

# Look for -o with argument, and -I and -D boolean arguments
options = {
  :p => 6042,
  :b => '0.0.0.0',
  :s => false,
  :k => '/path/to/server.key',
  :c => '/path/to/server.crt',
}
tls_options = {}
OptionParser.new do |opts|
  opts.banner = "Usage: websocket-server.rb [options]"

  opts.on("-p", "--port [OPT]", "port of websocket server") do |p|
    options[:p] = p
  end
  opts.on("-b", "--bind [OPT]", "bind address") do |b|
    options[:b] = b
  end
  opts.on("-s", "--secure", "enable secure connections") do |s|
    options[:s] = s
  end
  opts.on("-k", "--private-key [OPT]", "/path/to/server.key for secure connections") do |k|
    tls_options[:private_key_file] = k
  end
  opts.on("-c", "--certificate [OPT]", "/path/to/server.crt for secure connections") do |c|
    tls_options[:cert_chain_file] = c
  end
end.parse!

puts "Starting websocket server on #{ options[:b] }:#{ options[:p] } (secure:#{ options[:s].to_s })"
#puts options.inspect

@clients = {}
EventMachine.run {
  EventMachine::WebSocket.start( :host => options[:b], :port => options[:p], :secure => options[:s], :tls_options => tls_options ) do |ws|

    # register client connection
    ws.onopen {
      client_id = ws.object_id
      puts 'Client ' + client_id.to_s + ' connected'

      if !@clients.include? client_id
        @clients[client_id] = {
          :websocket => ws,
        } 
      end
    }

    # unregister client connection
    ws.onclose {
      client_id = ws.object_id
      puts 'Client ' + client_id.to_s + ' disconnected'

      # removed from current client list
      if @clients.include? client_id
        @clients.delete client_id
      end

      Session.destory( client_id )
    }

    # manage messages
    ws.onmessage { |msg|

      client_id = ws.object_id
      puts 'From Client ' + client_id.to_s + ' received message: ' + msg
      data = JSON.parse(msg)

      # get session
      if data['action'] == 'login'
        @clients[client_id][:session] = data['session']
        Session.create( client_id, data['session'] )

      # ping
      elsif data['action'] == 'ping'
        @clients[client_id][:last_ping] = Time.now
        ws.send( '[{"action":"pong"}]' )
      end
    }
  end

  EventMachine.add_periodic_timer(0.2) {
    puts "loop"
    @clients.each { |client_id, client|
      log 'checking waiting data...', client_id
      begin
        queue = Session.queue( client_id )
        if queue && queue[0]
#          log "send " + queue.inspect, client_id
          log "send data to client", client_id
          client[:websocket].send( queue.to_json )
        end
      rescue => e
        log 'problem:' + e.inspect, client_id
      end
    }
  }
  
  def log( data, client_id )
    puts "#{Time.now}:client(#{ client_id }) #{ data }"
  end

}