package net.minecraft.world.level.block.entity;

import com.mojang.serialization.Codec;
import net.minecraft.core.BlockPos;
import net.minecraft.core.FrontAndTop;
import net.minecraft.core.Holder;
import net.minecraft.core.HolderLookup;
import net.minecraft.core.Registry;
import net.minecraft.core.registries.Registries;
import net.minecraft.data.worldgen.Pools;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.game.ClientboundBlockEntityDataPacket;
import net.minecraft.resources.ResourceKey;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.util.StringRepresentable;
import net.minecraft.world.level.block.JigsawBlock;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.levelgen.structure.pools.JigsawPlacement;
import net.minecraft.world.level.levelgen.structure.pools.StructureTemplatePool;
import net.minecraft.world.level.levelgen.structure.templatesystem.StructureTemplate;

public class JigsawBlockEntity extends BlockEntity {
	public static final Codec<ResourceKey<StructureTemplatePool>> POOL_CODEC = ResourceKey.codec(Registries.TEMPLATE_POOL);
	public static final ResourceLocation EMPTY_ID = ResourceLocation.withDefaultNamespace("empty");
	private static final int DEFAULT_PLACEMENT_PRIORITY = 0;
	private static final int DEFAULT_SELECTION_PRIORITY = 0;
	public static final String TARGET = "target";
	public static final String POOL = "pool";
	public static final String JOINT = "joint";
	public static final String PLACEMENT_PRIORITY = "placement_priority";
	public static final String SELECTION_PRIORITY = "selection_priority";
	public static final String NAME = "name";
	public static final String FINAL_STATE = "final_state";
	public static final String DEFAULT_FINAL_STATE = "minecraft:air";
	private ResourceLocation name = EMPTY_ID;
	private ResourceLocation target = EMPTY_ID;
	private ResourceKey<StructureTemplatePool> pool = Pools.EMPTY;
	private JigsawBlockEntity.JointType joint = JigsawBlockEntity.JointType.ROLLABLE;
	private String finalState = "minecraft:air";
	private int placementPriority = 0;
	private int selectionPriority = 0;

	public JigsawBlockEntity(BlockPos pos, BlockState blockState) {
		super(BlockEntityType.JIGSAW, pos, blockState);
	}

	public ResourceLocation getName() {
		return this.name;
	}

	public ResourceLocation getTarget() {
		return this.target;
	}

	public ResourceKey<StructureTemplatePool> getPool() {
		return this.pool;
	}

	public String getFinalState() {
		return this.finalState;
	}

	public JigsawBlockEntity.JointType getJoint() {
		return this.joint;
	}

	public int getPlacementPriority() {
		return this.placementPriority;
	}

	public int getSelectionPriority() {
		return this.selectionPriority;
	}

	public void setName(ResourceLocation name) {
		this.name = name;
	}

	public void setTarget(ResourceLocation target) {
		this.target = target;
	}

	public void setPool(ResourceKey<StructureTemplatePool> pool) {
		this.pool = pool;
	}

	public void setFinalState(String finalState) {
		this.finalState = finalState;
	}

	public void setJoint(JigsawBlockEntity.JointType joint) {
		this.joint = joint;
	}

	public void setPlacementPriority(int placementPriority) {
		this.placementPriority = placementPriority;
	}

	public void setSelectionPriority(int selectionPriority) {
		this.selectionPriority = selectionPriority;
	}

	@Override
	protected void saveAdditional(CompoundTag tag, HolderLookup.Provider registries) {
		super.saveAdditional(tag, registries);
		tag.store("name", ResourceLocation.CODEC, this.name);
		tag.store("target", ResourceLocation.CODEC, this.target);
		tag.store("pool", POOL_CODEC, this.pool);
		tag.putString("final_state", this.finalState);
		tag.store("joint", JigsawBlockEntity.JointType.CODEC, this.joint);
		tag.putInt("placement_priority", this.placementPriority);
		tag.putInt("selection_priority", this.selectionPriority);
	}

	@Override
	protected void loadAdditional(CompoundTag tag, HolderLookup.Provider registries) {
		super.loadAdditional(tag, registries);
		this.name = (ResourceLocation)tag.read("name", ResourceLocation.CODEC).orElse(EMPTY_ID);
		this.target = (ResourceLocation)tag.read("target", ResourceLocation.CODEC).orElse(EMPTY_ID);
		this.pool = (ResourceKey<StructureTemplatePool>)tag.read("pool", POOL_CODEC).orElse(Pools.EMPTY);
		this.finalState = tag.getStringOr("final_state", "minecraft:air");
		this.joint = (JigsawBlockEntity.JointType)tag.read("joint", JigsawBlockEntity.JointType.CODEC)
			.orElseGet(() -> StructureTemplate.getDefaultJointType(this.getBlockState()));
		this.placementPriority = tag.getIntOr("placement_priority", 0);
		this.selectionPriority = tag.getIntOr("selection_priority", 0);
	}

	public ClientboundBlockEntityDataPacket getUpdatePacket() {
		return ClientboundBlockEntityDataPacket.create(this);
	}

	@Override
	public CompoundTag getUpdateTag(HolderLookup.Provider registries) {
		return this.saveCustomOnly(registries);
	}

	public void generate(ServerLevel level, int maxDepth, boolean keepJigsaws) {
		BlockPos blockPos = this.getBlockPos().relative(((FrontAndTop)this.getBlockState().getValue(JigsawBlock.ORIENTATION)).front());
		Registry<StructureTemplatePool> registry = level.registryAccess().lookupOrThrow(Registries.TEMPLATE_POOL);
		Holder<StructureTemplatePool> holder = registry.getOrThrow(this.pool);
		JigsawPlacement.generateJigsaw(level, holder, this.target, maxDepth, blockPos, keepJigsaws);
	}

	public static enum JointType implements StringRepresentable {
		ROLLABLE("rollable"),
		ALIGNED("aligned");

		public static final StringRepresentable.EnumCodec<JigsawBlockEntity.JointType> CODEC = StringRepresentable.fromEnum(JigsawBlockEntity.JointType::values);
		private final String name;

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

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

		public Component getTranslatedName() {
			return Component.translatable("jigsaw_block.joint." + this.name);
		}
	}
}