require "testingbot/version"

module Selenium
  module Client
    module Protocol
      attr_writer :client_key, :client_secret
      
        def client_key
          if @client_key.nil?
            @client_key, @client_secret = get_testingbot_credentials
          end
          @client_key
        end
        
        def client_secret
          if @client_secret.nil?
            @client_key, @client_secret = get_testingbot_credentials
          end
          @client_secret
        end
        
        def get_testingbot_credentials
          if File.exists?(File.expand_path("~/.testingbot"))
            str = File.open(File.expand_path("~/.testingbot")) { |f| f.readline }.chomp
            str.split(':')
          else
            raise "Please run the testingbot install tool first"
          end
        end
        
        # add custom parameters for testingbot.com
        def http_request_for_testingbot(verb, args)
          data = http_request_for_original(verb, args)
          data << "&client_key=#{client_key}&client_secret=#{client_secret}"
        end
        
        alias http_request_for_original http_request_for
        alias http_request_for http_request_for_testingbot
    end
  end
end

module Selenium
  module Client
    module Base
      alias :close_current_browser_session_old :close_current_browser_session
      attr_accessor :session_id_backup
      
      def close_current_browser_session
        @session_id_backup = @session_id
        close_current_browser_session_old
      end
    end
  end
end

if defined?(Spec)
  Spec::Runner.configure do |config|
    config.prepend_after(:each) do
      if File.exists?(File.expand_path("~/.testingbot"))
        str = File.open(File.expand_path("~/.testingbot")) { |f| f.readline }.chomp
        client_key, client_secret = str.split(':')
      
        params = {
          "session_id" => @selenium_driver.session_id_backup,
          "client_key" => client_key,
          "client_secret" => client_secret,
          "status_message" => @execution_error,
          "success" => !actual_failure?,
          "name" => description.to_s,
          "kind" => 2
        }
        
        url = URI.parse('http://localhost:3000/hq')
        http = Net::HTTP.new(url.host, url.port)
        response = http.post(url.path, params.map { |k, v| "#{k.to_s}=#{v}" }.join("&"))
      end
    end
  end
end

if defined?(Test::Unit::TestCase)
  module TestingBot
    class TestingBot::TestCase < Test::Unit::TestCase
      alias :run_teardown_old :run_teardown
      alias :handle_exception_old :handle_exception
      
      attr_accessor :exception
      
      def run_teardown
        client_key, client_secret = get_testingbot_credentials
        params = {
          "session_id" => browser.session_id,
          "client_key" => client_key,
          "client_secret" => client_secret,
          "status_message" => @exception,
          "success" => passed?,
          "name" => self.to_s,
          "kind" => 2
        }
        
        url = URI.parse('http://localhost:3000/hq')
        http = Net::HTTP.new(url.host, url.port)
        response = http.post(url.path, params.map { |k, v| "#{k.to_s}=#{v}" }.join("&"))
        run_teardown_old
      end #def run_teardown
      
      def handle_exception(e)
        @exception = e.to_s
        handle_exception_old(e)
      end
      
      def get_testingbot_credentials
        if File.exists?(File.expand_path("~/.testingbot"))
          str = File.open(File.expand_path("~/.testingbot")) { |f| f.readline }.chomp
          str.split(':')
        else
          raise "Please run the testingbot install tool first"
        end
      end
    end
  end
end