fix: remove guild on leave

SushiWaUmai · Jun 27, 2023 · 66428ef · 66428ef
1 parent fb172a3
commit 66428ef
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/bot/bot.go b/bot/bot.go
@@ -150,3 +150,11 @@ func handleInteraction(s *discordgo.Session, i *discordgo.InteractionCreate) {
 		c.Execute(s, i)
 	}
 }
+
+func handleGuildLeave(s *discordgo.Session, g *discordgo.GuildDelete) {
+	err := db.DeleteGuild(g.ID)
+
+	if err != nil {
+		log.Printf("Error deleting guild channels for guild %s: %v\n", g.ID, err)
+	}
+}
diff --git a/db/db.go b/db/db.go
@@ -65,6 +65,10 @@ func DeleteGuildChannels(guildID string, channelID string) error {
 	return db.Where("guild_id = ? AND channel_id = ?", guildID, channelID).Delete(&GuildChannel{}).Error
 }
 
+func DeleteGuild(guildID string) error {
+	return db.Where("guild_id = ?", guildID).Delete(&GuildChannel{}).Error
+}
+
 func GetGuildChannel(guildID string, channelID string, category string) (*GuildChannel, error) {
 	var guildChannel GuildChannel
 	err := db.Where("guild_id = ? AND channel_id = ? AND category = ?", guildID, channelID, category).First(&guildChannel).Error