1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
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;
}
}
|
