#$ neutron_plugin 01

import babelizer

import russian

def handler_babel_babel(type, source, parameters):

	splitdata = string.split(parameters)

	if len(splitdata) >= 3:

		from_lang = splitdata[0]

		to_lang = splitdata[1]

		body = string.join(splitdata[2:])

		try:

			reply = babelizer.translate(body, from_lang, to_lang)

		except babelizer.LanguageNotAvailableError:

			reply = 'Invalid Language'

		except babelizer.BabelfishChangedError:

			print str(sys.exc_info()[0]) + ' - ' + str(sys.exc_info()[1])

			reply = 'Unknown Error'

	else:

		reply = 'Syntax Error'

	if not reply:

		reply = 'No Results'

	reply = unicode(reply, 'utf-8')

	smsg(type, source, reply)

def handler_babel_tre(type, source, parameters):

	handler_babel_babel(type, source, u'russian english ' + parameters)

def handler_babel_translate(type, source, parameters):

	splitdata = string.split(parameters)

	if len(splitdata) >= 3:

		from_lang = splitdata[0]

		to_lang = splitdata[1]

		body = string.join(splitdata[2:])

		try:

			reply = russian.translate(body, from_lang, to_lang)

		except russian.LanguageNotAvailableError:

			reply = 'Invalid Language'

		except russian.BabelfishChangedError:

			print str(sys.exc_info()[0]) + ' - ' + str(sys.exc_info()[1])

			reply = 'Unknown Error'

	else:

		reply = 'Syntax Error'

	if not reply:

		reply = 'No Results'

	smsg(type, source, reply)

def handler_babel_enru(type, source, parameters):

	handler_babel_translate(type, source, u'e r ' + parameters)

def handler_babel_babelize(type, source, parameters):

	splitdata = string.split(parameters)

	if len(splitdata) >= 3:

		from_lang = splitdata[0]

		through_lang = splitdata[1]

		body = string.join(splitdata[2:])

		try:

			results = babelizer.babelize(body, from_lang, through_lang)

			reply = ''

			for result in results:

				reply += '\n' + result.decode('utf-8', 'replace')

		except babelizer.LanguageNotAvailableError:

			reply = 'Invalid Language'

		except babelizer.BabelfishChangedError:

			print str(sys.exc_info()[0]) + ' - ' + str(sys.exc_info()[1])

			reply = 'Unknown Error'

	else:

		reply = 'Syntax Error'

	if not reply:

		reply = 'No Results'

	smsg(type, source, reply)

def handler_babel_esperanto(type, source, parameters):

	formdata = 'al=2&de=3&al-k=4&teksto=' + urllib.quote(parameters)

	import socket

	s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

	s.connect(('lingvo.org', 80))

	s.send("""POST /cgi-bin/eoxx/eoxx HTTP/1.1

User-Agent: Mozilla/5.0 (compatible; Konqueror/3; Linux 2.4.18)

Pragma: no-cache

Cache-control: no-cache

Accept: text/*, image/jpeg, image/png, image/*, */*

Accept-Encoding: x-gzip, gzip, identity

Accept-Charset: ISO-8859-1, utf-8;q=0.5, *;q=0.5

Accept-Language: en, POSIX

Host: lingvo.org

Content-Type: application/x-www-form-urlencoded

Content-Length: """ + str(len(formdata)) + '\n\n' + formdata + '\n\n')

	pagedata = s.recv(1024)

	s.close()

	#formdata = urllib.urlencode({'al': '2', 'de': '3', 'al-k': '4', 'teksto': parameters})

	#f = urllib.urlopen('http://lingvo.org/cgi-bin/eoxx/eoxx', formdata)

	#pagedata = f.read()

	#print pagedata

	try:

		reply = string.split(string.split(pagedata, 'lang="en">')[1], '</font>')[0].strip()

	except:

		reply = 'Unknown Error'

	smsg(type, source, reply)

register_command_handler(handler_babel_babel, '!babel', 0, 'Translate a phrase from one language to another using AltaVista BabelFish.', '!babel <fromlang> <tolang> <phrase>', ['!babel english spanish hello', '!babel fr en bonjour'])

register_command_handler(handler_babel_translate, '!translate', 0, 'Translate a phrase from one language to another using translate.ru.', '!translate <fromlang> <tolang> <phrase>', ['!translate e s hello', '!translate f e bonjour'])

# examples: register_command_handler(handler_babel_tre, '!tre', 0, 'Translate a phrase from Russian to English (shortcut).', '!tre <phrase>', ['!tre ????????'])

# register_command_handler(handler_babel_enru, '!enru', 0, 'Translate a phrase from English to Russian (shortcut).', '!enru <phrase>', ['!enru the tortoise'])

# register_command_handler(handler_babel_esperanto, '!esperanto', 0, 'Translate a phrase from Esperanto to English.', '!esperanto <phrase>', ['!esperanto Saluton!'])

#register_command_handler(handler_babel_babelize, '!babelize', 0, 'Translate a phrase back and forth from one language to another (gives interesting results).', '!babelize <fromlang> <throughlang> <phrase>', ['!babelize english spanish hello'])