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

package com.intellectualcrafters.plot.commands;

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.UUID;

import org.bukkit.entity.Player;

import com.intellectualcrafters.plot.PlotSquared;
import com.intellectualcrafters.plot.database.MySQL;
import com.intellectualcrafters.plot.database.SQLManager;
import com.intellectualcrafters.plot.object.Plot;
import com.intellectualcrafters.plot.util.StringComparison;
import com.intellectualcrafters.plot.util.TaskManager;
import com.intellectualcrafters.plot.util.bukkit.BukkitPlayerFunctions;
import com.intellectualcrafters.plot.util.bukkit.UUIDHandler;

/**
 * Created 2014-11-15 for PlotSquared
 *
 * @author Citymonstret
 */
public class Database extends SubCommand {
    final String[] tables = new String[] { "plot_trusted", "plot_ratings", "plot_comments" };
    
    public Database() {
        super(Command.DATABASE, "Convert/Backup Storage", "database [type] [...details]", CommandCategory.DEBUG, false);
    }
    
    private static boolean sendMessageU(final UUID uuid, final String msg) {
        if (uuid == null) {
            PlotSquared.log(msg);
        } else {
            final Player p = UUIDHandler.uuidWrapper.getPlayer(uuid);
            if ((p != null) && p.isOnline()) {
                return BukkitPlayerFunctions.sendMessage(p, msg);
            } else {
                return sendMessageU(null, msg);
            }
        }
        return true;
    }
    
    public static void insertPlots(final SQLManager manager, final UUID requester, final Connection c) {
        final java.util.Set<Plot> plots = PlotSquared.getPlots();
        TaskManager.runTaskAsync(new Runnable() {
            @Override
            public void run() {
                try {
                    final ArrayList<Plot> ps = new ArrayList<>();
                    for (final Plot p : plots) {
                        ps.add(p);
                    }
                    manager.createPlots(ps);
                    manager.createAllSettingsAndHelpers(ps);
                    sendMessageU(requester, "&6Database conversion finished");
                } catch (final Exception e) {
                    sendMessageU(requester, "Failed to insert plot objects, see stacktrace for info");
                    e.printStackTrace();
                }
                try {
                    c.close();
                } catch (final SQLException e) {
                    e.printStackTrace();
                }
            }
        });
    }
    
    @Override
    public boolean execute(final Player plr, final String... args) {
        if (args.length < 1) {
            return sendMessage(plr, "/plot database [sqlite/mysql]");
        }
        final String type = new StringComparison(args[0], new String[] { "mysql", "sqlite" }).getBestMatch().toLowerCase();
        switch (type) {
            case "mysql":
                if (args.length < 6) {
                    return sendMessage(plr, "/plot database mysql [host] [port] [username] [password] [database] {prefix}");
                }
                final String host = args[1];
                final String port = args[2];
                final String username = args[3];
                final String password = args[4];
                final String database = args[5];
                String prefix = "";
                if (args.length > 6) {
                    prefix = args[6];
                }
                Connection n;
                try {
                    n = new MySQL(PlotSquared.THIS, host, port, database, username, password).openConnection();
                    // Connection
                    if (n.isClosed()) {
                        return sendMessage(plr, "Failed to open connection");
                    }
                } catch (SQLException | ClassNotFoundException e) {
                    e.printStackTrace();
                    return sendMessage(plr, "Failed to open connection, read stacktrace for info");
                }
                final SQLManager manager = new SQLManager(n, prefix);
                try {
                    final DatabaseMetaData meta = n.getMetaData();
                    ResultSet set = meta.getTables(null, null, prefix + "plot", null);
                    if (!set.next()) {
                        manager.createTables("mysql", true);
                    } else {
                        for (final String s : this.tables) {
                            set = meta.getTables(null, null, prefix + s, null);
                            if (!set.next()) {
                                manager.createTables("mysql", false);
                            }
                        }
                    }
                } catch (final SQLException e) {
                    e.printStackTrace();
                    return sendMessage(plr, "Could not create the required tables and/or load the database") && sendMessage(plr, "Please see the stacktrace for more information");
                }
                UUID requester = null;
                if (plr != null) {
                    requester = UUIDHandler.getUUID(plr);
                }
                insertPlots(manager, requester, n);
                break;
            case "sqlite":
                if (args.length < 2) {
                    return sendMessage(plr, "/plot database sqlite [file name]");
                }
                sendMessage(plr, "This is not supported yet");
                break;
            default:
                return sendMessage(plr, "Unknown database type");
        }
        return false;
    }
    
    private boolean sendMessage(final Player player, final String msg) {
        if (player == null) {
            PlotSquared.log(msg);
        } else {
            MainUtil.sendMessage(BukkitUtil.getPlayer(player), msg);
        }
        return true;
    }
}