require 'open-uri'
require 'nokogiri'

#
# Module method definitions
#
module GSearchParser

  # Entry method for performing a web search
  def GSearchParser.webSearch(query)
    webSearch = GoogleWebSearch.new(query)
  end

end

#
# Google Web Search class
#
class GoogleWebSearch
  attr_accessor :results, :nextURI
  @currentPage
  
  # Class initializer
  def initialize(query)
    # Initialize variables
    @results = Array.new

    # Update the results list: (Fetch, Store, and Parse)
    updateResults("http://google.com/search?sourceid=chrome&q=#{query}")
  end

  # Update the WebSearch results array by performing a Fetch, Store, Parse routine
  def updateResults(url)
    # Fetch
    searchPage = fetchPage(url)

    # Store
    @currentPage = searchPage

    # Parse
    parseCurrentPage
  end

  # Fetch the page from a URL
  def fetchPage(url)
    Nokogiri::HTML(open(url, 'User-Agent' => 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.19 (KHTML, like Gecko) Chrome/18.0.1025.152 Safari/535.19'))
  end

  # Parse the current page and populate results
  def parseCurrentPage
    # Initialize local variables
    currentResults = Array.new

    # Iterate over each Google result list element 
    @currentPage.css('li.g').each do |result|
      # Extract the title
      title = result.css('h3.r a').first.inner_html

      # Extract the content. There is the possibility for
      # the content to be nil, so check for this
      content = result.css('span.st').first.nil? ? '' : result.css('span.st').first.inner_html

      # Extract the URI
      uri = result.css('cite').first.inner_html

      # Ignore YouTube videos for websearch
      unless uri.index('www.youtube.com').nil? 
        next
      end

      # Create a new Result object and append to the array
      currentResults << Result.new(title, content, uri)
    end
    @results += currentResults
    return currentResults
  end

  # Parse the results from the next page and append to results list
  def nextResults
    # Parse next result page link from the currently marked one
    nextPagePath = @currentPage.at_css("table#nav tr td.cur").next_sibling().at_css("a")['href']

    # Construct the URI
    @nextURI = "http://www.google.com" + nextPagePath

    # Update results
    updateResults(@nextURI)
  end

  # Iterator over results
  def each(&blk)
    @results.each(&blk)
  end

end # GoogleWebSearch

#
# Result class
#
class Result
  attr_accessor :title, :content, :uri

  # Class initializer
  def initialize(title, content, uri)
    @title = title
    @content = content
    @uri = uri
  end

end # Result