import com.llamalad7.mixinextras.injector.ModifyExpressionValue;
import com.llamalad7.mixinextras.sugar.Local;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.world.item.ItemStack;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Unique;
import org.spongepowered.asm.mixin.injection.At;
import shadows.apotheosis.adventure.affix.Affix;
import shadows.apotheosis.adventure.loot.LootRarity;

import java.util.Set;

@Mixin(value = LootRarity.LootRule.class, remap = false)
public abstract class LootRarityMixin {
    @ModifyExpressionValue(method = "lambda$execute$3", at = @At(value = "INVOKE", target = "Ljava/util/Set;contains(Ljava/lang/Object;)Z"))
    private static boolean filterAffixes(boolean doesContain, final ItemStack stack, final LootRarity rarity, final Set<Affix> currentAffixes, @Local(argsOnly = true) final Affix affix) {
        if (doesContain) {
            return true;
        }

        String groupId = trialanderro$getGroupId(affix);

        if (!groupId.isBlank()) {
            for (Affix currentAffix : currentAffixes) {
                if (trialanderro$getGroupId(currentAffix).equals(groupId)) {
                    return true;
                }
            }
        }

        return false;
    }

    @Unique
    private static String trialanderro$getGroupId(final Affix affix) {
        if (affix == null) {
            return "";
        }

        ResourceLocation id = affix.getId();
        String[] path = id.getPath().split("/");

        if (path.length > 1 && path[0].equals("group")) {
            return path[1];
        }

        return "";
    }
}