require "rubygems"
require "watir-webdriver"
require "headless"

module Cello
  module Structure
    class Browser
      attr_accessor :context
      attr_reader :browser

      def initialize(browser)
        @headless = Headless.new
        @headless.start
        @browser = Watir::Browser.new browser
      end
      
      def visit
        @browser.goto @context.get_url
      end

      def context(page)
        @context = page.new(@browser)
      end

      def search(text)
       @browser.text.include? text 
      end

      def close
        @browser.close
        @headless.destroy
      end

      def get_screenshot
        @browser.driver.save_screenshot 'screenshot.png'
      end

      def title
        @browser.title
      end

      def response_time
        #pending
      end

      def method_missing(method_name, *arguments)
        @context.send(method_name, *arguments)
      end
    end
  end
end