Plex-FAWE/worldedit-core/src/main/java/com/sk89q/worldedit/command/HistorySubCommands.java

package com.sk89q.worldedit.command;

import static com.sk89q.worldedit.internal.command.CommandUtil.checkCommandArgument;

import com.boydti.fawe.Fawe;
import com.boydti.fawe.FaweAPI;
import com.boydti.fawe.config.Caption;
import com.boydti.fawe.config.Settings;
import com.boydti.fawe.database.DBHandler;
import com.boydti.fawe.database.RollbackDatabase;
import com.boydti.fawe.logging.rollback.RollbackOptimizedHistory;
import com.boydti.fawe.object.RegionWrapper;
import com.boydti.fawe.object.changeset.DiskStorageHistory;
import com.boydti.fawe.util.MainUtil;
import com.google.common.base.Function;
import com.google.common.collect.Lists;
import com.sk89q.worldedit.LocalSession;
import com.sk89q.worldedit.WorldEditException;
import com.sk89q.worldedit.command.argument.Arguments;
import com.sk89q.worldedit.command.util.CommandPermissions;
import com.sk89q.worldedit.command.util.CommandPermissionsConditionGenerator;
import com.sk89q.worldedit.command.util.annotation.Confirm;
import com.sk89q.worldedit.entity.Player;
import com.sk89q.worldedit.extension.platform.Actor;
import com.sk89q.worldedit.internal.annotation.AllowedRegion;
import com.sk89q.worldedit.internal.annotation.Time;
import com.sk89q.worldedit.math.BlockVector2;
import com.sk89q.worldedit.math.BlockVector3;
import com.sk89q.worldedit.regions.Region;
import com.sk89q.worldedit.util.Countable;
import com.sk89q.worldedit.util.Direction;
import com.sk89q.worldedit.util.Identifiable;
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.event.HoverEvent;
import com.sk89q.worldedit.world.World;
import com.sk89q.worldedit.world.block.BlockState;
import java.io.File;
import java.lang.ref.Reference;
import java.lang.ref.SoftReference;
import java.util.List;
import java.util.Objects;
import java.util.UUID;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.atomic.LongAdder;
import java.util.function.Supplier;
import javax.annotation.Nullable;
import org.enginehub.piston.annotation.Command;
import org.enginehub.piston.annotation.CommandContainer;
import org.enginehub.piston.annotation.param.Arg;
import org.enginehub.piston.annotation.param.ArgFlag;
import org.enginehub.piston.annotation.param.Switch;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Range;

@CommandContainer(superTypes = CommandPermissionsConditionGenerator.Registration.class)
public class HistorySubCommands {

    private final HistoryCommands parent;

    public HistorySubCommands(HistoryCommands parent) {
        this.parent = parent;
    }

    @Command(
            name = "restore",
            aliases = {"rerun"},
            desc = "Rerun edits" +
                    " - The time uses s, m, h, d, y.\n" +
                    " - Import from disk: /history import"
    )
    @CommandPermissions("worldedit.history.redo")
    @Confirm
    public synchronized void rerun(Player player, World world, RollbackDatabase database,
                                   @AllowedRegion Region[] allowedRegions,
                                   @ArgFlag(name = 'u', desc = "String user", def="me") UUID other,
                                   @ArgFlag(name = 'r', def = "0", desc = "radius")
                                   @Range(from = 0, to = Integer.MAX_VALUE) int radius,
                                   @ArgFlag(name = 't', desc = "Time e.g. 20s", def = "0")
                                   @Time long timeDiff) throws WorldEditException {
        rollback(player, world, database, allowedRegions, other, radius, timeDiff, true);
    }

