#$ neutron_plugin 01

QUIZ_FILE = 'static/quizdata.txt'
QUIZ_TOTAL_LINES = 16197
QUIZ_TIME_LIMIT = 60
QUIZ_IDLE_LIMIT = 5

QUIZ_RECURSIVE_MAX = 20
QUIZ_SCORES = {}
QUIZ_CURRENT_ANSWER = {}
QUIZ_CURRENT_HINT = {}
QUIZ_CURRENT_TIME = {}
QUIZ_IDLENESS = {}

def quiz_timer(groupchat, start_time):
	time.sleep(QUIZ_TIME_LIMIT)
	if QUIZ_CURRENT_TIME.has_key(groupchat) and QUIZ_CURRENT_ANSWER.has_key(groupchat) and start_time == QUIZ_CURRENT_TIME[groupchat]:
		QUIZ_CURRENT_ANSWER[groupchat]
		msg(groupchat, '[QUIZ] Time is up! ' + str(QUIZ_TIME_LIMIT) + ' seconds have elapsed. The correct answer was: ' + QUIZ_CURRENT_ANSWER[groupchat])
		if QUIZ_IDLENESS.has_key(groupchat):
			QUIZ_IDLENESS[groupchat] += 1
		else:
			QUIZ_IDLENESS[groupchat] = 1
		if QUIZ_IDLENESS[groupchat] >= QUIZ_IDLE_LIMIT:
			msg(groupchat, '[QUIZ] Quiz has automatically ended due to inactivity for ' + str(QUIZ_IDLE_LIMIT) + ' questions. Please type !quiz_start to restart the quiz.')
			del QUIZ_CURRENT_ANSWER[groupchat]
			quiz_list_scores(groupchat)
		else:
			quiz_ask_question(groupchat)

def quiz_new_question():
	line_num = random.randrange(16197)
	fp = file(QUIZ_FILE)
	for n in range(line_num + 1):
		if n == line_num:
			try:
				(question, answer) = string.split(fp.readline().strip(), '|', 1)
				return (unicode(question), unicode(answer))
			except:
				QUIZ_RECURSIVE_MAX -= 1
				if QUIZ_RECURSIVE_MAX:
					return quiz_new_question()
				else:
					QUIZ_RECURSIVE_MAX = 20
					return ('Parsing Error: Line ' + str(n), '')
		else:
			fp.readline()

def quiz_ask_question(groupchat):
	(question, answer) = quiz_new_question()
	QUIZ_CURRENT_ANSWER[groupchat] = answer
	QUIZ_CURRENT_HINT[groupchat] = None
	QUIZ_CURRENT_TIME[groupchat] = time.time()
	thread.start_new(quiz_timer, (groupchat, QUIZ_CURRENT_TIME[groupchat]))
	msg(groupchat, '[QUIZ] New Question: ' + question)

def quiz_answer_question(groupchat, nick, answer):
	if QUIZ_CURRENT_ANSWER.has_key(groupchat):
		if QUIZ_CURRENT_ANSWER[groupchat] == answer:
			if QUIZ_IDLENESS.has_key(groupchat):
				del QUIZ_IDLENESS[groupchat]
			answer_time = int(time.time() - QUIZ_CURRENT_TIME[groupchat])
			points = QUIZ_TIME_LIMIT / answer_time / 3 + 1
			msg(groupchat, '[QUIZ] ' + nick + ' was correct for +' + str(points) + ' points! The answer was: ' + answer)			
			if not QUIZ_SCORES.has_key(groupchat):
				QUIZ_SCORES[groupchat] = {}
			if QUIZ_SCORES[groupchat].has_key(nick):
				QUIZ_SCORES[groupchat][nick] += points
			else:
				QUIZ_SCORES[groupchat][nick] = points
			quiz_list_scores(groupchat)
			quiz_ask_question(groupchat)

def quiz_list_scores(groupchat):
	if QUIZ_SCORES.has_key(groupchat):
		if QUIZ_SCORES[groupchat]:
			if QUIZ_IDLENESS.has_key(groupchat):
				del QUIZ_IDLENESS[groupchat]
			result = '[QUIZ] Current Scores'
			for nick in QUIZ_SCORES[groupchat]:
				result += '\n' + nick + ': ' + str(QUIZ_SCORES[groupchat][nick])
			msg(groupchat, result)

def handler_quiz_start(type, source, parameters):
	groupchat = get_groupchat(source)
	if not groupchat:
		smsg(type, source, 'Not in groupchat.')
		return
	if QUIZ_CURRENT_ANSWER.has_key(groupchat):
		smsg(type, source, 'Quiz already exists.')
		return
	QUIZ_SCORES[groupchat] = {}
	if QUIZ_IDLENESS.has_key(groupchat):
		del QUIZ_IDLENESS[groupchat]
	msg(groupchat, '[QUIZ] Quiz has begun. Scores reset.')
	quiz_ask_question(groupchat)

def handler_quiz_stop(type, source, parameters):
	groupchat = get_groupchat(source)
	if QUIZ_CURRENT_ANSWER.has_key(groupchat):
		del QUIZ_CURRENT_ANSWER[groupchat]
		msg(groupchat, '[QUIZ] Quiz has ended.')
		quiz_list_scores(groupchat)
	else:
		smsg(type, source, 'No quiz to stop.')

def handler_quiz_hint(type, source, parameters):
	groupchat = get_groupchat(source)
	if QUIZ_CURRENT_ANSWER.has_key(groupchat):
		if QUIZ_IDLENESS.has_key(groupchat):
			del QUIZ_IDLENESS[groupchat]
		if QUIZ_CURRENT_HINT[groupchat] == None:
			QUIZ_CURRENT_HINT[groupchat] = 0
			#msg(groupchat, '[QUIZ] Answer has ' + str(len(QUIZ_CURRENT_ANSWER[groupchat])) + ' letters.')
		QUIZ_CURRENT_HINT[groupchat] += 1
		hint = QUIZ_CURRENT_ANSWER[groupchat][0:QUIZ_CURRENT_HINT[groupchat]]
		hint += '...'
		#hint += ' _' * (len(QUIZ_CURRENT_ANSWER[groupchat]) - QUIZ_CURRENT_HINT[groupchat])
		msg(groupchat, '[QUIZ] Hint: ' + hint)
	else:
		smsg(type, source, 'No quiz exists.')

def handler_quiz_scores(type, source, parameters):
	groupchat = get_groupchat(source)
	if QUIZ_CURRENT_ANSWER.has_key(groupchat):
		quiz_list_scores(groupchat)
	else:
		smsg(type, source, 'No quiz exists.')

def handler_quiz_message(type, source, body):
	groupchat = get_groupchat(source)
	if groupchat and QUIZ_CURRENT_ANSWER.has_key(groupchat):
		quiz_answer_question(source[1], source[2], body.strip())

register_command_handler(handler_quiz_start, '!quiz_start', 0, 'Starts a quiz in the groupchat.', '!quiz_start', ['!quiz_start'])
register_command_handler(handler_quiz_stop, '!quiz_stop', 0, 'Stops a quiz in the groupchat.', '!quiz_stop', ['!quiz_stop'])
register_command_handler(handler_quiz_hint, '!hint', 0, 'Asks for a hint on the current quiz question.', '!hint', ['!hint'])
register_command_handler(handler_quiz_scores, '!quiz_scores', 0, 'Request scores for current quiz.', '!quiz_scores', ['!quiz_scores'])
register_message_handler(handler_quiz_message)