EntityDamageSourceGun.java

package com.flansmod.common.guns;

import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.util.ChatComponentText;
import net.minecraft.util.EntityDamageSourceIndirect;
import net.minecraft.util.IChatComponent;

import com.flansmod.common.FlansMod;
import com.flansmod.common.PlayerHandler;
import com.flansmod.common.network.PacketKillMessage;
import com.flansmod.common.teams.Team;
import com.flansmod.common.types.InfoType;

public class EntityDamageSourceGun extends EntityDamageSourceIndirect {

	public InfoType weapon;
	public EntityPlayer shooter;
	public boolean headshot;
	
	public EntityDamageSourceGun(String s, Entity entity, EntityPlayer player, InfoType wep, boolean head) 
	{
		super(s, entity, player);
		weapon = wep;
		shooter = player;
		headshot = head;
	}
	
	@Override 
	public Entity getSourceOfDamage()
	{
		return shooter;
	}

	@Override
	public IChatComponent getDeathMessage(EntityLivingBase living)
    {
		if(!(living instanceof EntityPlayer) || shooter == null || PlayerHandler.getPlayerData(shooter) == null)
		{
			return super.getDeathMessage(living);
		}
		EntityPlayer player = (EntityPlayer)living;

    	Team killedTeam = PlayerHandler.getPlayerData(player).team;
    	Team killerTeam = PlayerHandler.getPlayerData(shooter).team;
    	
    	FlansMod.getPacketHandler().sendToDimension(new PacketKillMessage(headshot, weapon, (killedTeam == null ? "f" : killedTeam.textColour) + player.getName(),  (killerTeam == null ? "f" : killerTeam.textColour) + shooter.getName()), living.dimension);
    	
        return new ChatComponentText("#flansmod");//flanDeath." + weapon.shortName + "." + (killedTeam == null ? "f" : killedTeam.textColour) + player.getCommandSenderName() + "." + (killerTeam == null ? "f" : killerTeam.textColour) + shooter.getCommandSenderName());
    }
}