# encoding: utf-8
# UIAButton methods

# Find a button by text and optionally number.
# @param text [String, Integer] the text to exactly match. If int then the button at that index is returned.
# @param number [Integer] the occurance of the button matching text. Defaults to the first button.
# @return [Button] the button found with text and matching number
def button text, number=0
  # return button at index.
  return ele_index :button, text if text.is_a? Numeric

  number >= 1 ? button_num( text, number ) :
                find_ele_by_text_include( :button, text )
end

# Get an array of button texts or button elements if text is provided.
# @param text [String] the text to exactly match
# @return [Array<String>, Array<Buttons>] either an array of button texts or an array of button elements if text is provided.
def buttons text=nil
  text == nil ? find_eles_attr( :button, :text ) :
                find_eles_by_text_include( :button, text )
end

# Get the first button element.
# @return [Button]
def first_button
  first_ele :button
end

# Get the last button element.
# @return [Button]
def last_button
  last_ele :button
end

# Get the first button element that exactly matches text.
# @param text [String] the text to match exactly
# @return [Button]
def button_exact text
  find_ele_by_text :button, text
end

# Get all button elements that exactly match text.
# @param text [String] the text to match exactly
# @return [Array<Button>]
def buttons_exact text
  find_eles_by_text :button, text
end

# Get an array of button elements.
# @return [Array<Button>]
def e_buttons
  find_eles :button
end

# Expected to be called via button method.
#
# Get the button element exactly matching text and
# occurrence. number=2 means the 2nd occurrence.
#
# find the second Sign In button
#
# b = e_button 'Sign In', 2
#
# Button order will change in iOS vs Android
# so if there's no button found at number then
# return the first button.
#
# @param text [String] the text to match
# @param number [Integer] the button occurance to return. 1 = first button
# @return [Button] the button that matches text and number
def button_num text, number=1
  raise "Number must be >= 1" if number <= 0
  number = number - 1 # zero indexed

  result = nil

  elements = buttons text
  elements.size > number ? result = elements[number]
                         : result = elements.first

  result
end