import es
import popuplib
import keymenulib
import services

gPlayer = {}
hPlayer = {}

def load():
    punish_menu = popuplib.easymenu('punish_menu', None, menu)
    punish_menu.settitle('Select punishment amount:')
    punish_menu.addoption('hl_punish10', '-10 points')
    punish_menu.addoption('hl_punish25', '-25 points')
    punish_menu.addoption('hl_punish50', '-50 points')
    punish_menu.addoption('hl_punish75', '-75 points')
    punish_menu.addoption('hl_punish100', '-100 points')
    punish_menu.addoption('hl_punish150', '-150 points')
    punish_menu.addoption('hl_punish200', '-200 points')
    punish_menu.addoption('hl_punish300', '-300 points')
    punish_menu.addoption('hl_punish500', '-500 points')

    if not es.exists('saycommand', '!hlpunish'):
        es.regsaycmd('!hlpunish', 'hlpunish/call_menu')
        
    auth = services.use('auth') 
    auth.registerCapability('can_hlp_punish', auth.ADMIN)
    
def call_menu():
    user = int(es.getcmduserid()) 
    auth = services.use('auth')
    if auth.isUseridAuthorized(user, "can_hlp_punish"):
        if es.getargs() == None:
            hPlayer[user] = 'unknown reasons'
        else:
            hPlayer[user] = es.getargs()
        popuplib.send('punish_menu', user)
    else:
        es.tell(es.getcmduserid(), 'We can\'t trust you with that command! How dare you!')

def menu(userid, choice, popupname):
    gPlayer[userid] = choice
    es.server.cmd('es_xcreateplayerlist hlp_playerlist')
    es.set('hlp_menu_choice', 0)
    p_list = keymenulib.create('playerlist', 'hlp_menu_choice', player_choice, 'hlp_playerlist', '#keyvalue name', '#key', 'Current Players\nSelect a player to punish')
    p_list.send(userid)
    p_list.delete
    es.keygroupdelete('hlp_playerlist')

def player_choice(userid, choice, popupname):
    cut_string = gPlayer[userid].replace('hl_punish', '')
    es.msg('#multi', '#green' + es.getplayername(es.server_var['hlp_menu_choice']) + '#default has had#lightgreen ' + cut_string + '#default HLStats rank points removed for#lightgreen ' + hPlayer[userid] + '#default.')
    es.server.cmd('statlog ' + es.server_var['hlp_menu_choice'] + ' ' + gPlayer[userid])
    del gPlayer[userid]
    del hPlayer[userid]