uudate custom bot user info after fetching

merlinfuchs · Nov 30, 2023 · 24ae628 · 24ae628
1 parent 4d9960f
commit 24ae628
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 0 deletions.
diff --git a/embedg-server/api/handlers/custom_bots/handler.go b/embedg-server/api/handlers/custom_bots/handler.go
@@ -17,6 +17,7 @@ import (
 	"github.com/merlinfuchs/embed-generator/embedg-server/bot"
 	"github.com/merlinfuchs/embed-generator/embedg-server/db/postgres"
 	"github.com/merlinfuchs/embed-generator/embedg-server/util"
+	"github.com/rs/zerolog/log"
 	"github.com/spf13/viper"
 	"gopkg.in/guregu/null.v4"
 )
@@ -237,6 +238,19 @@ func (h *CustomBotsHandler) HandleGetCustomBot(c *fiber.Ctx) error {
 		}
 	}
 
+	customBot, err = h.pg.Q.UpdateCustomBotUser(c.Context(), postgres.UpdateCustomBotUserParams{
+		GuildID:           guildID,
+		UserName:          member.User.Username,
+		UserDiscriminator: member.User.Discriminator,
+		UserAvatar: sql.NullString{
+			String: member.User.Avatar,
+			Valid:  member.User.Avatar != "",
+		},
+	})
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to update custom bot user info")
+	}
+
 	guild, err := h.bot.State.Guild(guildID)
 	if err != nil {
 		return err

diff --git a/embedg-server/db/postgres/custom_bots.sql.go b/embedg-server/db/postgres/custom_bots.sql.go
diff --git a/embedg-server/db/postgres/queries/custom_bots.sql b/embedg-server/db/postgres/queries/custom_bots.sql
@@ -6,6 +6,9 @@ RETURNING *;
 -- name: UpdateCustomBotPresence :one
 UPDATE custom_bots SET gateway_status = $2, gateway_activity_type = $3, gateway_activity_name = $4, gateway_activity_state = $5, gateway_activity_url = $6 WHERE guild_id = $1 RETURNING *;
 
+-- name: UpdateCustomBotUser :one
+UPDATE custom_bots SET user_name = $2, user_discriminator = $3, user_avatar = $4 WHERE guild_id = $1 RETURNING *;
+
 -- name: UpdateCustomBotTokenInvalid :one
 UPDATE custom_bots SET token_invalid = $2 WHERE guild_id = $1 RETURNING *;