package net.aufdemrand.denizen.npc.traits;
import net.aufdemrand.denizen.Settings;
import net.aufdemrand.denizen.npc.dNPCRegistry;
import net.aufdemrand.denizen.objects.dPlayer;
import net.aufdemrand.denizen.scripts.containers.core.AssignmentScriptContainer;
import net.aufdemrand.denizen.utilities.DenizenAPI;
import net.aufdemrand.denizen.utilities.debugging.dB;
import net.aufdemrand.denizencore.objects.aH;
import net.aufdemrand.denizencore.scripts.ScriptRegistry;
import net.aufdemrand.denizencore.utilities.text.StringHolder;
import net.citizensnpcs.api.command.exception.CommandException;
import net.citizensnpcs.api.exception.NPCLoadException;
import net.citizensnpcs.api.persistence.Persist;
import net.citizensnpcs.api.trait.Trait;
import net.citizensnpcs.api.util.DataKey;
import net.citizensnpcs.api.util.Paginator;
import net.citizensnpcs.util.Messages;
import org.bukkit.ChatColor;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.LivingEntity;
import org.bukkit.entity.Player;
import org.bukkit.entity.Projectile;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.entity.EntityDamageByEntityEvent;
public class AssignmentTrait extends Trait {
// Saved to the C2 saves.yml
@Persist
private String assignment = "";
public AssignmentTrait() {
super("assignment");
}
/**
* Checks to see if the NPCs assignment is still a valid script on load of NPC.
*/
@Override
public void load(DataKey key) throws NPCLoadException {
// Check to make sure assignment is still valid. Throw a dB error if not.
if (hasAssignment()) {
dB.echoError("Missing assignment '" + assignment + "' for NPC '"
+ npc.getName() + "/" + npc.getId() + "! Perhaps the script has been removed?");
}
npc.getTrait(ConstantsTrait.class).rebuildAssignmentConstants();
}
// <--[action]
// @Actions
// assignment
//
// @Triggers when the NPC receives an assignment via '/npc assign --set Name'.
//
// @Context
// None
//
// -->
/**
* Sets the NPCs Assignment Script and fires an 'On Assignment:' action. Can specify a player for
* context with the action.
*
* @param assignment the name of the Assignment Script, case in-sensitive
* @param player the player adding the assignment, can be null
* @return false if the assignment is invalid
*/
public boolean setAssignment(String assignment, dPlayer player) {
if (ScriptRegistry.containsScript(assignment, AssignmentScriptContainer.class)) {
this.assignment = assignment.toUpperCase();
// Add Constants/Trigger trait if not already added to the NPC.
if (!npc.hasTrait(ConstantsTrait.class)) {
npc.addTrait(ConstantsTrait.class);
}
if (!npc.hasTrait(TriggerTrait.class)) {
npc.addTrait(TriggerTrait.class);
}
if (Settings.healthTraitEnabledByDefault()) {
if (!npc.hasTrait(HealthTrait.class)) {
npc.addTrait(HealthTrait.class);
}
}
// Reset Constants
npc.getTrait(ConstantsTrait.class).rebuildAssignmentConstants();
// 'On Assignment' action.
dNPCRegistry.getDenizen(npc).action("assignment", player);
return true;
}
else {
return false;
}
}
/**
* Gets the name of the current Assignment Script assigned to this NPC.
*
* @return assignment script name, null if not set or assignment is invalid
*/
public AssignmentScriptContainer getAssignment() {
if (hasAssignment() && ScriptRegistry.containsScript(assignment, AssignmentScriptContainer.class)) {
return ScriptRegistry.getScriptContainer(assignment);
}
else {
return null;
}
}
/**
* Checks to see if this NPC currently has an assignment.
*
* @return true if NPC has an assignment and it is valid
*/
public boolean hasAssignment() {
if (assignment == null || assignment.equals("")) {
return false;
}
return ScriptRegistry.containsScript(assignment);
}
// <--[action]
// @Actions
// remove assignment
//
// @Triggers when the NPC loses its assignment.
//
// @Context
// None
//
// -->
/**
* Removes the current assignment and fires an 'On Remove Assignment:' action. Can specify a player for
* context with the action.
*
* @param player the player removing the assignment, can be null
*/
public void removeAssignment(dPlayer player) {
dNPCRegistry.getDenizen(npc).action("remove assignment", player);
assignment = "";
}
public void describe(CommandSender sender, int page) throws CommandException {
AssignmentScriptContainer assignmentScript = ScriptRegistry.getScriptContainer(assignment);
Paginator paginator = new Paginator().header("Assignment");
paginator.addLine("<e>Current assignment: " + (hasAssignment() ? this.assignment : "None.") + "");
paginator.addLine("");
if (!hasAssignment()) {
paginator.sendPage(sender, page);
return;
}
// Interact Scripts
boolean entriesPresent = false;
paginator.addLine(ChatColor.GRAY + "Interact Scripts:");
paginator.addLine("<e>Key: <a>Priority <b>Name");
if (assignmentScript.contains("INTERACT SCRIPTS")) {
entriesPresent = true;
for (String scriptEntry : assignmentScript.getStringList("INTERACT SCRIPTS")) {
String[] split = scriptEntry.split(" ", 2);
if (split.length == 2 && aH.matchesInteger(split[0])) {
paginator.addLine("<a>" + split[0] + "<b> " + split[1]);
}
else {
paginator.addLine("<b>" + scriptEntry);
}
}
}
if (!entriesPresent) {
paginator.addLine("<c>No Interact Scripts assigned.");
}
paginator.addLine("");
if (!entriesPresent) {
if (!paginator.sendPage(sender, page)) {
throw new CommandException(Messages.COMMAND_PAGE_MISSING);
}
return;
}
// Scheduled Activities
entriesPresent = false;
paginator.addLine(ChatColor.GRAY + "Scheduled Scripts:");
paginator.addLine("<e>Key: <a>Time <b>Name");
if (assignmentScript.contains("SCHEDULED ACTIVITIES")) {
entriesPresent = true;
for (String scriptEntry : assignmentScript.getStringList("SCHEDULED ACTIVITIES")) {
paginator.addLine("<a>" + scriptEntry.split(" ")[0] + "<b> " + scriptEntry.split(" ", 2)[1]);
}
}
if (!entriesPresent) {
paginator.addLine("<c>No scheduled scripts activities.");
}
paginator.addLine("");
// Actions
entriesPresent = false;
paginator.addLine(ChatColor.GRAY + "Actions:");
paginator.addLine("<e>Key: <a>Action name <b>Script Size");
if (assignmentScript.contains("ACTIONS")) {
entriesPresent = true;
}
if (entriesPresent) {
for (StringHolder action : assignmentScript.getConfigurationSection("ACTIONS").getKeys(false)) {
paginator.addLine("<a>" + action.str + " <b>" + assignmentScript.getStringList("ACTIONS." + action.str).size());
}
}
else {
paginator.addLine("<c>No actions defined in the assignment.");
}
paginator.addLine("");
if (!paginator.sendPage(sender, page)) {
throw new CommandException(Messages.COMMAND_PAGE_MISSING, page);
}
}
// <--[action]
// @Actions
// hit
// hit on <entity>
//
// @Triggers when the NPC hits an enemy.
//
// @Context
// None
//
// -->
// <--[action]
// @Actions
// kill
// kill of <entity>
//
// @Triggers when the NPC kills an enemy.
//
// @Context
// None
//
// -->
// Listen for this NPC's hits on entities
@EventHandler(priority = EventPriority.MONITOR)
public void onHit(EntityDamageByEntityEvent event) {
// Check if the damager is this NPC
if (event.getDamager() != npc.getEntity()) {
// If the damager is not this NPC, the damager could still be a
// projectile shot by this NPC, in which case we want to continue
if (event.getDamager() instanceof Projectile) {
if (((Projectile) event.getDamager()).getShooter() != npc.getEntity()) {
return;
}
}
else {
return;
}
}
dPlayer player = null;
// Check if the entity hit by this NPC is a player
if (event.getEntity() instanceof Player) {
player = dPlayer.mirrorBukkitPlayer((Player) event.getEntity());
}
// TODO: Context containing the entity hit
DenizenAPI.getDenizenNPC(npc).action("hit", player);
DenizenAPI.getDenizenNPC(npc).action("hit on " + event.getEntityType().name(), player);
if (event.getEntity() instanceof LivingEntity) {
if (((LivingEntity) event.getEntity()).getHealth() - event.getFinalDamage() <= 0) {
DenizenAPI.getDenizenNPC(npc).action("kill", player);
DenizenAPI.getDenizenNPC(npc).action("kill of " + event.getEntityType().name(), player);
}
}
// All done!
}
}