CustomNPC-Plus/src/main/java/noppes/npcs/scripted/entity/ScriptVillager.java

package noppes.npcs.scripted.entity;

import net.minecraft.entity.passive.EntityVillager;
import noppes.npcs.api.entity.IEntityLivingBase;
import noppes.npcs.api.entity.IVillager;
import noppes.npcs.scripted.NpcAPI;
import noppes.npcs.scripted.constants.EntityType;

public class ScriptVillager<T extends EntityVillager> extends ScriptLiving<T> implements IVillager {

	public ScriptVillager(T entity) {
		super(entity);
	}

    @Override
    public int getProfession() {
        return entity.getProfession();
    }

    @Override
    public boolean getIsTrading() {
        return entity.isTrading();
    }

    @Override
    public IEntityLivingBase getCustomer()
    {
        return (IEntityLivingBase) NpcAPI.Instance().getPlayer(entity.getCustomer().getCommandSenderName());
    }

	@Override
	public int getType(){
		return EntityType.VILLAGER;
	}

    @Override
    public boolean typeOf(int type){
        return type == EntityType.VILLAGER || super.typeOf(type);
    }

}