1 files changed, 83 insertions, 0 deletions
diff --git a/src/main/java/moe/nea/funnyteleporters/ColouredChestState.java b/src/main/java/moe/nea/funnyteleporters/ColouredChestState.java
new file mode 100644
index 0000000..9e18b16
--- /dev/null
+++ b/src/main/java/moe/nea/funnyteleporters/ColouredChestState.java
@@ -0,0 +1,83 @@
+package moe.nea.funnyteleporters;
+
+import com.mojang.serialization.Codec;
+import com.mojang.serialization.codecs.RecordCodecBuilder;
+import net.minecraft.inventory.Inventories;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NbtCompound;
+import net.minecraft.registry.RegistryWrapper;
+import net.minecraft.server.MinecraftServer;
+import net.minecraft.util.DyeColor;
+import net.minecraft.util.Identifier;
+import net.minecraft.util.collection.DefaultedList;
+import net.minecraft.util.dynamic.Codecs;
+import net.minecraft.world.PersistentState;
+import net.minecraft.world.World;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+import java.util.StringJoiner;
+
+import static moe.nea.funnyteleporters.FunnyRegistry.MODID;
+
+public class ColouredChestState extends PersistentState {
+
+	public Map<List<DyeColor>, DefaultedList<ItemStack>> inventories = new HashMap<>();
+
+	private static String joinDyeIndex(List<DyeColor> dyes) {
+		StringJoiner j = new StringJoiner(";");
+		dyes.forEach(it -> j.add(it.getName()));
+		return j.toString();
+	}
+
+	private static List<DyeColor> splitDyeIndex(String string) {
+		List<DyeColor> dyes = new ArrayList<>(3);
+		for (String s : string.split(";")) {
+			dyes.add(DyeColor.byName(s, DyeColor.BLACK));
+		}
+		return Collections.unmodifiableList(dyes);
+	}
+
+	public static final String ColouredChestInventories = "ColouredChestInventories";
+
+	@Override
+	public NbtCompound writeNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
+		var compound = new NbtCompound();
+		inventories.forEach((dyeColors, itemStacks) ->
+			                    compound.put(joinDyeIndex(dyeColors), Inventories.writeNbt(new NbtCompound(), itemStacks, registryLookup))
+		);
+		nbt.put(ColouredChestInventories, compound);
+		return nbt;
+	}
+
+	public static ColouredChestState createFromNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
+		var state = new ColouredChestState();
+		var compound = nbt.getCompound(ColouredChestInventories);
+		compound.getKeys().forEach(key -> {
+			                           var list = DefaultedList.ofSize(27, ItemStack.EMPTY);
+			                           Inventories.readNbt(compound.getCompound(key), list, registryLookup);
+			                           state.inventories.put(splitDyeIndex(key), list);
+		                           }
+		);
+		return state;
+	}
+
+	public DefaultedList<ItemStack> getDelegate(List<DyeColor> dyeColors) {
+		return inventories.computeIfAbsent(dyeColors, ignored -> DefaultedList.ofSize(27, ItemStack.EMPTY));
+	}
+
+	public static Type<ColouredChestState> TYPE = new Type<>(ColouredChestState::new, ColouredChestState::createFromNbt, null);
+
+	public static Identifier identifier = Identifier.of(MODID, "coloured_chests");
+
+	public static ColouredChestState getServerState(MinecraftServer server) {
+		var persistentStateWorld = Objects.requireNonNull(server.getWorld(World.OVERWORLD)).getPersistentStateManager(); // Force all data saving to be done in the overworld
+		var state = persistentStateWorld.getOrCreate(TYPE, identifier.toString());
+		state.markDirty(); // Always save
+		return state;
+	}
+}