Auto-eat blacklist (TerraFirmaGreg-Team#228)

Author: Mqrius

src/main/java/su/terrafirmagreg/core/common/data/TFGTags.java

@@ -36,6 +36,8 @@ public static final class Items {
         public static final TagKey<Item> ColdProtectionEquipment = createItemTag("tfg:cold_protection_equipment");
         public static final TagKey<Item> FloatingProtectionEquipment = createItemTag("tfg:floating_protection_equipment");
 
+        public static final TagKey<Item> AutoEatBlacklist = createItemTag("tfg:auto_eat_blacklist");
+
         //Block Interaction tags for use in EMI
         public static final TagKey<Item> INTERACTIONBRICK = createItemTag("tfg:interaction/brick");
         public static final TagKey<Item> INTERACTIONBRICKSTAIR = createItemTag("tfg:interaction/brick_stairs");

src/main/java/su/terrafirmagreg/core/mixins/common/gtceu/QuarkTechSuiteMixin.java

@@ -7,19 +7,35 @@
 
 import com.gregtechceu.gtceu.api.capability.IElectricItem;
 import com.gregtechceu.gtceu.common.item.armor.QuarkTechSuite;
+import com.llamalad7.mixinextras.injector.wrapoperation.Operation;
+import com.llamalad7.mixinextras.injector.wrapoperation.WrapOperation;
 
+import net.dries007.tfc.common.TFCTags;
+import net.minecraft.world.entity.LivingEntity;
 import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.food.FoodProperties;
+import net.minecraft.world.item.ItemStack;
 
-// Cancel QuarkTech helmet feeding when player is in PlayerRevive bleeding state
-// This prevents food from being consumed without effect
+import su.terrafirmagreg.core.common.data.TFGTags;
 
 @Mixin(value = QuarkTechSuite.class, remap = false)
 public class QuarkTechSuiteMixin {
 
+    // Cancel QuarkTech helmet feeding when player is in PlayerRevive bleeding state
+    // This prevents food from being consumed without effect
     @Inject(method = "supplyFood", at = @At("HEAD"), remap = false, cancellable = true)
     private void tfg$preventQuarkFeedingWhenBleeding(IElectricItem item, Player player, CallbackInfoReturnable<Boolean> cir) {
         if (player.getPersistentData().getBoolean("playerrevive:bleeding") || player.level().isClientSide) {
             cir.setReturnValue(false);
         }
     }
+
+    // Only eat food, don't eat blacklisted food
+    @WrapOperation(method = "supplyFood", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/item/ItemStack;getFoodProperties(Lnet/minecraft/world/entity/LivingEntity;)Lnet/minecraft/world/food/FoodProperties;", remap = true), remap = false)
+    private FoodProperties tfg$checkFoodTag(ItemStack stack, LivingEntity player, Operation<FoodProperties> original) {
+        if (stack.is(TFCTags.Items.FOODS) || !stack.is(TFGTags.Items.AutoEatBlacklist)) {
+            return original.call(stack, player);
+        }
+        return null;
+    }
 }

src/main/java/su/terrafirmagreg/core/mixins/common/sophisticatedbackpacks/FeedingUpgradeWrapperMixin.java

@@ -3,26 +3,38 @@
 import javax.annotation.Nullable;
 
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Overwrite;
 import org.spongepowered.asm.mixin.Pseudo;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
 import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
 
+import net.dries007.tfc.common.TFCTags;
 import net.minecraft.core.BlockPos;
 import net.minecraft.world.entity.Entity;
+import net.minecraft.world.entity.LivingEntity;
 import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.level.Level;
 
-// Cancel backpack feeding upgrade tick when player is in PlayerRevive bleeding state
-// This prevents food from being consumed without effect
+import su.terrafirmagreg.core.common.data.TFGTags;
 
 @Pseudo
 @Mixin(targets = "net.p3pp3rf1y.sophisticatedcore.upgrades.feeding.FeedingUpgradeWrapper", remap = false)
 public class FeedingUpgradeWrapperMixin {
+
+    // Cancel backpack feeding upgrade tick when player is in PlayerRevive bleeding state
+    // This prevents food from being consumed without effect
     @Inject(method = "tick", at = @At("HEAD"), remap = false, cancellable = true)
     private void tfg$preventFeedingWhenBleeding(@Nullable Entity entity, Level level, BlockPos pos, CallbackInfo ci) {
         if (entity instanceof Player player && player.getPersistentData().getBoolean("playerrevive:bleeding")) {
             ci.cancel();
         }
     }
+
+    // Only eat food, don't eat blacklisted food
+    @Overwrite(remap = false)
+    private static boolean isEdible(ItemStack stack, LivingEntity player) {
+        return stack.is(TFCTags.Items.FOODS) && !stack.is(TFGTags.Items.AutoEatBlacklist);
+    }
 }