package net.minecraft.world.attribute.modifier;

import com.mojang.serialization.Codec;
import java.util.Map;
import net.minecraft.util.StringRepresentable;
import net.minecraft.world.attribute.EnvironmentAttribute;
import net.minecraft.world.attribute.LerpFunction;

public interface AttributeModifier<Subject, Argument> {
	Map<AttributeModifier.OperationId, AttributeModifier<Boolean, ?>> BOOLEAN_LIBRARY = Map.of(
		AttributeModifier.OperationId.AND,
		BooleanModifier.AND,
		AttributeModifier.OperationId.NAND,
		BooleanModifier.NAND,
		AttributeModifier.OperationId.OR,
		BooleanModifier.OR,
		AttributeModifier.OperationId.NOR,
		BooleanModifier.NOR,
		AttributeModifier.OperationId.XOR,
		BooleanModifier.XOR,
		AttributeModifier.OperationId.XNOR,
		BooleanModifier.XNOR
	);
	Map<AttributeModifier.OperationId, AttributeModifier<Float, ?>> FLOAT_LIBRARY = Map.of(
		AttributeModifier.OperationId.ALPHA_BLEND,
		FloatModifier.ALPHA_BLEND,
		AttributeModifier.OperationId.ADD,
		FloatModifier.ADD,
		AttributeModifier.OperationId.SUBTRACT,
		FloatModifier.SUBTRACT,
		AttributeModifier.OperationId.MULTIPLY,
		FloatModifier.MULTIPLY,
		AttributeModifier.OperationId.MINIMUM,
		FloatModifier.MINIMUM,
		AttributeModifier.OperationId.MAXIMUM,
		FloatModifier.MAXIMUM
	);
	Map<AttributeModifier.OperationId, AttributeModifier<Integer, ?>> RGB_COLOR_LIBRARY = Map.of(
		AttributeModifier.OperationId.ALPHA_BLEND,
		ColorModifier.ALPHA_BLEND,
		AttributeModifier.OperationId.ADD,
		ColorModifier.ADD,
		AttributeModifier.OperationId.SUBTRACT,
		ColorModifier.SUBTRACT,
		AttributeModifier.OperationId.MULTIPLY,
		ColorModifier.MULTIPLY_RGB,
		AttributeModifier.OperationId.BLEND_TO_GRAY,
		ColorModifier.BLEND_TO_GRAY
	);
	Map<AttributeModifier.OperationId, AttributeModifier<Integer, ?>> ARGB_COLOR_LIBRARY = Map.of(
		AttributeModifier.OperationId.ALPHA_BLEND,
		ColorModifier.ALPHA_BLEND,
		AttributeModifier.OperationId.ADD,
		ColorModifier.ADD,
		AttributeModifier.OperationId.SUBTRACT,
		ColorModifier.SUBTRACT,
		AttributeModifier.OperationId.MULTIPLY,
		ColorModifier.MULTIPLY_ARGB,
		AttributeModifier.OperationId.BLEND_TO_GRAY,
		ColorModifier.BLEND_TO_GRAY
	);
	Map<AttributeModifier.OperationId, AttributeModifier<Integer, ?>> INTEGER_LIBRARY = Map.of(
		AttributeModifier.OperationId.ADD,
		IntegerModifier.ADD,
		AttributeModifier.OperationId.SUBTRACT,
		IntegerModifier.SUBTRACT,
		AttributeModifier.OperationId.MULTIPLY,
		IntegerModifier.MULTIPLY,
		AttributeModifier.OperationId.MINIMUM,
		IntegerModifier.MINIMUM,
		AttributeModifier.OperationId.MAXIMUM,
		IntegerModifier.MAXIMUM
	);

	static <Value> AttributeModifier<Value, Value> override() {
		return AttributeModifier.OverrideModifier.INSTANCE;
	}

	Subject apply(Subject subject, Argument argument);

	Codec<Argument> argumentCodec(EnvironmentAttribute<Subject> attribute);

	LerpFunction<Argument> argumentKeyframeLerp(EnvironmentAttribute<Subject> attribute);

	public static enum OperationId implements StringRepresentable {
		OVERRIDE("override"),
		ALPHA_BLEND("alpha_blend"),
		ADD("add"),
		SUBTRACT("subtract"),
		MULTIPLY("multiply"),
		BLEND_TO_GRAY("blend_to_gray"),
		MINIMUM("minimum"),
		MAXIMUM("maximum"),
		AND("and"),
		NAND("nand"),
		OR("or"),
		NOR("nor"),
		XOR("xor"),
		XNOR("xnor");

		public static final Codec<AttributeModifier.OperationId> CODEC = StringRepresentable.fromEnum(AttributeModifier.OperationId::values);
		private final String name;

		private OperationId(final String name) {
			this.name = name;
		}

		@Override
		public String getSerializedName() {
			return this.name;
		}
	}

	public record OverrideModifier<Value>() implements AttributeModifier<Value, Value> {
		private static final AttributeModifier.OverrideModifier<?> INSTANCE = new AttributeModifier.OverrideModifier();

		@Override
		public Value apply(final Value subject, final Value argument) {
			return argument;
		}

		@Override
		public Codec<Value> argumentCodec(final EnvironmentAttribute<Value> attribute) {
			return attribute.valueCodec();
		}

		@Override
		public LerpFunction<Value> argumentKeyframeLerp(final EnvironmentAttribute<Value> attribute) {
			return attribute.type().keyframeLerp();
		}
	}
}