/
EntityTrades.java
97 lines (80 loc) · 2.98 KB
/
EntityTrades.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
package com.denizenscript.denizen.objects.properties.entity;
import com.denizenscript.denizen.objects.EntityTag;
import com.denizenscript.denizen.objects.TradeTag;
import com.denizenscript.denizencore.objects.Mechanism;
import com.denizenscript.denizencore.objects.core.ListTag;
import com.denizenscript.denizencore.objects.ObjectTag;
import com.denizenscript.denizencore.objects.properties.Property;
import com.denizenscript.denizencore.tags.Attribute;
import org.bukkit.inventory.Merchant;
import org.bukkit.inventory.MerchantRecipe;
import java.util.ArrayList;
public class EntityTrades implements Property {
public static boolean describes(ObjectTag entity) {
return entity instanceof EntityTag
&& ((EntityTag) entity).getBukkitEntity() instanceof Merchant;
}
public static EntityTrades getFrom(ObjectTag entity) {
if (!describes(entity)) {
return null;
}
return new EntityTrades((EntityTag) entity);
}
public static final String[] handledTags = new String[] {
"trades"
};
public static final String[] handledMechs = new String[] {
"trades"
};
public ListTag getTradeRecipes() {
ArrayList<TradeTag> recipes = new ArrayList<>();
for (MerchantRecipe recipe : ((Merchant) entity.getBukkitEntity()).getRecipes()) {
recipes.add(new TradeTag(recipe).duplicate());
}
return new ListTag(recipes);
}
private EntityTag entity;
public EntityTrades(EntityTag entity) {
this.entity = entity;
}
public String getPropertyString() {
return getTradeRecipes().identify();
}
public String getPropertyId() {
return "trades";
}
public ObjectTag getObjectAttribute(Attribute attribute) {
if (attribute == null) {
return null;
}
// <--[tag]
// @attribute <EntityTag.trades>
// @returns ListTag(TradeTag)
// @mechanism EntityTag.trades
// @description
// Returns a list of the Villager's trade recipes.
// -->
if (attribute.startsWith("trades")) {
return getTradeRecipes().getObjectAttribute(attribute.fulfill(1));
}
return null;
}
public void adjust(Mechanism mechanism) {
// <--[mechanism]
// @object EntityTag
// @name trades
// @input ListTag(TradeTag)
// @description
// Sets the trades that the entity will offer.
// @tags
// <EntityTag.trades>
// -->
if (mechanism.matches("trades")) {
ArrayList<MerchantRecipe> recipes = new ArrayList<>();
for (TradeTag recipe : mechanism.valueAsType(ListTag.class).filter(TradeTag.class, mechanism.context)) {
recipes.add(recipe.getRecipe());
}
((Merchant) entity.getBukkitEntity()).setRecipes(recipes);
}
}
}