PreventStabby/src/main/java/me/youhavetrouble/preventstabby/util/DatabaseSQLite.java

package me.youhavetrouble.preventstabby.util;

import me.youhavetrouble.preventstabby.PreventStabby;
import me.youhavetrouble.preventstabby.players.PlayerData;

import java.io.File;
import java.sql.*;
import java.util.UUID;

public class DatabaseSQLite {

    private final String url;
    private final File folder;

    public DatabaseSQLite(String url, File folder) {
        this.url = url;
        this.folder = folder;

    }

    public boolean createDatabaseFile() {
        this.folder.mkdir();
        try (Connection conn = DriverManager.getConnection(url)) {
            if (conn != null) {
                DatabaseMetaData meta = conn.getMetaData();
                System.out.println("The driver name is " + meta.getDriverName());
                Statement statement = conn.createStatement();
                String sql = "CREATE TABLE IF NOT EXISTS `players` (`player_uuid` varchar(36) UNIQUE PRIMARY KEY, `pvpenabled` boolean);";
                statement.execute(sql);
                conn.close();
            }
        } catch (SQLException e) {
            System.out.println(e.getMessage());
            return false;
        }
        return true;
    }

    public boolean testConnection() {
        Connection conn = null;
        try {
            conn = DriverManager.getConnection(url);
            System.out.println("Connection to SQLite has been established.");
            if (conn != null) {
                conn.close();
                return true;
            }
        } catch (SQLException e) {
            System.out.println(e.getMessage());
        }
        return false;
    }

    public PlayerData getPlayerInfo(UUID uuid) {
        try {
            Connection conn = DriverManager.getConnection(url);
            Statement insertnewuser = conn.createStatement();
            try {
                String newuserdata = "INSERT OR IGNORE INTO `players` (player_uuid, pvpenabled) VALUES ('" + uuid.toString() + "', " + PreventStabby.getPlugin().getConfigCache().isPvp_enabled_by_default() + ")";
                insertnewuser.execute(newuserdata);
            } catch (SQLException e) {
                if (e.getErrorCode() != 19) {
                    e.printStackTrace();
                }
            }
            Statement statement = conn.createStatement();
            String sql = "SELECT * FROM `players` WHERE `player_uuid` = '" + uuid + "';";
            statement.execute(sql);
            ResultSet result = statement.getResultSet();
            boolean state = result.getBoolean("pvpenabled");
            conn.close();
            return new PlayerData(uuid, state);
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return null;
    }

    public void updatePlayerInfo(UUID uuid, PlayerData data) {
        try {
            Connection conn = DriverManager.getConnection(url);
            Statement insertnewuser = conn.createStatement();
            try {
                String newuserdata = "UPDATE `players` SET pvpenabled = "+data.isPvpEnabled()+" WHERE `player_uuid` = '"+uuid.toString()+"';";
                insertnewuser.execute(newuserdata);
            } catch (SQLException e) {
                PreventStabby.getPlugin().getLogger().severe("Error while saving player data!");
                e.printStackTrace();
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
    }

}