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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
package moe.nea.funnyteleporters;
import com.mojang.serialization.Codec;
import com.mojang.serialization.DataResult;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import net.minecraft.block.BlockState;
import net.minecraft.block.entity.BlockEntity;
import net.minecraft.block.entity.BlockEntityType;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.inventory.Inventories;
import net.minecraft.inventory.Inventory;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NbtCompound;
import net.minecraft.nbt.NbtOps;
import net.minecraft.registry.RegistryWrapper;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.server.world.ServerWorld;
import net.minecraft.util.DyeColor;
import net.minecraft.util.collection.DefaultedList;
import net.minecraft.util.math.BlockPos;
import java.util.List;
public class ColouredChestBlockEntity extends BlockEntity implements Inventory {
public record ExtraData(
List<DyeColor> frequency
) {
public static Codec<ExtraData> CODEC = RecordCodecBuilder.create(instance -> instance.group(
DyeColor.CODEC.listOf().validate(it -> it.size() == 3 ? DataResult.success(it) : (it.size() > 3 ? DataResult.error(() -> "List " + it + "too long", it.subList(0, 3)) : DataResult.error(() -> "List " + it + " too short"))).fieldOf("frequency").forGetter(ExtraData::frequency)
).apply(instance, ExtraData::new));
}
public ColouredChestBlockEntity(BlockPos blockPos, BlockState blockState) {
this(FunnyRegistry.COLOURED_CHEST_ENTITY, blockPos, blockState);
}
protected ColouredChestBlockEntity(BlockEntityType<?> blockEntityType, BlockPos blockPos, BlockState blockState) {
super(blockEntityType, blockPos, blockState);
}
@Override
protected void writeNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
super.writeNbt(nbt, registryLookup);
var result = ExtraData.CODEC.encodeStart(NbtOps.INSTANCE, extra);
if (result.hasResultOrPartial()) {
nbt.put("extra", result.getPartialOrThrow());
}
}
ExtraData extra = new ExtraData(List.of(DyeColor.BLUE, DyeColor.BLUE, DyeColor.BLUE));
public DefaultedList<ItemStack> getInventory() {
assert world instanceof ServerWorld;
return ColouredChestState.getServerState(world.getServer()).getDelegate(extra.frequency());
}
public int size() {
return getInventory().size();
}
@Override
public boolean isEmpty() {
for (ItemStack itemStack : getInventory()) {
if (!itemStack.isEmpty()) return false;
}
return true;
}
@Override
public ItemStack getStack(int slot) {
if (0 <= slot && slot < size())
return getInventory().get(slot);
return ItemStack.EMPTY;
}
@Override
public ItemStack removeStack(int slot, int amount) {
ItemStack itemStack = Inventories.splitStack(this.getInventory(), slot, amount);
if (!itemStack.isEmpty()) {
markDirty();
}
return itemStack;
}
@Override
public ItemStack removeStack(int slot) {
markDirty();
return Inventories.removeStack(getInventory(), slot);
}
@Override
public void setStack(int slot, ItemStack stack) {
if (0 <= slot && slot < size()) {
markDirty();
getInventory().set(slot, stack);
}
}
@Override
public boolean canPlayerUse(PlayerEntity player) {
return true;
}
@Override
protected void readNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
super.readNbt(nbt, registryLookup);
if (nbt.contains("extra")) {
var extra = ExtraData.CODEC.decode(NbtOps.INSTANCE, nbt.get("extra"));
if (extra.hasResultOrPartial()) {
this.extra = extra.getPartialOrThrow().getFirst();
}
}
}
@Override
public void clear() {
markDirty();
getInventory().clear();
}
public void openScreen(ServerPlayerEntity player) {
new ColouredChestConfigGUI(this, player).open();
}
}
|
