Examples with SnapshotRestore com.sk89q.worldedit.world.snapshot.experimental.SnapshotRestore used on opensource projects

Example 1 with SnapshotRestore

use of com.sk89q.worldedit.world.snapshot.experimental.SnapshotRestore in project FastAsyncWorldEdit by IntellectualSites.

the class SnapshotUtilCommands method restore.

@Command(name = "restore", aliases = { "/restore" }, desc = "Restore the selection from a snapshot")
@Logging(REGION)
@CommandPermissions("worldedit.snapshots.restore")
public void restore(Actor actor, World world, LocalSession session, EditSession editSession, @Arg(name = "snapshot", desc = "The snapshot to restore", def = "") String snapshotName, // FAWE start - biome and entity restore
@Switch(name = 'b', desc = "If biomes should be restored. If restoring from pre-1.15 to 1.15+, biomes may not be " + "exactly the same due to 3D biomes.") boolean restoreBiomes, @Switch(name = 'e', desc = "If entities should be restored. Will cause issues with duplicate entities if all " + "original entities were not removed.") boolean restoreEntities) throws // FAWE end
WorldEditException, IOException {
    LocalConfiguration config = we.getConfiguration();
    checkSnapshotsConfigured(config);
    if (config.snapshotRepo != null) {
        // FAWE start - biome and entity restore
        legacy.restore(actor, world, session, editSession, snapshotName, restoreBiomes, restoreEntities);
        // FAWE end
        return;
    }
    Region region = session.getSelection(world);
    Snapshot snapshot;
    if (snapshotName != null) {
        URI uri = resolveSnapshotName(config, snapshotName);
        Optional<Snapshot> snapOpt = config.snapshotDatabase.getSnapshot(uri);
        if (!snapOpt.isPresent()) {
            actor.print(Caption.of("worldedit.restore.not-available"));
            return;
        }
        snapshot = snapOpt.get();
    } else {
        snapshot = session.getSnapshotExperimental();
    }
    // No snapshot set?
    if (snapshot == null) {
        try (Stream<Snapshot> snapshotStream = config.snapshotDatabase.getSnapshotsNewestFirst(world.getName())) {
            snapshot = snapshotStream.findFirst().orElse(null);
        }
        if (snapshot == null) {
            actor.print(Caption.of("worldedit.restore.none-for-specific-world", TextComponent.of(world.getName())));
            return;
        }
    }
    actor.print(Caption.of("worldedit.restore.loaded", TextComponent.of(snapshot.getInfo().getDisplayName())));
    try {
        // Restore snapshot
        // FAWE start - biome and entity restore
        SnapshotRestore restore = new SnapshotRestore(snapshot, editSession, region, restoreBiomes, restoreEntities);
        // FAWE end
        restore.restore();
        if (restore.hadTotalFailure()) {
            String error = restore.getLastErrorMessage();
            if (!restore.getMissingChunks().isEmpty()) {
                actor.print(Caption.of("worldedit.restore.chunk-not-present"));
            } else if (error != null) {
                actor.print(Caption.of("worldedit.restore.block-place-failed"));
                actor.print(Caption.of("worldedit.restore.block-place-error", TextComponent.of(error)));
            } else {
                actor.print(Caption.of("worldedit.restore.chunk-load-failed"));
            }
        } else {
            actor.print(Caption.of("worldedit.restore.restored", TextComponent.of(restore.getMissingChunks().size()), TextComponent.of(restore.getErrorChunks().size())));
        }
    } finally {
        try {
            snapshot.close();
        } catch (IOException ignored) {
        }
    }
}