#include <amxmodx>
#include <fakemeta>

new vModel[] = "models/v_snowball.mdl"
new pModel[] = "models/p_snowball.mdl"
new wModel[] = "models/w_snowball.mdl"

new wOldModel[] = "models/w_smokegrenade.mdl"

public plugin_init()
{
    register_forward(FM_SetModel, "forwardSetModel")
    register_event("CurWeapon", "EventCurrentWeapon", "be", "1=1")
}

public plugin_precache()
{
    precache_model(vModel)
    precache_model(pModel)
    precache_model(wModel)
}

public EventCurrentWeapon(id)
{
    new szWeaponId = read_data(2)

    if(szWeaponId != CSW_SMOKEGRENADE)
    return PLUGIN_CONTINUE

    set_pev(id, pev_viewmodel2, vModel)
    set_pev(id, pev_weaponmodel2, pModel)

    return PLUGIN_CONTINUE
}

public forwardSetModel(ent, model[])
{
    if(!pev_valid(ent))
    return FMRES_IGNORED

    if(!equali(model, wOldModel))
    return FMRES_IGNORED

    new szClassName[33]
    pev(ent, pev_classname, szClassName, 32)

    if(equal(szClassName, "weaponbox") || equal(szClassName, "armoury_entity") || equal(szClassName, "grenade"))
    {
        engfunc(EngFunc_SetModel, ent, wModel)
        return FMRES_SUPERCEDE
    }
    return FMRES_IGNORED
}