Plex/src/main/java/me/totalfreedom/plex/storage/SQLConnection.java

package me.totalfreedom.plex.storage;

import java.io.File;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import me.totalfreedom.plex.Plex;

public class SQLConnection
{
    private final Plex plugin = Plex.get();

    private Connection connection;

    public Connection getCon()
    {
        String host = plugin.config.getString("data.central.hostname");
        int port = plugin.config.getInt("data.central.port");
        String username = plugin.config.getString("data.central.user");
        String password = plugin.config.getString("data.central.password");
        String database = plugin.config.getString("data.central.db");

        try
        {
            if (plugin.config.getString("data.central.storage").equalsIgnoreCase("sqlite"))
            {
                connection = DriverManager.getConnection("jdbc:sqlite:" + new File(plugin.getDataFolder(), "database.db").getAbsolutePath());
                Plex.get().setStorageType(StorageType.SQLITE);
            }
            else if (plugin.config.getString("data.central.storage").equalsIgnoreCase("mysql"))
            {
                connection = DriverManager.getConnection("jdbc:mysql://" + host + ":" + port + "/" + database, username, password);
                Plex.get().setStorageType(StorageType.SQL);
            }
        }
        catch (SQLException throwables)
        {
            throwables.printStackTrace();
        }

        try
        {
            connection.prepareStatement("CREATE TABLE IF NOT EXISTS `players` (\n" +
                    "\t`uuid` VARCHAR(46),\n" +
                    "\t`name` VARCHAR(18),\n" +
                    "\t`login_msg` VARCHAR(256),\n" +
                    "\t`prefix` VARCHAR(30),\n" +
                    "\t`rank` VARCHAR(256),\n" +
                    "\t`ips` VARCHAR(65535),\n" +
                    "\t`coins` BIGINT,\n" +
                    "\tPRIMARY KEY (`uuid`)\n" +
                    ");").execute();
        }
        catch (SQLException throwables)
        {
            throwables.printStackTrace();
        }
        return connection;
    }
}