LibJF/src/main/java/io/gitlab/jfronny/libjf/mixin/RecipeManagerMixin.java

51 lines
2.1 KiB
Java
Raw Normal View History

package io.gitlab.jfronny.libjf.mixin;
import com.google.common.collect.ImmutableMap;
import com.google.gson.JsonObject;
import io.gitlab.jfronny.libjf.Libjf;
import io.gitlab.jfronny.libjf.data.RecipeUtil;
import net.minecraft.recipe.Recipe;
import net.minecraft.recipe.RecipeManager;
import net.minecraft.util.Identifier;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.ModifyVariable;
import org.spongepowered.asm.mixin.injection.Redirect;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.Map;
@Mixin(RecipeManager.class)
public class RecipeManagerMixin {
@ModifyVariable(method = "apply", at = @At(value = "INVOKE_ASSIGN", target = "Ljava/util/Set;iterator()Ljava/util/Iterator;", ordinal = 0, remap = false))
private Iterator<Map.Entry<Identifier, JsonObject>> filterIterator(Iterator<Map.Entry<Identifier, JsonObject>> iterator) {
ArrayList<Map.Entry<Identifier, JsonObject>> replacement = new ArrayList<>();
while(iterator.hasNext()) {
Map.Entry<Identifier, JsonObject> cur = iterator.next();
Identifier recipeId = cur.getKey();
if (RecipeUtil.isIdBlocked(recipeId)) {
Libjf.LOGGER.info("Blocking recipe by identifier: " + recipeId);
} else {
replacement.add(cur);
}
}
return replacement.iterator();
}
@Redirect(method = "apply", at = @At(value = "INVOKE", target = "Lcom/google/common/collect/ImmutableMap$Builder;put(Ljava/lang/Object;Ljava/lang/Object;)Lcom/google/common/collect/ImmutableMap$Builder;", remap = false))
private ImmutableMap.Builder<Identifier, Recipe<?>> onPutRecipe(ImmutableMap.Builder<Identifier, Recipe<?>> builder, Object key, Object value) {
Identifier id = (Identifier) key;
Recipe<?> recipe = (Recipe<?>) value;
if (RecipeUtil.isOutputBlocked(recipe.getOutput())) {
Libjf.LOGGER.info("Blocked recipe by predicate: " + recipe.getId());
return builder;
}
return builder.put(id, recipe);
}
}