package net.minecraft.world.entity.vehicle.minecart;

import java.util.Objects;
import net.minecraft.commands.CommandSource;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.syncher.EntityDataAccessor;
import net.minecraft.network.syncher.EntityDataSerializers;
import net.minecraft.network.syncher.SynchedEntityData;
import net.minecraft.server.level.ServerLevel;
import net.minecraft.server.permissions.LevelBasedPermissionSet;
import net.minecraft.world.InteractionHand;
import net.minecraft.world.InteractionResult;
import net.minecraft.world.entity.EntityType;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.Items;
import net.minecraft.world.level.BaseCommandBlock;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.Blocks;
import net.minecraft.world.level.block.state.BlockState;
import net.minecraft.world.level.storage.ValueInput;
import net.minecraft.world.level.storage.ValueOutput;
import net.minecraft.world.phys.Vec3;

public class MinecartCommandBlock extends AbstractMinecart {
	private static final EntityDataAccessor<String> DATA_ID_COMMAND_NAME = SynchedEntityData.defineId(MinecartCommandBlock.class, EntityDataSerializers.STRING);
	private static final EntityDataAccessor<Component> DATA_ID_LAST_OUTPUT = SynchedEntityData.defineId(
		MinecartCommandBlock.class, EntityDataSerializers.COMPONENT
	);
	private final BaseCommandBlock commandBlock = new MinecartCommandBlock.MinecartCommandBase();
	private static final int ACTIVATION_DELAY = 4;
	private int lastActivated;

	public MinecartCommandBlock(final EntityType<? extends MinecartCommandBlock> type, final Level level) {
		super(type, level);
	}

	@Override
	protected Item getDropItem() {
		return Items.MINECART;
	}

	@Override
	public ItemStack getPickResult() {
		return new ItemStack(Items.COMMAND_BLOCK_MINECART);
	}

	@Override
	protected void defineSynchedData(final SynchedEntityData.Builder entityData) {
		super.defineSynchedData(entityData);
		entityData.define(DATA_ID_COMMAND_NAME, "");
		entityData.define(DATA_ID_LAST_OUTPUT, CommonComponents.EMPTY);
	}

	@Override
	protected void readAdditionalSaveData(final ValueInput input) {
		super.readAdditionalSaveData(input);
		this.commandBlock.load(input);
		this.getEntityData().set(DATA_ID_COMMAND_NAME, this.getCommandBlock().getCommand());
		this.getEntityData().set(DATA_ID_LAST_OUTPUT, this.getCommandBlock().getLastOutput());
	}

	@Override
	protected void addAdditionalSaveData(final ValueOutput output) {
		super.addAdditionalSaveData(output);
		this.commandBlock.save(output);
	}

	@Override
	public BlockState getDefaultDisplayBlockState() {
		return Blocks.COMMAND_BLOCK.defaultBlockState();
	}

	public BaseCommandBlock getCommandBlock() {
		return this.commandBlock;
	}

	@Override
	public void activateMinecart(final ServerLevel level, final int xt, final int yt, final int zt, final boolean state) {
		if (state && this.tickCount - this.lastActivated >= 4) {
			this.getCommandBlock().performCommand(level);
			this.lastActivated = this.tickCount;
		}
	}

	@Override
	public InteractionResult interact(final Player player, final InteractionHand hand, final Vec3 location) {
		if (!player.canUseGameMasterBlocks()) {
			return InteractionResult.PASS;
		} else {
			if (player.level().isClientSide()) {
				player.openMinecartCommandBlock(this);
			}

			return InteractionResult.SUCCESS;
		}
	}

	@Override
	public void onSyncedDataUpdated(final EntityDataAccessor<?> accessor) {
		super.onSyncedDataUpdated(accessor);
		if (DATA_ID_LAST_OUTPUT.equals(accessor)) {
			try {
				this.commandBlock.setLastOutput(this.getEntityData().get(DATA_ID_LAST_OUTPUT));
			} catch (Throwable var3) {
			}
		} else if (DATA_ID_COMMAND_NAME.equals(accessor)) {
			this.commandBlock.setCommand(this.getEntityData().get(DATA_ID_COMMAND_NAME));
		}
	}

	private class MinecartCommandBase extends BaseCommandBlock {
		private MinecartCommandBase() {
			Objects.requireNonNull(MinecartCommandBlock.this);
			super();
		}

		@Override
		public void onUpdated(final ServerLevel level) {
			MinecartCommandBlock.this.getEntityData().set(MinecartCommandBlock.DATA_ID_COMMAND_NAME, this.getCommand());
			MinecartCommandBlock.this.getEntityData().set(MinecartCommandBlock.DATA_ID_LAST_OUTPUT, this.getLastOutput());
		}

		@Override
		public CommandSourceStack createCommandSourceStack(final ServerLevel level, final CommandSource source) {
			return new CommandSourceStack(
				source,
				MinecartCommandBlock.this.position(),
				MinecartCommandBlock.this.getRotationVector(),
				level,
				LevelBasedPermissionSet.GAMEMASTER,
				this.getName().getString(),
				MinecartCommandBlock.this.getDisplayName(),
				level.getServer(),
				MinecartCommandBlock.this
			);
		}

		@Override
		public boolean isValid() {
			return !MinecartCommandBlock.this.isRemoved();
		}
	}
}