googlecomparer.rb

module ProPublica
  class GoogleComparer
    BASE_URL = "https://ajax.googleapis.com/ajax/services/search/web?v=1.0&q="
    WEB_TITLE_DELIMITERS = /(-|\|).*$/
    
    def initialize(facility_name)
      @facility_name = facility_name
      url_encoded_facility_name = CGI.escape(@facility_name)
      @results = JSON.parse(RestClient.get(BASE_URL + url_encoded_facility_name))['responseData']['results']
    end
    
    def strip_goog(str)
      str.gsub(/<\/?[^>]*>/, "").gsub(WEB_TITLE_DELIMITERS,'')
    end
    
    def penalize_uppercase(s1, s2)
      expected_uppercase = s1.split(/ /).size
      gotten_uppercase = s2.scan(/([A-Z])/).size
      penalty = gotten_uppercase - expected_uppercase
    end
    
    def parse
      @evaluated_results = {}
      @results.each_with_index do |result,idx|
        evaluated_result = Levenshtein.distance(@facility_name, strip_goog(result['title'].upcase)) + 
                           penalize_uppercase(@facility_name, strip_goog(result['title']))
        @evaluated_results[evaluated_result] = strip_goog(result['title'])
      end
      @evaluated_results
    end
    
    def winner_value
      @evaluated_results.min[1].strip
    end

  end
end