/
LeashCommand.java
85 lines (67 loc) · 3.3 KB
/
LeashCommand.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package net.aufdemrand.denizen.scripts.commands.entity;
import java.util.List;
import net.aufdemrand.denizen.exceptions.CommandExecutionException;
import net.aufdemrand.denizen.exceptions.InvalidArgumentsException;
import net.aufdemrand.denizen.objects.aH;
import net.aufdemrand.denizen.objects.dEntity;
import net.aufdemrand.denizen.objects.dList;
import net.aufdemrand.denizen.scripts.ScriptEntry;
import net.aufdemrand.denizen.scripts.commands.AbstractCommand;
import net.aufdemrand.denizen.utilities.debugging.dB;
import net.aufdemrand.denizen.utilities.debugging.dB.Messages;
/**
* Leashes a list of entities to another entity.
*
* @author Alain Blanquet
*/
public class LeashCommand extends AbstractCommand {
@Override
public void parseArgs(ScriptEntry scriptEntry) throws InvalidArgumentsException {
for (aH.Argument arg : aH.interpret(scriptEntry.getArguments())) {
if (!scriptEntry.hasObject("cancel")
&& arg.matches("cancel, stop")) {
scriptEntry.addObject("cancel", "");
}
else if (!scriptEntry.hasObject("entities")
&& arg.matchesArgumentList(dEntity.class)) {
scriptEntry.addObject("entities", ((dList) arg.asType(dList.class)).filter(dEntity.class));
}
else if (!scriptEntry.hasObject("holder")
&& arg.matchesArgumentType(dEntity.class)
&& arg.matchesPrefix("holder, h")) {
scriptEntry.addObject("holder", arg.asType(dEntity.class));
}
}
// Check to make sure required arguments have been filled
if (!scriptEntry.hasObject("entities"))
throw new InvalidArgumentsException(Messages.ERROR_MISSING_OTHER, "ENTITIES");
if (!scriptEntry.hasObject("cancel")) {
scriptEntry.defaultObject("holder",
scriptEntry.hasNPC() ? scriptEntry.getNPC().getDenizenEntity() : null,
scriptEntry.hasPlayer() ? scriptEntry.getPlayer().getDenizenEntity() : null);
}
}
@SuppressWarnings("unchecked")
@Override
public void execute(final ScriptEntry scriptEntry) throws CommandExecutionException {
// Get objects
List<dEntity> entities = (List<dEntity>) scriptEntry.getObject("entities");
dEntity holder = (dEntity) scriptEntry.getObject("holder");
Boolean cancel = scriptEntry.hasObject("cancel");
// Report to dB
dB.report(getName(), (cancel == true ? aH.debugObj("cancel", cancel) : "") +
aH.debugObj("entities", entities.toString()) +
(holder != null ? aH.debugObj("holder", holder) : ""));
// Go through all the entities and leash/unleash them
for (dEntity entity : entities) {
if (entity.isSpawned() && entity.isLivingEntity()) {
if (cancel) {
entity.getLivingEntity().setLeashHolder(null);
}
else {
entity.getLivingEntity().setLeashHolder(holder.getBukkitEntity());
}
}
}
}
}