require 'yaml'
require 'rufus/drivers/iOS_device'
require 'rufus/drivers/iOS_simulator'
require 'selenium-webdriver'
require 'erb'
require 'selenium/webdriver/safari/extension'

module Rufus
  class Driver

    def initialize
      raise 'No config.yml found' if !File.exists?('config.yml')
      @config = YAML.load(ERB.new(File.read('config.yml')).result)
      @url = url(@config)
    end

    def start
      driver.get @url
    end

    def quit
      driver.quit
      @selenium = nil
    end

    def config
      @config
    end

    def server_url
      @url
    end

    def find(locator)
      how = locator.keys[0].to_sym
      what = locator[how]
      begin
        driver.find_element(how, what)
      rescue Selenium::WebDriver::Error::NoSuchElementError
        return nil
      end
    end

    def click(locator)
      find(locator).click
    end

    def press_button name
      click(:name => name)
    end

    def enabled?(locator)
      find(locator).enabled?
    end

    def displayed?(locator)
      find(locator).displayed?
    end

    def orientation
      driver.orientation.to_s
    end

    def rotate(orientation)
      driver.rotate orientation
    end

    def type(keys, name)
      element = find(:name => name)
      element.click
      sleep 1
      element.send_keys keys
    end

    def sequence(*names, times)
      timed_sequence(names, times, 1)
    end

    def buttons
      buttons = []
      elements = elements_by_tag 'UIAButton'
      elements.each do |element|
        buttons << element.text
      end
      buttons
    end

    def text_fields
      fields = []
      elements = elements_by_tag 'UIATextField'
      elements.each do |element|
        fields << element.text
      end
      fields
    end

    def labels
      labels = []
      elements = elements_by_tag 'UIAStaticText'
      elements.each do |element|
        labels << element.text
      end
      labels
    end

    def timed_sequence(names, times, seconds)
      current = 0
      until current == times
        names.each do |name|
          click(:name => name)
          sleep seconds

        end
        current += 1
        puts "sequence #{current} completed"
      end
    end

    def find_alert(locator)
      alert = nil
      all_elements.each do |element|
        if is_alert?(element)
          alert = element if match?(element, locator[:name])
        end
      end
      alert
    end

    def click_alert(button)
      if alert_shown?
        click_alert_button(button)
      end
    end

    def alert_shown?
      all_elements.each do |element|
        if is_alert?(element)
          return true
        end
      end
      false
    end

    def class_for(element)
      element.tag_name
    end

    def match?(element, name)
      element.attribute(:name).eql? name
    end

    def page_source
      driver.page_source
    end

    def all_elements
      elements_by_tag('UIAElement')
    end

    def elements_by_tag(name)
      driver.find_elements(:tag_name, name)
    end

    private

    def url(config)
      if config["appium_url"].nil? || config["appium_url"].eql?("")
        'http://127.0.0.1:4723/wd/hub'
      else
        config["appium_url"]
      end
    end

    def click_alert_button(button)
      all_elements.each do |element|
        element.click if is_table_view_cell?(element) && match?(element, button)
      end
    end

    def is_alert?(element)
      class_for(element).eql?('UIAAlert')
    end

    def is_table_view_cell?(element)
      class_for(element).eql?('UIATableCell')
    end

    def driver
      if use_device
        @selenium ||= Rufus::Drivers::IOS_Device.for(@config,@url)
      else
        @selenium ||= Rufus::Drivers::IOS_Simulator.for(@config,@url)
      end
    end

    def use_device
      @config["use_physical"] == true
    end
  end
end