194 lines
9.7 KiB
Java
194 lines
9.7 KiB
Java
package me.wiefferink.areashop.handlers;
|
|
|
|
import com.sk89q.worldedit.EditSession;
|
|
import com.sk89q.worldedit.IncompleteRegionException;
|
|
import com.sk89q.worldedit.LocalSession;
|
|
import com.sk89q.worldedit.MaxChangedBlocksException;
|
|
import com.sk89q.worldedit.Vector;
|
|
import com.sk89q.worldedit.bukkit.BukkitAdapter;
|
|
import com.sk89q.worldedit.extent.clipboard.BlockArrayClipboard;
|
|
import com.sk89q.worldedit.extent.clipboard.Clipboard;
|
|
import com.sk89q.worldedit.extent.clipboard.io.ClipboardFormat;
|
|
import com.sk89q.worldedit.extent.clipboard.io.ClipboardReader;
|
|
import com.sk89q.worldedit.extent.clipboard.io.ClipboardWriter;
|
|
import com.sk89q.worldedit.extent.transform.BlockTransformExtent;
|
|
import com.sk89q.worldedit.function.mask.Mask;
|
|
import com.sk89q.worldedit.function.mask.Mask2D;
|
|
import com.sk89q.worldedit.function.operation.ForwardExtentCopy;
|
|
import com.sk89q.worldedit.function.operation.Operations;
|
|
import com.sk89q.worldedit.regions.CuboidRegion;
|
|
import com.sk89q.worldedit.regions.Region;
|
|
import com.sk89q.worldedit.session.ClipboardHolder;
|
|
import com.sk89q.worldedit.util.io.Closer;
|
|
import com.sk89q.worldguard.protection.regions.ProtectedRegion;
|
|
import com.sk89q.worldguard.protection.regions.RegionType;
|
|
import me.wiefferink.areashop.interfaces.AreaShopInterface;
|
|
import me.wiefferink.areashop.interfaces.GeneralRegionInterface;
|
|
import me.wiefferink.areashop.interfaces.WorldEditInterface;
|
|
import me.wiefferink.areashop.interfaces.WorldEditSelection;
|
|
import org.apache.commons.lang.exception.ExceptionUtils;
|
|
import org.bukkit.entity.Player;
|
|
|
|
import java.io.BufferedInputStream;
|
|
import java.io.BufferedOutputStream;
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
|
|
// Future: this class could use the schematic save/paste api of FAWE: https://github.com/boy0001/FastAsyncWorldedit/wiki/Pasting-a-schematic
|
|
// This would not allow for shaping the schematic to a polygon region though
|
|
public class FastAsyncWorldEditHandler extends WorldEditInterface {
|
|
|
|
public FastAsyncWorldEditHandler(AreaShopInterface pluginInterface) {
|
|
super(pluginInterface);
|
|
}
|
|
|
|
@Override
|
|
public WorldEditSelection getPlayerSelection(Player player) {
|
|
try {
|
|
Region region = pluginInterface.getWorldEdit().getSession(player).getSelection(BukkitAdapter.adapt(player.getWorld()));
|
|
return new WorldEditSelection(
|
|
player.getWorld(),
|
|
BukkitAdapter.adapt(player.getWorld(), region.getMinimumPoint()),
|
|
BukkitAdapter.adapt(player.getWorld(), region.getMaximumPoint())
|
|
);
|
|
} catch (IncompleteRegionException e) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean restoreRegionBlocks(File rawFile, GeneralRegionInterface regionInterface) {
|
|
// TODO implement using the FastAsyncWorldEdit api to paste async
|
|
File file = null;
|
|
ClipboardFormat format = null;
|
|
for (ClipboardFormat formatOption : ClipboardFormat.values()) {
|
|
if (new File(rawFile.getAbsolutePath() + "." + formatOption.getExtension()).exists()) {
|
|
file = new File(rawFile.getAbsolutePath() + "." + formatOption.getExtension());
|
|
format = formatOption;
|
|
}
|
|
}
|
|
if(file == null) {
|
|
pluginInterface.getLogger().info("Did not restore region " + regionInterface.getName() + ", schematic file does not exist: " + rawFile.getAbsolutePath());
|
|
return false;
|
|
}
|
|
pluginInterface.debugI("Trying to restore region", regionInterface.getName(), " from file", file.getAbsolutePath(), "with format", format.name());
|
|
|
|
com.sk89q.worldedit.world.World world = null;
|
|
if(regionInterface.getName() != null) {
|
|
world = BukkitAdapter.adapt(regionInterface.getWorld());
|
|
}
|
|
if(world == null) {
|
|
pluginInterface.getLogger().info("Did not restore region " + regionInterface.getName() + ", world not found: " + regionInterface.getWorldName());
|
|
return false;
|
|
}
|
|
EditSession editSession = pluginInterface.getWorldEdit().getWorldEdit().getEditSessionFactory().getEditSession(world, pluginInterface.getConfig().getInt("maximumBlocks"));
|
|
editSession.enableQueue();
|
|
ProtectedRegion region = regionInterface.getRegion();
|
|
// Get the origin and size of the region
|
|
Vector origin = new Vector(region.getMinimumPoint().getBlockX(), region.getMinimumPoint().getBlockY(), region.getMinimumPoint().getBlockZ());
|
|
|
|
// Read the schematic and paste it into the world
|
|
try(Closer closer = Closer.create()) {
|
|
FileInputStream fis = closer.register(new FileInputStream(file));
|
|
BufferedInputStream bis = closer.register(new BufferedInputStream(fis));
|
|
ClipboardReader reader = format.getReader(bis);
|
|
|
|
LocalSession session = new LocalSession(pluginInterface.getWorldEdit().getLocalConfiguration());
|
|
Clipboard clipboard = reader.read();
|
|
if(clipboard.getDimensions().getY() != regionInterface.getHeight()
|
|
|| clipboard.getDimensions().getX() != regionInterface.getWidth()
|
|
|| clipboard.getDimensions().getZ() != regionInterface.getDepth()) {
|
|
pluginInterface.getLogger().warning("Size of the region " + regionInterface.getName() + " is not the same as the schematic to restore!");
|
|
pluginInterface.debugI("schematic|region, x:" + clipboard.getDimensions().getX() + "|" + regionInterface.getWidth() + ", y:" + clipboard.getDimensions().getY() + "|" + regionInterface.getHeight() + ", z:" + clipboard.getDimensions().getZ() + "|" + regionInterface.getDepth());
|
|
}
|
|
clipboard.setOrigin(clipboard.getMinimumPoint());
|
|
ClipboardHolder clipboardHolder = new ClipboardHolder(clipboard);
|
|
session.setBlockChangeLimit(pluginInterface.getConfig().getInt("maximumBlocks"));
|
|
session.setClipboard(clipboardHolder);
|
|
|
|
// Build operation
|
|
BlockTransformExtent extent = new BlockTransformExtent(clipboardHolder.getClipboard());
|
|
ForwardExtentCopy copy = new ForwardExtentCopy(extent, clipboard.getRegion(), clipboard.getOrigin(), editSession, origin);
|
|
copy.setTransform(clipboardHolder.getTransform());
|
|
// Mask to region (for polygon and other weird shaped regions)
|
|
// TODO make this more efficient (especially for polygon regions)
|
|
if(region.getType() != RegionType.CUBOID) {
|
|
copy.setSourceMask(new Mask() {
|
|
@Override
|
|
public boolean test(Vector vector) {
|
|
return region.contains(vector);
|
|
}
|
|
|
|
@Override
|
|
public Mask2D toMask2D() {
|
|
return null;
|
|
}
|
|
});
|
|
}
|
|
Operations.completeLegacy(copy);
|
|
} catch(MaxChangedBlocksException e) {
|
|
pluginInterface.getLogger().warning("exceeded the block limit while restoring schematic of " + regionInterface.getName() + ", limit in exception: " + e.getBlockLimit() + ", limit passed by AreaShop: " + pluginInterface.getConfig().getInt("maximumBlocks"));
|
|
return false;
|
|
} catch(IOException e) {
|
|
pluginInterface.getLogger().warning("An error occured while restoring schematic of " + regionInterface.getName() + ", enable debug to see the complete stacktrace");
|
|
pluginInterface.debugI(ExceptionUtils.getStackTrace(e));
|
|
return false;
|
|
} catch (Exception e) {
|
|
pluginInterface.getLogger().warning("crashed during restore of " + regionInterface.getName());
|
|
pluginInterface.debugI(ExceptionUtils.getStackTrace(e));
|
|
return false;
|
|
}
|
|
editSession.flushQueue();
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public boolean saveRegionBlocks(File file, GeneralRegionInterface regionInterface) {
|
|
// TODO implement using the FastAsyncWorldEdit api to save async
|
|
ClipboardFormat format = ClipboardFormat.STRUCTURE;
|
|
// TODO allow selecting FAWE format in the config? (when enabled you cannot go back to vanilla WorldEdit easily)
|
|
|
|
file = new File(file.getAbsolutePath() + "." + format.getExtension());
|
|
pluginInterface.debugI("Trying to save region", regionInterface.getName(), " to file", file.getAbsolutePath(), "with format", format.name());
|
|
com.sk89q.worldedit.world.World world = null;
|
|
if(regionInterface.getWorld() != null) {
|
|
world = BukkitAdapter.adapt(regionInterface.getWorld());
|
|
}
|
|
if(world == null) {
|
|
pluginInterface.getLogger().warning("Did not save region " + regionInterface.getName() + ", world not found: " + regionInterface.getWorldName());
|
|
return false;
|
|
}
|
|
EditSession editSession = pluginInterface.getWorldEdit().getWorldEdit().getEditSessionFactory().getEditSession(world, pluginInterface.getConfig().getInt("maximumBlocks"));
|
|
|
|
// Create a clipboard
|
|
CuboidRegion selection = new CuboidRegion(world, regionInterface.getRegion().getMinimumPoint(), regionInterface.getRegion().getMaximumPoint());
|
|
BlockArrayClipboard clipboard = new BlockArrayClipboard(selection);
|
|
clipboard.setOrigin(regionInterface.getRegion().getMinimumPoint());
|
|
ForwardExtentCopy copy = new ForwardExtentCopy(editSession, new CuboidRegion(world, regionInterface.getRegion().getMinimumPoint(), regionInterface.getRegion().getMaximumPoint()), clipboard, regionInterface.getRegion().getMinimumPoint());
|
|
try {
|
|
Operations.completeLegacy(copy);
|
|
} catch(MaxChangedBlocksException e) {
|
|
pluginInterface.getLogger().warning("Exceeded the block limit while saving schematic of " + regionInterface.getName() + ", limit in exception: " + e.getBlockLimit() + ", limit passed by AreaShop: " + pluginInterface.getConfig().getInt("maximumBlocks"));
|
|
return false;
|
|
}
|
|
|
|
try(Closer closer = Closer.create()) {
|
|
FileOutputStream fos = closer.register(new FileOutputStream(file));
|
|
BufferedOutputStream bos = closer.register(new BufferedOutputStream(fos));
|
|
ClipboardWriter writer = closer.register(format.getWriter(bos));
|
|
writer.write(clipboard);
|
|
} catch(IOException e) {
|
|
pluginInterface.getLogger().warning("An error occured while saving schematic of " + regionInterface.getName() + ", enable debug to see the complete stacktrace");
|
|
pluginInterface.debugI(ExceptionUtils.getStackTrace(e));
|
|
return false;
|
|
} catch (Exception e) {
|
|
pluginInterface.getLogger().warning("Crashed during save of " + regionInterface.getName());
|
|
pluginInterface.debugI(ExceptionUtils.getStackTrace(e));
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
}
|