require 'test/unit'

module TestCentricity
  class AppUIElement
    include Test::Unit::Assertions

    attr_reader   :parent, :locator, :context, :type, :name

    def initialize(name, parent, locator, context)
      @name       = name
      @parent     = parent
      @locator    = locator
      @context    = context
      @type       = nil
    end

    def get_object_type
      @type
    end

    def get_locator
      @locator
    end

    def get_name
      @name
    end

    def click
      obj = element
      object_not_found_exception(obj)
      obj.click
    end

    def tap
      obj = element
      object_not_found_exception(obj)
      actions = Appium::TouchAction.new
      actions.tap(element: obj).perform
    end

    def double_tap
      obj = element
      object_not_found_exception(obj)
      actions = Appium::TouchAction.new
      actions.double_tap(element: obj).perform
    end

    def set(value)
      obj = element
      object_not_found_exception(obj)
      if value.is_a?(Array)
        obj.send_keys(value[0])
        if value[1].is_a?(Integer)
          press_keycode(value[1])
        else
          obj.send_keys(value[1])
        end
      elsif value.is_a?(String)
        obj.send_keys(value)
      end
    end

    def send_keys(value)
      obj = element
      object_not_found_exception(obj)
      obj.send_keys(value)
    end

    def clear
      obj = element
      object_not_found_exception(obj)
      obj.clear
    end

    def get_value
      obj = element
      object_not_found_exception(obj)
      if AppiumConnect.is_webview?
        case obj.tag_name.downcase
        when 'input', 'select', 'textarea'
          obj.value
        else
          obj.text
        end
      else
        obj.value
      end
    end

    def get_caption
      obj = element
      object_not_found_exception(obj)
      if AppiumConnect.is_webview?
        case obj.tag_name.downcase
        when 'input', 'select', 'textarea'
          obj.value
        else
          obj.text
        end
      elsif Environ.device_os == :ios
        case obj.tag_name
        when 'XCUIElementTypeNavigationBar'
          obj.attribute('name')
        else
          obj.attribute('label')
        end
      else
        obj.text
      end
    end

    def exists?
      begin
        element.displayed?
      rescue
        false
      end
    end

    def visible?
      if exists?
        element.displayed?
      else
        false
      end
    end

    def hidden?
      !visible?
    end

    def enabled?
      obj = element
      object_not_found_exception(obj)
      obj.enabled?
    end

    def disabled?
      !enabled?
    end

    def selected?
      obj = element
      object_not_found_exception(obj)
      obj.selected?
    end

    def tag_name
      obj = element
      object_not_found_exception(obj)
      obj.tag_name
    end

    def get_attribute(attrib)
      obj = element
      object_not_found_exception(obj)
      obj.attribute(attrib)
    end

    # Wait until the object exists, or until the specified wait time has expired. If the wait time is nil, then the wait
    # time will be Environ.default_max_wait_time.
    #
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   run_button.wait_until_exists(0.5)
    #
    def wait_until_exists(seconds = nil)
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { exists? }
    rescue
      raise "Could not find UI #{object_ref_message} after #{timeout} seconds" unless exists?
    end

    # Wait until the object no longer exists, or until the specified wait time has expired. If the wait time is nil, then
    # the wait time will be Environ.default_max_wait_time.
    #
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   logout_button.wait_until_gone(5)
    #
    def wait_until_gone(seconds = nil)
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { !exists? }
    rescue
      raise "UI #{object_ref_message} remained visible after #{timeout} seconds" if exists?
    end

    # Wait until the object is visible, or until the specified wait time has expired. If the wait time is nil, then the
    # wait time will be Environ.default_max_wait_time.
    #
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   run_button.wait_until_visible(0.5)
    #
    def wait_until_visible(seconds = nil)
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { visible? }
    rescue
      raise "Could not find UI #{object_ref_message} after #{timeout} seconds" unless visible?
    end

    # Wait until the object is hidden, or until the specified wait time has expired. If the wait time is nil, then the
    # wait time will be Environ.default_max_wait_time.
    #
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   run_button.wait_until_hidden(10)
    #
    def wait_until_hidden(seconds = nil)
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { hidden? }
    rescue
      raise "UI #{object_ref_message} remained visible after #{timeout} seconds" if visible?
    end

    # Wait until the object's value equals the specified value, or until the specified wait time has expired. If the wait
    # time is nil, then the wait time will be Environ.default_max_wait_time.
    #
    # @param value [String or Hash] value expected or comparison hash
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   card_authorized_label.wait_until_value_is('Card authorized', 5)
    #     or
    #   total_weight_field.wait_until_value_is({ :greater_than => '250' }, 5)
    #
    def wait_until_value_is(value, seconds = nil)
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { compare(value, get_value) }
    rescue
      raise "Value of UI #{object_ref_message} failed to equal '#{value}' after #{timeout} seconds" unless get_value == value
    end

    # Wait until the object's value changes to a different value, or until the specified wait time has expired. If the
    # wait time is nil, then the wait time will be Environ.default_max_wait_time.
    #
    # @param seconds [Integer or Float] wait time in seconds
    # @example
    #   basket_grand_total_label.wait_until_value_changes(5)
    #
    def wait_until_value_changes(seconds = nil)
      value = get_value
      timeout = seconds.nil? ? Environ.default_max_wait_time : seconds
      wait = Selenium::WebDriver::Wait.new(timeout: timeout)
      wait.until { get_value != value }
    rescue
      raise "Value of UI #{object_ref_message} failed to change from '#{value}' after #{timeout} seconds" if get_value == value
    end

    def width
      obj = element
      object_not_found_exception(obj)
      obj.size.width
    end

    # Return height of object.
    #
    # @return [Integer]
    # @example
    #   button_height = my_button.height
    #
    def height
      obj = element
      object_not_found_exception(obj)
      obj.size.height
    end

    # Return x coordinate of object's location.
    #
    # @return [Integer]
    # @example
    #   button_x = my_button.x
    #
    def x_loc
      obj = element
      object_not_found_exception(obj)
      obj.location.x
    end

    # Return y coordinate of object's location.
    #
    # @return [Integer]
    # @example
    #   button_y = my_button.y
    #
    def y_loc
      obj = element
      object_not_found_exception(obj)
      obj.location.y
    end

    def scroll(direction)
      obj = element
      object_not_found_exception(obj)
      if Environ.device_os == :ios
        execute_script('mobile: scroll', direction: direction.to_s, element: obj)
      else
        case direction
        when :down
          x = obj.size.width.to_i / 2
          bottom = obj.size.height
          bottom = window_size[:height].to_i if bottom > window_size[:height]
          start_pt = [x, bottom]
          delta_pt = [0, 0 - (bottom - 20)]

        when :up
          x = obj.size.width.to_i / 2
          top = obj.location.y - (obj.size.height / 2)
          top = 0 if top < 0
          start_pt = [x, top]
          delta_pt = [0, obj.size.height / 2]
        end
        puts "Swipe start_pt = #{start_pt} / delta_pt = #{delta_pt}" if ENV['DEBUG']
        Appium::TouchAction.new.swipe(start_x: start_pt[0], start_y: start_pt[1], delta_x: delta_pt[0], delta_y: delta_pt[1]).perform
      end
    end

    def swipe(direction)
      obj = element
      object_not_found_exception(obj)
      if Environ.device_os == :ios
        execute_script('mobile: swipe', direction: direction.to_s, element: obj)
      else
        actions = Appium::TouchAction.new
        actions.scroll(element: obj).perform
      end
    end

    private

    def element
      if @context == :section
        parent_obj = nil
        locators = @parent.get_locator
        locators.each do |parent_locator|
          if parent_obj.nil?
            parent_obj = find_element(parent_locator.keys[0], parent_locator.values[0])
          else
            parent_obj = parent_obj.find_element(parent_locator.keys[0], parent_locator.values[0])
          end
        end
        puts "Found parent object '#{@parent.get_name}' - #{@parent.get_locator}" if ENV['DEBUG']
        obj = parent_obj.find_element(@locator.keys[0], @locator.values[0])
        puts "Found object '#{@name}' - #{@locator}" if ENV['DEBUG']
        obj
      else
        obj = find_element(@locator.keys[0], @locator.values[0])
        puts "Found object '#{@name}' - #{@locator}" if ENV['DEBUG']
        obj
      end
    rescue
      puts "Did not find object '#{@name}' - #{@locator}" if ENV['DEBUG']
      nil
    end

    def object_not_found_exception(obj)
      @type.nil? ? object_type = 'Object' : object_type = @type
      raise ObjectNotFoundError.new("#{object_type} named '#{@name}' (#{get_locator}) not found") unless obj
    end

    def object_ref_message
      "object '#{@name}' (#{get_locator})"
    end

    def compare(expected, actual)
      if expected.is_a?(Hash)
        result = false
        expected.each do |key, value|
          case key
          when :lt, :less_than
            result = actual < value
          when :lt_eq, :less_than_or_equal
            result = actual <= value
          when :gt, :greater_than
            result = actual > value
          when :gt_eq, :greater_than_or_equal
            result = actual >= value
          when :not_equal
            result = actual != value
          end
        end
      else
        result = expected == actual
      end
      result
    end
  end
end