#$ neutron_plugin 01

import google

def google_remove_html(text):

	nobold = text.replace('<b>', '').replace('</b>', '')

	nobreaks = nobold.replace('<br>', ' ')

	noescape = nobreaks.replace('&amp;', '&').replace('&lt;', '<').replace('&gt;', '>').replace('&quot;', '"')

	return noescape

def google_search(query):

	data = google.doGoogleSearch(query)

	try:

		first = data.results[0]

		url = first.URL

		title = google_remove_html(first.title)

		if first.summary:

			summary = google_remove_html(first.summary)

		else:

			summary = google_remove_html(first.snippet)

		searchtime = str(round(data.meta.searchTime, 3))

		total = str(data.meta.estimatedTotalResultsCount)

		return url + ' - ' + title + ' - ' + summary + ' (' + searchtime + 'sec) (' + total + ' sites)'

	except:

		return 'No Results'

def handler_google_google(type, source, parameters):

	results = google_search(parameters)

	smsg(type, source, results)

def handler_google_spell(type, source, parameters):

	correction = google.doSpellingSuggestion(parameters)

	if not correction:

		correction = 'No Suggestion'

	smsg(type, source, correction)

def handler_google_jepsearch(type, source, parameters):

	results = google_search(parameters + ' site:www.xmpp.org "/exensions/xep-"')

	smsg(type, source, results)

register_command_handler(handler_google_google, '!google', 0, 'Looks up search terms on google.', '!google <query>', ['!google "mike mintz"'])

register_command_handler(handler_google_spell, '!spell', 0, 'Returns a spelling suggestion from google.', '!spell <query>', ['!spell "pithon nutron"'])

register_command_handler(handler_google_jepsearch, '!jepsearch', 0, 'Searches google for a JEP.', '!jep <query>', ['!google "jep-0001"'])