kiafumi/src/main/java/moe/oko/Kiafumi/model/KiafumiDB.java

package moe.oko.Kiafumi.model;

import moe.oko.Kiafumi.Kiafumi;
import net.dv8tion.jda.api.entities.Guild;

import java.sql.*;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import static moe.oko.Kiafumi.Kiafumi.*;

/**
 * Kiafumi DB Class
 * Basically our helpful MySQL functions that pertain to data persistence.
 * @author Kay, with moral support from Yodabird
 * @apiNote  I HATE SQL I HATE SQL AAAAAAAAAAAAAA
 */
public class KiafumiDB {
    //Our actual MySQL Connection, this is created on class construction.
    private Connection connection;

    //The prepared statement strings
    private final String CREATE_SERVERINFO_TABLE = "CREATE TABLE IF NOT EXISTS `serverInfo`" +
            "(`id` LONGTEXT NOT NULL,`welcomeEnabled` TINYINT NOT NULL,`welcomeChannel` LONGTEXT NULL, joinRole LONGTEXT NULL, `serverProtected` TINYINT NOT NULL, `modRole` LONGTEXT NULL);";
    private final String INSERT_SERVERINFO_DEFAULT = "INSERT INTO serverInfo(id,welcomeEnabled,welcomeChannel, joinRole, serverProtected, modRole) values (?,?,?,?,?, ?)";
    private final String SERVERINFO_EXISTS = "select * from serverInfo where id = ?";
    private final String SERVER_INFO_LOAD = "select * from serverInfo";
    //1 = welcomeEnabled, 2 = welcomeChannel, 3 = joinRole, 4 = serverProtected, 5 = Guild_ID (non-changeable)
    private final String SERVER_INFO_MODIFY = "UPDATE `serverInfo` SET welcomeEnabled=?, welcomeChannel=?, joinRole=?,serverProtected=?, modRole=? WHERE id = ?";

    /**
     * KiafumiDB Config Constructor
     * @param username - username to access database
     * @param password - password to access database
     * @param host - host for server
     * @param port - port for server (3306 is default)
     * @param db - database name
     */
    public KiafumiDB(String username, String password, String host, int port, String db) {
        try {
            connection = DriverManager.getConnection("jdbc:mysql://" + host + ":" + port + "/" + db, username, password);
        } catch (Exception ex) {
            error("Failed to initialize the MySQL instance. Contact a developer.");
            ex.printStackTrace();
            return;
        }

        initializeTables();
    }

    /**
     * Table initialization function, ran on every startup.
     */
    private void initializeTables() {
        try {
            connection.prepareStatement(CREATE_SERVERINFO_TABLE).execute();
        } catch (Exception ex) {
            ex.printStackTrace();
            error("Failed to initialize SQL tables. They may need to be created manually.");
        }
    }

    /**
     * Creates the default server information when the bot joins a discord.
     * @param guild - the guild that the bot joined
     * @return whether the function succeeded.
     */
    public boolean createServerInformation(Guild guild) {
        try {
            PreparedStatement prep = connection.prepareStatement(SERVERINFO_EXISTS);
            prep.setInt(1, (int) guild.getIdLong());
            ResultSet rsCheck = prep.executeQuery();
            while (rsCheck.next()) {
                //Server already exists, no need to initialize a section for it.
                if(rsCheck.getInt(1) != 0) {
                    info("Server already existed. Skipping initialization.");
                    return true;
                }
            }
            //Proceed with making defaults.
            PreparedStatement ps = connection.prepareStatement(INSERT_SERVERINFO_DEFAULT);
            ps.setString(1, guild.getId());
            ps.setInt(2, 0); //default is false.
            ps.setNull(3, Types.LONGVARCHAR);
            ps.setNull(4, Types.LONGVARCHAR);
            ps.setInt(5, 0); //Falseeeee
            ps.setNull(6, Types.LONGVARCHAR);
            ps.execute();
            return true;
        } catch (Exception ex) {
            error("Failed to create default server info for guild " + guild.getId());
            ex.printStackTrace();
            return false;
        }
    }

    /**
     * Loads all the server information from MySQL into memory.
     * @return - a list of all servers loaded from MySQL.
     */
    public List<Server> loadServerInformation() {
        List<Server> servers = new ArrayList<>();
        try {
            PreparedStatement ps = connection.prepareStatement(SERVER_INFO_LOAD);
            ResultSet rs = ps.executeQuery();
            while(rs.next()) {
                info("Starting new load for server: " + rs.getString(1));
                String id = rs.getString(1);
                boolean welcomeEnabled = false;
                switch(rs.getInt(2)) {
                    case 0:
                        welcomeEnabled = false;
                    case 1:
                        welcomeEnabled = true;
                }
                String welcomeChannel = rs.getString(3);
                String joinRole = rs.getString(4);
                boolean protectionEnabled = false;
                switch(rs.getInt(5)) {
                    case 0:
                        protectionEnabled = false;
                    case 1:
                        protectionEnabled = true;
                }
                String modRole = rs.getString(6);
                Server server = new Server(id, welcomeEnabled, welcomeChannel, joinRole, protectionEnabled, modRole);
                debug("Loaded " + server + "from database.");
                servers.add(server);
            }
            return servers;
        } catch (Exception ex) {
            ex.printStackTrace();
            error("Failed to load server information, check stack.");
            return null;
        }
    }

    /**
     * Saves modified servers into persistent MySQL server.
     * @return - whether the method succeeded.
     */
    public boolean saveServerInformation() {
        Collection<Server> servers = Kiafumi.instance.getServerManager().getServers();
        info("Starting save on " + servers.size() + " servers.");
        try {
            //1 = welcomeEnabled, 2 = welcomeChannel, 3 = joinRole, 4 = serverProtected, 5 = modRole 6 = Guild_ID (non-changeable)
            PreparedStatement ps = connection.prepareStatement(SERVER_INFO_MODIFY);
            int i = 0;
            for (Server server : servers) {
                if(!server.isModified()) { continue; } //Skip, non modified server.
                info("Starting save on modified " + server);
                if(server.isWelcomeEnabled()) {
                    ps.setInt(1, 1);
                } else {
                    ps.setInt(1, 0);
                }
                if(server.getWelcomeChannel() != null) {
                    ps.setString(2, server.getWelcomeChannel());
                } else {
                    ps.setNull(2, Types.LONGVARCHAR);
                }
                if(server.getJoinRole() != null) {
                    ps.setString(3, server.getJoinRole());
                } else {
                    ps.setNull(3, Types.LONGVARCHAR);
                }
                if(server.isServerProtected()) {
                    ps.setInt(4, 1);
                } else {
                    ps.setInt(4, 0);
                }
                if(server.getModRole() != null) {
                    ps.setString(5, server.getModRole());
                } else {
                    ps.setNull(5, Types.LONGVARCHAR);
                }
                ps.setString(6, server.getId());
                ps.addBatch();
                i++;
            }
            info("Total Batches: " + i + ". Starting SQL save.");
            ps.executeBatch();
            return true;
        } catch (Exception ex) {
            error("Failed to persist server data. Check stack.");
            ex.printStackTrace();
            return false;
        }
    }
}