package com.modguard.manager; import com.google.gson.Gson; import com.google.gson.GsonBuilder; import com.google.gson.reflect.TypeToken; import com.modguard.ModGuard; import com.modguard.data.PlayerData; import com.modguard.data.Punishment; import com.modguard.data.PunishmentType; import org.bukkit.Bukkit; import org.bukkit.entity.Player; 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.stream.Collectors; public class PunishmentManager { private final ModGuard plugin; private final Gson gson; private final File punishmentsFile; private final File playersFile; private Map punishments; private Map players; public PunishmentManager(ModGuard plugin) { this.plugin = plugin; this.gson = new GsonBuilder().setPrettyPrinting().create(); File dataFolder = new File(this.plugin.getDataFolder(), "data"); if (!dataFolder.exists()) { dataFolder.mkdirs(); } this.punishmentsFile = new File(dataFolder, "punishments.json"); this.playersFile = new File(dataFolder, "players.json"); this.punishments = new HashMap<>(); this.players = new HashMap<>(); this.load(); } public void load() { this.loadPunishments(); this.loadPlayers(); } private void loadPunishments() { if (!this.punishmentsFile.exists()) { return; } try (FileReader reader = new FileReader(this.punishmentsFile)) { Type type = new TypeToken>(){}.getType(); Map loaded = this.gson.fromJson(reader, type); if (loaded != null) { this.punishments = loaded; } } catch (IOException e) { e.printStackTrace(); } } private void loadPlayers() { if (!this.playersFile.exists()) { return; } try (FileReader reader = new FileReader(this.playersFile)) { Type type = new TypeToken>(){}.getType(); Map loaded = this.gson.fromJson(reader, type); if (loaded != null) { this.players = loaded; } } catch (IOException e) { e.printStackTrace(); } } public void save() { this.savePunishments(); this.savePlayers(); } private void savePunishments() { try (FileWriter writer = new FileWriter(this.punishmentsFile)) { this.gson.toJson(this.punishments, writer); } catch (IOException e) { e.printStackTrace(); } } private void savePlayers() { try (FileWriter writer = new FileWriter(this.playersFile)) { this.gson.toJson(this.players, writer); } catch (IOException e) { e.printStackTrace(); } } public Punishment createPunishment(UUID player, PunishmentType type, String reason, long duration, UUID issuedBy, String issuedByName) { long expiresTime = duration > 0 ? System.currentTimeMillis() + duration : 0; Punishment punishment = new Punishment(player, type, reason, System.currentTimeMillis(), expiresTime, issuedBy, issuedByName); this.punishments.put(punishment.getId(), punishment); PlayerData data = this.getPlayerData(player); data.addPunishment(punishment.getId()); this.save(); return punishment; } public PlayerData getPlayerData(UUID player) { return this.players.computeIfAbsent(player, PlayerData::new); } public Punishment getActiveMute(UUID player) { return this.punishments.values().stream() .filter(p -> p.getPlayer().equals(player)) .filter(p -> p.getType() == PunishmentType.MUTE) .filter(Punishment::isActive) .filter(p -> !p.isExpired()) .findFirst() .orElse(null); } public Punishment getActiveBan(UUID player) { return this.punishments.values().stream() .filter(p -> p.getPlayer().equals(player)) .filter(p -> p.getType() == PunishmentType.BAN) .filter(Punishment::isActive) .filter(p -> !p.isExpired()) .findFirst() .orElse(null); } public Punishment getActiveBanByIp(String ip) { return this.punishments.values().stream() .filter(p -> p.getType() == PunishmentType.BAN) .filter(Punishment::isActive) .filter(p -> !p.isExpired()) .filter(p -> ip.equals(p.getIp())) .findFirst() .orElse(null); } public boolean isMuted(UUID player) { return this.getActiveMute(player) != null; } public boolean isBanned(UUID player) { return this.getActiveBan(player) != null; } public void removeMute(UUID player) { Punishment mute = this.getActiveMute(player); if (mute != null) { mute.setActive(false); this.save(); } } public void removeBan(UUID player) { Punishment ban = this.getActiveBan(player); if (ban != null) { ban.setActive(false); this.save(); } } public List getHistory(UUID player) { PlayerData data = this.getPlayerData(player); return data.getHistory().stream() .map(this.punishments::get) .filter(Objects::nonNull) .sorted(Comparator.comparingLong(Punishment::getIssuedTime).reversed()) .collect(Collectors.toList()); } public void checkExpired() { boolean changed = false; for (Punishment punishment : this.punishments.values()) { if (punishment.isActive() && punishment.isExpired()) { punishment.setActive(false); changed = true; Player player = Bukkit.getPlayer(punishment.getPlayer()); if (player != null) { if (punishment.getType() == PunishmentType.MUTE) { player.sendMessage(this.plugin.getConfigManager().getMessage("unmuted")); } } } } if (changed) { this.save(); } } public void checkWarningDecay() { long decayTime = this.plugin.getConfigManager().getWarningDecayHours() * 60 * 60 * 1000L; long now = System.currentTimeMillis(); boolean changed = false; for (PlayerData data : this.players.values()) { if (data.getWarnings() > 0 && data.getLastWarningTime() > 0) { if (now - data.getLastWarningTime() >= decayTime) { data.resetWarnings(); changed = true; } } } if (changed) { this.save(); } } public Punishment getPunishment(UUID id) { return this.punishments.get(id); } }