package com.example.advanceddiscordsync.manager; import com.example.advanceddiscordsync.AdvancedDiscordSync; import com.example.advanceddiscordsync.data.LinkCode; import com.example.advanceddiscordsync.data.PlayerLinkData; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.reflect.TypeToken; import java.io.*; import java.lang.reflect.Type; import java.util.*; import java.util.concurrent.ConcurrentHashMap; public class PlayerLinkManager { private final AdvancedDiscordSync plugin; private final Map playerToDiscord; private final Map discordToPlayer; private final Map pendingCodes; private final File dataFile; private final Gson gson; public PlayerLinkManager(AdvancedDiscordSync plugin) { this.plugin = plugin; this.playerToDiscord = new ConcurrentHashMap<>(); this.discordToPlayer = new ConcurrentHashMap<>(); this.pendingCodes = new ConcurrentHashMap<>(); this.dataFile = new File(plugin.getDataFolder(), "links.json"); this.gson = new GsonBuilder().setPrettyPrinting().create(); loadLinks(); startCleanupTask(); } public void loadLinks() { if (!dataFile.exists()) { return; } try (Reader reader = new FileReader(dataFile)) { Type type = new TypeToken>(){}.getType(); List links = gson.fromJson(reader, type); if (links != null) { playerToDiscord.clear(); discordToPlayer.clear(); for (PlayerLinkData link : links) { UUID uuid = UUID.fromString(link.getMinecraftUuid()); playerToDiscord.put(uuid, link.getDiscordId()); discordToPlayer.put(link.getDiscordId(), uuid); } } } catch (IOException e) { plugin.getLogger().severe("Failed to load player links: " + e.getMessage()); } } public void saveLinks() { try { if (!plugin.getDataFolder().exists()) { plugin.getDataFolder().mkdirs(); } List links = new ArrayList<>(); for (Map.Entry entry : playerToDiscord.entrySet()) { PlayerLinkData data = new PlayerLinkData(); data.setMinecraftUuid(entry.getKey().toString()); data.setDiscordId(entry.getValue()); data.setLinkedAt(System.currentTimeMillis()); links.add(data); } try (Writer writer = new FileWriter(dataFile)) { gson.toJson(links, writer); } } catch (IOException e) { plugin.getLogger().severe("Failed to save player links: " + e.getMessage()); } } public String generateCode(UUID playerUuid) { String code = generateRandomCode(); long expiry = System.currentTimeMillis() + (plugin.getConfigManager().getCodeExpiry() * 1000L); pendingCodes.put(code, new LinkCode(playerUuid, expiry)); return code; } public boolean verifyCode(String code, String discordId) { LinkCode linkCode = pendingCodes.get(code); if (linkCode == null) { return false; } if (System.currentTimeMillis() > linkCode.getExpiry()) { pendingCodes.remove(code); return false; } UUID playerUuid = linkCode.getPlayerUuid(); if (isLinked(playerUuid)) { unlink(playerUuid); } if (isDiscordLinked(discordId)) { unlinkDiscord(discordId); } playerToDiscord.put(playerUuid, discordId); discordToPlayer.put(discordId, playerUuid); pendingCodes.remove(code); saveLinks(); return true; } public boolean isLinked(UUID playerUuid) { return playerToDiscord.containsKey(playerUuid); } public boolean isDiscordLinked(String discordId) { return discordToPlayer.containsKey(discordId); } public String getDiscordId(UUID playerUuid) { return playerToDiscord.get(playerUuid); } public UUID getPlayerUuid(String discordId) { return discordToPlayer.get(discordId); } public void unlink(UUID playerUuid) { String discordId = playerToDiscord.remove(playerUuid); if (discordId != null) { discordToPlayer.remove(discordId); saveLinks(); } } public void unlinkDiscord(String discordId) { UUID playerUuid = discordToPlayer.remove(discordId); if (playerUuid != null) { playerToDiscord.remove(playerUuid); saveLinks(); } } public Set getAllLinkedPlayers() { return new HashSet<>(playerToDiscord.keySet()); } public Set getAllLinkedDiscordIds() { return new HashSet<>(discordToPlayer.keySet()); } private String generateRandomCode() { int length = plugin.getConfigManager().getCodeLength(); String chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"; Random random = new Random(); StringBuilder code = new StringBuilder(); for (int i = 0; i < length; i++) { code.append(chars.charAt(random.nextInt(chars.length()))); } return code.toString(); } private void startCleanupTask() { plugin.getServer().getScheduler().runTaskTimerAsynchronously(plugin, () -> { long now = System.currentTimeMillis(); pendingCodes.entrySet().removeIf(entry -> now > entry.getValue().getExpiry()); }, 1200L, 1200L); } }