package com.leaderrank.database; import com.leaderrank.model.ChampionHistory; import com.leaderrank.model.ChampionStats; import com.leaderrank.model.TitleData; import java.sql.*; import java.util.*; import java.util.concurrent.CompletableFuture; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; public class Database { private final String path; private Connection connection; private final ExecutorService executor; public Database(String path) { this.path = path; this.executor = Executors.newSingleThreadExecutor(); } public CompletableFuture connect() { return CompletableFuture.runAsync(() -> { try { Class.forName("org.sqlite.JDBC"); connection = DriverManager.getConnection("jdbc:sqlite:" + path); createTables(); } catch (Exception e) { throw new RuntimeException("Failed to connect to database", e); } }, executor); } public CompletableFuture disconnect() { return CompletableFuture.runAsync(() -> { try { if (connection != null && !connection.isClosed()) { connection.close(); } executor.shutdown(); } catch (SQLException e) { throw new RuntimeException("Failed to disconnect from database", e); } }); } private void createTables() throws SQLException { String playerTitles = """ CREATE TABLE IF NOT EXISTS player_titles ( uuid TEXT PRIMARY KEY, active_title TEXT, owned_titles TEXT ) """; String championHistory = """ CREATE TABLE IF NOT EXISTS champion_history ( id INTEGER PRIMARY KEY AUTOINCREMENT, player_uuid TEXT NOT NULL, player_name TEXT NOT NULL, title TEXT NOT NULL, leaderboard_id TEXT NOT NULL, obtained_at INTEGER NOT NULL, lost_at INTEGER, duration_held INTEGER DEFAULT 0 ) """; String championStats = """ CREATE TABLE IF NOT EXISTS champion_stats ( player_uuid TEXT PRIMARY KEY, times_champion INTEGER DEFAULT 0, longest_streak INTEGER DEFAULT 0, total_time_held INTEGER DEFAULT 0, current_streak INTEGER DEFAULT 0 ) """; String currentChampions = """ CREATE TABLE IF NOT EXISTS current_champions ( leaderboard_id TEXT PRIMARY KEY, player_uuid TEXT, player_name TEXT, title TEXT, since INTEGER ) """; try (Statement stmt = connection.createStatement()) { stmt.execute(playerTitles); stmt.execute(championHistory); stmt.execute(championStats); stmt.execute(currentChampions); } } public CompletableFuture loadPlayerData(UUID uuid) { return CompletableFuture.supplyAsync(() -> { String sql = "SELECT active_title, owned_titles FROM player_titles WHERE uuid = ?"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, uuid.toString()); ResultSet rs = stmt.executeQuery(); TitleData data = new TitleData(uuid); if (rs.next()) { String activeTitle = rs.getString("active_title"); String ownedTitles = rs.getString("owned_titles"); data.setActiveTitle(activeTitle); if (ownedTitles != null && !ownedTitles.isEmpty()) { String[] titles = ownedTitles.split(","); for (String title : titles) { data.addTitle(title); } } } return data; } catch (SQLException e) { throw new RuntimeException("Failed to load player data", e); } }, executor); } public CompletableFuture savePlayerData(TitleData data) { return CompletableFuture.runAsync(() -> { String sql = "REPLACE INTO player_titles (uuid, active_title, owned_titles) VALUES (?, ?, ?)"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, data.getPlayerUuid().toString()); stmt.setString(2, data.getActiveTitle()); stmt.setString(3, String.join(",", data.getOwnedTitles())); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to save player data", e); } }, executor); } public CompletableFuture> getCurrentChampions() { return CompletableFuture.supplyAsync(() -> { Map champions = new HashMap<>(); String sql = "SELECT leaderboard_id, player_uuid FROM current_champions"; try (Statement stmt = connection.createStatement(); ResultSet rs = stmt.executeQuery(sql)) { while (rs.next()) { String leaderboardId = rs.getString("leaderboard_id"); String uuidStr = rs.getString("player_uuid"); if (uuidStr != null) { champions.put(leaderboardId, UUID.fromString(uuidStr)); } } } catch (SQLException e) { throw new RuntimeException("Failed to load current champions", e); } return champions; }, executor); } public CompletableFuture setCurrentChampion(String leaderboardId, UUID playerUuid, String playerName, String title) { return CompletableFuture.runAsync(() -> { String sql = "REPLACE INTO current_champions (leaderboard_id, player_uuid, player_name, title, since) VALUES (?, ?, ?, ?, ?)"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, leaderboardId); stmt.setString(2, playerUuid.toString()); stmt.setString(3, playerName); stmt.setString(4, title); stmt.setLong(5, System.currentTimeMillis()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to set current champion", e); } }, executor); } public CompletableFuture addChampionHistory(UUID playerUuid, String playerName, String title, String leaderboardId) { return CompletableFuture.runAsync(() -> { String sql = "INSERT INTO champion_history (player_uuid, player_name, title, leaderboard_id, obtained_at) VALUES (?, ?, ?, ?, ?)"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, playerUuid.toString()); stmt.setString(2, playerName); stmt.setString(3, title); stmt.setString(4, leaderboardId); stmt.setLong(5, System.currentTimeMillis()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to add champion history", e); } }, executor); } public CompletableFuture endChampionHistory(UUID playerUuid, String leaderboardId) { return CompletableFuture.runAsync(() -> { String sql = "UPDATE champion_history SET lost_at = ?, duration_held = ? - obtained_at WHERE player_uuid = ? AND leaderboard_id = ? AND lost_at IS NULL"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { long now = System.currentTimeMillis(); stmt.setLong(1, now); stmt.setLong(2, now); stmt.setString(3, playerUuid.toString()); stmt.setString(4, leaderboardId); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to end champion history", e); } }, executor); } public CompletableFuture> getPlayerHistory(UUID playerUuid) { return CompletableFuture.supplyAsync(() -> { List history = new ArrayList<>(); String sql = "SELECT * FROM champion_history WHERE player_uuid = ? ORDER BY obtained_at DESC"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, playerUuid.toString()); ResultSet rs = stmt.executeQuery(); while (rs.next()) { history.add(new ChampionHistory( rs.getInt("id"), UUID.fromString(rs.getString("player_uuid")), rs.getString("player_name"), rs.getString("title"), rs.getString("leaderboard_id"), rs.getLong("obtained_at"), rs.getObject("lost_at") != null ? rs.getLong("lost_at") : null, rs.getLong("duration_held") )); } } catch (SQLException e) { throw new RuntimeException("Failed to get player history", e); } return history; }, executor); } public CompletableFuture getPlayerStats(UUID playerUuid) { return CompletableFuture.supplyAsync(() -> { String sql = "SELECT * FROM champion_stats WHERE player_uuid = ?"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, playerUuid.toString()); ResultSet rs = stmt.executeQuery(); if (rs.next()) { return new ChampionStats( playerUuid, rs.getInt("times_champion"), rs.getLong("longest_streak"), rs.getLong("total_time_held"), rs.getLong("current_streak") ); } } catch (SQLException e) { throw new RuntimeException("Failed to get player stats", e); } return new ChampionStats(playerUuid, 0, 0, 0, 0); }, executor); } public CompletableFuture updatePlayerStats(ChampionStats stats) { return CompletableFuture.runAsync(() -> { String sql = "REPLACE INTO champion_stats (player_uuid, times_champion, longest_streak, total_time_held, current_streak) VALUES (?, ?, ?, ?, ?)"; try (PreparedStatement stmt = connection.prepareStatement(sql)) { stmt.setString(1, stats.getPlayerUuid().toString()); stmt.setInt(2, stats.getTimesChampion()); stmt.setLong(3, stats.getLongestStreak()); stmt.setLong(4, stats.getTotalTimeHeld()); stmt.setLong(5, stats.getCurrentStreak()); stmt.executeUpdate(); } catch (SQLException e) { throw new RuntimeException("Failed to update player stats", e); } }, executor); } }