package io.github.papercore.database; import com.zaxxer.hikari.HikariConfig; import com.zaxxer.hikari.HikariDataSource; import io.github.papercore.PaperCore; import io.github.papercore.models.*; import io.github.papercore.util.LocationUtil; import org.bukkit.Location; import java.io.File; import java.sql.*; import java.time.Instant; import java.util.ArrayList; import java.util.List; import java.util.UUID; import java.util.concurrent.CompletableFuture; public class SQLiteDatabase implements Database { private final PaperCore plugin; private HikariDataSource dataSource; public SQLiteDatabase(PaperCore plugin) { this.plugin = plugin; } @Override public void initialize() { File dataFolder = plugin.getDataFolder(); if (!dataFolder.exists()) { dataFolder.mkdirs(); } File dbFile = new File(dataFolder, plugin.getConfigManager().getSQLiteFile()); HikariConfig config = new HikariConfig(); config.setJdbcUrl("jdbc:sqlite:" + dbFile.getAbsolutePath()); config.setMaximumPoolSize(10); config.setConnectionTestQuery("SELECT 1"); dataSource = new HikariDataSource(config); createTables(); } private void createTables() { try (Connection conn = dataSource.getConnection(); Statement stmt = conn.createStatement()) { stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS temp_bans (" + "player_id TEXT PRIMARY KEY," + "reason TEXT NOT NULL," + "banner TEXT NOT NULL," + "banned_at INTEGER NOT NULL," + "expires_at INTEGER NOT NULL)" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS mutes (" + "player_id TEXT PRIMARY KEY," + "reason TEXT NOT NULL," + "muter TEXT NOT NULL," + "muted_at INTEGER NOT NULL," + "expires_at INTEGER NOT NULL)" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS homes (" + "player_id TEXT NOT NULL," + "name TEXT NOT NULL," + "location TEXT NOT NULL," + "PRIMARY KEY (player_id, name))" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS warps (" + "name TEXT PRIMARY KEY," + "location TEXT NOT NULL)" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS spawn (" + "id INTEGER PRIMARY KEY," + "location TEXT NOT NULL)" ); stmt.executeUpdate( "CREATE TABLE IF NOT EXISTS moderation_logs (" + "id INTEGER PRIMARY KEY AUTOINCREMENT," + "type TEXT NOT NULL," + "target TEXT NOT NULL," + "moderator TEXT NOT NULL," + "reason TEXT," + "timestamp INTEGER NOT NULL)" ); } catch (SQLException e) { throw new RuntimeException("Failed to create tables", e); } } @Override public void close() { if (dataSource != null && !dataSource.isClosed()) { dataSource.close(); } } @Override public CompletableFuture saveTempBan(TempBan tempBan) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "REPLACE INTO temp_bans (player_id, reason, banner, banned_at, expires_at) VALUES (?, ?, ?, ?, ?)")) { stmt.setString(1, tempBan.getPlayerId().toString()); stmt.setString(2, tempBan.getReason()); stmt.setString(3, tempBan.getBanner()); stmt.setLong(4, tempBan.getBannedAt().getEpochSecond()); stmt.setLong(5, tempBan.getExpiresAt().getEpochSecond()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save temp ban", e); } }); } @Override public CompletableFuture getTempBan(UUID playerId) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT * FROM temp_bans WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return TempBan.builder() .playerId(UUID.fromString(rs.getString("player_id"))) .reason(rs.getString("reason")) .banner(rs.getString("banner")) .bannedAt(Instant.ofEpochSecond(rs.getLong("banned_at"))) .expiresAt(Instant.ofEpochSecond(rs.getLong("expires_at"))) .build(); } return null; } catch (SQLException e) { throw new RuntimeException("Failed to get temp ban", e); } }); } @Override public CompletableFuture removeTempBan(UUID playerId) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "DELETE FROM temp_bans WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to remove temp ban", e); } }); } @Override public CompletableFuture> getActiveTempBans() { return CompletableFuture.supplyAsync(() -> { List bans = new ArrayList<>(); try (Connection conn = dataSource.getConnection(); Statement stmt = conn.createStatement(); ResultSet rs = stmt.executeQuery("SELECT * FROM temp_bans")) { while (rs.next()) { bans.add(TempBan.builder() .playerId(UUID.fromString(rs.getString("player_id"))) .reason(rs.getString("reason")) .banner(rs.getString("banner")) .bannedAt(Instant.ofEpochSecond(rs.getLong("banned_at"))) .expiresAt(Instant.ofEpochSecond(rs.getLong("expires_at"))) .build()); } } catch (SQLException e) { throw new RuntimeException("Failed to get active temp bans", e); } return bans; }); } @Override public CompletableFuture saveMute(Mute mute) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "REPLACE INTO mutes (player_id, reason, muter, muted_at, expires_at) VALUES (?, ?, ?, ?, ?)")) { stmt.setString(1, mute.getPlayerId().toString()); stmt.setString(2, mute.getReason()); stmt.setString(3, mute.getMuter()); stmt.setLong(4, mute.getMutedAt().getEpochSecond()); stmt.setLong(5, mute.getExpiresAt().getEpochSecond()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save mute", e); } }); } @Override public CompletableFuture getMute(UUID playerId) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT * FROM mutes WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return Mute.builder() .playerId(UUID.fromString(rs.getString("player_id"))) .reason(rs.getString("reason")) .muter(rs.getString("muter")) .mutedAt(Instant.ofEpochSecond(rs.getLong("muted_at"))) .expiresAt(Instant.ofEpochSecond(rs.getLong("expires_at"))) .build(); } return null; } catch (SQLException e) { throw new RuntimeException("Failed to get mute", e); } }); } @Override public CompletableFuture removeMute(UUID playerId) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "DELETE FROM mutes WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to remove mute", e); } }); } @Override public CompletableFuture saveHome(Home home) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "REPLACE INTO homes (player_id, name, location) VALUES (?, ?, ?)")) { stmt.setString(1, home.getPlayerId().toString()); stmt.setString(2, home.getName()); stmt.setString(3, LocationUtil.locationToString(home.getLocation())); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save home", e); } }); } @Override public CompletableFuture getHome(UUID playerId, String name) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT * FROM homes WHERE player_id = ? AND name = ?")) { stmt.setString(1, playerId.toString()); stmt.setString(2, name); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return Home.builder() .playerId(UUID.fromString(rs.getString("player_id"))) .name(rs.getString("name")) .location(LocationUtil.stringToLocation(rs.getString("location"))) .build(); } return null; } catch (SQLException e) { throw new RuntimeException("Failed to get home", e); } }); } @Override public CompletableFuture> getHomes(UUID playerId) { return CompletableFuture.supplyAsync(() -> { List homes = new ArrayList<>(); try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT * FROM homes WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); ResultSet rs = stmt.executeQuery(); while (rs.next()) { homes.add(Home.builder() .playerId(UUID.fromString(rs.getString("player_id"))) .name(rs.getString("name")) .location(LocationUtil.stringToLocation(rs.getString("location"))) .build()); } } catch (SQLException e) { throw new RuntimeException("Failed to get homes", e); } return homes; }); } @Override public CompletableFuture deleteHome(UUID playerId, String name) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "DELETE FROM homes WHERE player_id = ? AND name = ?")) { stmt.setString(1, playerId.toString()); stmt.setString(2, name); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to delete home", e); } }); } @Override public CompletableFuture getHomeCount(UUID playerId) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT COUNT(*) FROM homes WHERE player_id = ?")) { stmt.setString(1, playerId.toString()); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return rs.getInt(1); } return 0; } catch (SQLException e) { throw new RuntimeException("Failed to get home count", e); } }); } @Override public CompletableFuture saveWarp(Warp warp) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "REPLACE INTO warps (name, location) VALUES (?, ?)")) { stmt.setString(1, warp.getName()); stmt.setString(2, LocationUtil.locationToString(warp.getLocation())); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save warp", e); } }); } @Override public CompletableFuture getWarp(String name) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT * FROM warps WHERE name = ?")) { stmt.setString(1, name); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return Warp.builder() .name(rs.getString("name")) .location(LocationUtil.stringToLocation(rs.getString("location"))) .build(); } return null; } catch (SQLException e) { throw new RuntimeException("Failed to get warp", e); } }); } @Override public CompletableFuture> getWarps() { return CompletableFuture.supplyAsync(() -> { List warps = new ArrayList<>(); try (Connection conn = dataSource.getConnection(); Statement stmt = conn.createStatement(); ResultSet rs = stmt.executeQuery("SELECT * FROM warps")) { while (rs.next()) { warps.add(Warp.builder() .name(rs.getString("name")) .location(LocationUtil.stringToLocation(rs.getString("location"))) .build()); } } catch (SQLException e) { throw new RuntimeException("Failed to get warps", e); } return warps; }); } @Override public CompletableFuture deleteWarp(String name) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "DELETE FROM warps WHERE name = ?")) { stmt.setString(1, name); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to delete warp", e); } }); } @Override public CompletableFuture saveSpawn(Location location) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "REPLACE INTO spawn (id, location) VALUES (1, ?)")) { stmt.setString(1, LocationUtil.locationToString(location)); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save spawn", e); } }); } @Override public CompletableFuture getSpawn() { return CompletableFuture.supplyAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "SELECT location FROM spawn WHERE id = 1")) { ResultSet rs = stmt.executeQuery(); if (rs.next()) { return LocationUtil.stringToLocation(rs.getString("location")); } return null; } catch (SQLException e) { throw new RuntimeException("Failed to get spawn", e); } }); } @Override public CompletableFuture logModeration(ModerationLog log) { return CompletableFuture.runAsync(() -> { try (Connection conn = dataSource.getConnection(); PreparedStatement stmt = conn.prepareStatement( "INSERT INTO moderation_logs (type, target, moderator, reason, timestamp) VALUES (?, ?, ?, ?, ?)")) { stmt.setString(1, log.getType()); stmt.setString(2, log.getTarget()); stmt.setString(3, log.getModerator()); stmt.setString(4, log.getReason()); stmt.setLong(5, log.getTimestamp().getEpochSecond()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to log moderation action", e); } }); } }