    @Command(
            name = "rollback",
            desc = "Undo a specific edit. " +
                    " - The time uses s, m, h, d, y."
    )
    @CommandPermissions("worldedit.history.undo")
    @Confirm
    public synchronized void rollback(Player player, World world, RollbackDatabase database,
                                      @AllowedRegion Region[] allowedRegions,
                                      @ArgFlag(name = 'u', desc = "String user", def = "") UUID other,
                                      @ArgFlag(name = 'r', def = "0", desc = "radius")
                                      @Range(from = 0, to = Integer.MAX_VALUE) int radius,
                                      @ArgFlag(name = 't', desc = "Time e.g. 20s", def = "0") @Time long timeDiff,
                                      @Switch(name = 'f', desc = "Restore instead of rollback") boolean restore) throws WorldEditException {
        if (!Settings.IMP.HISTORY.USE_DATABASE) {
            player.print(Caption.of("fawe.error.setting.disable" , "history.use-database (Import with /history import )"));
            return;
        }
        checkCommandArgument(radius > 0, "Radius must be >= 0");
        checkCommandArgument(timeDiff > 0, "Time must be >= 0");

        if (other == null) other = player.getUniqueId();
        if (!other.equals(player.getUniqueId())) {
            player.checkPermission("worldedit.history.undo.other");
        }
        if (other == Identifiable.EVERYONE) other = null;
        Location origin = player.getLocation();
        BlockVector3 bot = origin.toBlockPoint().subtract(radius, radius, radius);
        BlockVector3 top = origin.toBlockPoint().add(radius, radius, radius);
        bot = bot.clampY(0, world.getMaxY());
        top = top.clampY(0, world.getMaxY());
        // TODO mask the regions bot / top to the bottom and top coord in the allowedRegions
        // TODO: then mask the edit to the bot / top
//        if (allowedRegions.length != 1 || !allowedRegions[0].isGlobal()) {
//            finalQueue = new MaskedIQueueExtent(SetQueue.IMP.getNewQueue(fp.getWorld(), true, false), allowedRegions);
//        } else {
//            finalQueue = SetQueue.IMP.getNewQueue(fp.getWorld(), true, false);
//        }
        int count = 0;
        UUID finalOther = other;
        long minTime = System.currentTimeMillis() - timeDiff;
        for (Supplier<RollbackOptimizedHistory> supplier : database.getEdits(other, minTime, bot, top, !restore, restore)) {
            count++;
            RollbackOptimizedHistory edit = supplier.get();
            edit.undo(player, allowedRegions);
            String path = edit.getWorld().getName() + "/" + finalOther + "-" + edit.getIndex();
            player.print(Caption.of("fawe.worldedit.rollback.rollback.element", path));
        }
        player.print(Caption.of("fawe.worldedit.tool.tool.inspect.info.footer" , count));
    }

    @Command(
            name = "import",
            desc = "Import history into the database" +
                    " - The time uses s, m, h, d, y.\n" +
                    " - Import from disk: /history import"
    )
    @CommandPermissions("fawe.rollback.import")
    @Confirm
    public synchronized void importdb(Actor actor) throws WorldEditException {
        File folder = MainUtil.getFile(Fawe.imp().getDirectory(), Settings.IMP.PATHS.HISTORY);
        if (folder.exists()) {
            for (File worldFolder : Objects.requireNonNull(folder.listFiles())) {
                if (worldFolder != null && worldFolder.isDirectory()) {
                    String worldName = worldFolder.getName();
                    World world = FaweAPI.getWorld(worldName);
                    if (world != null) {
                        for (File userFolder : worldFolder.listFiles()) {
                            if (!userFolder.isDirectory()) {
                                continue;
                            }
                            String userUUID = userFolder.getName();
                            try {
                                UUID uuid = UUID.fromString(userUUID);
                                for (File historyFile : userFolder.listFiles()) {
                                    String name = historyFile.getName();
                                    if (!name.endsWith(".bd")) {
                                        continue;
                                    }
                                    RollbackOptimizedHistory rollback = new RollbackOptimizedHistory(
                                            world, uuid,
                                            Integer.parseInt(
                                                    name.substring(0, name.length() - 3)));
                                    DiskStorageHistory.DiskStorageSummary summary = rollback
                                            .summarize(RegionWrapper.GLOBAL(), false);
                                    if (summary != null) {
                                        rollback.setDimensions(
                                                BlockVector3.at(summary.minX, 0, summary.minZ),
                                                BlockVector3
                                                        .at(summary.maxX, 255, summary.maxZ));
                                        rollback.setTime(historyFile.lastModified());
                                        RollbackDatabase db = DBHandler.IMP
                                                .getDatabase(world);
                                        db.logEdit(rollback);
                                        actor.print("Logging: " + historyFile);
                                    }
                                }
                            } catch (IllegalArgumentException e) {
                                e.printStackTrace();
                            }
                        }
                    }
                }
            }
            actor.print("Done import!");
        }
    }

