TF-PlotSquared/PlotSquared/src/main/java/com/intellectualcrafters/plot/commands/DebugClaimTest.java

////////////////////////////////////////////////////////////////////////////////////////////////////
// PlotSquared - A plot manager and world generator for the Bukkit API                             /
// Copyright (c) 2014 IntellectualSites/IntellectualCrafters                                       /
//                                                                                                 /
// 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, write to the Free Software Foundation,                         /
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA                               /
//                                                                                                 /
// You can contact us via: support@intellectualsites.com                                           /
////////////////////////////////////////////////////////////////////////////////////////////////////

package com.intellectualcrafters.plot.commands;

import com.google.common.collect.BiMap;
import com.intellectualcrafters.plot.PlotMain;
import com.intellectualcrafters.plot.config.C;
import com.intellectualcrafters.plot.database.DBFunc;
import com.intellectualcrafters.plot.events.PlayerClaimPlotEvent;
import com.intellectualcrafters.plot.flag.FlagManager;
import com.intellectualcrafters.plot.object.*;
import com.intellectualcrafters.plot.util.PlayerFunctions;
import com.intellectualcrafters.plot.util.PlotHelper;
import com.intellectualcrafters.plot.util.UUIDHandler;
import org.bukkit.Bukkit;
import org.bukkit.Chunk;
import org.bukkit.Location;
import org.bukkit.World;
import org.bukkit.block.Block;
import org.bukkit.block.Sign;
import org.bukkit.entity.Player;

import java.util.ArrayList;
import java.util.UUID;

/**
 * @author Citymonstret
 */
public class DebugClaimTest extends SubCommand {

    public DebugClaimTest() {
        super(Command.DEBUGCLAIMTEST, "If you accidentally delete your database, this command will attempt to restore all plots based on the data from the plot signs. Execution time may vary", "debugclaimtest", CommandCategory.DEBUG, false);
    }

    @SuppressWarnings("unused")
    public static boolean claimPlot(final Player player, final Plot plot, final boolean teleport) {
        return claimPlot(player, plot, teleport, "");
    }

    public static boolean claimPlot(final Player player, final Plot plot, final boolean teleport, @SuppressWarnings("unused") final String schematic) {
        final PlayerClaimPlotEvent event = new PlayerClaimPlotEvent(player, plot);
        Bukkit.getPluginManager().callEvent(event);
        if (!event.isCancelled()) {
            PlotHelper.createPlot(player, plot);
            PlotHelper.setSign(player, plot);
            PlayerFunctions.sendMessage(player, C.CLAIMED);
            if (teleport) {
                PlotMain.teleportPlayer(player, player.getLocation(), plot);
            }
            plot.settings.setFlags(FlagManager.parseFlags(PlotMain.getWorldSettings(player.getWorld()).DEFAULT_FLAGS));
        }
        return event.isCancelled();
    }

    @Override
    public boolean execute(final Player plr, final String... args) {
        if (plr == null) {
            if (args.length < 3) {
                return !PlayerFunctions.sendMessage(null, "If you accidentally delete your database, this command will attempt to restore all plots based on the data from the plot signs. \n\n&cMissing world arg /plot debugclaimtest {world} {PlotId min} {PlotId max}");
            }
            final World world = Bukkit.getWorld(args[0]);
            if ((world == null) || !PlotMain.isPlotWorld(world)) {
                return !PlayerFunctions.sendMessage(null, "&cInvalid plot world!");
            }

            PlotId min, max;

            try {
                final String[] split1 = args[1].split(";");
                final String[] split2 = args[2].split(";");

                min = new PlotId(Integer.parseInt(split1[0]), Integer.parseInt(split1[1]));
                max = new PlotId(Integer.parseInt(split2[0]), Integer.parseInt(split2[1]));
            } catch (final Exception e) {
                return !PlayerFunctions.sendMessage(null, "&cInvalid min/max values. &7The values are to Plot IDs in the format &cX;Y &7where X,Y are the plot coords\nThe conversion will only check the plots in the selected area.");
            }
            PlayerFunctions.sendMessage(null, "&3Sign Block&8->&3PlotSquared&8: &7Beginning sign to plot conversion. This may take a while...");
            PlayerFunctions.sendMessage(null, "&3Sign Block&8->&3PlotSquared&8: Found an excess of 250,000 chunks. Limiting search radius... (~3.8 min)");

            final PlotManager manager = PlotMain.getPlotManager(world);
            final PlotWorld plotworld = PlotMain.getWorldSettings(world);

            final ArrayList<Plot> plots = new ArrayList<>();

            for (final PlotId id : PlayerFunctions.getPlotSelectionIds(world, min, max)) {
                final Plot plot = PlotHelper.getPlot(world, id);
                final boolean contains = PlotMain.getPlots(world).containsKey(plot.id);
                if (contains) {
                    PlayerFunctions.sendMessage(null, " - &cDB Already contains: " + plot.id);
                    continue;
                }

                final Location loc = manager.getSignLoc(world, plotworld, plot);

                final Chunk chunk = world.getChunkAt(loc);

                if (!chunk.isLoaded()) {
                    final boolean result = chunk.load(false);
                    if (!result) {
                        continue;
                    }
                }

                final Block block = world.getBlockAt(loc);
                if (block != null) {
                    if (block.getState() instanceof Sign) {
                        final Sign sign = (Sign) block.getState();
                        String line = sign.getLine(2);
                        if ((line != null) && (line.length() > 2)) {
                            line = line.substring(2);

                            final BiMap<StringWrapper, UUID> map = UUIDHandler.getUuidMap();

                            UUID uuid = (map.get(new StringWrapper(line)));

                            if (uuid == null) {
                                for (final StringWrapper string : map.keySet()) {
                                    if (string.value.toLowerCase().startsWith(line.toLowerCase())) {
                                        uuid = map.get(string);
                                        break;
                                    }
                                }
                            }
                            if (uuid == null) {
                                uuid = UUIDHandler.getUUID(line);
                            }
                            if (uuid != null) {
                                PlayerFunctions.sendMessage(null, " - &aFound plot: " + plot.id + " : " + line);
                                plot.owner = uuid;
                                plot.hasChanged = true;
                                plots.add(plot);
                            } else {
                                PlayerFunctions.sendMessage(null, " - &cInvalid playername: " + plot.id + " : " + line);
                            }
                        }
                    }
                }
            }

            if (plots.size() > 0) {
                PlayerFunctions.sendMessage(null, "&3Sign Block&8->&3PlotSquared&8: &7Updating '" + plots.size() + "' plots!");
                DBFunc.createPlots(plots);
                DBFunc.createAllSettingsAndHelpers(plots);

                for (final Plot plot : plots) {
                    PlotMain.updatePlot(plot);
                }

                PlayerFunctions.sendMessage(null, "&3Sign Block&8->&3PlotSquared&8: &7Complete!");

            } else {
                PlayerFunctions.sendMessage(null, "No plots were found for the given search.");
            }

        } else {
            PlayerFunctions.sendMessage(plr, "This debug command can only be executed by console as it has been deemed unsafe if abused.");
        }
        return true;
    }
}