teampoltergeist--poltergeist/lib/capybara/poltergeist/server.rb

module Capybara::Poltergeist
  class Server
    attr_reader :port, :socket, :timeout

    def initialize(timeout = nil)
      @port    = Util.find_available_port
      @timeout = timeout
      start
    end

    def timeout=(sec)
      @timeout = @socket.timeout = sec
    end

    def start
      @socket = WebSocketServer.new(port, timeout)
    end

    def restart
      @socket.close
      @socket = WebSocketServer.new(port, timeout)
    end

    def send(message)
      @socket.send(message) or raise DeadClient.new(message)
    end
  end
end