package com.saveinventory.data; import com.saveinventory.compat.VersionAdapter; import net.minecraft.item.ItemStack; import net.minecraft.nbt.NbtCompound; import net.minecraft.nbt.NbtElement; import net.minecraft.nbt.NbtList; import net.minecraft.registry.RegistryWrapper; import net.minecraft.server.network.ServerPlayerEntity; import java.util.ArrayList; import java.util.List; public class SavedInventory { private final List mainInventory; private final List armorInventory; private final ItemStack offhandItem; private final int experience; private final int experienceLevel; private final float experienceProgress; public SavedInventory(List mainInventory, List armorInventory, ItemStack offhandItem, int experience, int experienceLevel, float experienceProgress) { this.mainInventory = mainInventory; this.armorInventory = armorInventory; this.offhandItem = offhandItem; this.experience = experience; this.experienceLevel = experienceLevel; this.experienceProgress = experienceProgress; } public static SavedInventory fromPlayer(ServerPlayerEntity player) { List mainInv = new ArrayList<>(); for (int i = 0; i < player.getInventory().main.size(); i++) { mainInv.add(player.getInventory().main.get(i).copy()); } List armorInv = new ArrayList<>(); for (int i = 0; i < player.getInventory().armor.size(); i++) { armorInv.add(player.getInventory().armor.get(i).copy()); } ItemStack offhand = player.getInventory().offHand.get(0).copy(); return new SavedInventory(mainInv, armorInv, offhand, player.totalExperience, player.experienceLevel, player.experienceProgress); } public void applyToPlayer(ServerPlayerEntity player) { player.getInventory().clear(); for (int i = 0; i < mainInventory.size() && i < player.getInventory().main.size(); i++) { player.getInventory().main.set(i, mainInventory.get(i).copy()); } for (int i = 0; i < armorInventory.size() && i < player.getInventory().armor.size(); i++) { player.getInventory().armor.set(i, armorInventory.get(i).copy()); } player.getInventory().offHand.set(0, offhandItem.copy()); player.totalExperience = experience; player.experienceLevel = experienceLevel; player.experienceProgress = experienceProgress; VersionAdapter.getInstance().syncPlayerInventory(player); } public NbtCompound toNbt(RegistryWrapper.WrapperLookup registryLookup) { NbtCompound nbt = new NbtCompound(); VersionAdapter adapter = VersionAdapter.getInstance(); NbtList mainInvList = new NbtList(); for (ItemStack item : mainInventory) { mainInvList.add(adapter.encodeItemStack(item, registryLookup)); } nbt.put("MainInventory", mainInvList); NbtList armorInvList = new NbtList(); for (ItemStack item : armorInventory) { armorInvList.add(adapter.encodeItemStack(item, registryLookup)); } nbt.put("ArmorInventory", armorInvList); nbt.put("Offhand", adapter.encodeItemStack(offhandItem, registryLookup)); nbt.putInt("Experience", experience); nbt.putInt("ExperienceLevel", experienceLevel); nbt.putFloat("ExperienceProgress", experienceProgress); return nbt; } public static SavedInventory fromNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) { VersionAdapter adapter = VersionAdapter.getInstance(); List mainInv = new ArrayList<>(); NbtList mainInvList = nbt.getList("MainInventory", NbtElement.COMPOUND_TYPE); for (int i = 0; i < mainInvList.size(); i++) { mainInv.add(adapter.decodeItemStack(mainInvList.getCompound(i), registryLookup)); } List armorInv = new ArrayList<>(); NbtList armorInvList = nbt.getList("ArmorInventory", NbtElement.COMPOUND_TYPE); for (int i = 0; i < armorInvList.size(); i++) { armorInv.add(adapter.decodeItemStack(armorInvList.getCompound(i), registryLookup)); } ItemStack offhand = adapter.decodeItemStack(nbt.getCompound("Offhand"), registryLookup); int exp = nbt.getInt("Experience"); int expLevel = nbt.getInt("ExperienceLevel"); float expProgress = nbt.getFloat("ExperienceProgress"); return new SavedInventory(mainInv, armorInv, offhand, exp, expLevel, expProgress); } }