mirror of
https://github.com/plexusorg/Plex-FAWE.git
synced 2024-06-01 17:01:45 +00:00
5e534ff9e2
Follow up to a9d1202ce1
197 lines
8.5 KiB
Java
197 lines
8.5 KiB
Java
/*
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
* Copyright (C) WorldEdit team and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package com.sk89q.worldedit.command;
|
|
|
|
import com.boydti.fawe.config.Caption;
|
|
import com.google.gson.JsonIOException;
|
|
import com.sk89q.worldedit.LocalSession;
|
|
import com.sk89q.worldedit.WorldEdit;
|
|
import com.sk89q.worldedit.WorldEditException;
|
|
import com.sk89q.worldedit.command.util.CommandPermissions;
|
|
import com.sk89q.worldedit.command.util.CommandPermissionsConditionGenerator;
|
|
import com.sk89q.worldedit.command.util.Logging;
|
|
import com.sk89q.worldedit.entity.Player;
|
|
import com.sk89q.worldedit.extension.platform.Actor;
|
|
import com.sk89q.worldedit.internal.anvil.ChunkDeleter;
|
|
import com.sk89q.worldedit.internal.anvil.ChunkDeletionInfo;
|
|
import com.sk89q.worldedit.math.BlockVector2;
|
|
import com.sk89q.worldedit.regions.CuboidRegion;
|
|
import com.sk89q.worldedit.regions.Region;
|
|
import com.sk89q.worldedit.util.Location;
|
|
import com.sk89q.worldedit.util.formatting.component.PaginationBox;
|
|
import com.sk89q.worldedit.util.formatting.text.Component;
|
|
import com.sk89q.worldedit.util.formatting.text.TextComponent;
|
|
import com.sk89q.worldedit.util.formatting.text.TranslatableComponent;
|
|
import com.sk89q.worldedit.util.formatting.text.event.ClickEvent;
|
|
import com.sk89q.worldedit.util.formatting.text.format.TextColor;
|
|
import com.sk89q.worldedit.world.World;
|
|
import com.sk89q.worldedit.world.storage.LegacyChunkStore;
|
|
import com.sk89q.worldedit.world.storage.McRegionChunkStore;
|
|
import org.enginehub.piston.annotation.Command;
|
|
import org.enginehub.piston.annotation.CommandContainer;
|
|
import org.enginehub.piston.annotation.param.ArgFlag;
|
|
import org.enginehub.piston.exception.StopExecutionException;
|
|
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.time.ZonedDateTime;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
|
|
import static com.google.common.base.Preconditions.checkNotNull;
|
|
import static com.sk89q.worldedit.command.util.Logging.LogMode.REGION;
|
|
import static com.sk89q.worldedit.internal.anvil.ChunkDeleter.DELCHUNKS_FILE_NAME;
|
|
|
|
/**
|
|
* Commands for working with chunks.
|
|
*/
|
|
@CommandContainer(superTypes = CommandPermissionsConditionGenerator.Registration.class)
|
|
public class ChunkCommands {
|
|
|
|
private final WorldEdit worldEdit;
|
|
|
|
public ChunkCommands(WorldEdit worldEdit) {
|
|
checkNotNull(worldEdit);
|
|
this.worldEdit = worldEdit;
|
|
}
|
|
|
|
@Command(
|
|
name = "chunkinfo",
|
|
aliases = { "/chunkinfo" },
|
|
desc = "Get information about the chunk you're inside"
|
|
)
|
|
@CommandPermissions("worldedit.chunkinfo")
|
|
public void chunkInfo(Player player) {
|
|
Location pos = player.getBlockLocation();
|
|
int chunkX = (int) Math.floor(pos.getBlockX() / 16.0);
|
|
int chunkZ = (int) Math.floor(pos.getBlockZ() / 16.0);
|
|
|
|
final BlockVector2 chunkPos = BlockVector2.at(chunkX, chunkZ);
|
|
player.print(Caption.of("worldedit.chunkinfo.chunk", TextComponent.of(chunkX), TextComponent.of(chunkZ)));
|
|
player.print(Caption.of("worldedit.chunkinfo.old-filename", TextComponent.of(LegacyChunkStore.getFilename(chunkPos))));
|
|
player.print(Caption.of("worldedit.chunkinfo.mcregion-filename", TextComponent.of(McRegionChunkStore.getFilename(chunkPos))));
|
|
}
|
|
|
|
@Command(
|
|
name = "listchunks",
|
|
aliases = { "/listchunks" },
|
|
desc = "List chunks that your selection includes"
|
|
)
|
|
@CommandPermissions("worldedit.listchunks")
|
|
public void listChunks(Actor actor, World world, LocalSession session,
|
|
@ArgFlag(name = 'p', desc = "Page number.", def = "1") int page) throws WorldEditException {
|
|
final Region region = session.getSelection(world);
|
|
|
|
actor.print(Caption.of("worldedit.listchunks.listfor", TextComponent.of(actor.getName())));
|
|
actor.print(new ChunkListPaginationBox(region).create(page));
|
|
}
|
|
|
|
@Command(
|
|
name = "delchunks",
|
|
aliases = { "/delchunks" },
|
|
desc = "Delete chunks that your selection includes"
|
|
)
|
|
@CommandPermissions("worldedit.delchunks")
|
|
@Logging(REGION)
|
|
public void deleteChunks(Actor actor, World world, LocalSession session,
|
|
@ArgFlag(name = 'o', desc = "Only delete chunks older than the specified time.")
|
|
ZonedDateTime beforeTime) throws WorldEditException {
|
|
Path worldDir = world.getStoragePath();
|
|
if (worldDir == null) {
|
|
throw new StopExecutionException(TextComponent.of("Couldn't find world folder for this world."));
|
|
}
|
|
|
|
File chunkFile = worldEdit.getWorkingDirectoryFile(DELCHUNKS_FILE_NAME);
|
|
Path chunkPath = chunkFile.toPath();
|
|
ChunkDeletionInfo currentInfo = null;
|
|
if (Files.exists(chunkPath)) {
|
|
try {
|
|
currentInfo = ChunkDeleter.readInfo(chunkFile.toPath());
|
|
} catch (IOException e) {
|
|
throw new StopExecutionException(TextComponent.of("Error reading existing chunk file."));
|
|
}
|
|
}
|
|
if (currentInfo == null) {
|
|
currentInfo = new ChunkDeletionInfo();
|
|
currentInfo.batches = new ArrayList<>();
|
|
}
|
|
|
|
ChunkDeletionInfo.ChunkBatch newBatch = new ChunkDeletionInfo.ChunkBatch();
|
|
newBatch.worldPath = worldDir.toAbsolutePath().normalize().toString();
|
|
newBatch.backup = true;
|
|
final Region selection = session.getSelection(world);
|
|
if (selection instanceof CuboidRegion) {
|
|
newBatch.minChunk = selection.getMinimumPoint().shr(4).toBlockVector2();
|
|
newBatch.maxChunk = selection.getMaximumPoint().shr(4).toBlockVector2();
|
|
} else {
|
|
// this has a possibility to OOM for very large selections still
|
|
Set<BlockVector2> chunks = selection.getChunks();
|
|
newBatch.chunks = new ArrayList<>(chunks);
|
|
}
|
|
if (beforeTime != null) {
|
|
newBatch.deletionPredicates = new ArrayList<>();
|
|
ChunkDeletionInfo.DeletionPredicate timePred = new ChunkDeletionInfo.DeletionPredicate();
|
|
timePred.property = "modification";
|
|
timePred.comparison = "<";
|
|
timePred.value = String.valueOf((int) beforeTime.toOffsetDateTime().toEpochSecond());
|
|
newBatch.deletionPredicates.add(timePred);
|
|
}
|
|
currentInfo.batches.add(newBatch);
|
|
|
|
try {
|
|
ChunkDeleter.writeInfo(currentInfo, chunkPath);
|
|
} catch (IOException | JsonIOException e) {
|
|
throw new StopExecutionException(TextComponent.of("Failed to write chunk list: " + e.getMessage()));
|
|
}
|
|
|
|
actor.print(TextComponent.of(String.format("%d chunk(s) have been marked for deletion the next time the server starts.",
|
|
newBatch.getChunkCount())));
|
|
if (currentInfo.batches.size() > 1) {
|
|
actor.printDebug(TextComponent.of(String.format("%d chunks total marked for deletion. (May have overlaps).",
|
|
currentInfo.batches.stream().mapToInt(ChunkDeletionInfo.ChunkBatch::getChunkCount).sum())));
|
|
}
|
|
actor.print(TextComponent.of("You can mark more chunks for deletion, or to stop now, run: ", TextColor.LIGHT_PURPLE)
|
|
.append(TextComponent.of("/stop", TextColor.AQUA)
|
|
.clickEvent(ClickEvent.of(ClickEvent.Action.SUGGEST_COMMAND, "/stop"))));
|
|
}
|
|
|
|
private static class ChunkListPaginationBox extends PaginationBox {
|
|
//private final Region region;
|
|
private final List<BlockVector2> chunks = null;
|
|
|
|
ChunkListPaginationBox(Region region) {
|
|
super("Selected Chunks", "/listchunks -p %page%");
|
|
}
|
|
|
|
@Override
|
|
public Component getComponent(int number) {
|
|
return TextComponent.of(chunks.get(number).toString());
|
|
}
|
|
|
|
@Override
|
|
public int getComponentsSize() {
|
|
return chunks.size();
|
|
}
|
|
}
|
|
}
|