// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.core.command.commands.world.chunk;

import com.hypixel.hytale.component.Ref;
import com.hypixel.hytale.server.core.universe.world.storage.ChunkStore;
import com.hypixel.hytale.math.vector.Vector2i;
import com.hypixel.hytale.server.core.universe.world.chunk.WorldChunk;
import com.hypixel.hytale.server.core.Message;
import com.hypixel.hytale.math.util.ChunkUtil;
import com.hypixel.hytale.component.ComponentAccessor;
import com.hypixel.hytale.server.core.universe.world.storage.EntityStore;
import com.hypixel.hytale.component.Store;
import com.hypixel.hytale.server.core.universe.world.World;
import com.hypixel.hytale.server.core.command.system.CommandContext;
import com.hypixel.hytale.server.core.command.system.arguments.types.ArgTypes;
import javax.annotation.Nonnull;
import com.hypixel.hytale.server.core.command.system.arguments.types.RelativeChunkPosition;
import com.hypixel.hytale.server.core.command.system.arguments.system.RequiredArg;
import com.hypixel.hytale.server.core.command.system.basecommands.AbstractWorldCommand;

public class ChunkMarkSaveCommand extends AbstractWorldCommand
{
    @Nonnull
    private final RequiredArg<RelativeChunkPosition> chunkPosArg;
    
    public ChunkMarkSaveCommand() {
        super("marksave", "server.commands.chunk.marksave.desc");
        this.chunkPosArg = this.withRequiredArg("x z", "server.commands.chunk.marksave.position.desc", ArgTypes.RELATIVE_CHUNK_POSITION);
    }
    
    @Override
    protected void execute(@Nonnull final CommandContext context, @Nonnull final World world, @Nonnull final Store<EntityStore> store) {
        final RelativeChunkPosition chunkPosition = this.chunkPosArg.get(context);
        final Vector2i position = chunkPosition.getChunkPosition(context, store);
        final ChunkStore chunkStore = world.getChunkStore();
        final Store<ChunkStore> chunkStoreStore = chunkStore.getStore();
        final long chunkIndex = ChunkUtil.indexChunk(position.x, position.y);
        final Ref<ChunkStore> chunkRef = chunkStore.getChunkReference(chunkIndex);
        if (chunkRef == null || !chunkRef.isValid()) {
            context.sendMessage(Message.translation("server.commands.chunk.load.loading").param("chunkX", position.x).param("chunkZ", position.y).param("worldName", world.getName()));
            world.getChunkAsync(position.x, position.y).thenAccept(worldChunk -> world.execute(() -> {
                worldChunk.markNeedsSaving();
                context.sendMessage(Message.translation("server.commands.chunk.marksave.loaded").param("x", position.x).param("z", position.y).param("worldName", world.getName()));
            }));
            return;
        }
        final WorldChunk worldChunkComponent = chunkStoreStore.getComponent(chunkRef, WorldChunk.getComponentType());
        assert worldChunkComponent != null;
        worldChunkComponent.markNeedsSaving();
        context.sendMessage(Message.translation("server.commands.chunk.marksave.markingAlreadyLoaded").param("x", position.x).param("z", position.y).param("worldName", world.getName()));
    }
}