package com.vertexreplay.storage; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.vertexreplay.VertexReplayPlugin; import com.vertexreplay.core.RecordingSession; import com.vertexreplay.core.ReplayMetadata; import com.vertexreplay.util.FileUtil; import java.io.*; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.TimeUnit; public class StorageManager { private final VertexReplayPlugin plugin; private final Gson gson; private final File replaysDir; private final File flaggedDir; private final File metadataDir; private final Map> metadataCache; private final ExecutorService executor; public StorageManager(VertexReplayPlugin plugin) { this.plugin = plugin; this.gson = new GsonBuilder().setPrettyPrinting().create(); this.metadataCache = new ConcurrentHashMap<>(); this.executor = Executors.newFixedThreadPool(plugin.getConfigManager().getThreadPoolSize()); String basePath = plugin.getConfig().getString("storage.base-path", "plugins/VertexReplay/replays"); String flaggedPath = plugin.getConfig().getString("storage.flagged-path", "plugins/VertexReplay/replays/staff/flagged"); this.replaysDir = new File(basePath); this.flaggedDir = new File(flaggedPath); this.metadataDir = new File(plugin.getDataFolder(), "metadata"); FileUtil.ensureDirectory(replaysDir); FileUtil.ensureDirectory(flaggedDir); FileUtil.ensureDirectory(metadataDir); loadAllMetadata(); } public void saveReplay(RecordingSession session) { executor.execute(() -> { try { ReplayMetadata metadata = createMetadata(session); File targetDir = session.getReason() != null ? flaggedDir : replaysDir; File playerDir = new File(targetDir, session.getPlayerUuid().toString()); FileUtil.ensureDirectory(playerDir); File replayFile = new File(playerDir, metadata.getReplayId().toString() + ".replay"); byte[] replayBytes = serializeReplay(session); byte[] encryptedBytes = plugin.getSecurityManager().encrypt(replayBytes); try (FileOutputStream fos = new FileOutputStream(replayFile)) { fos.write(encryptedBytes); } metadata.setFilePath(replayFile.getAbsolutePath()); metadata.setFileSize(replayFile.length()); metadata.setChecksum(plugin.getSecurityManager().calculateChecksum(encryptedBytes)); saveMetadata(metadata); plugin.getLogger().info("Saved replay: " + metadata.getReplayId() + " for " + session.getPlayerName() + " (" + metadata.getFileSizeMB() + "MB)"); } catch (Exception e) { plugin.getLogger().severe("Failed to save replay: " + e.getMessage()); e.printStackTrace(); } }); } public void autoSaveReplay(RecordingSession session) { saveReplay(session); } private ReplayMetadata createMetadata(RecordingSession session) { ReplayMetadata metadata = new ReplayMetadata(); metadata.setReplayId(session.getSessionId()); metadata.setPlayerUuid(session.getPlayerUuid()); metadata.setPlayerName(session.getPlayerName()); metadata.setReplayName("Replay_" + System.currentTimeMillis()); metadata.setStartTime(session.getStartTime()); metadata.setEndTime(System.currentTimeMillis()); metadata.setDuration(session.getDuration()); metadata.setEncrypted(plugin.getConfigManager().isEncryptionEnabled()); metadata.setFlagged(session.getReason() != null); metadata.setFlagReason(session.getReason()); metadata.setPacketCount(session.getReplayData().getPacketCount()); metadata.setCreatedAt(System.currentTimeMillis()); metadata.setLocked(false); metadata.setFavorite(false); return metadata; } private byte[] serializeReplay(RecordingSession session) throws IOException { ByteArrayOutputStream baos = new ByteArrayOutputStream(); ObjectOutputStream oos = new ObjectOutputStream(baos); oos.writeObject(session.getReplayData()); oos.close(); return baos.toByteArray(); } private void saveMetadata(ReplayMetadata metadata) { UUID playerUuid = metadata.getPlayerUuid(); File metadataFile = new File(metadataDir, playerUuid.toString() + ".json"); List playerMetadata = metadataCache.computeIfAbsent(playerUuid, k -> new ArrayList<>()); playerMetadata.add(metadata); try (FileWriter writer = new FileWriter(metadataFile)) { gson.toJson(playerMetadata, writer); } catch (IOException e) { plugin.getLogger().severe("Failed to save metadata: " + e.getMessage()); } } private void loadAllMetadata() { File[] metadataFiles = metadataDir.listFiles((dir, name) -> name.endsWith(".json")); if (metadataFiles == null) return; for (File file : metadataFiles) { try (FileReader reader = new FileReader(file)) { ReplayMetadata[] metadata = gson.fromJson(reader, ReplayMetadata[].class); if (metadata != null && metadata.length > 0) { UUID playerUuid = metadata[0].getPlayerUuid(); metadataCache.put(playerUuid, new ArrayList<>(Arrays.asList(metadata))); } } catch (Exception e) { plugin.getLogger().warning("Failed to load metadata from " + file.getName()); } } plugin.getLogger().info("Loaded metadata for " + metadataCache.size() + " players"); } public List getPlayerReplays(UUID playerUuid, boolean includeFlagged) { List replays = metadataCache.getOrDefault(playerUuid, new ArrayList<>()); if (!includeFlagged) { replays = new ArrayList<>(replays); replays.removeIf(ReplayMetadata::isFlagged); } return replays; } public List getAllFlaggedReplays() { List flagged = new ArrayList<>(); for (List playerReplays : metadataCache.values()) { for (ReplayMetadata metadata : playerReplays) { if (metadata.isFlagged()) { flagged.add(metadata); } } } return flagged; } public int getReplayCount(UUID playerUuid) { return metadataCache.getOrDefault(playerUuid, new ArrayList<>()).size(); } public boolean deleteReplay(UUID replayId) { for (Map.Entry> entry : metadataCache.entrySet()) { Iterator iterator = entry.getValue().iterator(); while (iterator.hasNext()) { ReplayMetadata metadata = iterator.next(); if (metadata.getReplayId().equals(replayId)) { File replayFile = new File(metadata.getFilePath()); if (replayFile.exists()) { replayFile.delete(); } iterator.remove(); saveMetadataForPlayer(entry.getKey()); return true; } } } return false; } private void saveMetadataForPlayer(UUID playerUuid) { List metadata = metadataCache.get(playerUuid); if (metadata == null) return; File metadataFile = new File(metadataDir, playerUuid.toString() + ".json"); try (FileWriter writer = new FileWriter(metadataFile)) { gson.toJson(metadata, writer); } catch (IOException e) { plugin.getLogger().severe("Failed to save metadata: " + e.getMessage()); } } public void shutdown() { executor.shutdown(); try { if (!executor.awaitTermination(30, TimeUnit.SECONDS)) { executor.shutdownNow(); } } catch (InterruptedException e) { executor.shutdownNow(); } } }