module GenericTest
  # Represents a web page state at a particular point of time
  class Page
    # @return [Array] List of links
    attr_accessor :links
    # @return [Array] List of emails
    attr_accessor :emails
    attr_accessor :url
    attr_accessor :html
    attr_accessor :title

    # Store state of page when it is loaded
    # @param [Watir::Browser] browser Watir browser
    def initialize(browser)
      self.emails, self.links = browser.links.partition do |link|
        link.href.start_with?('mailto:')
      end
      links.reject! { |link| link.href.empty? }
      emails.collect! { |link| link.href.split(':').last }
      self.url = browser.url
      self.html = browser.html
      self.title = browser.title
      puts "Found #{links.count} links, #{emails.count} emails at #{url} (#{title})"
    end
  end
end