package net.minecraft.advancements;

import com.mojang.serialization.Codec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.util.Optional;
import net.minecraft.core.ClientAsset;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.ComponentSerialization;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.world.item.ItemStackTemplate;

public class DisplayInfo {
	public static final Codec<DisplayInfo> CODEC = RecordCodecBuilder.create(
		i -> i.group(
				ItemStackTemplate.CODEC.fieldOf("icon").forGetter(DisplayInfo::getIcon),
				ComponentSerialization.CODEC.fieldOf("title").forGetter(DisplayInfo::getTitle),
				ComponentSerialization.CODEC.fieldOf("description").forGetter(DisplayInfo::getDescription),
				ClientAsset.ResourceTexture.CODEC.optionalFieldOf("background").forGetter(DisplayInfo::getBackground),
				AdvancementType.CODEC.optionalFieldOf("frame", AdvancementType.TASK).forGetter(DisplayInfo::getType),
				Codec.BOOL.optionalFieldOf("show_toast", true).forGetter(DisplayInfo::shouldShowToast),
				Codec.BOOL.optionalFieldOf("announce_to_chat", true).forGetter(DisplayInfo::shouldAnnounceChat),
				Codec.BOOL.optionalFieldOf("hidden", false).forGetter(DisplayInfo::isHidden)
			)
			.apply(i, DisplayInfo::new)
	);
	public static final StreamCodec<RegistryFriendlyByteBuf, DisplayInfo> STREAM_CODEC = StreamCodec.ofMember(
		DisplayInfo::serializeToNetwork, DisplayInfo::fromNetwork
	);
	private final Component title;
	private final Component description;
	private final ItemStackTemplate icon;
	private final Optional<ClientAsset.ResourceTexture> background;
	private final AdvancementType type;
	private final boolean showToast;
	private final boolean announceChat;
	private final boolean hidden;
	private float x;
	private float y;

	public DisplayInfo(
		final ItemStackTemplate icon,
		final Component title,
		final Component description,
		final Optional<ClientAsset.ResourceTexture> background,
		final AdvancementType type,
		final boolean showToast,
		final boolean announceChat,
		final boolean hidden
	) {
		this.title = title;
		this.description = description;
		this.icon = icon;
		this.background = background;
		this.type = type;
		this.showToast = showToast;
		this.announceChat = announceChat;
		this.hidden = hidden;
	}

	public void setLocation(final float x, final float y) {
		this.x = x;
		this.y = y;
	}

	public Component getTitle() {
		return this.title;
	}

	public Component getDescription() {
		return this.description;
	}

	public ItemStackTemplate getIcon() {
		return this.icon;
	}

	public Optional<ClientAsset.ResourceTexture> getBackground() {
		return this.background;
	}

	public AdvancementType getType() {
		return this.type;
	}

	public float getX() {
		return this.x;
	}

	public float getY() {
		return this.y;
	}

	public boolean shouldShowToast() {
		return this.showToast;
	}

	public boolean shouldAnnounceChat() {
		return this.announceChat;
	}

	public boolean isHidden() {
		return this.hidden;
	}

	private void serializeToNetwork(final RegistryFriendlyByteBuf output) {
		ComponentSerialization.TRUSTED_STREAM_CODEC.encode(output, this.title);
		ComponentSerialization.TRUSTED_STREAM_CODEC.encode(output, this.description);
		ItemStackTemplate.STREAM_CODEC.encode(output, this.icon);
		output.writeEnum(this.type);
		int flags = 0;
		if (this.background.isPresent()) {
			flags |= 1;
		}

		if (this.showToast) {
			flags |= 2;
		}

		if (this.hidden) {
			flags |= 4;
		}

		output.writeInt(flags);
		this.background.map(ClientAsset::id).ifPresent(output::writeIdentifier);
		output.writeFloat(this.x);
		output.writeFloat(this.y);
	}

	private static DisplayInfo fromNetwork(final RegistryFriendlyByteBuf input) {
		Component title = ComponentSerialization.TRUSTED_STREAM_CODEC.decode(input);
		Component description = ComponentSerialization.TRUSTED_STREAM_CODEC.decode(input);
		ItemStackTemplate icon = ItemStackTemplate.STREAM_CODEC.decode(input);
		AdvancementType frame = input.readEnum(AdvancementType.class);
		int flags = input.readInt();
		Optional<ClientAsset.ResourceTexture> background = (flags & 1) != 0 ? Optional.of(new ClientAsset.ResourceTexture(input.readIdentifier())) : Optional.empty();
		boolean showToast = (flags & 2) != 0;
		boolean hidden = (flags & 4) != 0;
		DisplayInfo info = new DisplayInfo(icon, title, description, background, frame, showToast, false, hidden);
		info.setLocation(input.readFloat(), input.readFloat());
		return info;
	}
}