#$ neutron_plugin 01

CURRENT_POLL = {}

def handler_vote_vote(type, source, parameters):
	global CURRENT_POLL
	if GROUPCHATS.has_key(source[1]) and GROUPCHATS[source[1]].has_key(source[2]):
		if CURRENT_POLL:
			if isadmin(GROUPCHATS[source[1]][source[2]]['jid']) or not GROUPCHATS[source[1]][source[2]]['jid'] in CURRENT_POLL['jids']:
				if CURRENT_POLL['options'].has_key(parameters.strip().lower()):
					CURRENT_POLL['options'][parameters.strip().lower()] += 1
					CURRENT_POLL['jids'].append(GROUPCHATS[source[1]][source[2]]['jid'])
					smsg(type, source, 'Vote Counted')
				else:
					smsg(type, source, 'Option Not Available')
			else:
				smsg(type, source, 'You already voted.')
		else:
			smsg(type, source, 'There\'s no poll going on right now.')
	else:
		smsg(type, source, 'You must vote from the groupchat.')

def handler_vote_newpoll(type, source, parameters):
	global CURRENT_POLL
	if CURRENT_POLL:
		poll_text = 'New Poll (' + source[2] + ') - ' + CURRENT_POLL['question'] + '\n'
		for option in CURRENT_POLL['options'].keys():
			poll_text += '  *  ' + option + '\n'
		poll_text += 'To vote, type or message me the following: !vote <option>'
		msg(source[1], poll_text)
	else:
		CURRENT_POLL = {'options': {}, 'question': parameters, 'jids': []}
		smsg(type, source, 'Poll Created')

def handler_vote_polloption(type, source, parameters):
	global CURRENT_POLL
	if CURRENT_POLL:
		CURRENT_POLL['options'][parameters.strip().lower()] = 0
		smsg(type, source, 'Option Added')
	else:
		smsg(type, source, 'Error: No Poll')

def handler_vote_endpoll(type, source, parameters):
	global CURRENT_POLL
	if CURRENT_POLL:
		poll_text = 'Poll Results (' + source[2] + ') - ' + CURRENT_POLL['question'] + '\n'
		num = 1
		for option in CURRENT_POLL['options'].keys():
			poll_text += str(CURRENT_POLL['options'][option]) + ' - ' + option + '\n'
			num += 1
		msg(source[1], poll_text)
		CURRENT_POLL = {}
	else:
		smsg(type, source, 'Error: No Poll')

register_command_handler(handler_vote_vote, '!vote', 0, 'Casts a vote for current poll.', '!vote <option>', ['!vote yes'])
register_command_handler(handler_vote_newpoll, '!newpoll', 100, 'Creates a new poll, or if options are added, submits the poll to current channel.', '!newpoll [question]', ['!newpoll Do you like turtles?', '!newpoll'])
register_command_handler(handler_vote_polloption, '!polloption', 100, 'Adds an option to the current poll.', '!polloption <option>', ['!polloption yes'])
register_command_handler(handler_vote_endpoll, '!endpoll', 100, 'Ends the poll and returns the results.', '!endpoll]', ['!endpoll'])