package net.mineflare.link.storage; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.reflect.TypeToken; import net.mineflare.link.MinecraftDiscordLink; import net.mineflare.link.model.LinkedAccount; import net.mineflare.link.model.PendingCode; import org.bukkit.Bukkit; import java.io.File; import java.io.FileReader; import java.io.FileWriter; import java.io.IOException; import java.lang.reflect.Type; import java.util.*; import java.util.concurrent.ConcurrentHashMap; public class JsonStorage { private final MinecraftDiscordLink plugin; private final Gson gson; private final File dataFolder; private final File pendingCodesFile; private final File linkedAccountsFile; private final Map pendingCodes; private final Map linkedAccounts; private final Map discordToMinecraft; private final Map nameToAccount; public JsonStorage(MinecraftDiscordLink plugin) { this.plugin = plugin; this.gson = new GsonBuilder().setPrettyPrinting().create(); this.dataFolder = new File(plugin.getDataFolder(), "data"); this.pendingCodesFile = new File(dataFolder, "pending-codes.json"); this.linkedAccountsFile = new File(dataFolder, "linked-accounts.json"); this.pendingCodes = new ConcurrentHashMap<>(); this.linkedAccounts = new ConcurrentHashMap<>(); this.discordToMinecraft = new ConcurrentHashMap<>(); this.nameToAccount = new ConcurrentHashMap<>(); } public void load() { if (!dataFolder.exists()) { dataFolder.mkdirs(); } loadPendingCodes(); loadLinkedAccounts(); plugin.getLogger().info("Loaded " + pendingCodes.size() + " pending codes and " + linkedAccounts.size() + " linked accounts"); } private void loadPendingCodes() { if (!pendingCodesFile.exists()) { return; } try (FileReader reader = new FileReader(pendingCodesFile)) { Type type = new TypeToken>(){}.getType(); Map loaded = gson.fromJson(reader, type); if (loaded != null) { pendingCodes.putAll(loaded); } } catch (IOException e) { plugin.getLogger().severe("Failed to load pending codes: " + e.getMessage()); e.printStackTrace(); } } private void loadLinkedAccounts() { if (!linkedAccountsFile.exists()) { return; } try (FileReader reader = new FileReader(linkedAccountsFile)) { Type type = new TypeToken>(){}.getType(); Map loaded = gson.fromJson(reader, type); if (loaded != null) { linkedAccounts.putAll(loaded); for (LinkedAccount account : loaded.values()) { discordToMinecraft.put(account.getDiscordId(), account.getMinecraftUuid()); nameToAccount.put(account.getMinecraftName().toLowerCase(), account); } } } catch (IOException e) { plugin.getLogger().severe("Failed to load linked accounts: " + e.getMessage()); e.printStackTrace(); } } public void savePendingCodes() { try (FileWriter writer = new FileWriter(pendingCodesFile)) { gson.toJson(pendingCodes, writer); } catch (IOException e) { plugin.getLogger().severe("Failed to save pending codes: " + e.getMessage()); e.printStackTrace(); } } public void saveLinkedAccounts() { try (FileWriter writer = new FileWriter(linkedAccountsFile)) { gson.toJson(linkedAccounts, writer); } catch (IOException e) { plugin.getLogger().severe("Failed to save linked accounts: " + e.getMessage()); e.printStackTrace(); } } public void savePendingCode(PendingCode pendingCode) { pendingCodes.put(pendingCode.getCode(), pendingCode); savePendingCodes(); } public Optional getPendingCode(String code) { return Optional.ofNullable(pendingCodes.get(code)); } public void deletePendingCode(String code) { pendingCodes.remove(code); savePendingCodes(); } public void deletePendingCodesByUUID(UUID uuid) { pendingCodes.entrySet().removeIf(entry -> entry.getValue().getMinecraftUuid().equals(uuid)); savePendingCodes(); } public void deletePendingCodesByName(String name) { pendingCodes.entrySet().removeIf(entry -> entry.getValue().getMinecraftName().equalsIgnoreCase(name)); savePendingCodes(); } public void deletePendingCodesByIdentifier(String identifier) { if (Bukkit.getServer().getOnlineMode()) { try { UUID uuid = UUID.fromString(identifier); deletePendingCodesByUUID(uuid); } catch (IllegalArgumentException e) { deletePendingCodesByName(identifier); } } else { deletePendingCodesByName(identifier); } } public void cleanExpiredCodes() { long now = System.currentTimeMillis(); int removed = 0; Iterator> iterator = pendingCodes.entrySet().iterator(); while (iterator.hasNext()) { Map.Entry entry = iterator.next(); if (entry.getValue().getExpiresAt() < now) { iterator.remove(); removed++; } } if (removed > 0) { savePendingCodes(); plugin.getLogger().info("Cleaned " + removed + " expired codes"); } } public void saveLinkedAccount(LinkedAccount account) { linkedAccounts.put(account.getMinecraftUuid(), account); discordToMinecraft.put(account.getDiscordId(), account.getMinecraftUuid()); nameToAccount.put(account.getMinecraftName().toLowerCase(), account); saveLinkedAccounts(); } public Optional getLinkedAccountByUUID(UUID uuid) { return Optional.ofNullable(linkedAccounts.get(uuid)); } public Optional getLinkedAccountByName(String name) { return Optional.ofNullable(nameToAccount.get(name.toLowerCase())); } public Optional getLinkedAccountByDiscordId(String discordId) { UUID uuid = discordToMinecraft.get(discordId); if (uuid == null) { return Optional.empty(); } return Optional.ofNullable(linkedAccounts.get(uuid)); } public void deleteLinkedAccount(UUID uuid) { LinkedAccount account = linkedAccounts.remove(uuid); if (account != null) { discordToMinecraft.remove(account.getDiscordId()); nameToAccount.remove(account.getMinecraftName().toLowerCase()); saveLinkedAccounts(); } } public void deleteLinkedAccountByName(String name) { LinkedAccount account = nameToAccount.remove(name.toLowerCase()); if (account != null) { linkedAccounts.remove(account.getMinecraftUuid()); discordToMinecraft.remove(account.getDiscordId()); saveLinkedAccounts(); } } }