From 620ed4bbffeaf77d245689605ffa7c8e96c5c820 Mon Sep 17 00:00:00 2001
From: ayn2op <ayn2op@gmail.com>
Date: Thu, 17 Jul 2025 00:12:29 -0400
Subject: [PATCH] feat: add handler for message update event

---
 cmd/state.go | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/cmd/state.go b/cmd/state.go
index d8770f6..b73e577 100644
--- a/cmd/state.go
+++ b/cmd/state.go
@@ -33,6 +33,7 @@ func openState(token string) error {
 	discordState.AddHandler(onRaw)
 	discordState.AddHandler(onReady)
 	discordState.AddHandler(onMessageCreate)
+	discordState.AddHandler(onMessageUpdate)
 	discordState.AddHandler(onMessageDelete)
 	discordState.AddHandler(onReadUpdate)
 
@@ -158,21 +159,27 @@ func onReady(r *gateway.ReadyEvent) {
 	app.Draw()
 }
 
-func onMessageCreate(msg *gateway.MessageCreateEvent) {
-	if app.guildsTree.selectedChannelID == msg.ChannelID {
-		app.messagesList.createMsg(msg.Message)
+func onMessageCreate(message *gateway.MessageCreateEvent) {
+	if app.guildsTree.selectedChannelID == message.ChannelID {
+		app.messagesList.createMsg(message.Message)
 		app.Draw()
 	}
 
-	if err := notifications.HandleIncomingMessage(discordState, msg, app.cfg); err != nil {
+	if err := notifications.HandleIncomingMessage(discordState, message, app.cfg); err != nil {
 		slog.Error("Notification failed", "err", err)
 	}
 }
 
-func onMessageDelete(msg *gateway.MessageDeleteEvent) {
-	if app.guildsTree.selectedChannelID == msg.ChannelID {
+func onMessageUpdate(message *gateway.MessageUpdateEvent) {
+	if app.guildsTree.selectedChannelID == message.ChannelID {
+		onMessageDelete(&gateway.MessageDeleteEvent{ID: message.ID, ChannelID: message.ChannelID, GuildID: message.GuildID})
+	}
+}
+
+func onMessageDelete(message *gateway.MessageDeleteEvent) {
+	if app.guildsTree.selectedChannelID == message.ChannelID {
 		app.messagesList.reset()
-		app.messagesList.drawMsgs(msg.ChannelID)
+		app.messagesList.drawMsgs(message.ChannelID)
 		app.Draw()
 	}
 }