grrrisu/em-box

require 'rbconfig'
require 'json'
require 'rubygems'
require 'eventmachine'

require_relative '../em_box'

module EMBox

  class Server

    at_exit { puts "server exiting..." }

    def initialize options = {}
      @client_class          = options[:client_class] || EMBox::Client::Base
      @ruby                  = options[:ruby] || File.join(RbConfig::CONFIG['bindir'], RbConfig::CONFIG['ruby_install_name'])
      @client_configuration  = options[:client_configuration]
      @client_ready_timeout  = options[:client_ready_timeout]
    end

    def start &block
      EM.run do
        yield self
      end
    end

    def stop
      EM.stop
    end

    # agent includes HasClient
    def start_agent agent, &block
      em_lib = File.expand_path(File.dirname(__FILE__) + '/../em_box.rb')
      libs   = %W{rubygems json eventmachine #{em_lib}}.join(' -r')
      cmd    = "#{@ruby} -r#{libs}"
      cmd   += " -e '#{@client_class}.new(\"#{agent.agent_class}\", \"#{agent.agent_file}\""
      cmd   += ", \"#{@client_configuration}\"" if @client_configuration
      cmd   += ")'"
      agent.connection      = EM.popen(cmd, ClientConnection)
      agent.server          = self
      agent.start_callback  = block if block_given?
      stop_client_if_not_ready(agent)
      agent
    end

    def stop_client_if_not_ready agent
      if @client_ready_timeout
        EM::add_timer(@client_ready_timeout) do
          unless agent.connection.status == :ready
            agent.connection.close_connection
          end
        end
      end
    end

  end

end