module Faye
  class WebSocket

    class Client
      include API

      attr_reader :headers, :status

      def initialize(url, protocols = nil, options = {})
        @driver = ::WebSocket::Driver.client(self, :max_length => options[:max_length], :protocols => protocols)

        [:open, :error].each do |event|
          @driver.on(event) do
            @headers = @driver.headers
            @status  = @driver.status
          end
        end

        super(options)

        @url = url
        @uri = URI.parse(url)

        port = @uri.port || (@uri.scheme == 'wss' ? 443 : 80)
        EventMachine.connect(@uri.host, port, Connection) do |conn|
          @stream = conn
          conn.parent = self
        end
      rescue => error
        event = Event.create('error', :message => "Network error: #{url}: #{error.message}")
        event.init_event('error', false, false)
        dispatch_event(event)
        finalize(error.message, 1006)
      end

    private

      def on_connect
        @stream.start_tls if @uri.scheme == 'wss'
        @driver.start
      end

      module Connection
        attr_accessor :parent

        def connection_completed
          parent.__send__(:on_connect)
        end

        def receive_data(data)
          parent.__send__(:parse, data)
        end

        def unbind
          parent.__send__(:finalize, '', 1006)
        end

        def write(data)
          send_data(data) rescue nil
        end
      end
    end

  end
end