1 files changed, 33 insertions, 0 deletions
diff --git a/src/main/kotlin/moe/nea/notenoughupdates/mixins/MixinMessageHandler.kt b/src/main/kotlin/moe/nea/notenoughupdates/mixins/MixinMessageHandler.kt
new file mode 100644
index 0000000..b6151b9
--- /dev/null
+++ b/src/main/kotlin/moe/nea/notenoughupdates/mixins/MixinMessageHandler.kt
@@ -0,0 +1,33 @@
+package moe.nea.notenoughupdates.mixins
+
+import org.spongepowered.asm.mixin.Mixin
+import org.spongepowered.asm.mixin.injection.At
+import org.spongepowered.asm.mixin.injection.Inject
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo
+import net.minecraft.client.network.message.MessageHandler
+import net.minecraft.network.message.MessageType
+import net.minecraft.network.message.SignedMessage
+import net.minecraft.text.Text
+import moe.nea.notenoughupdates.events.ServerChatLineReceivedEvent
+
+@Mixin(MessageHandler::class)
+class MixinMessageHandler {
+    @Inject(method = ["onChatMessage"], at = [At("HEAD")], cancellable = true)
+    fun onOnChatMessage(message: SignedMessage, params: MessageType.Parameters, ci: CallbackInfo) {
+        val decoratedText = params.applyChatDecoration(message.unsignedContent.orElse(message.content))
+        val event = ServerChatLineReceivedEvent(decoratedText)
+        if (ServerChatLineReceivedEvent.publish(event).cancelled) {
+            ci.cancel()
+        }
+    }
+
+    @Inject(method = ["onGameMessage"], at = [At("HEAD")], cancellable = true)
+    fun onOnGameMessage(message: Text, overlay: Boolean, ci: CallbackInfo) {
+        if (!overlay) {
+            val event = ServerChatLineReceivedEvent(message)
+            if (ServerChatLineReceivedEvent.publish(event).cancelled) {
+                ci.cancel()
+            }
+        }
+    }
+}