    @Command(
            name = "summary",
            aliases = {"info", "summarize"},
            desc = "Summarize an edit"
    )
    @CommandPermissions("worldedit.history.find")
    public synchronized void summary(Player player, RollbackDatabase database, Arguments arguments,
                                     @Arg(desc = "Player uuid/name") UUID other,
                                     @Arg(desc = "edit index") Integer index,
                                     @ArgFlag(name = 'p', desc = "Page to view.", def = "-1") int page) throws WorldEditException, ExecutionException, InterruptedException {
        RollbackOptimizedHistory edit = database.getEdit(other, index).get();
        if (edit == null) {
            player.print(TranslatableComponent.of("fawe.worldedit.schematic.schematic.none"));
            return;
        }
        Location origin = player.getLocation();

        String name = Fawe.imp().getName(edit.getUUID());
        String cmd = edit.getCommand();
        BlockVector3 pos1 = edit.getMinimumPoint();
        BlockVector3 pos2 = edit.getMaximumPoint();

        double distanceX = Math.min( Math.abs(pos1.getX() - origin.getX()), Math.abs(pos2.getX() - origin.getX()));
        double distanceZ = Math.min( Math.abs(pos1.getZ() - origin.getZ()), Math.abs(pos2.getZ() - origin.getZ()));
        int distance = (int) Math.sqrt(distanceX * distanceX + distanceZ * distanceZ);

        BlockVector2 dirVec = BlockVector2.at(edit.getOriginX() - origin.getX(), edit.getOriginZ() - origin.getZ());
        Direction direction = Direction.findClosest(dirVec.toVector3(), Direction.Flag.ALL);

        long seconds = (System.currentTimeMillis() - edit.getBDFile().lastModified()) / 1000;
        String timeStr = MainUtil.secToTime(seconds);

        int size = edit.size();

        String pageCommand = "/" + arguments.get().replaceAll("-p [0-9]+", "").trim();
        List<Countable<BlockState>> list = null;
        Reference<List<Countable<BlockState>>> cached = player.getMeta(pageCommand);
        if (cached != null) {
            list = cached.get();
        }
        if (list == null) {
            DiskStorageHistory.DiskStorageSummary summary = edit.summarize(null, false);
            if (summary != null) {
                list = summary.getBlockDistributionWithData();
                player.setMeta(pageCommand, new SoftReference<>(list));
            }
        }
        boolean biomes = edit.getBioFile().exists();
        boolean createdEnts = edit.getEnttFile().exists();
        boolean removedEnts = edit.getEntfFile().exists();
        boolean createdTiles = edit.getNbttFile().exists();
        boolean removedTiles = edit.getNbtfFile().exists();
        System.out.println("TODO FIXME move to translations");
        if (page == -1) {
            player.print("name: " + name
                    + ", cmd: " + edit.getCommand()
                    + ", dist: " + distance + "m " + direction.name()
                    + ", time: " + timeStr + " ago"
                    + ", size: " + size + " blocks"
                    + ", biomes: " + biomes
                    + ", +entity: " + createdEnts
                    + ", -entity: " + removedEnts
                    + ", +tile: " + createdTiles
                    + ", -tile: " + removedTiles
                    + ", disk: " + (edit.getSizeOnDisk() / 1000) + "mb"
                    + ", min " + edit.getMinimumPoint()
                    + ", max " + edit.getMaximumPoint()
            );
        }
        page = 1;
        if (list != null) {
            SelectionCommands.BlockDistributionResult pages = new SelectionCommands.BlockDistributionResult((List) list, true, pageCommand);
            player.print(pages.create(page));
        }
    }

