#$ neutron_plugin 01

import thread

import time

LOCALDB_FILE = 'dynamic/localdb.txt'

def handler_userinfo_join(groupchat, nick):

	GROUPCHATS[groupchat][nick] = {'idle': time.time(), 'jid': groupchat + '/' + nick, 'fullname': None, 'nickname': None, 'email': None, 'url': None, 'client': None, 'version': None, 'os': None}

	browse_iq = xmpp.Iq(groupchat + '/' + nick, 'get')

	browse_iq.setQueryNS('jabber:iq:browse')

	JCON.send(browse_iq)

	time.sleep(2)

	version_iq = xmpp.Iq(groupchat + '/' + nick, 'get')

	version_iq.setQueryNS('jabber:iq:version')

	JCON.send(version_iq)

	time.sleep(2)

	vcard_iq = xmpp.Iq(groupchat + '/' + nick, 'get')

	vcard_iq.setQueryNS('vcard-temp')

	JCON.send(vcard_iq)

def handler_userinfo_message(type, source, body):

	if type == 'public':

		if GROUPCHATS.has_key(source[1]) and GROUPCHATS[source[1]].has_key(source[2]):

			GROUPCHATS[source[1]][source[2]]['idle'] = time.time()

def handler_userinfo_iq(iq):

	type = iq.getType()

	namespace = iq.getQueryNS()

	if not namespace:

		if iq.getTag('user'):

			namespace = iq.getTag('user').getNamespace()

		elif iq.getTag('VCARD') or iq.getTag('vCard') or iq.getTag('vcard'):

			vcard = iq.getTag('VCARD') or iq.getTag('vCard') or iq.getTag('vcard')

			namespace = vcard.getNamespace()

	groupchat = iq.getFrom().getStripped()

	nick = iq.getFrom().getResource()

	if GROUPCHATS.has_key(groupchat) and GROUPCHATS[groupchat].has_key(nick):

		if type == 'result':

			if namespace == 'jabber:iq:browse':

				if iq.getTag('user'):

					GROUPCHATS[groupchat][nick]['jid'] = iq.getTag('user').getTag('user').getAttr('jid')

					#fp = open('userlog.txt', 'a')

					#fp.write(groupchat + '/' + nick + '\n' + GROUPCHATS[groupchat][nick]['jid'] + '\n\n')

					#fp.close()

					#if string.split(GROUPCHATS[groupchat][nick]['jid'], '/')[0] in ADMINS:

					fp = open(LOCALDB_FILE, 'r')

					localdb = eval(fp.read())

					fp.close()

					if GROUPCHATS[groupchat][nick]['jid'] and localdb.has_key(string.split(GROUPCHATS[groupchat][nick]['jid'], '/')[0]):

						msg(groupchat, nick + ': ' + localdb[string.split(string.split(GROUPCHATS[groupchat][nick]['jid'], '/')[0], '/')[0]])

					#elif localdb.has_key(nick):

					#	msg(groupchat, nick + ': ' + localdb[nick])

					#else:

					#	msg(groupchat, nick + ': ' + 'Welcome')

			elif namespace == 'vcard-temp':

				vcard = iq.getTag('VCARD') or iq.getTag('vCard') or iq.getTag('vcard')

				if vcard.getTag('FN'):

					GROUPCHATS[groupchat][nick]['fullname'] = vcard.getTag('FN').getData()

				elif vcard.getTag('N') and vcard.getTag('N').getTag('GIVEN') and vcard.getTag('N').getTag('FAMILY'):

					GROUPCHATS[groupchat][nick]['fullname'] = vcard.getTag('N').getTag('GIVEN').getData() + ' ' + vcard.getTag('N').getTag('FAMILY').getData()

				if vcard.getTag('NICKNAME'):

					GROUPCHATS[groupchat][nick]['nickname'] = vcard.getTag('NICKNAME').getData()

				if vcard.getTag('EMAIL'):

					GROUPCHATS[groupchat][nick]['email'] = vcard.getTag('EMAIL').getData()

				if vcard.getTag('URL'):

					GROUPCHATS[groupchat][nick]['url'] = vcard.getTag('URL').getData()

			elif namespace == 'jabber:iq:version':

				info = iq.getTag('query')

				GROUPCHATS[groupchat][nick]['client'] = info.getTag('name').getData()

				GROUPCHATS[groupchat][nick]['version'] = info.getTag('version').getData()

				GROUPCHATS[groupchat][nick]['os'] = info.getTag('os').getData()

