package net.minecraft.world.entity.ai.attributes;

import com.mojang.logging.LogUtils;
import com.mojang.serialization.Codec;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.DataResult.Error;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import io.netty.buffer.ByteBuf;
import java.util.function.IntFunction;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.nbt.NbtOps;
import net.minecraft.nbt.Tag;
import net.minecraft.network.codec.ByteBufCodecs;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.ByIdMap;
import net.minecraft.util.StringRepresentable;
import net.minecraft.util.ByIdMap.OutOfBoundsStrategy;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;

public record AttributeModifier(ResourceLocation id, double amount, AttributeModifier.Operation operation) {
	private static final Logger LOGGER = LogUtils.getLogger();
	public static final MapCodec<AttributeModifier> MAP_CODEC = RecordCodecBuilder.mapCodec(
		instance -> instance.group(
				ResourceLocation.CODEC.fieldOf("id").forGetter(AttributeModifier::id),
				Codec.DOUBLE.fieldOf("amount").forGetter(AttributeModifier::amount),
				AttributeModifier.Operation.CODEC.fieldOf("operation").forGetter(AttributeModifier::operation)
			)
			.apply(instance, AttributeModifier::new)
	);
	public static final Codec<AttributeModifier> CODEC = MAP_CODEC.codec();
	public static final StreamCodec<ByteBuf, AttributeModifier> STREAM_CODEC = StreamCodec.composite(
		ResourceLocation.STREAM_CODEC,
		AttributeModifier::id,
		ByteBufCodecs.DOUBLE,
		AttributeModifier::amount,
		AttributeModifier.Operation.STREAM_CODEC,
		AttributeModifier::operation,
		AttributeModifier::new
	);

	public CompoundTag save() {
		DataResult<Tag> dataResult = CODEC.encode(this, NbtOps.INSTANCE, new CompoundTag());
		return (CompoundTag)dataResult.getOrThrow();
	}

	@Nullable
	public static AttributeModifier load(CompoundTag nbt) {
		DataResult<AttributeModifier> dataResult = CODEC.parse(NbtOps.INSTANCE, nbt);
		if (dataResult.isSuccess()) {
			return dataResult.getOrThrow();
		} else {
			LOGGER.warn("Unable to create attribute: {}", ((Error)dataResult.error().get()).message());
			return null;
		}
	}

	public boolean is(ResourceLocation id) {
		return id.equals(this.id);
	}

	public static enum Operation implements StringRepresentable {
		ADD_VALUE("add_value", 0),
		ADD_MULTIPLIED_BASE("add_multiplied_base", 1),
		ADD_MULTIPLIED_TOTAL("add_multiplied_total", 2);

		public static final IntFunction<AttributeModifier.Operation> BY_ID = ByIdMap.continuous(AttributeModifier.Operation::id, values(), OutOfBoundsStrategy.ZERO);
		public static final StreamCodec<ByteBuf, AttributeModifier.Operation> STREAM_CODEC = ByteBufCodecs.idMapper(BY_ID, AttributeModifier.Operation::id);
		public static final Codec<AttributeModifier.Operation> CODEC = StringRepresentable.fromEnum(AttributeModifier.Operation::values);
		private final String name;
		private final int id;

		private Operation(final String name, final int value) {
			this.name = name;
			this.id = value;
		}

		public int id() {
			return this.id;
		}

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