PlotSquared/src/main/java/com/intellectualcrafters/plot/util/BO3Handler.java

package com.intellectualcrafters.plot.util;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.StandardOpenOption;
import java.util.HashMap;
import java.util.List;
import java.util.Map.Entry;

import com.intellectualcrafters.plot.PS;
import com.intellectualcrafters.plot.config.Settings;
import com.intellectualcrafters.plot.flag.FlagManager;
import com.intellectualcrafters.plot.generator.ClassicPlotWorld;
import com.intellectualcrafters.plot.object.BO3;
import com.intellectualcrafters.plot.object.ChunkLoc;
import com.intellectualcrafters.plot.object.Location;
import com.intellectualcrafters.plot.object.Plot;
import com.intellectualcrafters.plot.object.PlotBlock;
import com.intellectualcrafters.plot.object.PlotPlayer;
import com.intellectualcrafters.plot.object.PlotWorld;

public class BO3Handler
{

    /**
     * @see #saveBO3(null, Plot)
     * @param plot
     * @return if successfully exported
     */
    public static boolean saveBO3(final Plot plot)
    {
        return saveBO3(null, plot);
    }

    public static boolean contains(final PlotBlock[] blocks, final PlotBlock block)
    {
        for (final PlotBlock item : blocks)
        {
            if (item.equals(block)) { return true; }
        }
        return false;
    }

    /**
     * Save a plot as a BO3 file<br>
     *  - Use null for the player object if no player is applicable
     * @param plr
     * @param plot
     * @return
     */
    public static boolean saveBO3(final PlotPlayer plr, final Plot plot)
    {
        final PlotWorld plotworld = PS.get().getPlotWorld(plot.world);
        if (!(plotworld instanceof ClassicPlotWorld) || (plotworld.TYPE != 0))
        {
            MainUtil.sendMessage(plr, "BO3 exporting only supports type 0 classic generation.");
            return false;
        }
        final String alias = plot.toString();
        final Location pos1 = plot.getBottom();
        final Location pos2 = plot.getTop();
        final ClassicPlotWorld cpw = (ClassicPlotWorld) plotworld;
        final int height = cpw.PLOT_HEIGHT;

        final int cx = (pos1.getX() + pos2.getX()) / 2;
        final int cz = (pos1.getZ() + pos2.getZ()) / 2;

        final HashMap<ChunkLoc, BO3> map = new HashMap<>();

        boolean content = false;
        for (int x = pos1.getX(); x <= pos2.getX(); x++)
        {
            final int X = ((x + 7) - cx) >> 4;
        final int xx = (x - cx) % 16;
        for (int z = pos1.getZ(); z <= pos2.getZ(); z++)
        {
            final int Z = ((z + 7) - cz) >> 4;
        final int zz = (z - cz) % 16;
        final ChunkLoc loc = new ChunkLoc(X, Z);
        BO3 bo3 = map.get(loc);
        for (int y = 1; y < height; y++)
        {
            final PlotBlock block = BlockManager.manager.getBlock(new Location(plot.world, x, y, z));
            if ((block != null) && !contains(cpw.MAIN_BLOCK, block))
            {
                if (bo3 == null)
                {
                    bo3 = new BO3(alias, loc);
                    map.put(loc, bo3);
                    content = true;
                }
                bo3.addBlock(xx, y - height - 1, zz, block);
            }
        }
        final PlotBlock floor = BlockManager.manager.getBlock(new Location(plot.world, x, height, z));
        if ((floor != null) && !contains(cpw.TOP_BLOCK, floor))
        {
            if (bo3 == null)
            {
                bo3 = new BO3(alias, loc);
                map.put(loc, bo3);
                content = true;
            }
            bo3.addBlock(xx, -1, zz, floor);
        }
        for (int y = height + 1; y < 256; y++)
        {
            final PlotBlock block = BlockManager.manager.getBlock(new Location(plot.world, x, y, z));
            if ((block != null) && (block.id != 0))
            {
                if (bo3 == null)
                {
                    bo3 = new BO3(alias, loc);
                    map.put(loc, bo3);
                    content = true;
                }
                bo3.addBlock(xx, y - height - 1, zz, block);
            }
        }
        }
        }

        if (!content)
        {
            MainUtil.sendMessage(plr, "No content found!");
            return false;
        }

        for (final Entry<ChunkLoc, BO3> entry : map.entrySet())
        {
            final ChunkLoc chunk = entry.getKey();
            final BO3 bo3 = entry.getValue();
            if ((chunk.x == 0) && (chunk.z == 0))
            {
                continue;
            }
            int x = chunk.x;
            int z = chunk.z;
            if (Math.abs(chunk.x) > Math.abs(chunk.z))
            {
                x += chunk.x > 0 ? -1 : 1;
            }
            else
            {
                z += chunk.z > 0 ? -1 : 1;
            }
            ChunkLoc parentLoc = new ChunkLoc(x, z);
            if (!map.containsKey(parentLoc))
            {
                parentLoc = null;
                for (final Entry<ChunkLoc, BO3> entry2 : map.entrySet())
                {
                    final ChunkLoc other = entry2.getKey();
                    if (((other.x == (chunk.x - 1)) && (other.z == chunk.z)) || ((other.z == (chunk.z - 1)) && (other.x == chunk.x)))
                    {
                        parentLoc = other;
                    }
                }
                if (parentLoc == null)
                {
                    MainUtil.sendMessage(plr, "Exporting BO3 cancelled due to detached chunk: " + chunk + " - Make sure you only have one object per plot");
                    return false;
                }
            }
            map.get(parentLoc).addChild(bo3);
        }

        for (final Entry<ChunkLoc, BO3> entry : map.entrySet())
        {
            save(plot, entry.getValue());
        }

        MainUtil.sendMessage(plr, "BO3 exporting was successful!");
        return true;
    }

    public static boolean save(final Plot plot, final BO3 bo3)
    {
        final File base = getBaseFile(plot.world);
        try
        {
            final List<String> lines = Files.readAllLines(base.toPath(), StandardCharsets.UTF_8);
            for (int i = 0; i < lines.size(); i++)
            {
                final String line = lines.get(i).trim();
                final String result = StringMan.replaceAll(line, "%owner%", MainUtil.getName(plot.owner), "%alias%", plot.toString(), "%blocks%", bo3.getBlocks(), "%branches%", bo3.getChildren(),
                "%flags%", StringMan.join(FlagManager.getPlotFlags(plot).values(), ","));
                if (!StringMan.isEqual(result, line))
                {
                    lines.set(i, result);
                }
            }
            File bo3File;
            if ((bo3.getLoc().x == 0) && (bo3.getLoc().z == 0))
            {
                bo3File = new File(base.getParentFile(), bo3.getName() + ".bo3");
            }
            else
            {
                bo3File = new File(base.getParentFile(), bo3.getName() + "_" + bo3.getLoc().x + "_" + bo3.getLoc().z + ".bo3");
            }
            bo3File.createNewFile();
            Files.write(bo3File.toPath(), StringMan.join(lines, System.getProperty("line.separator")).getBytes(), StandardOpenOption.WRITE);
            return true;
        }
        catch (final Exception e)
        {
            e.printStackTrace();
            return false;
        }
    }

    public static File getBaseFile(final String category)
    {
        final File base = new File(PS.get().IMP.getDirectory(), Settings.BO3_SAVE_PATH + File.separator + category + File.separator + "base.yml");
        if (!base.exists())
        {
            PS.get().copyFile("base.yml", Settings.BO3_SAVE_PATH + File.separator + category);
        }
        return base;
    }
}