package com.zelchat.database; import com.zelchat.ZelChat; import com.zaxxer.hikari.HikariConfig; import com.zaxxer.hikari.HikariDataSource; import org.bukkit.configuration.file.FileConfiguration; import java.io.File; import java.sql.Connection; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.util.*; import java.util.concurrent.CompletableFuture; public class DatabaseManager { private final ZelChat plugin; private HikariDataSource dataSource; private String tableUsers; private String tableData; private String tableInfractions; private String tableIgnore; public DatabaseManager(ZelChat plugin) { this.plugin = plugin; } public void connect() { FileConfiguration config = plugin.getConfigManager().get("database.yml"); String type = config.getString("type", "H2"); tableUsers = config.getString("table-users", "zelchat_users"); tableData = config.getString("table-data", "zelchat_data"); tableInfractions = config.getString("table-infractions", "zelchat_infractions"); tableIgnore = config.getString("table-ignore", "zelchat_ignore"); HikariConfig hikariConfig = new HikariConfig(); if (type.equalsIgnoreCase("H2")) { File dbFile = new File(plugin.getDataFolder(), "data"); hikariConfig.setJdbcUrl("jdbc:h2:" + dbFile.getAbsolutePath()); hikariConfig.setDriverClassName("org.h2.Driver"); } else if (type.equalsIgnoreCase("MYSQL")) { String host = config.getString("host", "127.0.0.1"); String port = config.getString("port", "3306"); String database = config.getString("database", "zelchat"); String user = config.getString("user", "zelchat"); String password = config.getString("password", "user_password"); hikariConfig.setJdbcUrl("jdbc:mysql://" + host + ":" + port + "/" + database); hikariConfig.setUsername(user); hikariConfig.setPassword(password); hikariConfig.setDriverClassName("com.mysql.cj.jdbc.Driver"); hikariConfig.addDataSourceProperty("useSSL", "false"); hikariConfig.addDataSourceProperty("characterEncoding", "utf8mb4"); } hikariConfig.setMaximumPoolSize(10); hikariConfig.setConnectionTimeout(30000); hikariConfig.setLeakDetectionThreshold(60000); dataSource = new HikariDataSource(hikariConfig); createTables(); } private void createTables() { FileConfiguration config = plugin.getConfigManager().get("database.yml"); String dbType = config.getString("type", "H2"); String autoIncrement = dbType.equalsIgnoreCase("MYSQL") ? "AUTO_INCREMENT" : "IDENTITY"; try (Connection conn = getConnection()) { conn.createStatement().executeUpdate( "CREATE TABLE IF NOT EXISTS " + tableUsers + " (" + "uuid VARCHAR(36) PRIMARY KEY, " + "username VARCHAR(16) NOT NULL)" ); conn.createStatement().executeUpdate( "CREATE TABLE IF NOT EXISTS " + tableData + " (" + "uuid VARCHAR(36) PRIMARY KEY, " + "chat_toggled BOOLEAN DEFAULT FALSE, " + "pings_enabled BOOLEAN DEFAULT TRUE, " + "messages_closed BOOLEAN DEFAULT FALSE, " + "spy_enabled BOOLEAN DEFAULT FALSE, " + "staffchat_toggled BOOLEAN DEFAULT FALSE)" ); conn.createStatement().executeUpdate( "CREATE TABLE IF NOT EXISTS " + tableInfractions + " (" + "id INT " + autoIncrement + " PRIMARY KEY, " + "uuid VARCHAR(36) NOT NULL, " + "points INT DEFAULT 0, " + "last_message VARCHAR(256), " + "last_infraction_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP)" ); conn.createStatement().executeUpdate( "CREATE TABLE IF NOT EXISTS " + tableIgnore + " (" + "id INT " + autoIncrement + " PRIMARY KEY, " + "player_uuid VARCHAR(36) NOT NULL, " + "ignored_uuid VARCHAR(36) NOT NULL, " + "UNIQUE KEY unique_pair (player_uuid, ignored_uuid))" ); } catch (SQLException e) { plugin.getLogger().severe("Failed to create tables: " + e.getMessage()); } } public Connection getConnection() throws SQLException { return dataSource.getConnection(); } public void disconnect() { if (dataSource != null && !dataSource.isClosed()) { dataSource.close(); } } public CompletableFuture registerUser(UUID uuid, String username) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableUsers + " (uuid, username) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE username = ?")) { ps.setString(1, uuid.toString()); ps.setString(2, username); ps.setString(3, username); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to register user: " + e.getMessage()); } }); } public CompletableFuture getChatToggled(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT chat_toggled FROM " + tableData + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getBoolean("chat_toggled"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get chat toggle: " + e.getMessage()); } return false; }); } public CompletableFuture setChatToggled(UUID uuid, boolean toggled) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableData + " (uuid, chat_toggled) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE chat_toggled = ?")) { ps.setString(1, uuid.toString()); ps.setBoolean(2, toggled); ps.setBoolean(3, toggled); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set chat toggle: " + e.getMessage()); } }); } public CompletableFuture getPingsEnabled(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT pings_enabled FROM " + tableData + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getBoolean("pings_enabled"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get pings enabled: " + e.getMessage()); } return true; }); } public CompletableFuture setPingsEnabled(UUID uuid, boolean enabled) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableData + " (uuid, pings_enabled) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE pings_enabled = ?")) { ps.setString(1, uuid.toString()); ps.setBoolean(2, enabled); ps.setBoolean(3, enabled); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set pings enabled: " + e.getMessage()); } }); } public CompletableFuture getMessagesClosed(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT messages_closed FROM " + tableData + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getBoolean("messages_closed"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get messages closed: " + e.getMessage()); } return false; }); } public CompletableFuture setMessagesClosed(UUID uuid, boolean closed) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableData + " (uuid, messages_closed) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE messages_closed = ?")) { ps.setString(1, uuid.toString()); ps.setBoolean(2, closed); ps.setBoolean(3, closed); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set messages closed: " + e.getMessage()); } }); } public CompletableFuture addIgnore(UUID playerUuid, UUID ignoredUuid) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT IGNORE INTO " + tableIgnore + " (player_uuid, ignored_uuid) VALUES (?, ?)")) { ps.setString(1, playerUuid.toString()); ps.setString(2, ignoredUuid.toString()); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to add ignore: " + e.getMessage()); } }); } public CompletableFuture removeIgnore(UUID playerUuid, UUID ignoredUuid) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "DELETE FROM " + tableIgnore + " WHERE player_uuid = ? AND ignored_uuid = ?")) { ps.setString(1, playerUuid.toString()); ps.setString(2, ignoredUuid.toString()); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to remove ignore: " + e.getMessage()); } }); } public CompletableFuture isIgnoring(UUID playerUuid, UUID ignoredUuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT 1 FROM " + tableIgnore + " WHERE player_uuid = ? AND ignored_uuid = ?")) { ps.setString(1, playerUuid.toString()); ps.setString(2, ignoredUuid.toString()); ResultSet rs = ps.executeQuery(); return rs.next(); } catch (SQLException e) { plugin.getLogger().warning("Failed to check ignore: " + e.getMessage()); } return false; }); } public CompletableFuture> getIgnoredPlayers(UUID playerUuid) { return CompletableFuture.supplyAsync(() -> { Set ignored = new HashSet<>(); try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT ignored_uuid FROM " + tableIgnore + " WHERE player_uuid = ?")) { ps.setString(1, playerUuid.toString()); ResultSet rs = ps.executeQuery(); while (rs.next()) { ignored.add(UUID.fromString(rs.getString("ignored_uuid"))); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get ignored players: " + e.getMessage()); } return ignored; }); } public CompletableFuture clearIgnores(UUID playerUuid) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "DELETE FROM " + tableIgnore + " WHERE player_uuid = ?")) { ps.setString(1, playerUuid.toString()); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to clear ignores: " + e.getMessage()); } }); } public CompletableFuture getInfractionPoints(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT points FROM " + tableInfractions + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getInt("points"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get infraction points: " + e.getMessage()); } return 0; }); } public CompletableFuture addInfractionPoints(UUID uuid, int points) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableInfractions + " (uuid, points) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE points = points + ?, last_infraction_time = CURRENT_TIMESTAMP")) { ps.setString(1, uuid.toString()); ps.setInt(2, points); ps.setInt(3, points); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to add infraction points: " + e.getMessage()); } }); } public CompletableFuture getLastMessage(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT last_message FROM " + tableInfractions + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getString("last_message"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get last message: " + e.getMessage()); } return null; }); } public CompletableFuture setLastMessage(UUID uuid, String message) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableInfractions + " (uuid, last_message) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE last_message = ?")) { ps.setString(1, uuid.toString()); ps.setString(2, message); ps.setString(3, message); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set last message: " + e.getMessage()); } }); } public CompletableFuture getSpyEnabled(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT spy_enabled FROM " + tableData + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getBoolean("spy_enabled"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get spy enabled: " + e.getMessage()); } return false; }); } public CompletableFuture setSpyEnabled(UUID uuid, boolean enabled) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableData + " (uuid, spy_enabled) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE spy_enabled = ?")) { ps.setString(1, uuid.toString()); ps.setBoolean(2, enabled); ps.setBoolean(3, enabled); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set spy enabled: " + e.getMessage()); } }); } public CompletableFuture getStaffChatToggled(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT staffchat_toggled FROM " + tableData + " WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getBoolean("staffchat_toggled"); } } catch (SQLException e) { plugin.getLogger().warning("Failed to get staffchat toggle: " + e.getMessage()); } return false; }); } public CompletableFuture setStaffChatToggled(UUID uuid, boolean toggled) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "INSERT INTO " + tableData + " (uuid, staffchat_toggled) VALUES (?, ?) " + "ON DUPLICATE KEY UPDATE staffchat_toggled = ?")) { ps.setString(1, uuid.toString()); ps.setBoolean(2, toggled); ps.setBoolean(3, toggled); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().warning("Failed to set staffchat toggle: " + e.getMessage()); } }); } }