Fortress Forever - Headshot reward lua?

This code is untested. It should give 20 team points per headshot (10 for regular kill, 10 bonus for headshots).

Code:

TEAM_POINTS_PER_KILL = 10

BONUS_TEAM_POINTS_PER_HEADSHOT = 10



player_onheadshot( shot_entity, shooter_entity )

        if IsPlayer(shooter_entity) then

                local player = CastToPlayer(shooter_entity)

                local team = player:GetTeam()

                team:AddScore(BONUS_TEAM_POINTS_PER_HEADSHOT)

        end

end



player_killed( player, damageinfo )

        -- Entity that is attacking

        local attacker = damageinfo:GetAttacker()



        -- If no attacker do nothing

        if not attacker then

                return

        end



        -- If attacker not a player do nothing

        if not IsPlayer(attacker) then 

                return

        end



        local playerAttacker = CastToPlayer(attacker)

        local weapon = damageinfo:GetInflictor():GetClassName()



        -- If attacked by sniper rifle, add points

        if weapon == "ff_weapon_sniperrifle" then

                local team = playerAttacker:GetTeam()

                -- Add score to team

                team:AddScore(TEAM_POINTS_PER_KILL)

        end

end