generated from FabricMC/fabric-example-mod
-
-
Notifications
You must be signed in to change notification settings - Fork 6
/
ShulkerBoxBlockEntityMixin.java
85 lines (75 loc) · 3.07 KB
/
ShulkerBoxBlockEntityMixin.java
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
package de.rubixdev.enchantedshulkers.mixin;
import de.rubixdev.enchantedshulkers.config.WorldConfig;
import de.rubixdev.enchantedshulkers.interfaces.EnchantableBlockEntity;
import net.minecraft.block.BlockState;
import net.minecraft.block.entity.BlockEntityType;
import net.minecraft.block.entity.LootableContainerBlockEntity;
import net.minecraft.block.entity.ShulkerBoxBlockEntity;
import net.minecraft.nbt.NbtCompound;
import net.minecraft.nbt.NbtElement;
import net.minecraft.nbt.NbtList;
import net.minecraft.network.listener.ClientPlayPacketListener;
import net.minecraft.network.packet.Packet;
import net.minecraft.network.packet.s2c.play.BlockEntityUpdateS2CPacket;
import net.minecraft.text.MutableText;
import net.minecraft.text.Style;
import net.minecraft.text.Text;
import net.minecraft.util.Formatting;
import net.minecraft.util.math.BlockPos;
import org.jetbrains.annotations.Nullable;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Unique;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.Redirect;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
@Mixin(ShulkerBoxBlockEntity.class)
public abstract class ShulkerBoxBlockEntityMixin extends LootableContainerBlockEntity
implements EnchantableBlockEntity {
@Unique
private NbtList enchantments = new NbtList();
protected ShulkerBoxBlockEntityMixin(BlockEntityType<?> blockEntityType, BlockPos blockPos, BlockState blockState) {
super(blockEntityType, blockPos, blockState);
}
@Override
public NbtList getEnchantments() {
return this.enchantments;
}
@Override
public void setEnchantments(NbtList enchantments) {
this.enchantments = enchantments;
}
@Inject(method = "readNbt", at = @At("TAIL"))
public void readNbt(NbtCompound nbt, CallbackInfo ci) {
if (nbt.contains("Enchantments", NbtElement.LIST_TYPE)) {
setEnchantments(nbt.getList("Enchantments", NbtElement.COMPOUND_TYPE));
}
}
@Inject(method = "writeNbt", at = @At("TAIL"))
public void writeNbt(NbtCompound nbt, CallbackInfo ci) {
nbt.put("Enchantments", this.enchantments);
}
@Override
public NbtCompound toInitialChunkDataNbt() {
return this.toClientNbt();
}
@Nullable
@Override
public Packet<ClientPlayPacketListener> toUpdatePacket() {
return BlockEntityUpdateS2CPacket.create(this);
}
@Redirect(
method = "getContainerName",
at =
@At(
value = "INVOKE",
target =
"Lnet/minecraft/text/Text;translatable(Ljava/lang/String;)Lnet/minecraft/text/MutableText;"))
public MutableText getContainerName(String key) {
MutableText text = Text.translatable(key);
if (WorldConfig.coloredNames() && !enchantments.isEmpty()) {
text.setStyle(Style.EMPTY.withFormatting(Formatting.AQUA));
}
return text;
}
}