import es
import cmdlib
import random
import gamethread

admin_command_prefix = '#.#'


JoinedTeam = {}

def load():
    cmdlib.registerSayCommand(admin_command_prefix, admin_command, 'Execute admin command')

def unload():
    cmdlib.unregisterSayCommand(admin_command_prefix)

def es_map_start(ev):
    JoinedTeam.clear()

def player_team(ev):
    player = int(ev['userid'])
    if player not in JoinedTeam:
        JoinedTeam[player] = "yes"
        es.server.queuecmd("es_xchangeteam %s 1" %(player))

def player_spawn(ev):
    player = int(ev['userid'])
    if player not in JoinedTeam:
        JoinedTeam[player] = "yes"
        es.server.queuecmd("es_xchangeteam %s 1" %(player))

def player_disconnect(ev):
    player = int(ev['userid'])
    if player in JoinedTeam:
        del JoinedTeam[player]

def admin_command(userid, args):
    if args[0] == 'cm':
        if (es.exists('map',args[1])):
            es.server.changelevel(args[1])
        else:
            if (args[1] == "aa"):
                es.server.queuecmd('changelevel aim_arena_reloaded')
            elif (args[1] == "air"):
                es.server.queuecmd('changelevel dm_airfusion')
            elif (args[1] == "asylum"):
                es.server.queuecmd('changelevel dm_asylum_rc2')
            elif (args[1] == "atrium"):
                es.server.queuecmd('changelevel dm_atrium_b3')
            elif (args[1] == "bio"):
                es.server.queuecmd('changelevel dm_biohazard_cal')
            elif (args[1] == "cav"):
                es.server.queuecmd('changelevel dm_caverns_r1')
            elif (args[1] == "col"):
                es.server.queuecmd('changelevel dm_collapse_r1')
            elif (args[1] == "dec"):
                es.server.queuecmd('changelevel dm_decoy_r1')
            elif (args[1] == "fr"):
                es.server.queuecmd('changelevel dm_frenzy_cu')
            elif (args[1] == "hx"):
                es.server.queuecmd('changelevel dm_helix')
            elif (args[1] == "hydro"):
                es.server.queuecmd('changelevel dm_hydroponicz_cu')
            elif (args[1] == "inf"):
                es.server.queuecmd('changelevel dm_infinity')
            elif (args[1] == "int"):
                es.server.queuecmd('changelevel dm_intensity_b1')
            elif (args[1] == "alive"):
                es.server.queuecmd('changelevel dm_is_alive')
            elif (args[1] == "ld"):
                es.server.queuecmd('changelevel dm_lockdown')
            elif (args[1] == "lock"):
                es.server.queuecmd('changelevel dm_lockdown')
            elif (args[1] == "ld5"):
                es.server.queuecmd('changelevel dm_lockdown_r5')
            elif (args[1] == "lock5"):
                es.server.queuecmd('changelevel dm_lockdown_r5')
            elif (args[1] == "la"):
                es.server.queuecmd('changelevel dm_lostarena_rpg')
            elif (args[1] == "lv"):
                es.server.queuecmd('changelevel dm_lostvillage_r1')
            elif (args[1] == "octa"):
                es.server.queuecmd('changelevel dm_octagon')
            elif (args[1] == "ow"):
                es.server.queuecmd('changelevel dm_overwatch_cu')
            elif (args[1] == "over"):
                es.server.queuecmd('changelevel dm_overwatch_cu')
            elif (args[1] == "qt"):
                es.server.queuecmd('changelevel dm_qtech_r3')
            elif (args[1] == "qtech"):
                es.server.queuecmd('changelevel dm_qtech_r3')
            elif (args[1] == "resi"):
                es.server.queuecmd('changelevel dm_resident_r1')
            elif (args[1] == "tig"):
                es.server.queuecmd('changelevel dm_tigcrik_r2')
            elif (args[1] == "under"):
                es.server.queuecmd('changelevel dm_undercastle_r1')
            elif (args[1] == "volt"):
                es.server.queuecmd('changelevel dm_voltage_test')
            elif (args[1] == "zeta"):
                es.server.queuecmd('changelevel dm_zeta_rc2')
            else:
                es.tell(userid, '#multi', '#greenNo such map: \x05%s' %(args[1]))
    elif args[0] == 'start':
        es.server.queuecmd('exec '+args[1])
        WasConfigFile = 0
        if (args[1] == "cu1v1"): WasConfigFile = 1
        if (args[1] == "cu1v1.cfg"): WasConfigFile = 1
        if (args[1] == "cu2v2"): WasConfigFile = 1
        if (args[1] == "cu2v2.cfg"): WasConfigFile = 1
        if (WasConfigFile == 1):
            d1 = random.randint(0, 9)
            d2 = random.randint(0, 9)
            df = "%i%i" %(d1, d2)
            es.server.queuecmd("sv_password %s" %(df))
            gamethread.delayed(7, es.msg,("#multi", "#greenPassword was set to: \x05%s" %(df)))
    elif args[0] == 'teamplay':
        es.server.queuecmd('mp_teamplay '+args[1])
    elif args[0] == 'record':
        es.server.queuecmd('tv_record '+args[1])
    elif args[0] == 'voice':
        es.server.queuecmd('sv_voiceenable '+args[1])
    elif args[0] == 'restart':
        es.server.queuecmd('mp_restartgame 10')