package dev.loki.lovisual.mixins;

import net.fabricmc.loader.api.FabricLoader;
import org.objectweb.asm.tree.ClassNode;
import org.spongepowered.asm.mixin.extensibility.IMixinConfigPlugin;
import org.spongepowered.asm.mixin.extensibility.IMixinInfo;
import dev.loki.lovisual.util.logging.DebugLog;

import java.util.List;
import java.util.Set;

public final class LoVisualMixinPlugin implements IMixinConfigPlugin {
    // Sodium 26.2 uses Mojang RenderPipeline for terrain on both GL and Vulkan backends.
    private static final boolean STUB_SODIUM = false;
    private static final boolean STUB_IRIS = false;
    private static final Set<String> DISABLED_SODIUM_SHADOW_GRAPH_MIXINS = Set.of();
    private static final Set<String> OPTIONAL_SODIUM_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumBlockRendererMixin",
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumChunkMeshFormatsMixin",
            "dev.loki.lovisual.mixins.sodium.vertex.SodiumChunkVertexMixin",
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumShaderChunkRendererMixin",
            "dev.loki.lovisual.mixins.sodium.vertex.SodiumVertexConsumerTrackerMixin",
            "dev.loki.lovisual.mixins.sodium.placement.SodiumPlaceCullMixin",
            "dev.loki.lovisual.mixins.sodium.placement.SodiumPlaceHideMixin",
            "dev.loki.lovisual.mixins.optimize.entity.SodiumEntityRendererMixin"
    );
    private static final Set<String> OPTIONAL_IRIS_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.iris.render.IrisCommonUniformsMixin",
            "dev.loki.lovisual.mixins.iris.render.IrisGameRendererInteropMixin",
            "dev.loki.lovisual.mixins.iris.hand.IrisHandRendererAccessor",
            "dev.loki.lovisual.mixins.iris.hand.IrisHandRendererMixin",
            "dev.loki.lovisual.mixins.iris.processor.IrisIncludeProcessorMixin",
            "dev.loki.lovisual.mixins.iris.render.IrisShaderPackLoadMixin",
            "dev.loki.lovisual.mixins.iris.render.IrisRenderingPipelineFinalizeMixin",
            "dev.loki.lovisual.mixins.iris.hand.IrisVanillaHandInteropMixin"
    );
    private static final Set<String> DISABLED_WITH_IRIS_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumBlockRendererMixin",
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumChunkMeshFormatsMixin",
            "dev.loki.lovisual.mixins.sodium.vertex.SodiumChunkVertexMixin",
            "dev.loki.lovisual.mixins.sodium.terrain.SodiumShaderChunkRendererMixin"
    );
    private static final Set<String> OPTIONAL_XAERO_MINIMAP_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.xaero.minimap.AbstractWaypointRenderProviderMixin"
    );
    private static final Set<String> OPTIONAL_XAERO_WORLDMAP_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.xaero.worldmap.WaypointRenderProviderMixin"
    );
    private static final Set<String> OPTIONAL_MORECULLING_MIXINS = Set.of(
            "dev.loki.lovisual.mixins.moreculling.CullingUtilsMixin"
    );
    private static volatile boolean loggedSodiumDecision;

    @Override
    public void onLoad(String mixinPackage) {
    }

    @Override
    public String getRefMapperConfig() {
        return null;
    }

    @Override
    public boolean shouldApplyMixin(String targetClassName, String mixinClassName) {
        if (STUB_SODIUM && OPTIONAL_SODIUM_MIXINS.contains(mixinClassName)) {
            return false;
        }
        if (STUB_IRIS && OPTIONAL_IRIS_MIXINS.contains(mixinClassName)) {
            return false;
        }
        if (DISABLED_SODIUM_SHADOW_GRAPH_MIXINS.contains(mixinClassName)) {
            return false;
        }
        if (OPTIONAL_XAERO_MINIMAP_MIXINS.contains(mixinClassName)) {
            return FabricLoader.getInstance().isModLoaded("xaerominimap");
        }
        if (OPTIONAL_XAERO_WORLDMAP_MIXINS.contains(mixinClassName)) {
            return FabricLoader.getInstance().isModLoaded("xaeroworldmap");
        }
        if (OPTIONAL_MORECULLING_MIXINS.contains(mixinClassName)) {
            return FabricLoader.getInstance().isModLoaded("moreculling");
        }
        if (OPTIONAL_IRIS_MIXINS.contains(mixinClassName)) {
            return FabricLoader.getInstance().isModLoaded("iris");
        }
        if (DISABLED_WITH_IRIS_MIXINS.contains(mixinClassName)) {
            return !FabricLoader.getInstance().isModLoaded("iris");
        }
        if (!OPTIONAL_SODIUM_MIXINS.contains(mixinClassName)) {
            return true;
        }
        boolean sodiumLoaded = FabricLoader.getInstance().isModLoaded("sodium");
        if (!loggedSodiumDecision) {
            loggedSodiumDecision = true;
            DebugLog.warn("[SodiumCompat] optional mixins %s", sodiumLoaded ? "enabled" : "disabled");
        }
        return sodiumLoaded;
    }

    @Override
    public void acceptTargets(Set<String> myTargets, Set<String> otherTargets) {
    }

    @Override
    public List<String> getMixins() {
        return null;
    }

    @Override
    public void preApply(String targetClassName, ClassNode targetClass, String mixinClassName, IMixinInfo mixinInfo) {
    }

    @Override
    public void postApply(String targetClassName, ClassNode targetClass, String mixinClassName, IMixinInfo mixinInfo) {
    }
}