module Polonium
  class WebrickSeleniumServerRunner < ServerRunner
    attr_accessor :socket, :dispatch_servlet, :environment_path, :server

    protected
    def start_server
      socket.do_not_reverse_lookup = true # patch for OS X

      @server = configuration.create_webrick_server
      mount_parameters = {
        :port            => configuration.internal_app_server_port,
        :ip              => configuration.internal_app_server_host,
        :environment     => configuration.rails_env.dup,
        :server_root     => configuration.server_root,
        :server_type     => WEBrick::SimpleServer,
        :charset         => "UTF-8",
        :mime_types      => WEBrick::HTTPUtils::DefaultMimeTypes,
        :working_directory => File.expand_path(configuration.rails_root.to_s)
      }
      server.mount('/', dispatch_servlet, mount_parameters)

      trap("INT") { stop_server }

      require @environment_path
      require "dispatcher"
      server.start
    end

    def stop_server
      server.shutdown
    end
  end
end