package com.tptapi.database; import com.tptapi.TpTapi; import com.tptapi.model.Home; import com.zaxxer.hikari.HikariConfig; import com.zaxxer.hikari.HikariDataSource; import org.bukkit.Bukkit; import org.bukkit.Location; import org.bukkit.World; import java.io.File; import java.sql.*; import java.util.*; import java.util.concurrent.CompletableFuture; import java.util.logging.Level; public class DatabaseManager { private final TpTapi plugin; private HikariDataSource dataSource; private final String type; public DatabaseManager(TpTapi plugin) { this.plugin = plugin; this.type = plugin.getConfig().getString("database.type", "sqlite").toLowerCase(); } public void initialize() { try { HikariConfig hikari = new HikariConfig(); if (type.equals("mysql")) { var cfg = plugin.getConfig(); String host = cfg.getString("database.mysql.host", "localhost"); int port = cfg.getInt("database.mysql.port", 3306); String database = cfg.getString("database.mysql.database", "tptapi"); String username = cfg.getString("database.mysql.username", "root"); String password = cfg.getString("database.mysql.password", ""); int poolSize = cfg.getInt("database.mysql.pool-size", 10); hikari.setJdbcUrl("jdbc:mysql://" + host + ":" + port + "/" + database + "?useSSL=false&characterEncoding=utf8"); hikari.setUsername(username); hikari.setPassword(password); hikari.setMaximumPoolSize(poolSize); hikari.setConnectionTimeout(10000); hikari.setIdleTimeout(600000); hikari.setMaxLifetime(1800000); } else { File dbFile = new File(plugin.getDataFolder(), "data.db"); plugin.getDataFolder().mkdirs(); hikari.setJdbcUrl("jdbc:sqlite:" + dbFile.getAbsolutePath()); hikari.setMaximumPoolSize(1); hikari.setConnectionTestQuery("SELECT 1"); } hikari.setPoolName("TpTapi-DB"); dataSource = new HikariDataSource(hikari); createTables(); } catch (Exception e) { plugin.getLogger().log(Level.SEVERE, "Failed to initialize database", e); } } private void createTables() throws SQLException { try (Connection conn = dataSource.getConnection(); Statement stmt = conn.createStatement()) { stmt.execute(""" CREATE TABLE IF NOT EXISTS homes ( uuid TEXT NOT NULL, name TEXT NOT NULL, world TEXT NOT NULL, x REAL NOT NULL, y REAL NOT NULL, z REAL NOT NULL, yaw REAL NOT NULL, pitch REAL NOT NULL, PRIMARY KEY (uuid, name) ) """); stmt.execute(""" CREATE TABLE IF NOT EXISTS tpa_toggles ( uuid TEXT PRIMARY KEY, enabled INTEGER NOT NULL DEFAULT 1 ) """); stmt.execute(""" CREATE TABLE IF NOT EXISTS cooldowns ( uuid TEXT NOT NULL, type TEXT NOT NULL, expires_at INTEGER NOT NULL, PRIMARY KEY (uuid, type) ) """); } } private Connection getConnection() throws SQLException { return dataSource.getConnection(); } private String upsertHomes() { if (type.equals("mysql")) { return "INSERT INTO homes (uuid, name, world, x, y, z, yaw, pitch) VALUES (?, ?, ?, ?, ?, ?, ?, ?) " + "ON DUPLICATE KEY UPDATE world=VALUES(world), x=VALUES(x), y=VALUES(y), z=VALUES(z), yaw=VALUES(yaw), pitch=VALUES(pitch)"; } return "INSERT OR REPLACE INTO homes (uuid, name, world, x, y, z, yaw, pitch) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"; } private String upsertTpaToggle() { if (type.equals("mysql")) { return "INSERT INTO tpa_toggles (uuid, enabled) VALUES (?, ?) ON DUPLICATE KEY UPDATE enabled=VALUES(enabled)"; } return "INSERT OR REPLACE INTO tpa_toggles (uuid, enabled) VALUES (?, ?)"; } private String upsertCooldown() { if (type.equals("mysql")) { return "INSERT INTO cooldowns (uuid, type, expires_at) VALUES (?, ?, ?) ON DUPLICATE KEY UPDATE expires_at=VALUES(expires_at)"; } return "INSERT OR REPLACE INTO cooldowns (uuid, type, expires_at) VALUES (?, ?, ?)"; } public CompletableFuture> getHomes(UUID uuid) { return CompletableFuture.supplyAsync(() -> { List homes = new ArrayList<>(); try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT name, world, x, y, z, yaw, pitch FROM homes WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); while (rs.next()) { World world = Bukkit.getWorld(rs.getString("world")); if (world == null) continue; Location loc = new Location(world, rs.getDouble("x"), rs.getDouble("y"), rs.getDouble("z"), rs.getFloat("yaw"), rs.getFloat("pitch")); homes.add(new Home(rs.getString("name"), uuid, loc)); } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to get homes", e); } return homes; }); } public CompletableFuture saveHome(Home home) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement(upsertHomes())) { ps.setString(1, home.getOwnerUuid().toString()); ps.setString(2, home.getName()); ps.setString(3, home.getLocation().getWorld().getName()); ps.setDouble(4, home.getLocation().getX()); ps.setDouble(5, home.getLocation().getY()); ps.setDouble(6, home.getLocation().getZ()); ps.setFloat(7, home.getLocation().getYaw()); ps.setFloat(8, home.getLocation().getPitch()); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to save home", e); } }); } public CompletableFuture deleteHome(UUID uuid, String name) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "DELETE FROM homes WHERE uuid = ? AND name = ?")) { ps.setString(1, uuid.toString()); ps.setString(2, name); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to delete home", e); } }); } public CompletableFuture getTpaToggle(UUID uuid) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT enabled FROM tpa_toggles WHERE uuid = ?")) { ps.setString(1, uuid.toString()); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getInt("enabled") == 1; } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to get TPA toggle", e); } return true; }); } public CompletableFuture setTpaToggle(UUID uuid, boolean enabled) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement(upsertTpaToggle())) { ps.setString(1, uuid.toString()); ps.setInt(2, enabled ? 1 : 0); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to set TPA toggle", e); } }); } public CompletableFuture getCooldownExpiry(UUID uuid, String type) { return CompletableFuture.supplyAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement( "SELECT expires_at FROM cooldowns WHERE uuid = ? AND type = ?")) { ps.setString(1, uuid.toString()); ps.setString(2, type); ResultSet rs = ps.executeQuery(); if (rs.next()) { return rs.getLong("expires_at"); } } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to get cooldown", e); } return 0L; }); } public CompletableFuture setCooldown(UUID uuid, String type, long expiresAt) { return CompletableFuture.runAsync(() -> { try (Connection conn = getConnection(); PreparedStatement ps = conn.prepareStatement(upsertCooldown())) { ps.setString(1, uuid.toString()); ps.setString(2, type); ps.setLong(3, expiresAt); ps.executeUpdate(); } catch (SQLException e) { plugin.getLogger().log(Level.SEVERE, "Failed to set cooldown", e); } }); } public void close() { if (dataSource != null && !dataSource.isClosed()) { dataSource.close(); } } }