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
|
package moe.nea.funnyteleporters;
import com.mojang.datafixers.util.Pair;
import com.mojang.serialization.Codec;
import net.minecraft.block.BlockState;
import net.minecraft.block.entity.BlockEntity;
import net.minecraft.block.entity.BlockEntityType;
import net.minecraft.item.Item;
import net.minecraft.item.ItemConvertible;
import net.minecraft.item.Items;
import net.minecraft.nbt.NbtCompound;
import net.minecraft.nbt.NbtOps;
import net.minecraft.registry.Registries;
import net.minecraft.registry.RegistryWrapper;
import net.minecraft.registry.entry.RegistryEntry;
import net.minecraft.util.math.BlockPos;
import java.util.Optional;
import java.util.TreeMap;
public class TeleporterNexusBlockEntity extends BlockEntity {
public TeleporterNexusBlockEntity(BlockPos pos, BlockState state) {this(FunnyRegistry.TELEPORTER_NEXUS_ENTITY, pos, state);}
protected TeleporterNexusBlockEntity(BlockEntityType<?> type, BlockPos pos, BlockState state) {
super(type, pos, state);
}
public Optional<String> getName(TeleporterDestination dest) {
var label = destinations.get(dest);
return label != null ? (label.label) : Optional.empty();
}
public ItemConvertible getIcon(TeleporterDestination dest) {
var label = destinations.get(dest);
return (label != null ? label.item : getDefaultItem()).value();
}
public record Label(
RegistryEntry<Item> item,
Optional<String> label
) {
}
public TreeMap<TeleporterDestination, Label> destinations = new TreeMap<>();
private static final Codec<TreeMap<TeleporterDestination, Label>> DESTINATION_CODEC =
TeleporterDestination.Labeled.CODEC
.listOf()
.xmap(
pairs -> {
var hash = new TreeMap<TeleporterDestination, Label>();
for (TeleporterDestination.Labeled pair : pairs) {
hash.put(pair.destination(), new Label(pair.item(), pair.name()));
}
return hash;
},
map -> map.entrySet().stream().map(it -> new TeleporterDestination.Labeled(it.getValue().item(), it.getKey(), it.getValue().label())).toList()
);
static RegistryEntry<Item> getDefaultItem() {
return Registries.ITEM.getEntry(Items.ENDER_PEARL);
}
@Override
protected void readNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
super.readNbt(nbt, registryLookup);
var result = DESTINATION_CODEC.decode(registryLookup.getOps(NbtOps.INSTANCE), nbt.get("destinations"));
result.resultOrPartial().map(Pair::getFirst).ifPresent(it -> destinations = it);
}
@Override
protected void writeNbt(NbtCompound nbt, RegistryWrapper.WrapperLookup registryLookup) {
super.writeNbt(nbt, registryLookup);
nbt.put("destinations", DESTINATION_CODEC.encodeStart(registryLookup.getOps(NbtOps.INSTANCE), destinations).getPartialOrThrow());
}
public void addDestination(TeleporterDestination destination) {
destinations.putIfAbsent(destination, new Label(getDefaultItem(), Optional.empty()));
markDirty();
}
public void removeDestination(TeleporterDestination destination) {
destinations.remove(destination);
markDirty();
}
public void setIcon(TeleporterDestination dest, Item item) {
var entry = Registries.ITEM.getEntry(item);
destinations.compute(dest, (key, old) -> old == null ? new Label(entry, Optional.empty()) : new Label(entry, old.label()));
markDirty();
}
public void setName(TeleporterDestination dest, Optional<String> string) {
destinations.compute(dest, (key, old) -> old == null ? new Label(getDefaultItem(), string) : new Label(old.item(), string));
markDirty();
}
}
|
