1 files changed, 54 insertions, 0 deletions
diff --git a/src/main/kotlin/events/registration/ChatEvents.kt b/src/main/kotlin/events/registration/ChatEvents.kt
new file mode 100644
index 0000000..4c1c63f
--- /dev/null
+++ b/src/main/kotlin/events/registration/ChatEvents.kt
@@ -0,0 +1,54 @@
+
+
+package moe.nea.firmament.events.registration
+
+import net.fabricmc.fabric.api.client.message.v1.ClientReceiveMessageEvents
+import net.fabricmc.fabric.api.event.player.AttackBlockCallback
+import net.fabricmc.fabric.api.event.player.UseBlockCallback
+import net.minecraft.text.Text
+import net.minecraft.util.ActionResult
+import moe.nea.firmament.events.AllowChatEvent
+import moe.nea.firmament.events.AttackBlockEvent
+import moe.nea.firmament.events.ModifyChatEvent
+import moe.nea.firmament.events.ProcessChatEvent
+import moe.nea.firmament.events.UseBlockEvent
+
+private var lastReceivedMessage: Text? = null
+
+fun registerFirmamentEvents() {
+    ClientReceiveMessageEvents.ALLOW_CHAT.register(ClientReceiveMessageEvents.AllowChat { message, signedMessage, sender, params, receptionTimestamp ->
+        lastReceivedMessage = message
+        !ProcessChatEvent.publish(ProcessChatEvent(message, false)).cancelled
+            && !AllowChatEvent.publish(AllowChatEvent(message)).cancelled
+    })
+    ClientReceiveMessageEvents.ALLOW_GAME.register(ClientReceiveMessageEvents.AllowGame { message, overlay ->
+        lastReceivedMessage = message
+        overlay || (!ProcessChatEvent.publish(ProcessChatEvent(message, false)).cancelled &&
+            !AllowChatEvent.publish(AllowChatEvent(message)).cancelled)
+    })
+    ClientReceiveMessageEvents.MODIFY_GAME.register(ClientReceiveMessageEvents.ModifyGame { message, overlay ->
+        if (overlay) message
+        else ModifyChatEvent.publish(ModifyChatEvent(message)).replaceWith
+    })
+    ClientReceiveMessageEvents.GAME_CANCELED.register(ClientReceiveMessageEvents.GameCanceled { message, overlay ->
+        if (!overlay && lastReceivedMessage !== message) {
+            ProcessChatEvent.publish(ProcessChatEvent(message, true))
+        }
+    })
+    ClientReceiveMessageEvents.CHAT_CANCELED.register(ClientReceiveMessageEvents.ChatCanceled { message, signedMessage, sender, params, receptionTimestamp ->
+        if (lastReceivedMessage !== message) {
+            ProcessChatEvent.publish(ProcessChatEvent(message, true))
+        }
+    })
+
+    AttackBlockCallback.EVENT.register(AttackBlockCallback { player, world, hand, pos, direction ->
+        if (AttackBlockEvent.publish(AttackBlockEvent(player, world, hand, pos, direction)).cancelled)
+            ActionResult.CONSUME
+        else ActionResult.PASS
+    })
+    UseBlockCallback.EVENT.register(UseBlockCallback { player, world, hand, hitResult ->
+        if (UseBlockEvent.publish(UseBlockEvent(player, world, hand, hitResult)).cancelled)
+            ActionResult.CONSUME
+        else ActionResult.PASS
+    })
+}