package me.kingdomsmp.managers; import me.kingdomsmp.KingdomSMP; import me.kingdomsmp.models.*; import java.io.File; import java.sql.*; import java.util.*; import java.util.concurrent.CompletableFuture; import java.util.function.Consumer; import java.util.logging.Level; public class DatabaseManager { private final KingdomSMP plugin; private Connection connection; private boolean useMySQL; public DatabaseManager(KingdomSMP plugin) { this.plugin = plugin; } public void initialize() { useMySQL = plugin.getConfig().getBoolean("database.mysql.enabled", false); if (useMySQL) { connectMySQL(); } else { connectSQLite(); } createTables(); } private void connectSQLite() { try { Class.forName("org.sqlite.JDBC"); File dbFile = new File(plugin.getDataFolder(), "kingdomsmp.db"); connection = DriverManager.getConnection("jdbc:sqlite:" + dbFile.getAbsolutePath()); plugin.getLogger().info("Connected to SQLite database."); } catch (Exception e) { plugin.getLogger().log(Level.SEVERE, "Failed to connect to SQLite", e); } } private void connectMySQL() { String host = plugin.getConfig().getString("database.mysql.host", "localhost"); int port = plugin.getConfig().getInt("database.mysql.port", 3306); String database = plugin.getConfig().getString("database.mysql.database", "kingdomsmp"); String username = plugin.getConfig().getString("database.mysql.username", "root"); String password = plugin.getConfig().getString("database.mysql.password", ""); boolean useSSL = plugin.getConfig().getBoolean("database.mysql.useSSL", false); try { Class.forName("com.mysql.cj.jdbc.Driver"); String url = "jdbc:mysql://" + host + ":" + port + "/" + database + "?useSSL=" + useSSL + "&autoReconnect=true"; connection = DriverManager.getConnection(url, username, password); plugin.getLogger().info("Connected to MySQL database."); } catch (Exception e) { plugin.getLogger().log(Level.SEVERE, "Failed to connect to MySQL, falling back to SQLite", e); useMySQL = false; connectSQLite(); createTables(); } } private void createTables() { executeAsync(conn -> { try (Statement stmt = conn.createStatement()) { stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS kingdoms (" + "id TEXT PRIMARY KEY, " + "name TEXT NOT NULL UNIQUE, " + "description TEXT DEFAULT '', " + "treasury REAL DEFAULT 0.0, " + "honor INTEGER DEFAULT 0, " + "level INTEGER DEFAULT 1, " + "king_uuid TEXT NOT NULL, " + "created_at LONG DEFAULT " + System.currentTimeMillis() + ")" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS kingdom_members (" + "kingdom_id TEXT NOT NULL, " + "player_uuid TEXT NOT NULL, " + "player_name TEXT DEFAULT '', " + "role TEXT NOT NULL DEFAULT 'RECRUIT', " + "joined_at LONG DEFAULT " + System.currentTimeMillis() + ", " + "PRIMARY KEY (kingdom_id, player_uuid))" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS player_data (" + "player_uuid TEXT PRIMARY KEY, " + "player_name TEXT DEFAULT '', " + "hearts REAL DEFAULT 30.0, " + "max_hearts INTEGER DEFAULT 30, " + "kingdom_id TEXT, " + "kingdom_role TEXT DEFAULT 'RECRUIT')" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS claims (" + "kingdom_id TEXT NOT NULL, " + "world_name TEXT NOT NULL, " + "chunk_x INTEGER NOT NULL, " + "chunk_z INTEGER NOT NULL, " + "PRIMARY KEY (world_name, chunk_x, chunk_z))" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS kingdom_relics (" + "kingdom_id TEXT NOT NULL, " + "relic_name TEXT NOT NULL, " + "count INTEGER DEFAULT 1, " + "PRIMARY KEY (kingdom_id, relic_name))" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS wars (" + "id TEXT PRIMARY KEY, " + "attacker_id TEXT NOT NULL, " + "defender_id TEXT NOT NULL, " + "status TEXT NOT NULL DEFAULT 'DECLARED', " + "declared_at LONG DEFAULT " + System.currentTimeMillis() + ", " + "started_at LONG DEFAULT 0)" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS war_rosters (" + "war_id TEXT NOT NULL, " + "player_uuid TEXT NOT NULL, " + "kingdom_id TEXT NOT NULL, " + "lives INTEGER DEFAULT 2, " + "eliminated INTEGER DEFAULT 0, " + "PRIMARY KEY (war_id, player_uuid))" ); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to create tables", e); } }); } // ---- ASYNC HELPERS ---- public void executeAsync(Consumer consumer) { CompletableFuture.runAsync(() -> { try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } consumer.accept(connection); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Database error", e); } }); } public CompletableFuture executeAsyncFuture(Consumer consumer) { return CompletableFuture.runAsync(() -> { try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } consumer.accept(connection); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Database error", e); } }); } // ---- KINGDOM OPERATIONS ---- public CompletableFuture saveKingdom(Kingdom kingdom) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO kingdoms (id, name, description, treasury, honor, level, king_uuid, created_at) " + "VALUES (?, ?, ?, ?, ?, ?, ?, ?)")) { ps.setString(1, kingdom.getId().toString()); ps.setString(2, kingdom.getName()); ps.setString(3, kingdom.getDescription()); ps.setDouble(4, kingdom.getTreasury()); ps.setInt(5, kingdom.getHonor()); ps.setInt(6, kingdom.getLevel()); ps.setString(7, kingdom.getKingUuid().toString()); ps.setLong(8, kingdom.getCreatedAt()); ps.executeUpdate(); } }); } public CompletableFuture deleteKingdom(UUID kingdomId) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement("DELETE FROM kingdoms WHERE id = ?")) { ps.setString(1, kingdomId.toString()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement("DELETE FROM kingdom_members WHERE kingdom_id = ?")) { ps.setString(1, kingdomId.toString()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement("DELETE FROM claims WHERE kingdom_id = ?")) { ps.setString(1, kingdomId.toString()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement("DELETE FROM kingdom_relics WHERE kingdom_id = ?")) { ps.setString(1, kingdomId.toString()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement("UPDATE player_data SET kingdom_id = NULL, kingdom_role = 'RECRUIT' WHERE kingdom_id = ?")) { ps.setString(1, kingdomId.toString()); ps.executeUpdate(); } }); } public CompletableFuture> loadAllKingdoms() { return CompletableFuture.supplyAsync(() -> { List kingdoms = new ArrayList<>(); try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } try (Statement stmt = connection.createStatement(); ResultSet rs = stmt.executeQuery("SELECT * FROM kingdoms")) { while (rs.next()) { Kingdom kingdom = new Kingdom( UUID.fromString(rs.getString("id")), rs.getString("name"), rs.getString("description"), rs.getDouble("treasury"), rs.getInt("honor"), rs.getInt("level"), UUID.fromString(rs.getString("king_uuid")), rs.getLong("created_at") ); kingdoms.add(kingdom); } } for (Kingdom kingdom : kingdoms) { loadMembers(kingdom); loadRelics(kingdom); loadClaims(kingdom); } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to load kingdoms", e); } return kingdoms; }); } private void loadMembers(Kingdom kingdom) throws SQLException { try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM kingdom_members WHERE kingdom_id = ?")) { ps.setString(1, kingdom.getId().toString()); try (ResultSet rs = ps.executeQuery()) { while (rs.next()) { kingdom.getMembers().add(new KingdomMember( UUID.fromString(rs.getString("player_uuid")), rs.getString("player_name"), KingdomRole.valueOf(rs.getString("role")), rs.getLong("joined_at") )); } } } } private void loadRelics(Kingdom kingdom) throws SQLException { try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM kingdom_relics WHERE kingdom_id = ?")) { ps.setString(1, kingdom.getId().toString()); try (ResultSet rs = ps.executeQuery()) { while (rs.next()) { try { Relic relic = Relic.valueOf(rs.getString("relic_name")); int count = rs.getInt("count"); for (int i = 0; i < count; i++) { kingdom.getRelics().add(relic); } } catch (IllegalArgumentException ignored) {} } } } } private void loadClaims(Kingdom kingdom) throws SQLException { try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM claims WHERE kingdom_id = ?")) { ps.setString(1, kingdom.getId().toString()); try (ResultSet rs = ps.executeQuery()) { while (rs.next()) { kingdom.getClaims().add(new ClaimedChunk( kingdom.getId(), rs.getString("world_name"), rs.getInt("chunk_x"), rs.getInt("chunk_z") )); } } } } public CompletableFuture saveMember(UUID kingdomId, KingdomMember member) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO kingdom_members (kingdom_id, player_uuid, player_name, role, joined_at) " + "VALUES (?, ?, ?, ?, ?)")) { ps.setString(1, kingdomId.toString()); ps.setString(2, member.getUuid().toString()); ps.setString(3, member.getPlayerName()); ps.setString(4, member.getRole().name()); ps.setLong(5, member.getJoinedAt()); ps.executeUpdate(); } }); } public CompletableFuture removeMember(UUID kingdomId, UUID playerUuid) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "DELETE FROM kingdom_members WHERE kingdom_id = ? AND player_uuid = ?")) { ps.setString(1, kingdomId.toString()); ps.setString(2, playerUuid.toString()); ps.executeUpdate(); } }); } // ---- CLAIM OPERATIONS ---- public CompletableFuture saveClaim(ClaimedChunk claim) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO claims (kingdom_id, world_name, chunk_x, chunk_z) VALUES (?, ?, ?, ?)")) { ps.setString(1, claim.getKingdomId().toString()); ps.setString(2, claim.getWorldName()); ps.setInt(3, claim.getX()); ps.setInt(4, claim.getZ()); ps.executeUpdate(); } }); } public CompletableFuture removeClaim(ClaimedChunk claim) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "DELETE FROM claims WHERE world_name = ? AND chunk_x = ? AND chunk_z = ?")) { ps.setString(1, claim.getWorldName()); ps.setInt(2, claim.getX()); ps.setInt(3, claim.getZ()); ps.executeUpdate(); } }); } public CompletableFuture getClaimAt(String worldName, int x, int z) { return CompletableFuture.supplyAsync(() -> { try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM claims WHERE world_name = ? AND chunk_x = ? AND chunk_z = ?")) { ps.setString(1, worldName); ps.setInt(2, x); ps.setInt(3, z); try (ResultSet rs = ps.executeQuery()) { if (rs.next()) { return new ClaimedChunk( UUID.fromString(rs.getString("kingdom_id")), rs.getString("world_name"), rs.getInt("chunk_x"), rs.getInt("chunk_z") ); } } } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to get claim", e); } return null; }); } // ---- RELIC OPERATIONS ---- public CompletableFuture saveRelic(UUID kingdomId, Relic relic) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT INTO kingdom_relics (kingdom_id, relic_name, count) VALUES (?, ?, 1) " + "ON CONFLICT(kingdom_id, relic_name) DO UPDATE SET count = count + 1")) { ps.setString(1, kingdomId.toString()); ps.setString(2, relic.name()); ps.executeUpdate(); } }); } public CompletableFuture removeRelic(UUID kingdomId, Relic relic) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "UPDATE kingdom_relics SET count = count - 1 WHERE kingdom_id = ? AND relic_name = ? AND count > 0")) { ps.setString(1, kingdomId.toString()); ps.setString(2, relic.name()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement( "DELETE FROM kingdom_relics WHERE kingdom_id = ? AND relic_name = ? AND count <= 0")) { ps.setString(1, kingdomId.toString()); ps.setString(2, relic.name()); ps.executeUpdate(); } }); } // ---- PLAYER DATA ---- public CompletableFuture savePlayerData(PlayerData data) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO player_data (player_uuid, player_name, hearts, max_hearts, kingdom_id, kingdom_role) " + "VALUES (?, ?, ?, ?, ?, ?)")) { ps.setString(1, data.getUuid().toString()); ps.setString(2, data.getPlayerName()); ps.setDouble(3, data.getHearts()); ps.setInt(4, data.getMaxHearts()); ps.setString(5, data.getKingdomId() != null ? data.getKingdomId().toString() : null); ps.setString(6, data.getKingdomRole().name()); ps.executeUpdate(); } }); } public CompletableFuture loadPlayerData(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM player_data WHERE player_uuid = ?")) { ps.setString(1, uuid.toString()); try (ResultSet rs = ps.executeQuery()) { if (rs.next()) { String kingdomIdStr = rs.getString("kingdom_id"); UUID kingdomId = kingdomIdStr != null ? UUID.fromString(kingdomIdStr) : null; return new PlayerData( UUID.fromString(rs.getString("player_uuid")), rs.getString("player_name"), rs.getDouble("hearts"), rs.getInt("max_hearts"), kingdomId, KingdomRole.valueOf(rs.getString("kingdom_role")) ); } } } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to load player data for " + uuid, e); } return null; }); } // ---- WAR OPERATIONS ---- public CompletableFuture saveWar(WarData war) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO wars (id, attacker_id, defender_id, status, declared_at, started_at) " + "VALUES (?, ?, ?, ?, ?, ?)")) { ps.setString(1, war.getId().toString()); ps.setString(2, war.getAttackingKingdomId().toString()); ps.setString(3, war.getDefendingKingdomId().toString()); ps.setString(4, war.getStatus().name()); ps.setLong(5, war.getDeclaredAt()); ps.setLong(6, war.getStartedAt()); ps.executeUpdate(); } // Save war rosters try (PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO war_rosters (war_id, player_uuid, kingdom_id, lives, eliminated) VALUES (?, ?, ?, ?, ?)")) { for (UUID playerUuid : war.getAttackers()) { ps.setString(1, war.getId().toString()); ps.setString(2, playerUuid.toString()); ps.setString(3, war.getAttackingKingdomId().toString()); ps.setInt(4, war.getWarLives().getOrDefault(playerUuid, 2)); ps.setInt(5, war.isEliminated(playerUuid) ? 1 : 0); ps.addBatch(); } for (UUID playerUuid : war.getDefenders()) { ps.setString(1, war.getId().toString()); ps.setString(2, playerUuid.toString()); ps.setString(3, war.getDefendingKingdomId().toString()); ps.setInt(4, war.getWarLives().getOrDefault(playerUuid, 2)); ps.setInt(5, war.isEliminated(playerUuid) ? 1 : 0); ps.addBatch(); } ps.executeBatch(); } }); } public CompletableFuture deleteWar(UUID warId) { return executeAsyncFuture(conn -> { try (PreparedStatement ps = conn.prepareStatement("DELETE FROM wars WHERE id = ?")) { ps.setString(1, warId.toString()); ps.executeUpdate(); } try (PreparedStatement ps = conn.prepareStatement("DELETE FROM war_rosters WHERE war_id = ?")) { ps.setString(1, warId.toString()); ps.executeUpdate(); } }); } public CompletableFuture> loadAllWars() { return CompletableFuture.supplyAsync(() -> { List wars = new ArrayList<>(); try { if (connection == null || connection.isClosed()) { if (useMySQL) connectMySQL(); else connectSQLite(); } try (Statement stmt = connection.createStatement(); ResultSet rs = stmt.executeQuery("SELECT * FROM wars WHERE status != 'FINISHED' AND status != 'CANCELLED'")) { while (rs.next()) { UUID warId = UUID.fromString(rs.getString("id")); WarData war = new WarData( warId, UUID.fromString(rs.getString("attacker_id")), UUID.fromString(rs.getString("defender_id")), WarData.WarStatus.valueOf(rs.getString("status")), new HashMap<>(), new ArrayList<>(), new HashSet<>(), new HashSet<>(), rs.getLong("declared_at"), rs.getLong("started_at") ); loadWarRoster(war); wars.add(war); } } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to load wars", e); } return wars; }); } private void loadWarRoster(WarData war) throws SQLException { try (PreparedStatement ps = connection.prepareStatement( "SELECT * FROM war_rosters WHERE war_id = ?")) { ps.setString(1, war.getId().toString()); try (ResultSet rs = ps.executeQuery()) { while (rs.next()) { UUID playerUuid = UUID.fromString(rs.getString("player_uuid")); UUID kingdomId = UUID.fromString(rs.getString("kingdom_id")); int lives = rs.getInt("lives"); boolean eliminated = rs.getInt("eliminated") == 1; if (kingdomId.equals(war.getAttackingKingdomId())) { war.getAttackers().add(playerUuid); } else { war.getDefenders().add(playerUuid); } if (!eliminated) { war.getWarLives().put(playerUuid, lives); } else { war.getEliminatedPlayers().add(playerUuid); } } } } } public void close() { try { if (connection != null && !connection.isClosed()) { connection.close(); } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Error closing database", e); } } }