feat: add welcome message functionality

- Introduced a new welcome service and repository for managing welcome messages, channels, and GIFs. - Added commands to set welcome channel, message, DM, and GIF through the bot's configuration. - Implemented event handling for new guild members to send welcome messages and DMs based on configuration. - Updated main application to include the new welcome service in the bot's services.
2026-03-18 05:07:11 +00:00
parent 6410619662
commit e6aa5def96
8 changed files with 489 additions and 6 deletions
@@ -0,0 +1,119 @@
+package events
+
+import (
+	"context"
+	"log"
+	"strconv"
+	"strings"
+
+	"velox-bot/internal/db/services"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+// HandleGuildMemberAdd sends a configurable welcome message in a guild channel
+// and an optional configurable DM to the joining user.
+func HandleGuildMemberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd, svc *services.Services) {
+	if s == nil || m == nil || svc == nil || svc.Welcome == nil {
+		return
+	}
+	if m.User == nil {
+		return
+	}
+
+	ctx := context.Background()
+	guildID := mustParseInt64(m.GuildID)
+
+	settings, err := svc.Welcome.GetSettings(ctx, guildID)
+	if err != nil {
+		log.Printf("welcome: failed to load settings for guild %s: %v", m.GuildID, err)
+		return
+	}
+	if settings == nil || settings.ChannelID == 0 {
+		// No welcome channel configured; nothing to do.
+		return
+	}
+
+	memberMention := m.User.Mention()
+
+	// Resolve guild/server name and user avatar for placeholders / visuals.
+	guildName := ""
+	userAvatarURL := ""
+	if g, err := s.State.Guild(m.GuildID); err == nil && g != nil {
+		guildName = g.Name
+	} else if g, err := s.Guild(m.GuildID); err == nil && g != nil {
+		guildName = g.Name
+	}
+	if m.User.Avatar != "" {
+		userAvatarURL = discordgo.EndpointUserAvatar(m.User.ID, m.User.Avatar)
+	}
+
+	// Build guild welcome message.
+	channelID := strconv.FormatInt(settings.ChannelID, 10)
+	welcomeMsg := strings.TrimSpace(settings.WelcomeMessage)
+	if welcomeMsg == "" {
+		// Default message when none configured.
+		// "<user>! Welcome to <server>! Have fun!"
+		welcomeMsg = "<user>! Welcome to <server>! Have fun!"
+	}
+	welcomeMsg = strings.ReplaceAll(welcomeMsg, "{user}", memberMention)
+	if guildName != "" {
+		welcomeMsg = strings.ReplaceAll(welcomeMsg, "{server}", guildName)
+		welcomeMsg = strings.ReplaceAll(welcomeMsg, "<server>", guildName)
+	}
+	welcomeMsg = strings.ReplaceAll(welcomeMsg, "<user>", memberMention)
+
+	embed := &discordgo.MessageEmbed{
+		Title:       "👋 Welcome!",
+		Description: welcomeMsg,
+		Color:       0xFFA500,
+		Footer: &discordgo.MessageEmbedFooter{
+			Text: "ID: " + m.User.ID,
+		},
+	}
+	if userAvatarURL != "" {
+		embed.Author = &discordgo.MessageEmbedAuthor{
+			Name:    m.User.Username,
+			IconURL: userAvatarURL,
+		}
+		embed.Thumbnail = &discordgo.MessageEmbedThumbnail{
+			URL: userAvatarURL,
+		}
+	}
+	gif := strings.TrimSpace(settings.WelcomeGIFURL)
+	if gif == "" {
+		// Default GIF when none configured.
+		gif = "https://images-ext-1.discordapp.net/external/uJ6XfdK2WwDnei3RmNWUqSiOVboC4mK9r78TtgVE_9g/https/media.giphy.com/media/XD9o33QG9BoMis7iM4/giphy.gif"
+	}
+	embed.Image = &discordgo.MessageEmbedImage{URL: gif}
+
+	if _, err := s.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{
+		Content: memberMention,
+		Embed:   embed,
+	}); err != nil {
+		log.Printf("welcome: failed to send channel welcome in guild %s: %v", m.GuildID, err)
+	}
+
+	// Optional DM.
+	dmText := strings.TrimSpace(settings.WelcomeDM)
+	if dmText == "" {
+		// Default DM when none configured.
+		// "Welcome to <server>! Have fun!"
+		dmText = "Welcome to <server>! Have fun!"
+	}
+	dmText = strings.ReplaceAll(dmText, "{user}", memberMention)
+	if guildName != "" {
+		dmText = strings.ReplaceAll(dmText, "{server}", guildName)
+		dmText = strings.ReplaceAll(dmText, "<server>", guildName)
+	}
+
+	dmCh, err := s.UserChannelCreate(m.User.ID)
+	if err != nil {
+		log.Printf("welcome: failed to create DM channel for user %s: %v", m.User.ID, err)
+		return
+	}
+	if _, err := s.ChannelMessageSend(dmCh.ID, dmText); err != nil {
+		log.Printf("welcome: failed to send DM welcome to user %s: %v", m.User.ID, err)
+	}
+}
+