194 lines
8.4 KiB
Java
194 lines
8.4 KiB
Java
/*
|
|
* Minecraft Forge
|
|
* Copyright (c) 2016-2019.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation version 2.1
|
|
* of the License.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
package net.minecraftforge.common.data;
|
|
|
|
import com.mojang.datafixers.util.Pair;
|
|
import net.minecraft.advancements.criterion.ItemPredicate;
|
|
import net.minecraft.data.DataGenerator;
|
|
import net.minecraft.data.LootTableProvider;
|
|
import net.minecraft.item.Item;
|
|
import net.minecraft.item.Items;
|
|
import net.minecraft.tags.ITag;
|
|
import net.minecraft.util.ResourceLocation;
|
|
import net.minecraft.loot.*;
|
|
import net.minecraft.loot.conditions.Alternative;
|
|
import net.minecraft.loot.conditions.ILootCondition;
|
|
import net.minecraft.loot.conditions.Inverted;
|
|
import net.minecraft.loot.conditions.MatchTool;
|
|
import net.minecraftforge.common.Tags;
|
|
import net.minecraftforge.fml.common.ObfuscationReflectionHelper;
|
|
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.function.BiConsumer;
|
|
import java.util.function.Consumer;
|
|
import java.util.function.Supplier;
|
|
import java.util.stream.Collectors;
|
|
|
|
/**
|
|
* Currently used only for replacing shears item to shears tag
|
|
*/
|
|
public class ForgeLootTableProvider extends LootTableProvider {
|
|
|
|
public ForgeLootTableProvider(DataGenerator gen) {
|
|
super(gen);
|
|
}
|
|
|
|
@Override
|
|
protected void validate(Map<ResourceLocation, LootTable> map, ValidationTracker validationtracker) {
|
|
// do not validate against all registered loot tables
|
|
}
|
|
|
|
@Override
|
|
protected List<Pair<Supplier<Consumer<BiConsumer<ResourceLocation, LootTable.Builder>>>, LootParameterSet>> getTables() {
|
|
return super.getTables().stream().map(pair -> {
|
|
// provides new consumer with filtering only changed loot tables and replacing condition item to condition tag
|
|
return new Pair<Supplier<Consumer<BiConsumer<ResourceLocation, LootTable.Builder>>>, LootParameterSet>(() -> replaceAndFilterChangesOnly(pair.getFirst().get()), pair.getSecond());
|
|
}).collect(Collectors.toList());
|
|
}
|
|
|
|
private Consumer<BiConsumer<ResourceLocation, LootTable.Builder>> replaceAndFilterChangesOnly(Consumer<BiConsumer<ResourceLocation, LootTable.Builder>> consumer) {
|
|
return (newConsumer) -> consumer.accept((resourceLocation, builder) -> {
|
|
if (findAndReplaceInLootTableBuilder(builder, Items.SHEARS, Tags.Items.SHEARS)) {
|
|
newConsumer.accept(resourceLocation, builder);
|
|
}
|
|
});
|
|
}
|
|
|
|
private boolean findAndReplaceInLootTableBuilder(LootTable.Builder builder, Item from, ITag.INamedTag<Item> to) {
|
|
List<LootPool> lootPools = ObfuscationReflectionHelper.getPrivateValue(LootTable.Builder.class, builder, "field_216041_a");
|
|
boolean found = false;
|
|
|
|
if (lootPools == null) {
|
|
throw new IllegalStateException(LootTable.Builder.class.getName() + " is missing field field_216041_a");
|
|
}
|
|
|
|
for (LootPool lootPool : lootPools) {
|
|
if (findAndReplaceInLootPool(lootPool, from, to)) {
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
private boolean findAndReplaceInLootPool(LootPool lootPool, Item from, ITag.INamedTag<Item> to) {
|
|
List<LootEntry> lootEntries = ObfuscationReflectionHelper.getPrivateValue(LootPool.class, lootPool, "field_186453_a");
|
|
List<ILootCondition> lootConditions = ObfuscationReflectionHelper.getPrivateValue(LootPool.class, lootPool, "field_186454_b");
|
|
boolean found = false;
|
|
|
|
if (lootEntries == null) {
|
|
throw new IllegalStateException(LootPool.class.getName() + " is missing field field_186453_a");
|
|
}
|
|
|
|
for (LootEntry lootEntry : lootEntries) {
|
|
if (lootEntry instanceof ParentedLootEntry) {
|
|
if (findAndReplaceInParentedLootEntry((ParentedLootEntry) lootEntry, from, to)) {
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (lootConditions == null) {
|
|
throw new IllegalStateException(LootPool.class.getName() + " is missing field field_186454_b");
|
|
}
|
|
|
|
for (int i = 0; i < lootConditions.size(); i++) {
|
|
ILootCondition lootCondition = lootConditions.get(i);
|
|
if (lootCondition instanceof MatchTool && checkMatchTool((MatchTool) lootCondition, from)) {
|
|
lootConditions.set(i, MatchTool.builder(ItemPredicate.Builder.create().tag(to)).build());
|
|
found = true;
|
|
} else if (lootCondition instanceof Inverted) {
|
|
ILootCondition invLootCondition = ObfuscationReflectionHelper.getPrivateValue(Inverted.class, (Inverted) lootCondition, "field_215981_a");
|
|
|
|
if (invLootCondition instanceof MatchTool && checkMatchTool((MatchTool) invLootCondition, from)) {
|
|
lootConditions.set(i, Inverted.builder(MatchTool.builder(ItemPredicate.Builder.create().tag(to))).build());
|
|
found = true;
|
|
} else if (invLootCondition instanceof Alternative && findAndReplaceInAlternative((Alternative) invLootCondition, from, to)) {
|
|
found = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
private boolean findAndReplaceInParentedLootEntry(ParentedLootEntry entry, Item from, ITag.INamedTag<Item> to) {
|
|
LootEntry[] lootEntries = ObfuscationReflectionHelper.getPrivateValue(ParentedLootEntry.class, entry, "field_216147_c");
|
|
boolean found = false;
|
|
|
|
if (lootEntries == null) {
|
|
throw new IllegalStateException(ParentedLootEntry.class.getName() + " is missing field field_216147_c");
|
|
}
|
|
|
|
for (LootEntry lootEntry : lootEntries) {
|
|
if (findAndReplaceInLootEntry(lootEntry, from, to)) {
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
private boolean findAndReplaceInLootEntry(LootEntry entry, Item from, ITag.INamedTag<Item> to) {
|
|
ILootCondition[] lootConditions = ObfuscationReflectionHelper.getPrivateValue(LootEntry.class, entry, "field_216144_d");
|
|
boolean found = false;
|
|
|
|
if (lootConditions == null) {
|
|
throw new IllegalStateException(LootEntry.class.getName() + " is missing field field_216144_d");
|
|
}
|
|
|
|
for (int i = 0; i < lootConditions.length; i++) {
|
|
if (lootConditions[i] instanceof Alternative && findAndReplaceInAlternative((Alternative) lootConditions[i], from, to)) {
|
|
found = true;
|
|
} else if (lootConditions[i] instanceof MatchTool && checkMatchTool((MatchTool) lootConditions[i], from)) {
|
|
lootConditions[i] = MatchTool.builder(ItemPredicate.Builder.create().tag(to)).build();
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
private boolean findAndReplaceInAlternative(Alternative alternative, Item from, ITag.INamedTag<Item> to) {
|
|
ILootCondition[] lootConditions = ObfuscationReflectionHelper.getPrivateValue(Alternative.class, alternative, "field_215962_a");
|
|
boolean found = false;
|
|
|
|
if (lootConditions == null) {
|
|
throw new IllegalStateException(Alternative.class.getName() + " is missing field field_215962_a");
|
|
}
|
|
|
|
for (int i = 0; i < lootConditions.length; i++) {
|
|
if (lootConditions[i] instanceof MatchTool && checkMatchTool((MatchTool) lootConditions[i], from)) {
|
|
lootConditions[i] = MatchTool.builder(ItemPredicate.Builder.create().tag(to)).build();
|
|
found = true;
|
|
}
|
|
}
|
|
|
|
return found;
|
|
}
|
|
|
|
private boolean checkMatchTool(MatchTool lootCondition, Item expected) {
|
|
ItemPredicate predicate = ObfuscationReflectionHelper.getPrivateValue(MatchTool.class, lootCondition, "field_216014_a");
|
|
Item item = ObfuscationReflectionHelper.getPrivateValue(ItemPredicate.class, predicate, "field_192496_b");
|
|
return item != null && item.equals(expected);
|
|
}
|
|
}
|