package com.enhanceddogs.managers; import com.enhanceddogs.EnhancedDogs; import com.enhanceddogs.data.DogData; import lombok.Getter; import org.bukkit.configuration.ConfigurationSection; import org.bukkit.configuration.file.FileConfiguration; import org.bukkit.configuration.file.YamlConfiguration; import org.bukkit.entity.Wolf; import org.bukkit.inventory.ItemStack; import java.io.File; import java.io.IOException; import java.util.*; import java.util.concurrent.ConcurrentHashMap; @Getter public class DataManager { private final EnhancedDogs plugin; private final Map dogDataMap; private final Map> ownerDogsMap; private final File dataFolder; public DataManager(EnhancedDogs plugin) { this.plugin = plugin; this.dogDataMap = new ConcurrentHashMap<>(); this.ownerDogsMap = new ConcurrentHashMap<>(); this.dataFolder = new File(plugin.getDataFolder(), "dogs"); if (!dataFolder.exists()) { dataFolder.mkdirs(); } } public void loadAllData() { File[] files = dataFolder.listFiles((dir, name) -> name.endsWith(".yml")); if (files == null) return; for (File file : files) { try { String fileName = file.getName().replace(".yml", ""); UUID dogUUID = UUID.fromString(fileName); DogData data = loadDogData(dogUUID); if (data != null) { dogDataMap.put(dogUUID, data); ownerDogsMap.computeIfAbsent(data.getOwnerUUID(), k -> new HashSet<>()).add(dogUUID); } } catch (Exception e) { plugin.getLogger().warning("Failed to load dog data from " + file.getName()); } } plugin.getLogger().info("Loaded " + dogDataMap.size() + " dogs"); } public void saveAllData() { for (DogData data : dogDataMap.values()) { saveDogData(data); } } public DogData getDogData(UUID dogUUID) { return dogDataMap.get(dogUUID); } public DogData getDogData(Wolf wolf) { return getDogData(wolf.getUniqueId()); } public void createDogData(Wolf wolf, UUID ownerUUID, String breed) { DogData data = new DogData(wolf.getUniqueId(), ownerUUID, breed); dogDataMap.put(wolf.getUniqueId(), data); ownerDogsMap.computeIfAbsent(ownerUUID, k -> new HashSet<>()).add(wolf.getUniqueId()); saveDogData(data); } public void removeDogData(UUID dogUUID) { DogData data = dogDataMap.remove(dogUUID); if (data != null) { Set ownerDogs = ownerDogsMap.get(data.getOwnerUUID()); if (ownerDogs != null) { ownerDogs.remove(dogUUID); } File file = new File(dataFolder, dogUUID.toString() + ".yml"); if (file.exists()) { file.delete(); } } } public Set getOwnerDogs(UUID ownerUUID) { return ownerDogsMap.getOrDefault(ownerUUID, new HashSet<>()); } public int getOwnerDogCount(UUID ownerUUID) { return getOwnerDogs(ownerUUID).size(); } public Collection getAllDogs() { return dogDataMap.values(); } private DogData loadDogData(UUID dogUUID) { File file = new File(dataFolder, dogUUID.toString() + ".yml"); if (!file.exists()) return null; FileConfiguration config = YamlConfiguration.loadConfiguration(file); UUID ownerUUID = UUID.fromString(config.getString("owner")); String breed = config.getString("breed"); DogData data = new DogData(dogUUID, ownerUUID, breed); data.setCustomName(config.getString("customName")); data.setLevel(config.getInt("level", 1)); data.setXp(config.getDouble("xp", 0)); data.setXpToNextLevel(config.getDouble("xpToNextLevel", 100)); data.setMaxHealth(config.getDouble("maxHealth", 20.0)); data.setDamage(config.getDouble("damage", 4.0)); data.setSpeed(config.getDouble("speed", 1.0)); data.setHunger(config.getInt("hunger", 100)); data.setHappiness(config.getInt("happiness", 100)); ConfigurationSection abilitiesSection = config.getConfigurationSection("abilities"); if (abilitiesSection != null) { for (String key : abilitiesSection.getKeys(false)) { data.getAbilityLevels().put(key, abilitiesSection.getInt(key)); } } data.setInventorySize(config.getInt("inventorySize", 9)); if (config.contains("inventory")) { List items = config.getList("inventory"); if (items != null) { for (Object obj : items) { if (obj instanceof ItemStack) { data.getInventory().add((ItemStack) obj); } } } } data.setTraits(config.getStringList("traits")); data.setSitting(config.getBoolean("sitting", false)); data.setGuarding(config.getBoolean("guarding", false)); if (config.contains("guardLocation")) { data.setGuardLocation(UUID.fromString(config.getString("guardLocation"))); } data.setLastFed(config.getLong("lastFed", System.currentTimeMillis())); data.setLastPetted(config.getLong("lastPetted", System.currentTimeMillis())); data.setLastBreed(config.getLong("lastBreed", 0)); data.setTotalKills(config.getInt("stats.totalKills", 0)); data.setPlayerKills(config.getInt("stats.playerKills", 0)); data.setBossKills(config.getInt("stats.bossKills", 0)); data.setItemsFetched(config.getInt("stats.itemsFetched", 0)); data.setTimePlayed(config.getLong("stats.timePlayed", 0)); return data; } private void saveDogData(DogData data) { File file = new File(dataFolder, data.getDogUUID().toString() + ".yml"); FileConfiguration config = new YamlConfiguration(); config.set("owner", data.getOwnerUUID().toString()); config.set("breed", data.getBreed()); config.set("customName", data.getCustomName()); config.set("level", data.getLevel()); config.set("xp", data.getXp()); config.set("xpToNextLevel", data.getXpToNextLevel()); config.set("maxHealth", data.getMaxHealth()); config.set("damage", data.getDamage()); config.set("speed", data.getSpeed()); config.set("hunger", data.getHunger()); config.set("happiness", data.getHappiness()); for (Map.Entry entry : data.getAbilityLevels().entrySet()) { config.set("abilities." + entry.getKey(), entry.getValue()); } config.set("inventorySize", data.getInventorySize()); config.set("inventory", data.getInventory()); config.set("traits", data.getTraits()); config.set("sitting", data.isSitting()); config.set("guarding", data.isGuarding()); if (data.getGuardLocation() != null) { config.set("guardLocation", data.getGuardLocation().toString()); } config.set("lastFed", data.getLastFed()); config.set("lastPetted", data.getLastPetted()); config.set("lastBreed", data.getLastBreed()); config.set("stats.totalKills", data.getTotalKills()); config.set("stats.playerKills", data.getPlayerKills()); config.set("stats.bossKills", data.getBossKills()); config.set("stats.itemsFetched", data.getItemsFetched()); config.set("stats.timePlayed", data.getTimePlayed()); try { config.save(file); } catch (IOException e) { plugin.getLogger().severe("Failed to save dog data for " + data.getDogUUID()); } } }