require 'page-object/elements'

module PageObject
  #
  # Watir implementation of the page object platform driver.  You should not use the
  # class directly.  Instead you should include the PageObject module in your page object
  # and use the methods dynamically added from the PageObject::Accessors module.
  #
  class WatirPageObject
    def initialize(browser)
      @browser = browser      
    end

    #
    # platform method to navigate to a provided url
    # See PageObject#navigate_to
    #
    def navigate_to(url)
      @browser.goto url
    end

    #
    # platform method to retrieve the text from the current page
    # See PageObject#text
    #
    def text
      @browser.text
    end

    #
    # platform method to retrieve the html for the current page
    # See PageObject#html
    #
    def html
      @browser.html
    end

    #
    # platform method to retrieve the title for the current page
    # See PageObject#title
    #
    def title
      @browser.title
    end

    #
    # platform method to get the value stored in a text field
    # See PageObject::Accessors#text_field
    #
    def text_field_value_for(identifier)
      identifier = Elements::TextField.watir_identifier_for identifier
      @browser.text_field(identifier).value
    end
    
    #
    # platform method to set the value for a text field
    # See PageObject::Accessors#text_field
    #
    def text_field_value_set(identifier, value)
      identifier = Elements::TextField.watir_identifier_for identifier
      @browser.text_field(identifier).set(value)
    end
    
    #
    # platform method to retrieve a text field element
    # See PageObject::Accessors#text_field
    #
    def text_field_for(identifier)
      identifier = Elements::TextField.watir_identifier_for identifier
      element = @browser.text_field(identifier)
      Elements::TextField.new(element, :platform => :watir)
    end
    
    #
    # platform method to get the value stored in a hidden field
    # See PageObject::Accessors#hidden_field
    #
    def hidden_field_value_for(identifier)
      identifier = Elements::HiddenField.watir_identifier_for identifier
      @browser.hidden(identifier).value
    end
    
    #
    # platform method to retrieve a hidden field element
    # See PageObject::Accessors#hidden_field
    #
    def hidden_field_for(identifier)
      identifier = Elements::HiddenField.watir_identifier_for identifier
      element = @browser.hidden(identifier)
      Elements::HiddenField.new(element, :platform => :watir)
    end    
    
    #
    # platform method to set text in a textarea
    # See PageObject::Accessors#text_area
    #
    def text_area_value_set(identifier, value)
      identifier = Elements::TextArea.watir_identifier_for identifier
      @browser.textarea(identifier).send_keys(value)
    end
    
    #
    # platform method to get the text from a textarea
    # See PageObject::Accessors#text_area
    #
    def text_area_value_for(identifier)
      identifier = Elements::TextArea.watir_identifier_for identifier
      @browser.textarea(identifier).value
    end
    
    #
    # platform method to get the text area element
    # See PageObject::Accessors#text_area
    #
    def text_area_for(identifier)
      identifier = Elements::TextArea.watir_identifier_for identifier
      element = @browser.textarea(identifier)
      Elements::TextArea.new(element, :platform => :watir)
    end
    
    #
    # platform method to get the currently selected value from a select list
    # See PageObject::Accessors#select_list
    #
    def select_list_value_for(identifier)
      identifier = Elements::SelectList.watir_identifier_for identifier
      @browser.select_list(identifier).value
    end
    
    #
    # platform method to select a value from a select list
    # See PageObject::Accessors#select_list
    #
    def select_list_value_set(identifier, value)
      identifier = Elements::SelectList.watir_identifier_for identifier
      @browser.select_list(identifier).select(value)
    end
    
    #
    # platform method to return the select list element
    # See PageObject::Accessors#select_list
    #
    def select_list_for(identifier)
      identifier = Elements::SelectList.watir_identifier_for identifier
      element = @browser.select_list(identifier)
      Elements::SelectList.new(element, :platform => :watir)
    end
        
    #
    # platform method to click a link
    # See PageObject::Accessors#link
    #
    def click_link_for(identifier)
      identifier = Elements::Link.watir_identifier_for identifier
      @browser.link(identifier).click if identifier
    end
    
    #
    # platform method to return a PageObject::Elements::Link object
    # see PageObject::Accessors#link
    #
    def link_for(identifier)
      identifier = Elements::Link.watir_identifier_for identifier
      element = @browser.link(identifier)
      Elements::Link.new(element, :platform => :watir)
    end

    #
    # platform method to check a checkbox
    # See PageObject::Accessors#checkbox
    #
    def check_checkbox(identifier)
      identifier = Elements::CheckBox.watir_identifier_for identifier
      @browser.checkbox(identifier).set
    end

    #
    # platform method to uncheck a checkbox
    # See PageObject::Accessors#checkbox
    #
    def uncheck_checkbox(identifier)
      identifier = Elements::CheckBox.watir_identifier_for identifier
      @browser.checkbox(identifier).clear
    end

    #
    # platform method to determine if a checkbox is checked
    # See PageObject::Accessors#checkbox
    #
    def checkbox_checked?(identifier)
      identifier = Elements::CheckBox.watir_identifier_for identifier
      @browser.checkbox(identifier).set?
    end
    
    #
    # platform method to return a PageObject::Elements::CheckBox element
    # See PageObject::Accessors#checkbox
    #
    def checkbox_for(identifier)
      identifier = Elements::CheckBox.watir_identifier_for identifier
      element = @browser.checkbox(identifier)
      Elements::CheckBox.new(element, :platform => :watir)
    end

    #
    # platform method to select a radio button
    # See PageObject::Accessors#radio_button
    #
    def select_radio(identifier)
      identifier = Elements::RadioButton.watir_identifier_for identifier
      @browser.radio(identifier).set
    end

    #
    # platform method to clear a radio button
    # See PageObject::Accessors#radio_button
    #
    def clear_radio(identifier)
      identifier = Elements::RadioButton.watir_identifier_for identifier
      @browser.radio(identifier).clear
    end

    #
    # platform method to determine if a radio button is selected
    # See PageObject::Accessors#radio_button
    #
    def radio_selected?(identifier)
      identifier = Elements::RadioButton.watir_identifier_for identifier
      @browser.radio(identifier).set?
    end
    
    #
    # platform method to return a PageObject::Eements::RadioButton element
    # See PageObject::Accessors#radio_button
    #
    def radio_button_for(identifier)
      identifier = Elements::RadioButton.watir_identifier_for identifier
      element = @browser.radio(identifier)
      PageObject::Elements::RadioButton.new(element, :platform => :watir)
    end
    
    #
    # platform method to return the text for a div
    # See PageObject::Accessors#div
    #
    def div_text_for(identifier)
      identifier = add_tagname_if_needed identifier, "div"
      identifier = Elements::Div.watir_identifier_for identifier
      @browser.div(identifier).text
    end
    
    #
    # platform method to return a PageObject::Elements::Div element
    # See PageObject::Accessors#div
    #
    def div_for(identifier)
      identifier = add_tagname_if_needed identifier, "div"
      identifier = Elements::Div.watir_identifier_for identifier
      element = @browser.div(identifier)
      PageObject::Elements::Div.new(element, :platform => :watir)
    end
    
    #
    # platform method to return the text for a span
    # See PageObject::Accessors#span
    #
    def span_text_for(identifier)
      identifier = add_tagname_if_needed identifier, "span"
      identifier = Elements::Span.watir_identifier_for identifier
      @browser.span(identifier).text
    end
    
    #
    # platform method to return a PageObject::Elements::Span element
    # See PageObject::Accessors#span
    #
    def span_for(identifier)
      identifier = add_tagname_if_needed identifier, "span"
      identifier = Elements::Span.watir_identifier_for identifier
      element = @browser.span(identifier)
      PageObject::Elements::Span.new(element, :platform => :watir)
    end
    
    #
    # platform method to click a button
    # See PageObject::Accessors#button
    #
    def click_button_for(identifier)
      identifier = Elements::Button.watir_identifier_for identifier
      @browser.button(identifier).click
    end
    
    #
    # platform method to retrieve a button element
    # See PageObject::Accessors#button
    #
    def button_for(identifier)
      identifier = Elements::Button.watir_identifier_for identifier
      element = @browser.button(identifier)
      PageObject::Elements::Button.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve a table element
    # See PageObject::Accessors#table
    #
    def table_for(identifier)
      identifier = add_tagname_if_needed identifier, "table"
      identifier = Elements::Table.watir_identifier_for identifier.clone
      element = @browser.table(identifier)
      PageObject::Elements::Table.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve the text from a table cell
    # See PageObject::Accessors#cell
    #
    def cell_text_for(identifier)
      identifier = add_tagname_if_needed identifier, "td"
      identifier = Elements::TableCell.watir_identifier_for identifier
      @browser.td(identifier).text
    end
    
    #
    # platform method to retrieve a table cell element
    # See PageObject::Accessors#cell
    #
    def cell_for(identifier)
      identifier = add_tagname_if_needed identifier, "td"
      identifier = Elements::TableCell.watir_identifier_for identifier
      element = @browser.td(identifier)
      PageObject::Elements::TableCell.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve an image element
    # See PageObject::Accessors#image
    #
    def image_for(identifier)
      identifier = Elements::Image.watir_identifier_for identifier
      element = @browser.image(identifier)
      PageObject::Elements::Image.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve a form element
    # See PageObject::Accessors#form
    #
    def form_for(identifier)
      identifier = Elements::Form.watir_identifier_for identifier
      element = @browser.form(identifier)
      PageObject::Elements::Form.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve the text from a list item
    # See PageObject::Accessors#list_item
    #
    def list_item_text_for(identifier)
      identifier = add_tagname_if_needed identifier, "li"
      identifier = Elements::ListItem.watir_identifier_for identifier
      @browser.li(identifier).text
    end
    
    #
    # platform method to retrieve a list item element
    # See PageObject::Accessors#list_item
    #
    def list_item_for(identifier)
      identifier = add_tagname_if_needed identifier, "li"
      identifier = Elements::ListItem.watir_identifier_for identifier
      element = @browser.li(identifier)
      PageObject::Elements::ListItem.new(element, :platform => :watir)
    end

    #
    # platform method to retrieve an unordered list element
    # See PageObject::Accessors#unordered_list
    #
    def unordered_list_for(identifier)
      identifier = add_tagname_if_needed identifier, "ul"
      identifier = Elements::UnorderedList.watir_identifier_for identifier
      element = @browser.ul(identifier)
      PageObject::Elements::UnorderedList.new(element, :platform => :watir)
    end
    
    #
    # platform method to retrieve an ordered list element
    # See PageObject::Accessors#ordered_list
    #
    def ordered_list_for(identifier)
      identifier = add_tagname_if_needed identifier, "ol"
      identifier = Elements::OrderedList.watir_identifier_for identifier
      element = @browser.ol(identifier)
      PageObject::Elements::OrderedList.new(element, :platform => :watir)
    end

    private
    
    def add_tagname_if_needed identifier, tag
      return identifier if identifier.length < 2 and not identifier[:name]
      identifier[:tag_name] = tag if identifier[:name]
      identifier
    end
  end
end