Vexes no longer take damage when chat data exists

CHANGELOG.md

@@ -14,6 +14,7 @@ All notable changes to **CreatureChat** are documented in this file. The format
 
 ### Fixed
 - Bees no longer forget their chat data when entering/leaving hives (writeNbt & readNbt modified)
+- Vexes no longer take damage when chat data exists
 
 ## [1.3.0] - 2025-01-14
 

src/main/java/com/owlmaddie/mixin/MixinVexEntity.java

+package com.owlmaddie.mixin;
+
+import com.owlmaddie.chat.ChatDataManager;
+import com.owlmaddie.chat.EntityChatData;
+import net.minecraft.entity.mob.VexEntity;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+/**
+ * Mixin to modify Vex behavior by setting `alive = false` if chat data exists.
+ */
+@Mixin(VexEntity.class)
+public abstract class MixinVexEntity {
+    @Shadow
+    private boolean alive;
+
+    @Inject(method = "tick", at = @At("HEAD"))
+    private void disableVexIfChatData(CallbackInfo ci) {
+        VexEntity vex = (VexEntity) (Object) this;
+
+        // Get chat data for this Vex
+        EntityChatData chatData = ChatDataManager.getServerInstance().getOrCreateChatData(vex.getUuidAsString());
+        if (this.alive && !chatData.characterSheet.isEmpty()) {
+            this.alive = false; // Prevents the Vex from ticking and taking damage
+        }
+    }
+}

src/main/resources/creaturechat.mixins.json

@@ -9,6 +9,7 @@
 		"MixinBucketable",
 		"MixinEntityChatData",
 		"MixinWitherEntity",
+		"MixinVexEntity",
 		"MixinVillagerEntity",
 		"MixinOnChat"
 	],