From 3fb0b2a22bae6c676ab8df0999a7dd5556288615 Mon Sep 17 00:00:00 2001
From: RaphiMC <50594595+RaphiMC@users.noreply.github.com>
Date: Sun, 26 Nov 2023 21:16:40 +0100
Subject: [PATCH] Fixed build error

---
 .../fixes/ArmorUpdateListener.java            | 27 +++++++++++++------
 1 file changed, 19 insertions(+), 8 deletions(-)

diff --git a/src/main/java/de/florianmichael/viafabricplus/fixes/ArmorUpdateListener.java b/src/main/java/de/florianmichael/viafabricplus/fixes/ArmorUpdateListener.java
index 731c42bc..6d7efa19 100644
--- a/src/main/java/de/florianmichael/viafabricplus/fixes/ArmorUpdateListener.java
+++ b/src/main/java/de/florianmichael/viafabricplus/fixes/ArmorUpdateListener.java
@@ -19,40 +19,51 @@
 
 package de.florianmichael.viafabricplus.fixes;
 
+import com.viaversion.viaversion.api.connection.UserConnection;
 import com.viaversion.viaversion.api.protocol.packet.PacketWrapper;
 import com.viaversion.viaversion.api.type.Type;
 import com.viaversion.viaversion.protocols.protocol1_9to1_8.ArmorType;
 import com.viaversion.viaversion.protocols.protocol1_9to1_8.ClientboundPackets1_9;
 import com.viaversion.viaversion.protocols.protocol1_9to1_8.Protocol1_9To1_8;
 import de.florianmichael.viafabricplus.ViaFabricPlus;
+import de.florianmichael.viafabricplus.injection.access.IClientConnection;
 import net.fabricmc.fabric.api.client.event.lifecycle.v1.ClientTickEvents;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.item.ItemStack;
 import net.minecraft.registry.Registries;
 
+import java.util.UUID;
+
 public class ArmorUpdateListener {
 
+    private static final UUID ARMOR_POINTS_UUID = UUID.fromString("2AD3F246-FEE1-4E67-B886-69FD380BB150");
+
+    private static double oldArmor = 0;
+
     public static void init() {
         ClientTickEvents.START_WORLD_TICK.register(world -> {
             if (MinecraftClient.getInstance().player != null) {
-                try {
-                    sendArmorUpdate();
-                } catch (Throwable t) {
-                    ViaFabricPlus.global().getLogger().error("Error sending armor update", t);
+                final UserConnection userConnection = ((IClientConnection) MinecraftClient.getInstance().getNetworkHandler().getConnection()).viaFabricPlus$getUserConnection();
+                if (userConnection != null) {
+                    try {
+                        sendArmorUpdate(userConnection);
+                    } catch (Throwable t) {
+                        ViaFabricPlus.global().getLogger().error("Error sending armor update", t);
+                    }
                 }
             }
         });
     }
 
-    public static void sendArmorUpdate() throws Exception {
+    public static void sendArmorUpdate(final UserConnection userConnection) throws Exception {
         int armor = 0;
         for (final ItemStack stack : MinecraftClient.getInstance().player.getInventory().armor) {
             armor += ArmorType.findByType(Registries.ITEM.getId(stack.getItem()).toString()).getArmorPoints();
         }
-        if (armor == this.oldArmor) return;
+        if (armor == oldArmor) return;
 
-        this.oldArmor = armor;
-        final PacketWrapper properties = PacketWrapper.create(ClientboundPackets1_9.ENTITY_PROPERTIES, MinecraftClient.getInstance().getNetworkHandler().getConnection().getUserConnection());
+        oldArmor = armor;
+        final PacketWrapper properties = PacketWrapper.create(ClientboundPackets1_9.ENTITY_PROPERTIES, userConnection);
         properties.write(Type.VAR_INT, MinecraftClient.getInstance().player.getId());
         properties.write(Type.INT, 1);
         properties.write(Type.STRING, "generic.armor");