update MessageChannels LastMessageID

disgoorg · Jun 1, 2022 · 33d0ea6 · 33d0ea6
1 parent 734eba7
commit 33d0ea6
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/discord/channel.go b/discord/channel.go
@@ -995,3 +995,22 @@ func ApplyGuildIDToChannel(channel GuildChannel, guildID snowflake.ID) GuildChan
 		panic("unknown channel type")
 	}
 }
+
+func ApplyLastMessageIDToChannel(channel MessageChannel, lastMessageID snowflake.ID) MessageChannel {
+	switch c := channel.(type) {
+	case GuildTextChannel:
+		c.lastMessageID = &lastMessageID
+		return c
+	case GuildVoiceChannel:
+		c.lastMessageID = &lastMessageID
+		return c
+	case GuildNewsChannel:
+		c.lastMessageID = &lastMessageID
+		return c
+	case GuildThread:
+		c.lastMessageID = &lastMessageID
+		return c
+	default:
+		panic("unknown channel type")
+	}
+}
diff --git a/handlers/message_create_handler.go b/handlers/message_create_handler.go
@@ -30,6 +30,10 @@ func (h *gatewayHandlerMessageCreate) HandleGatewayEvent(client bot.Client, sequ
 
 	client.Caches().Messages().Put(message.ChannelID, message.ID, message)
 
+	if channel, ok := client.Caches().Channels().GetMessageChannel(message.ChannelID); ok {
+		client.Caches().Channels().Put(message.ChannelID, discord.ApplyLastMessageIDToChannel(channel, message.ID))
+	}
+
 	genericEvent := events.NewGenericEvent(client, sequenceNumber, shardID)
 	client.EventManager().DispatchEvent(&events.MessageCreate{
 		GenericMessage: &events.GenericMessage{