package net.minecraftforge.cauldron.entity;

import cpw.mods.fml.common.registry.EntityRegistry;
import net.minecraft.entity.Entity;
import org.bukkit.craftbukkit.CraftServer;
import org.bukkit.craftbukkit.entity.CraftEntity;
import org.bukkit.entity.EntityType;
import org.ultramine.mods.bukkit.interfaces.IMixinEntityRegistry;

public class CraftCustomEntity extends CraftEntity
{

	public Class<? extends Entity> entityClass;
	public String entityName;

	public CraftCustomEntity(CraftServer server, net.minecraft.entity.Entity entity)
	{
		super(server, entity);
		this.entityClass = entity.getClass();
		this.entityName = ((IMixinEntityRegistry) EntityRegistry.instance()).getCustomEntityTypeName(entityClass);
		if(entityName == null)
			entityName = entity.getCommandSenderName();
	}

	@Override
	public net.minecraft.entity.Entity getHandle()
	{
		return (net.minecraft.entity.Entity) entity;
	}

	@Override
	public String toString()
	{
		return this.entityName;
	}

	public EntityType getType()
	{
		EntityType type = EntityType.fromName(this.entityName);
		if(type != null)
			return type;
		else return EntityType.UNKNOWN;
	}
}