/
ModifierSetWorktableRecipeBuilder.java
101 lines (87 loc) · 3.67 KB
/
ModifierSetWorktableRecipeBuilder.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
package slimeknights.tconstruct.library.recipe.worktable;
import com.google.gson.JsonObject;
import lombok.AccessLevel;
import lombok.AllArgsConstructor;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import lombok.experimental.Accessors;
import net.minecraft.data.recipes.FinishedRecipe;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.tags.TagKey;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.crafting.Ingredient;
import net.minecraft.world.item.crafting.RecipeSerializer;
import slimeknights.mantle.data.predicate.IJsonPredicate;
import slimeknights.tconstruct.common.TinkerTags;
import slimeknights.tconstruct.library.json.predicate.modifier.ModifierPredicate;
import slimeknights.tconstruct.library.modifiers.ModifierId;
import slimeknights.tconstruct.tools.TinkerModifiers;
import javax.annotation.Nullable;
import java.util.function.Consumer;
/** Builder for recipes to add or remove a modifier from a set in persistent data */
@RequiredArgsConstructor(access = AccessLevel.PRIVATE)
@AllArgsConstructor(access = AccessLevel.PRIVATE)
public class ModifierSetWorktableRecipeBuilder extends AbstractSizedIngredientRecipeBuilder<ModifierSetWorktableRecipeBuilder> {
private final ResourceLocation dataKey;
@Setter @Accessors(fluent = true)
private IJsonPredicate<ModifierId> modifierPredicate = ModifierPredicate.ANY;
private final boolean addToSet;
private Ingredient tools = Ingredient.EMPTY;
private boolean allowTraits = false;
/** Creates a new recipe for adding to a set */
public static ModifierSetWorktableRecipeBuilder setAdding(ResourceLocation dataKey) {
return new ModifierSetWorktableRecipeBuilder(dataKey, true);
}
/** Creates a new recipe for removing from a set */
public static ModifierSetWorktableRecipeBuilder setRemoving(ResourceLocation dataKey) {
return new ModifierSetWorktableRecipeBuilder(dataKey, false);
}
/** Sets the tool requirement for this recipe */
public ModifierSetWorktableRecipeBuilder setTools(Ingredient ingredient) {
this.tools = ingredient;
return this;
}
/** Sets the tool requirement for this recipe */
public ModifierSetWorktableRecipeBuilder setTools(TagKey<Item> tag) {
return this.setTools(Ingredient.of(tag));
}
/** Sets the recipe to allow traits */
public ModifierSetWorktableRecipeBuilder allowTraits() {
allowTraits = true;
return this;
}
@Override
public void save(Consumer<FinishedRecipe> consumer) {
save(consumer, dataKey);
}
@Override
public void save(Consumer<FinishedRecipe> consumer, ResourceLocation id) {
if (inputs.isEmpty()) {
throw new IllegalStateException("Must have at least one ingredient");
}
ResourceLocation advancementId = buildOptionalAdvancement(id, "modifiers");
consumer.accept(new Finished(id, advancementId));
}
private class Finished extends SizedFinishedRecipe {
public Finished(ResourceLocation id, @Nullable ResourceLocation advancementId) {
super(id, advancementId);
}
@Override
public void serializeRecipeData(JsonObject json) {
json.addProperty("data_key", dataKey.toString());
Ingredient ingredient = tools;
if (ingredient == Ingredient.EMPTY) {
ingredient = Ingredient.of(TinkerTags.Items.MODIFIABLE);
}
json.add("tools", ingredient.toJson());
super.serializeRecipeData(json);
json.add("modifier_predicate", ModifierPredicate.LOADER.serialize(modifierPredicate));
json.addProperty("add_to_set", addToSet);
json.addProperty("allow_traits", allowTraits);
}
@Override
public RecipeSerializer<?> getType() {
return TinkerModifiers.modifierSetWorktableSerializer.get();
}
}
}