-
-
Notifications
You must be signed in to change notification settings - Fork 102
/
EquipCommand.java
139 lines (105 loc) · 5.84 KB
/
EquipCommand.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
package net.aufdemrand.denizen.scripts.commands.entity;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.bukkit.entity.LivingEntity;
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.objects.dItem;
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;
import net.citizensnpcs.api.npc.NPC;
import net.citizensnpcs.api.trait.trait.Equipment;
/**
* Equip entities with items and armor.
*
* @author David Cernat
*/
public class EquipCommand extends AbstractCommand {
@Override
public void parseArgs(ScriptEntry scriptEntry) throws InvalidArgumentsException {
Map<String, dItem> equipment = new HashMap<String,dItem>();
// Initialize necessary fields
for (aH.Argument arg : aH.interpret(scriptEntry.getArguments())) {
if (!scriptEntry.hasObject("entities")
&& arg.matchesArgumentList(dEntity.class)) {
scriptEntry.addObject("entities", ((dList) arg.asType(dList.class)).filter(dEntity.class));
}
else if (arg.matchesArgumentType(dItem.class)
&& arg.matchesPrefix("head, helmet")) {
equipment.put("head", dItem.valueOf(arg.getValue()));
}
else if (arg.matchesArgumentType(dItem.class)
&& arg.matchesPrefix("chest, chestplate")) {
equipment.put("chest", dItem.valueOf(arg.getValue()));
}
else if (arg.matchesArgumentType(dItem.class)
&& arg.matchesPrefix("legs, leggings")) {
equipment.put("legs", dItem.valueOf(arg.getValue()));
}
else if (arg.matchesArgumentType(dItem.class)
&& arg.matchesPrefix("boots, feet")) {
equipment.put("boots", dItem.valueOf(arg.getValue()));
}
// Default to item in hand if no prefix is used
else if (arg.matchesArgumentType(dItem.class)) {
equipment.put("hand", dItem.valueOf(arg.getValue()));
}
else if (arg.matches("player") && scriptEntry.hasPlayer()) {
// Player arg for compatibility with old scripts
scriptEntry.addObject("entities", Arrays.asList(scriptEntry.getPlayer().getDenizenEntity()));
}
else dB.echoError(dB.Messages.ERROR_UNKNOWN_ARGUMENT, arg.raw_value);
}
// Make sure at least one equipment argument was used
if (equipment.isEmpty())
throw new InvalidArgumentsException(Messages.ERROR_MISSING_OTHER, "EQUIPMENT");
scriptEntry.addObject("equipment", equipment);
// Use player or NPC as default entity
scriptEntry.defaultObject("entities", (scriptEntry.hasNPC() ? Arrays.asList(scriptEntry.getNPC().getDenizenEntity()) : null),
(scriptEntry.hasPlayer() ? Arrays.asList(scriptEntry.getPlayer().getDenizenEntity()) : null));
}
@SuppressWarnings("unchecked")
@Override
public void execute(ScriptEntry scriptEntry)
throws CommandExecutionException {
Map<String, dItem> equipment = (Map<String, dItem>) scriptEntry.getObject("equipment");
List<dEntity> entities = (List<dEntity>) scriptEntry.getObject("entities");
// Report to dB
dB.report(getName(), aH.debugObj("entities", entities.toString()) +
aH.debugObj("equipment", equipment.toString()));
for (dEntity entity : entities) {
if (entity.isGeneric()) {
dB.echoError("Cannot equip generic entity " + entity.identify() + "!");
}
else if (entity.isNPC()) {
NPC npc = entity.getNPC();
if (npc != null) {
if (!npc.hasTrait(Equipment.class)) npc.addTrait(Equipment.class);
Equipment trait = npc.getTrait(Equipment.class);
if (equipment.get("hand") != null) trait.set(0, equipment.get("hand").getItemStack());
if (equipment.get("head") != null) trait.set(1, equipment.get("head").getItemStack());
if (equipment.get("chest") != null) trait.set(2, equipment.get("chest").getItemStack());
if (equipment.get("legs") != null) trait.set(3, equipment.get("legs").getItemStack());
if (equipment.get("boots") != null) trait.set(4, equipment.get("boots").getItemStack());
}
}
else {
LivingEntity livingEntity = entity.getLivingEntity();
if (livingEntity != null) {
if (equipment.get("hand") != null) livingEntity.getEquipment().setItemInHand(equipment.get("hand").getItemStack());
if (equipment.get("head") != null) livingEntity.getEquipment().setHelmet(equipment.get("head").getItemStack());
if (equipment.get("chest") != null) livingEntity.getEquipment().setChestplate(equipment.get("chest").getItemStack());
if (equipment.get("legs") != null) livingEntity.getEquipment().setLeggings(equipment.get("legs").getItemStack());
if (equipment.get("boots") != null) livingEntity.getEquipment().setBoots(equipment.get("boots").getItemStack());
}
}
}
}
}