/*
 * This file is part of the LoVisual Client distribution.
 * Copyright (c) 2026 pivosos2007.
 *
 * Licensed under the GNU General Public License v3.0.
 */

package dev.loki.lovisual.mixins;

import com.llamalad7.mixinextras.injector.ModifyExpressionValue;
import com.llamalad7.mixinextras.injector.ModifyReturnValue;
import com.llamalad7.mixinextras.sugar.Local;
import net.minecraft.client.renderer.feature.ItemFeatureRenderer;
import net.minecraft.client.renderer.rendertype.RenderType;
import net.minecraft.client.resources.model.geometry.BakedQuad;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Unique;
import org.spongepowered.asm.mixin.injection.At;

@Mixin(ItemFeatureRenderer.class)
public abstract class ItemFeatureRendererMixin {
    @Unique
    private static final int LOVISUAL_VIEW_FADE_DITHER_MARK = 0x00010001;
    @Unique
    private static final int LOVISUAL_VIEW_FADE_A2C_MARK = 0x00010100;

    @Unique
    private static boolean lovisual$isViewFadeMarked(int argb) {
        int alpha = (argb >>> 24) & 0xFF;
        if (alpha >= 255) {
            return false;
        }
        int rgbMarker = argb & 0x00010101;
        return rgbMarker == LOVISUAL_VIEW_FADE_DITHER_MARK || rgbMarker == LOVISUAL_VIEW_FADE_A2C_MARK;
    }

    @Unique
    private static boolean lovisual$hasViewFadeTint(int[] tints) {
        return tints != null && tints.length > 0 && lovisual$isViewFadeMarked(tints[0]);
    }

    @ModifyReturnValue(
            method = "getLayerColorSafe([ILnet/minecraft/client/resources/model/geometry/BakedQuad$MaterialInfo;)I",
            at = @At("RETURN")
    )
    private static int lovisual$applyViewFadeToUntintedItemQuads(int original,
                                                                  int[] tints,
                                                                  BakedQuad.MaterialInfo materialInfo) {
        if (original != -1 || !lovisual$hasViewFadeTint(tints)) {
            return original;
        }
        return tints[0];
    }

    @ModifyExpressionValue(
            method = "prepareMainSubmit",
            at = @At(value = "INVOKE", target = "Lnet/minecraft/client/resources/model/geometry/BakedQuad$MaterialInfo;itemRenderType()Lnet/minecraft/client/renderer/rendertype/RenderType;")
    )
    private RenderType lovisual$useTranslucentRenderTypeForViewFadeItem(RenderType original,
                                                                         @Local(argsOnly = true) ItemFeatureRenderer.Submit submit,
                                                                         @Local(ordinal = 0) BakedQuad.MaterialInfo materialInfo) {
        // Vanilla item translucent sheets render into ITEM_ENTITY_TARGET. For entity fade this can
        // corrupt the following world translucent composite, so item layers keep their own type.
        return original;
    }
}