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

require 'multi_json'

module Capybara::Poltergeist
  class Browser
    attr_reader :server, :client, :logger, :js_errors

    def initialize(server, client, logger = nil, js_errors = true)
      @server     = server
      @client     = client
      @logger     = logger
      @js_errors  = js_errors
    end

    def restart
      server.restart
      client.restart
    end

    def visit(url, headers)
      command 'visit', url, headers
    end

    def current_url
      command 'current_url'
    end

    def body
      command 'body'
    end

    def source
      command 'source'
    end

    def find(selector)
      result = command('find', selector)
      result['ids'].map { |id| [result['page_id'], id] }
    end

    def find_within(page_id, id, selector)
      command 'find_within', page_id, id, selector
    end

    def text(page_id, id)
      command 'text', page_id, id
    end

    def attribute(page_id, id, name)
      command 'attribute', page_id, id, name.to_s
    end

    def value(page_id, id)
      command 'value', page_id, id
    end

    def set(page_id, id, value)
      command 'set', page_id, id, value
    end

    def select_file(page_id, id, value)
      command 'select_file', page_id, id, value
    end

    def tag_name(page_id, id)
      command('tag_name', page_id, id).downcase
    end

    def visible?(page_id, id)
      command 'visible', page_id, id
    end

    def evaluate(script)
      command 'evaluate', script
    end

    def execute(script)
      command 'execute', script
    end

    def within_frame(id, &block)
      command 'push_frame', id
      val = yield
      command 'pop_frame'
      val
    end

    def click(page_id, id)
      command 'click', page_id, id
    end

    def drag(page_id, id, other_id)
      command 'drag', page_id, id, other_id
    end

    def select(page_id, id, value)
      command 'select', page_id, id, value
    end

    def trigger(page_id, id, event)
      command 'trigger', page_id, id, event.to_s
    end

    def reset
      command 'reset'
    end

    def render(path, options = {})
      command 'render', path.to_s, !!options[:full]
    end

    def resize(width, height)
      command 'resize', width, height
    end

    def network_traffic
      command('network_traffic').values.map do |event|
        NetworkTraffic::Request.new(
          event['request'],
          event['responseParts'].map { |response| NetworkTraffic::Response.new(response) }
        )
      end
    end

    def command(name, *args)
      message = { 'name' => name, 'args' => args }
      log message.inspect

      json = JSON.load(server.send(JSON.dump(message)))
      log json.inspect

      if json['error']
        if json['error']['name'] == 'Poltergeist.JavascriptError'
          error = JavascriptError.new(json['error'])
          if js_errors
            raise error
          else
            log error
          end
        else
          raise BrowserError.new(json['error'])
        end
      end
      json['response']

    rescue DeadClient
      restart
      raise
    end

    private
    def log(message)
      logger.puts message if logger
    end
  end
end