1 files changed, 34 insertions, 0 deletions
diff --git a/src/main/java/moe/nea/firmament/mixins/MixinMessageHandler.java b/src/main/java/moe/nea/firmament/mixins/MixinMessageHandler.java
new file mode 100644
index 0000000..34dd5a1
--- /dev/null
+++ b/src/main/java/moe/nea/firmament/mixins/MixinMessageHandler.java
@@ -0,0 +1,34 @@
+package moe.nea.firmament.mixins;
+
+import com.mojang.authlib.GameProfile;
+import moe.nea.firmament.events.ServerChatLineReceivedEvent;
+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 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;
+
+@Mixin(MessageHandler.class)
+public class MixinMessageHandler {
+    @Inject(method = "onChatMessage", cancellable = true, at = @At("HEAD"))
+    public void onOnChatMessage(SignedMessage message, GameProfile sender, MessageType.Parameters params, CallbackInfo ci) {
+        var decoratedText = params.applyChatDecoration(message.unsignedContent() != null ? message.unsignedContent() : message.getContent());
+        var event = new ServerChatLineReceivedEvent(decoratedText);
+        if (ServerChatLineReceivedEvent.Companion.publish(event).getCancelled()) {
+            ci.cancel();
+        }
+    }
+
+    @Inject(method = "onGameMessage", at = @At("HEAD"), cancellable = true)
+    public void onOnGameMessage(Text message, boolean overlay, CallbackInfo ci) {
+        if (!overlay) {
+            var event = new ServerChatLineReceivedEvent(message);
+            if (ServerChatLineReceivedEvent.Companion.publish(event).getCancelled()) {
+                ci.cancel();
+            }
+        }
+    }
+}