    @Command(
            name = "find",
            aliases = {"inspect", "search", "near"},
            desc = "Find nearby edits"
    )
    @CommandPermissions("worldedit.history.find")
    public synchronized void find(Player player, World world, RollbackDatabase database, Arguments arguments,
                                  @ArgFlag(name = 'u', desc = "String user") UUID other,
                                  @ArgFlag(name = 'r', def = "0", desc = "radius") int radius,
                                  @ArgFlag(name = 't', desc = "Time e.g. 20s", def = "0")
                                  @Time long timeDiff,
                                  @ArgFlag(name = 'p', desc = "Page to view.", def = "1") int page) throws WorldEditException {
        if (!Settings.IMP.HISTORY.USE_DATABASE) {
            player.print(Caption.of("fawe.error.setting.disable" , "history.use-database (Import with //history import )"));
            return;
        }
        checkCommandArgument(radius > 0, "Radius must be >= 0");
        checkCommandArgument(timeDiff > 0, "Time must be >= 0");

        Location origin = player.getLocation();
        String pageCommand = "/" + arguments.get().replaceAll("-p [0-9]+", "").trim();

        List<Supplier<RollbackOptimizedHistory>> list = null;
        Reference<List<Supplier<RollbackOptimizedHistory>>> cached = player.getMeta(pageCommand);
        if (cached != null) {
            list = cached.get();
        }
        if (list == null) {
            if (other == null) other = player.getUniqueId();
            if (!other.equals(player.getUniqueId())) {
                player.checkPermission("worldedit.history.undo.other");
            }
            if (other == Identifiable.EVERYONE) other = null;

            BlockVector3 bot = origin.toBlockPoint().subtract(radius, radius, radius);
            BlockVector3 top = origin.toBlockPoint().add(radius, radius, radius);
            bot = bot.clampY(0, world.getMaxY());
            top = top.clampY(0, world.getMaxY());

            LongAdder total = new LongAdder();

            long minTime = System.currentTimeMillis() - timeDiff;
            Iterable<Supplier<RollbackOptimizedHistory>> edits = database.getEdits(other, minTime, bot, top, false, false);
            list = Lists.newArrayList(edits);
            player.setMeta(pageCommand, new SoftReference<>(list));
        }

        PaginationBox pages = PaginationBox.fromStrings("Edits:", pageCommand, list, new Function<Supplier<RollbackOptimizedHistory>, Component>() {
            @NotNull
            @Override
            public Component apply(@Nullable Supplier<RollbackOptimizedHistory> input) {
                RollbackOptimizedHistory edit = input.get();
                UUID uuid = edit.getUUID();
                int index = edit.getIndex();
                if (!edit.isEmpty()) {
                    database.delete(uuid, index);
                    return TextComponent.empty();
                }
                String name = Fawe.imp().getName(edit.getUUID());

                String cmd = edit.getCommand();
                BlockVector3 pos1 = edit.getMinimumPoint();
                BlockVector3 pos2 = edit.getMaximumPoint();

                double distanceX = Math.min(Math.abs(pos1.getX() - origin.getX()), Math.abs(pos2.getX() - origin.getX()));
                double distanceZ = Math.min(Math.abs(pos1.getZ() - origin.getZ()), Math.abs(pos2.getZ() - origin.getZ()));
                int distance = (int) Math.sqrt(distanceX * distanceX + distanceZ * distanceZ);

                BlockVector2 dirVec = BlockVector2.at(edit.getOriginX() - origin.getX(), edit.getOriginZ() - origin.getZ());
                Direction direction = Direction.findClosest(dirVec.toVector3(), Direction.Flag.ALL);

                long seconds = (System.currentTimeMillis() - edit.getBDFile().lastModified()) / 1000;
                String timeStr = MainUtil.secToTime(seconds);

                int size = edit.size();

                TranslatableComponent elem = Caption.of("fawe.worldedit.history.find.element", name, timeStr, distance, direction.name(), cmd);

                String infoCmd = "//history summary " + uuid + " " + index;
                TranslatableComponent hover = Caption.of("fawe.worldedit.history.find.hover", size);
                elem = elem.hoverEvent(HoverEvent.of(HoverEvent.Action.SHOW_TEXT, hover));
                elem = elem.clickEvent(ClickEvent.of(ClickEvent.Action.RUN_COMMAND, infoCmd));

                return elem;
            }
        });
        player.print(pages.create(page));
    }

    @Command(
            name = "clear",
            desc = "Clear your history"
    )
    @CommandPermissions("worldedit.history.clear")
    public void clearHistory(Actor actor, LocalSession session) {
        parent.clearHistory(actor, session);
    }
}