package me.orden.plugin.database; import com.zaxxer.hikari.HikariConfig; import com.zaxxer.hikari.HikariDataSource; import me.orden.plugin.OrdenPlugin; import me.orden.plugin.kingdom.Kingdom; import me.orden.plugin.kingdom.KingdomType; import me.orden.plugin.player.OrdenPlayer; import me.orden.plugin.roles.Role; import org.bukkit.Bukkit; import org.bukkit.Location; import java.io.File; import java.sql.*; import java.util.UUID; import java.util.concurrent.CompletableFuture; public class SQLiteDatabase implements DatabaseManager { private final OrdenPlugin plugin; private HikariDataSource dataSource; public SQLiteDatabase(OrdenPlugin plugin) { this.plugin = plugin; } @Override public CompletableFuture initialize() { return CompletableFuture.runAsync(() -> { try { File dataFolder = plugin.getDataFolder(); if (!dataFolder.exists()) { dataFolder.mkdirs(); } HikariConfig config = new HikariConfig(); config.setJdbcUrl("jdbc:sqlite:" + new File(dataFolder, "orden.db").getAbsolutePath()); config.setMaximumPoolSize(5); config.setMinimumIdle(1); this.dataSource = new HikariDataSource(config); createTables(); } catch (Exception e) { e.printStackTrace(); } }); } private void createTables() throws SQLException { try (Connection conn = dataSource.getConnection()) { Statement stmt = conn.createStatement(); stmt.execute("CREATE TABLE IF NOT EXISTS players (" + "uuid TEXT PRIMARY KEY," + "username TEXT NOT NULL," + "kingdom TEXT," + "role TEXT NOT NULL," + "lives INTEGER NOT NULL," + "pvp_hearts INTEGER NOT NULL DEFAULT 10," + "live_mode INTEGER NOT NULL DEFAULT 0," + "newbie_protection BIGINT NOT NULL DEFAULT 0)"); stmt.execute("CREATE TABLE IF NOT EXISTS kingdoms (" + "name TEXT PRIMARY KEY," + "spawn_world TEXT," + "spawn_x REAL," + "spawn_y REAL," + "spawn_z REAL," + "spawn_yaw REAL," + "spawn_pitch REAL)"); stmt.execute("CREATE TABLE IF NOT EXISTS kingdom_members (" + "kingdom TEXT NOT NULL," + "player_uuid TEXT NOT NULL," + "is_king INTEGER NOT NULL DEFAULT 0," + "is_general INTEGER NOT NULL DEFAULT 0," + "PRIMARY KEY (kingdom, player_uuid))"); stmt.execute("CREATE TABLE IF NOT EXISTS alliances (" + "kingdom1 TEXT NOT NULL," + "kingdom2 TEXT NOT NULL," + "PRIMARY KEY (kingdom1, kingdom2))"); stmt.execute("CREATE TABLE IF NOT EXISTS eliminations (" + "uuid TEXT PRIMARY KEY," + "reason TEXT NOT NULL," + "timestamp BIGINT NOT NULL)"); for (KingdomType type : KingdomType.values()) { stmt.execute("INSERT OR IGNORE INTO kingdoms (name) VALUES ('" + type.name() + "')"); } } } @Override public CompletableFuture close() { return CompletableFuture.runAsync(() -> { if (dataSource != null && !dataSource.isClosed()) { dataSource.close(); } }); } @Override public CompletableFuture loadPlayer(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "SELECT * FROM players WHERE uuid = ?"); ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { OrdenPlayer player = new OrdenPlayer(uuid, rs.getString("username")); String kingdomStr = rs.getString("kingdom"); if (kingdomStr != null) { player.setKingdom(KingdomType.valueOf(kingdomStr)); } player.setRole(Role.valueOf(rs.getString("role"))); player.setLives(rs.getInt("lives")); player.setPvpHearts(rs.getInt("pvp_hearts")); player.setLiveMode(rs.getInt("live_mode") == 1); player.setNewbieProtectionExpiry(rs.getLong("newbie_protection")); return player; } return null; } catch (SQLException e) { e.printStackTrace(); return null; } }); } @Override public CompletableFuture savePlayer(OrdenPlayer player) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO players (uuid, username, kingdom, role, lives, pvp_hearts, live_mode, newbie_protection) " + "VALUES (?, ?, ?, ?, ?, ?, ?, ?)"); ps.setString(1, player.getUuid().toString()); ps.setString(2, player.getUsername()); ps.setString(3, player.getKingdom() != null ? player.getKingdom().name() : null); ps.setString(4, player.getRole().name()); ps.setInt(5, player.getLives()); ps.setInt(6, player.getPvpHearts()); ps.setInt(7, player.isLiveMode() ? 1 : 0); ps.setLong(8, player.getNewbieProtectionExpiry()); ps.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } }); } @Override public CompletableFuture loadKingdom(KingdomType type) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection()) { Kingdom kingdom = new Kingdom(type); PreparedStatement ps = conn.prepareStatement( "SELECT * FROM kingdoms WHERE name = ?"); ps.setString(1, type.name()); ResultSet rs = ps.executeQuery(); if (rs.next()) { String world = rs.getString("spawn_world"); if (world != null) { Location spawn = new Location( Bukkit.getWorld(world), rs.getDouble("spawn_x"), rs.getDouble("spawn_y"), rs.getDouble("spawn_z"), rs.getFloat("spawn_yaw"), rs.getFloat("spawn_pitch") ); kingdom.setSpawn(spawn); } } ps = conn.prepareStatement( "SELECT * FROM kingdom_members WHERE kingdom = ?"); ps.setString(1, type.name()); rs = ps.executeQuery(); while (rs.next()) { UUID memberUuid = UUID.fromString(rs.getString("player_uuid")); kingdom.addMember(memberUuid); if (rs.getInt("is_king") == 1) { kingdom.addKing(memberUuid); } if (rs.getInt("is_general") == 1) { kingdom.setGeneral(memberUuid); } } ps = conn.prepareStatement( "SELECT kingdom2 FROM alliances WHERE kingdom1 = ? " + "UNION SELECT kingdom1 FROM alliances WHERE kingdom2 = ?"); ps.setString(1, type.name()); ps.setString(2, type.name()); rs = ps.executeQuery(); while (rs.next()) { KingdomType ally = KingdomType.valueOf(rs.getString(1)); kingdom.addAlly(ally); } return kingdom; } catch (SQLException e) { e.printStackTrace(); return new Kingdom(type); } }); } @Override public CompletableFuture saveKingdom(Kingdom kingdom) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "UPDATE kingdoms SET spawn_world = ?, spawn_x = ?, spawn_y = ?, spawn_z = ?, spawn_yaw = ?, spawn_pitch = ? WHERE name = ?"); Location spawn = kingdom.getSpawn(); if (spawn != null) { ps.setString(1, spawn.getWorld().getName()); ps.setDouble(2, spawn.getX()); ps.setDouble(3, spawn.getY()); ps.setDouble(4, spawn.getZ()); ps.setFloat(5, spawn.getYaw()); ps.setFloat(6, spawn.getPitch()); } else { ps.setNull(1, Types.VARCHAR); ps.setNull(2, Types.REAL); ps.setNull(3, Types.REAL); ps.setNull(4, Types.REAL); ps.setNull(5, Types.REAL); ps.setNull(6, Types.REAL); } ps.setString(7, kingdom.getType().name()); ps.executeUpdate(); ps = conn.prepareStatement("DELETE FROM kingdom_members WHERE kingdom = ?"); ps.setString(1, kingdom.getType().name()); ps.executeUpdate(); for (UUID member : kingdom.getMembers()) { ps = conn.prepareStatement( "INSERT INTO kingdom_members (kingdom, player_uuid, is_king, is_general) VALUES (?, ?, ?, ?)"); ps.setString(1, kingdom.getType().name()); ps.setString(2, member.toString()); ps.setInt(3, kingdom.isKing(member) ? 1 : 0); ps.setInt(4, kingdom.isGeneral(member) ? 1 : 0); ps.executeUpdate(); } } catch (SQLException e) { e.printStackTrace(); } }); } @Override public CompletableFuture saveAlliance(KingdomType kingdom1, KingdomType kingdom2) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "INSERT OR IGNORE INTO alliances (kingdom1, kingdom2) VALUES (?, ?)"); ps.setString(1, kingdom1.name()); ps.setString(2, kingdom2.name()); ps.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } }); } @Override public CompletableFuture removeAlliance(KingdomType kingdom1, KingdomType kingdom2) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "DELETE FROM alliances WHERE (kingdom1 = ? AND kingdom2 = ?) OR (kingdom1 = ? AND kingdom2 = ?)"); ps.setString(1, kingdom1.name()); ps.setString(2, kingdom2.name()); ps.setString(3, kingdom2.name()); ps.setString(4, kingdom1.name()); ps.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } }); } @Override public CompletableFuture saveElimination(UUID uuid, String reason) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement( "INSERT OR REPLACE INTO eliminations (uuid, reason, timestamp) VALUES (?, ?, ?)"); ps.setString(1, uuid.toString()); ps.setString(2, reason); ps.setLong(3, System.currentTimeMillis()); ps.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } }); } @Override public CompletableFuture removePlayerBan(UUID uuid) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection()) { PreparedStatement ps = conn.prepareStatement("DELETE FROM eliminations WHERE uuid = ?"); ps.setString(1, uuid.toString()); ps.executeUpdate(); } catch (SQLException e) { e.printStackTrace(); } }); } }