package com.sixkingdoms; import net.minecraft.nbt.NbtCompound; import net.minecraft.nbt.NbtList; import net.minecraft.nbt.NbtElement; import net.minecraft.registry.RegistryWrapper; import net.minecraft.server.MinecraftServer; import net.minecraft.server.world.ServerWorld; import net.minecraft.world.PersistentState; import net.minecraft.world.World; import java.util.*; public class KingdomData extends PersistentState { private final Map playerData = new HashMap<>(); private final Map> kingdomMembers = new EnumMap<>(Kingdom.class); private final Map> kingdomKings = new EnumMap<>(Kingdom.class); public KingdomData() { for (Kingdom kingdom : Kingdom.values()) { kingdomMembers.put(kingdom, new ArrayList<>()); kingdomKings.put(kingdom, new ArrayList<>()); } } public PlayerKingdomData getPlayerData(UUID playerId) { return playerData.computeIfAbsent(playerId, k -> new PlayerKingdomData()); } public void setPlayerKingdom(UUID playerId, Kingdom kingdom, KingdomRole role) { PlayerKingdomData data = getPlayerData(playerId); if (data.hasKingdom()) { removeFromKingdom(playerId); } data.setKingdom(kingdom); data.setRole(role); data.resetHearts(); kingdomMembers.get(kingdom).add(playerId); if (role.isKing()) { kingdomKings.get(kingdom).add(playerId); } markDirty(); } public void removeFromKingdom(UUID playerId) { PlayerKingdomData data = getPlayerData(playerId); if (data.hasKingdom()) { Kingdom kingdom = data.getKingdom(); kingdomMembers.get(kingdom).remove(playerId); kingdomKings.get(kingdom).remove(playerId); data.reset(); markDirty(); } } public List getKingdomMembers(Kingdom kingdom) { return new ArrayList<>(kingdomMembers.get(kingdom)); } public List getKingdomKings(Kingdom kingdom) { return new ArrayList<>(kingdomKings.get(kingdom)); } public int getKingCount(Kingdom kingdom) { return kingdomKings.get(kingdom).size(); } public boolean canAddKing(Kingdom kingdom) { return getKingCount(kingdom) < 2; } @Override public NbtCompound writeNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registries) { NbtList playerList = new NbtList(); for (Map.Entry entry : playerData.entrySet()) { PlayerKingdomData data = entry.getValue(); if (data.hasKingdom()) { NbtCompound playerNbt = new NbtCompound(); playerNbt.putUuid("UUID", entry.getKey()); playerNbt.putString("Kingdom", data.getKingdom().name()); playerNbt.putString("Role", data.getRole().name()); playerNbt.putInt("Hearts", data.getHearts()); playerList.add(playerNbt); } } nbt.put("Players", playerList); return nbt; } public static KingdomData createFromNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registries) { KingdomData state = new KingdomData(); NbtList playerList = nbt.getList("Players", NbtElement.COMPOUND_TYPE); for (int i = 0; i < playerList.size(); i++) { NbtCompound playerNbt = playerList.getCompound(i); UUID playerId = playerNbt.getUuid("UUID"); Kingdom kingdom = Kingdom.valueOf(playerNbt.getString("Kingdom")); KingdomRole role = KingdomRole.valueOf(playerNbt.getString("Role")); int hearts = playerNbt.getInt("Hearts"); PlayerKingdomData data = new PlayerKingdomData(kingdom, role); data.setHearts(hearts); state.playerData.put(playerId, data); state.kingdomMembers.get(kingdom).add(playerId); if (role.isKing()) { state.kingdomKings.get(kingdom).add(playerId); } } return state; } private static final PersistentState.Type TYPE = new PersistentState.Type<>( KingdomData::new, KingdomData::createFromNbt, null); public static KingdomData getServerState(MinecraftServer server) { ServerWorld world = server.getWorld(World.OVERWORLD); return world.getPersistentStateManager().getOrCreate( TYPE, "sixkingdoms_data" ); } }