def handler_userinfo_whois(type, source, parameters):

	groupchat = source[1]

	if GROUPCHATS.has_key(groupchat) and GROUPCHATS[groupchat].has_key(parameters):

		info = GROUPCHATS[groupchat][parameters]

		reply = ''

		if info.has_key('jid') and info['jid']:

			reply += '<' + info['jid'] + '> '

		if info.has_key('fullname') and info['fullname']:

			reply += info['fullname'] + ' '

		if info.has_key('nickname') and info['nickname']:

			reply += '(' + info['nickname'] + ') '

		if info.has_key('email') and info['email']:

			reply += '<' + info['email'] + '> '

		if info.has_key('client') and info['client']:

			reply += '- ' + info['client'] + ' ' + info['version'] + ' - ' + info['os']

		reply += ' [' + parameters + ']'

		smsg(type, source, reply)

	else:

		smsg(type, source, 'User Not In Chat [' + parameters + ']')

def handler_userinfo_idle(type, source, parameters):

	if GROUPCHATS.has_key(source[1]) and GROUPCHATS[source[1]].has_key(parameters):

		nick = parameters

		groupchat = source[1]

		idletime = int(time.time() - GROUPCHATS[groupchat][nick]['idle'])

		reply = ''

		seconds = idletime % 60

		minutes = (idletime / 60) % 60

		hours = (idletime / 3600) % 60

		days = idletime / 216000

		if days: reply += str(days) + 'd '

		if hours: reply += str(hours) + 'h '

		if minutes: reply += str(minutes) + 'm '

		reply += str(seconds) + 's' + ' [' + parameters + ']'

	else:

		reply = 'Unknown' + ' [' + parameters + ']'

	smsg(type, source, reply)

def handler_userinfo_probe(type, source, parameters):

	iq = xmpp.Iq(parameters, 'get')

	iq.setQueryNS('jabber:iq:browse')

	response = JCON.SendAndWaitForResponse(iq)

	if response:

		iqtype = response.getType()

	else:

		iqtype = 'error'

	if iqtype == 'result':

		reply = 'Online [' + parameters + ']'

	else:

		reply = 'Offline [' + parameters + ']'

	smsg(type, source, reply)

def handler_userinfo_getlast(type, source, parameters):

	if GROUPCHATS.has_key(source[1]) and GROUPCHATS[source[1]].has_key(parameters):

		nick = parameters

		groupchat = source[1]

		last_iq = xmpp.Iq(groupchat + '/' + nick, 'get')

		last_iq.setQueryNS('jabber:iq:last')

		last_result = JCON.SendAndWaitForResponse(last_iq)

		if last_result and last_result.getType() == 'result':

			idletime = int(last_result.getTag('query').getAttr('seconds'))

			reply = ''

			seconds = idletime % 60

			minutes = (idletime / 60) % 60

			hours = (idletime / 3600) % 60

			days = idletime / 216000

			if days: reply += str(days) + 'd '

			if hours: reply += str(hours) + 'h '

			if minutes: reply += str(minutes) + 'm '

			reply += str(seconds) + 's' + ' [' + parameters + ']'

		else:

			reply = 'Unknown' + ' [' + parameters + ']'

	else:

		reply = 'Unknown'

	smsg(type, source, reply)

def handler_userinfo_gettime(type, source, parameters):

	if GROUPCHATS.has_key(source[1]) and GROUPCHATS[source[1]].has_key(parameters):

		nick = parameters

		groupchat = source[1]

		time_iq = xmpp.Iq(groupchat + '/' + nick, 'get')

		time_iq.setQueryNS('jabber:iq:time')

		time_result = JCON.SendAndWaitForResponse(time_iq)

		if time_result and time_result.getType() == 'result':

			timestring = time_result.getTag('query').getTag('display').getData()

			timezone = time_result.getTag('query').getTag('tz').getData()

			reply = timestring + ' - ' + timezone + ' [' + parameters + ']'

		else:

			reply = 'Unknown' + ' [' + parameters + ']'

	else:

		reply = 'Unknown' + ' [' + parameters + ']'

	smsg(type, source, reply)

register_join_handler(handler_userinfo_join)

register_message_handler(handler_userinfo_message)

register_iq_handler(handler_userinfo_iq)

register_command_handler(handler_userinfo_whois, '!whois', 0, 'Gives information on specified groupchat user.', '!whois [nick]', ['!whois mikem'])

register_command_handler(handler_userinfo_idle, '!idle', 0, 'Gives the idle time of a groupchat nick.', '!idle [nick]', ['!idle mikem'])

register_command_handler(handler_userinfo_probe, '!probe', 0, 'Tells if a server, agent, or user is online.', '!probe [JID]', ['!probe mikem@jabber.org', '!probe jabber.org', '!probe msn.jabber.org'])

register_command_handler(handler_userinfo_getlast, '!getlast', 0, 'Gives the iq:last idle reply of specified nick in groupchat.', '!getlast [nick]', ['!getlast mikem'])

register_command_handler(handler_userinfo_gettime, '!gettime', 0, 'Gives the iq:time time reply of specified nick in groupchat.', '!gettime [nick]', ['!gettime mikem'])