import es
import datetime
import cPickle
import os
import threading


vip_cmd = '!vip'
vip_weeks = 0
vip_hours = 1000
vip_minutes = 0


addonpath = es.getAddonPath('vipmod')
dbpath = os.path.join(addonpath, 'users.db')


vip_td = datetime.timedelta(weeks=vip_weeks, hours=vip_hours, minutes=vip_minutes)



vip_dict = {}
temp_data = {}

try:
    vip_dict = cPickle.load(open(dbpath, 'rb'))
except:
    pass

info = es.AddonInfo()
info.name = 'vipmod'
info.author = 'Brainsucker'
info.version = '1.0.0 (Unreleased)'

def load():
    es.ServerVar(info['name'], info['version']).makepublic()
    for p in es.getUseridList():
        addPlayerbyUserid(p)


def saveDB():
    threading.Thread(target=cPickle.dump, args=(vip_dict, open(dbpath, 'wb'))).start()

def addPlayerbySteamid(steamid):
    if not steamid in ['BOT', 'STEAM_ID_PENDING', 'STEAM_ID_LAN']:
        if not vip_dict.has_key(steamid):
            vip_dict[steamid] = {}
        if not vip_dict[steamid].has_key('playtime'):
            vip_dict[steamid]['playtime'] = datetime.timedelta()

        temp_data[steamid]['jointime'] = datetime.datetime.today()

def addPlayerbyUserid(userid):
    steamid = es.getplayersteamid(userid)
    addPlayerbySteamid(steamid)


def es_player_validated(ev):
    steamid = ev['networkid']
    addPlayerbySteamid(steamid)


def player_disconnect(ev):
    steamid = ev['es_steamid']
    current_time = datetime.datetime.today()
    if temp_data.has_key(steamid):
        playtime = current_time - temp_data[steamid]['jointime']
        vip_dict[steamid]['playtime'] += playtime

        saveDB()