package moe.nea.firmament.mixins.custommodels;

import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;
import com.llamalad7.mixinextras.injector.ModifyExpressionValue;
import com.llamalad7.mixinextras.injector.ModifyReturnValue;
import com.llamalad7.mixinextras.sugar.Local;
import moe.nea.firmament.features.texturepack.CustomModelOverrideParser;
import moe.nea.firmament.features.texturepack.ModelOverrideData;
import net.minecraft.client.render.model.json.ModelOverride;
import net.minecraft.util.Identifier;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;

import java.util.List;
import java.util.Map;

@Mixin(ModelOverride.Deserializer.class)
public class PatchOverrideDeserializer {

	@ModifyReturnValue(
		method = "deserialize(Lcom/google/gson/JsonElement;Ljava/lang/reflect/Type;Lcom/google/gson/JsonDeserializationContext;)Lnet/minecraft/client/render/model/json/ModelOverride;",
		at = @At(value = "RETURN"))
	private ModelOverride addCustomOverrides(ModelOverride original, @Local JsonObject jsonObject) {
		var originalData = (ModelOverrideData) (Object) original;
		originalData.setFirmamentOverrides(CustomModelOverrideParser.parseCustomModelOverrides(jsonObject));
		return original;
	}

	@ModifyExpressionValue(
		method = "deserializeMinPropertyValues(Lcom/google/gson/JsonObject;)Ljava/util/List;",
		at = @At(value = "INVOKE", target = "Ljava/util/Map$Entry;getValue()Ljava/lang/Object;"))
	private Object removeFirmamentPredicatesFromJsonIteration(Object original, @Local Map.Entry<String, JsonElement> entry) {
		if (entry.getKey().startsWith("firmament:")) return new JsonPrimitive(0F);
		return original;
	}

	@Inject(
		method = "deserializeMinPropertyValues",
		at = @At(value = "INVOKE", target = "Ljava/util/Map;entrySet()Ljava/util/Set;")
	)
	private void whatever(JsonObject object, CallbackInfoReturnable<List<ModelOverride.Condition>> cir,
	                      @Local Map<Identifier, Float> maps) {
		maps.entrySet().removeIf(it -> it.getKey().getNamespace().equals("firmament"));
	}
}