require 'cinch'
require 'ostruct'
require 'open-uri'
require 'json'
require 'cgi'

module Cinch
  module Plugins
    class Google
      include Cinch::Plugin

      set :prefix, /^~/
      set :plugin_name, 'google'
      set :help, <<-USAGE.gsub(/^ {6}/, '')
        Searches Google for results on image and web entries.
          Usage:
            * ~google <terms>: This will execute a Google WEB search, and return the top three results.
            * ~images <terms>: This will execute a Google Image search and returns the top three results.
          USAGE

      match /google (.+?) (.+)/i, method: :execute_w
      match /images (.+)/i, method: :execute_i

      def execute_w(m, query)
        query.gsub! /\s/, '+'
        data = search_w(m, query)
        return m.reply "No results found for #{query}." if data.empty?
        search_result(m, data)
      end

      def execute_i(m, query)
        query.gsub! /\s/, '+'
        data = search_i(m, query)
        return m.reply "No results found for #{query}." if data.empty?
        search_result(m, data)
      end

      def search_w(m, terms)
        logo = "12G4o8o12g9l4e"
        data = JSON.parse(open("http://ajax.googleapis.com/ajax/services/search/web?v=1.0&q=#{terms}").read)
        rdata = data['responseData']['results']
        results = []

        rdata.each{|i| results.push("%s: %s [ %s ]" % [logo, i['titleNoFormatting'], i['unescapedUrl']])}

        return results
      end

      def search_i(m, terms)
        logo = "12I4m8a12g9e4s"
        data = JSON.parse(open("http://ajax.googleapis.com/ajax/services/search/images?v=1.0&q=#{terms}").read)
        rdata = data['responseData']['results']
        results = []

        rdata.each{|i| results.push("%s: %s (%sx%s)" % [logo, i['unescapedUrl'], i['width'], i['height']])}

        return results
      end

      def search_result(m, data)
        data[0..2].each{|i| m.reply i}
      end

    end
  end
end