import es, playerlib, popuplib

info = es.AddonInfo()
info.name     	= "SurfTricks Extended"
info.basename 	= "surftricks_ext"
info.version  	= "HERPDERPDURR"
info.url      	= "microvnet.info"
info.description= "SUP3R L33T SURFTR1CKK5 3XT3ND3D"
info.author   	= "Capt. Micro"
info.contact  	= "arrowflyer@gmail.com"
info.tags		= "surf,tricks,surftricks,race,money"

tricklist = {
	'Awp2Awp':{
		'racable':True,
		'prize':500,
	},
	'Awp3':{
		'racable':True,
		'prize':500,
	},
	'Pikachu Gap':{
		'racable':True,
		'prize':500,
	},
	'Tobe Go Home':{
		'racable':True,
		'prize':500,
	},
	'Deagle to awp':{
		'racable':True,
		'prize':500,
	},
	'T1000':{
		'racable':True,
		'prize':500,
	},
	'T1000 to spawn':{
		'racable':True,
		'prize':500,
	},
	'T1000 to sign':{
		'racable':True,
		'prize':500,
	},
	'Awp4':{
		'racable':True,
		'prize':500,
	},
	'Main to sign':{
		'racable':True,
		'prize':500,
	},
	'Razor Gap':{
		'racable':True,
		'prize':500,
	},
	'Awp To Awp Over Main':{
		'racable':True,
		'prize':500,
	},
	'Spawn Hop Ct->T':{
		'racable':True,
		'prize':500,
	},
	'Spawn Hop T->Ct':{
		'racable':True,
		'prize':500,
	},
	'Nc Waave':{
		'racable':True,
		'prize':500,
	},
	'Double Hop':{
		'racable':True,
		'prize':500,
	},
	'Chrome Gap':{
		'racable':True,
		'prize':500,
	}
}

race_start_positions = {
	'default': {
		'player1': (-1073, 232, 6080),
		'player2': (-1073, -232, 6080)
	},
	'pikachu': {
		'player1': (-1040, 8.0, 6080.093750),
		'player2': (-1040, -7.9, 6080.093750)
	},
	'top': {
		'player1': (-1952, 150, 7272.093750),
		'player2': (-1952, -150, 7272.093750)
	},
}

race_start_pos = race_start_positions['default']
race_winner_prize = 1500
race_races = []
race_playerselect = None
race_trickselect = None
race_menutmp = {}

def race_newrace(userid1, userid2, trick):
	global race_races
	race_races += [{
		'userid1': str(userid1),
		'userid2': str(userid2),
		'trick': trick,
		'status': 'WAIT'
	}]

def race_startrace(raceindex):
	global race_races
	#this resets both players tricks (kinda hacky yo)
	es.event('initialize', 'player_spawn')
	es.event('setstring', 'player_spawn', 'userid', race['userid1'])
	es.event('fire', 'player_spawn')
	es.event('initialize', 'player_spawn')
	es.event('setstring', 'player_spawn', 'userid', race['userid2'])
	es.event('fire', 'player_spawn')
	#move players to start positions
	playerlib.getPlayer(race_races[raceindex]['userid1']
		).setLocation(race_start_pos['player1'])
	playerlib.getPlayer(race_races[raceindex]['userid2']
		).setLocation(race_start_pos['player2'])
	#FIRE THE BOOLET
	race_races[raceindex]['status'] = 'RACE'

def race_startrace(raceindex, winner):
	global race_races
	ply1 = playerlib.getPlayer(race_races[raceindex]['userid1'])
	ply2 = playerlib.getPlayer(race_races[raceindex]['userid2'])
	pwinner = playerlib.getPlayer(winner)
	es.msg("#multi", "#default[#lightgreenSurf Tricks Racing#default] " +
		"[#lightgreen%s vs #lightgreen%s#default]#green #default%s has won the race!"
		%(ply1.name,ply2.name,winner.name))
	pwinner.cash += race_winner_prize
	del race_races[raceindex]

def playermenu_handler(userid, choice, menuname):
	global race_trickselect
	global race_menutmp
	race_menutmp[userid] = choice
	race_trickselect.send(userid)

def trickmenu_handler(userid, choice, menuname):
	global race_menutmp
	opp = race_menutmp[userid]
	del race_menutmp[userid]
	race_newrace(userid, opp, choice)

def load():
	es.loadevents('addons/eventscripts/surftricks/surftricks_trick.res')

	#RACE trick selection menu
	global race_trickselect
	if (race_trickselect != None): race_trickselect.delete()
	race_trickselect = popuplib.easymenu('trickselect', None, trickmenu_handler)
	race_trickselect.settitle("[RACE] Trick Select")
	for trackname in tricklist:
		if (tracklist[trackname]['racable']):
			race_trickselect.addoption(trackname, trackname)

	#RACE player selection menu
	global race_playerselect
	if (race_playerselect != None): race_playerselect.delete()
	race_playerselect = popuplib.easymenu('playerselect', None, playermenu_handler)
	race_playerselect.settitle("[RACE] Player Select")
	for player in playerlib.getPlayerList('#human'):
		race_playerselect.addoption(player.userid, player.name)

def es_map_start(e):
	es.loadevents('addons/eventscripts/surftricks/surftricks_trick.res')

def unload():
	global race_playerselect
	if (race_playerselect != None): race_playerselect.delete()
	global race_trickselect
	if (race_trickselect != None): race_trickselect.delete()
	pass

def player_activate(e):
	#RACE player selection menu
	global race_playerselect
	if (race_playerselect != None): race_playerselect.delete()
	race_playerselect = popuplib.easymenu('playerselect', None, playermenu_handler)
	race_playerselect.settitle("[RACE] Player Select")
	for player in playerlib.getPlayerList('#human'):
		race_playerselect.addoption(player.userid, player.name)

def player_disconnect(e):
	#RACE player selection menu
	global race_playerselect
	if (race_playerselect != None): race_playerselect.delete()
	race_playerselect = popuplib.easymenu('playerselect', None, playermenu_handler)
	race_playerselect.settitle("[RACE] Player Select")
	for player in playerlib.getPlayerList('#human'):
		race_playerselect.addoption(player.userid, player.name)

def es_player_chat(e):
	text = e['text']
	ply = playerlib.getPlayer(e['userid'])
	if (text[0:5] == '!race'):
		args = text.split(' ')
		if (len(args) == 1):
			global race_playerselect
			race_playerselect.send(ply.userid)
		elif (len(args) == 2):
			if (args[1] == 'accept'):
				global race_races
				for race in race_races:
					if (race['userid2'] == str(ply.userid)):
						race_startrace(race_races.index(race))

def surftricks_trick(userid, trick, time):
	#Cash money prize
	ply = playerlib.getPlayer(userid)
	ply.cash += tracklist[trick]['prize']

	#Racing stuffffff
	global race_races
	for race in race_races:
		if (race['trick'] == trick and race['status'] == 'RACE'):
			if (race['ply1'] == str(userid) or race['ply2'] == str(userid)):
				race_endrace(race_races.index(race), str(userid))
				race['status'] = 'OVER'