import es
import spe

from spe import HookType
from spe import HookAction


def load():
    spe.detourFunction('OnTakeDamage', HookType.Pre, preOnTakeDamage)

def unload():
    spe.undetourFunction('OnTakeDamage', HookType.Pre, preOnTakeDamage)


def preOnTakeDamage(args):
    info = spe.makeObject('CTakeDamageInfo', args[0])
    if info.hAttacker == info.hInflictor:
        userid = es.getuserid(info.hAttacker)
        handle = es.getplayerprop(userid, 'CBaseCombatCharacter.m_hActiveWeapon')
        index  = es.getindexfromhandle(handle)

    else:
        index = es.getindexfromhandle(info.hInflictor)

    weapon = es.entitygetvalue(index, 'classname')
    info.flDamage *= 2
    return (HookAction.Continue, 0)