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

module Capybara::Poltergeist
  class Client
    PHANTOMJS_SCRIPT  = File.expand_path('../client/compiled/main.js', __FILE__)
    PHANTOMJS_VERSION = '1.6.0'
    PHANTOMJS_NAME    = 'phantomjs'

    def self.start(*args)
      client = new(*args)
      client.start
      client
    end

    attr_reader :pid, :port, :path, :window_size, :phantomjs_options

    def initialize(port, options = {})
      @port              = port
      @path              = options[:path]              || PHANTOMJS_NAME
      @window_size       = options[:window_size]       || [1024, 768]
      @phantomjs_options = options[:phantomjs_options] || []

      pid = Process.pid
      at_exit { stop if Process.pid == pid }
    end

    def start
      check_phantomjs_version
      @pid = Spawn.spawn(*command)
    end

    def stop
      if pid
        begin
          Process.kill('TERM', pid)
          Process.wait(pid)
        rescue Errno::ESRCH, Errno::ECHILD
          # Zed's dead, baby
        end

        @pid = nil
      end
    end

    def restart
      stop
      start
    end

    def command
      @command ||= begin
        parts = [path]
        parts.concat phantomjs_options
        parts << PHANTOMJS_SCRIPT
        parts << port
        parts.concat window_size
        parts
      end
    end

    private

    def check_phantomjs_version
      return if @phantomjs_version_checked

      version = `#{path} --version`.chomp

      if $? != 0
        raise PhantomJSFailed.new($?)
      elsif version < PHANTOMJS_VERSION
        raise PhantomJSTooOld.new(version)
      end

      @phantomjs_version_checked = true
    end
  end
end