Citizens2/src/net/citizensnpcs/npc/CitizensNPC.java

package net.citizensnpcs.npc;

import java.util.HashMap;
import java.util.Map;

import net.citizensnpcs.api.CitizensAPI;
import net.citizensnpcs.api.event.NPCDespawnEvent;
import net.citizensnpcs.api.event.NPCSpawnEvent;
import net.citizensnpcs.api.npc.trait.Character;
import net.citizensnpcs.api.npc.NPC;
import net.citizensnpcs.api.npc.pathfinding.Navigator;
import net.citizensnpcs.api.npc.trait.Trait;
import net.citizensnpcs.api.npc.trait.trait.LocationTrait;
import net.citizensnpcs.resources.lib.CraftNPC;
import net.citizensnpcs.util.Messaging;

import org.bukkit.Bukkit;
import org.bukkit.ChatColor;
import org.bukkit.Location;

public class CitizensNPC implements NPC {
    private final int id;
    private Character character = null;
    private final Map<Class<? extends Trait>, Trait> traits = new HashMap<Class<? extends Trait>, Trait>();
    private String name;
    private CraftNPC mcEntity;
    private boolean spawned;
    private final CitizensNPCManager manager;
    private final Factory factory = new Factory();

    public CitizensNPC(String name, Character character) {
        this.name = name;
        this.character = character;
        manager = (CitizensNPCManager) CitizensAPI.getNPCManager();
        id = manager.getUniqueID();
    }

    public CitizensNPC(int id, String name, Character character) {
        this.name = name;
        this.character = character;
        manager = (CitizensNPCManager) CitizensAPI.getNPCManager();
        this.id = id;
    }

    @Override
    public String getFullName() {
        return name;
    }

    @Override
    public String getName() {
        return ChatColor.stripColor(name);
    }

    @Override
    public void setName(String name) {
        this.name = name;
    }

    @Override
    public void addTrait(Class<? extends Trait> trait) {
        if (!hasTrait(trait))
            traits.put(trait, factory.create(trait));
        else
            Messaging.debug("The NPC already has the trait '" + trait.getName() + "'.");
    }

    @Override
    public Character getCharacter() {
        return character;
    }

    @Override
    public int getId() {
        return id;
    }

    @Override
    public Navigator getNavigator() {
        // TODO add default navigator
        return null;
    }

    @Override
    public <T extends Trait> T getTrait(Class<T> trait) {
        Trait t = traits.get(trait);
        if (t != null) {
            return trait.cast(t);
        }

        T create = factory.create(trait);
        addTrait(trait);
        return create;
    }

    @Override
    public Iterable<Trait> getTraits() {
        return traits.values();
    }

    @Override
    public boolean hasTrait(Class<? extends Trait> trait) {
        return traits.containsKey(trait);
    }

    @Override
    public void removeTrait(Class<? extends Trait> trait) {
        if (!hasTrait(trait)) {
            Messaging.debug("The NPC does not have a trait with the name of '" + trait.getName() + ".");
            return;
        }
        traits.remove(trait);
    }

    @Override
    public void setCharacter(Character character) {
        if (this.character.equals(character)) {
            Messaging.debug("The NPC already has the character '" + character.getName() + "'.");
            return;
        }
        this.character = character;
    }

    @Override
    public boolean isSpawned() {
        return spawned;
    }

    @Override
    public void spawn(Location loc) {
        if (isSpawned()) {
            Messaging.debug("The NPC is already spawned.");
            return;
        }

        NPCSpawnEvent spawnEvent = new NPCSpawnEvent(this, loc);
        Bukkit.getPluginManager().callEvent(spawnEvent);
        if (spawnEvent.isCancelled()) {
            return;
        }

        if (mcEntity == null)
            mcEntity = manager.spawn(this, loc);
        else
            manager.spawn(this, loc);

        // Set the location
        addTrait(LocationTrait.class);
        getTrait(LocationTrait.class).setLocation(loc);

        spawned = true;
    }

    @Override
    public void despawn() {
        if (!isSpawned()) {
            Messaging.debug("The NPC is already despawned.");
            return;
        }

        Bukkit.getPluginManager().callEvent(new NPCDespawnEvent(this));

        manager.despawn(this);
        mcEntity.die();
        spawned = false;
    }

    @Override
    public void remove() {
        if (isSpawned()) {
            despawn();
        }
        manager.remove(this);
    }

    public CraftNPC getHandle() {
        return mcEntity;
    }

    private class Factory {

        public <T extends Trait> T create(Class<T> clazz) {
            try {
                return clazz.cast(clazz.newInstance());
            } catch (Exception ex) {
                ex.printStackTrace();
                return null;
            }
        }
    }
}