templates: add msg forward guild checks

Signed-off-by: SoggySaussages <[email protected]>
botlabs-gg · Nov 15, 2024 · 374cc53 · 374cc53
1 parent 0c8ecfa
commit 374cc53
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/common/templates/context_funcs.go b/common/templates/context_funcs.go
@@ -65,6 +65,19 @@ func (c *Context) tmplSendDM(s ...interface{}) string {
 	}
 	msgSend.Components = append(serverInfo, msgSend.Components...)
 
+	if msgSend.Reference != nil {
+		if msgSend.Reference.Type == discordgo.MessageReferenceTypeForward {
+			if originChannel := c.ChannelArgNoDM(msgSend.Reference.ChannelID); originChannel != 0 {
+				hasPerms, _ := bot.BotHasPermissionGS(c.GS, originChannel, discordgo.PermissionViewChannel|discordgo.PermissionReadMessageHistory)
+				if !hasPerms {
+					msgSend.Reference = &discordgo.MessageReference{}
+				}
+			} else {
+				msgSend.Reference = &discordgo.MessageReference{}
+			}
+		}
+	}
+
 	channel, err := common.BotSession.UserChannelCreate(c.MS.User.ID)
 	if err != nil {
 		return ""
@@ -390,6 +403,19 @@ func (c *Context) tmplSendMessage(filterSpecialMentions bool, returnID bool) fun
 			msgSend.Components = append(serverInfo, msgSend.Components...)
 		}
 
+		if msgSend.Reference != nil {
+			if msgSend.Reference.Type == discordgo.MessageReferenceTypeForward {
+				if originChannel := c.ChannelArgNoDM(msgSend.Reference.ChannelID); originChannel != 0 {
+					hasPerms, _ := bot.BotHasPermissionGS(c.GS, originChannel, discordgo.PermissionViewChannel|discordgo.PermissionReadMessageHistory)
+					if !hasPerms {
+						msgSend.Reference = &discordgo.MessageReference{}
+					}
+				} else {
+					msgSend.Reference = &discordgo.MessageReference{}
+				}
+			}
+		}
+
 		m, err = common.BotSession.ChannelMessageSendComplex(cid, msgSend)
 
 		if err == nil && returnID {