Refactor/#24 사용자 랜덤 선택 기능 개선 (#25)

* refactor: Changed command options

- Added count query in random-user-pick command

* refactor: Changed `getRandomUserByChannel` Logic

- Changed logic to get user info with guild(server) information

* chore: Removed unused function

Author: tidavid1

internal/handler/commands.go

@@ -80,5 +80,14 @@ var commands = []*discordgo.ApplicationCommand{
 	{
 		Name:        commandRandomUserPick,
 		Description: "채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.",
+		Options: []*discordgo.ApplicationCommandOption{
+			{
+				Type:        discordgo.ApplicationCommandOptionInteger,
+				Name:        "count",
+				Description: "랜덤으로 선택할 사용자 수를 입력해주세요.",
+				Required:    false,
+				MinValue:    func() *float64 { v := 1.0; return &v }(),
+			},
+		},
 	},
 }

internal/handler/handlers.go

@@ -6,6 +6,7 @@ import (
 	"day-day-review/internal/util"
 	"fmt"
 	"log"
+	"math/rand"
 	"strconv"
 	"strings"
 	"time"
@@ -183,52 +184,54 @@ func getScrumsByDate(session *discordgo.Session, interaction *discordgo.Interact
 
 // getRandomUserByChannel 채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.
 func getRandomUserByChannel(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
-	channel, err := session.Channel(interaction.ChannelID)
-	// 채널 정보를 불러오는 중 오류가 발생하면 에러 메시지를 전송합니다.
-	if err != nil {
-		logErrorAndSendMessage(session, interaction, "채널을 불러오는 중 오류가 발생했습니다.", err)
-		return
+	count := 1
+	// 옵션을 받았을 경우 옵션 값을 사용합니다.
+	if options := interaction.ApplicationCommandData().Options; len(options) > 0 {
+		count = int(options[0].IntValue())
 	}
-	// 음성 채널이 아니면 에러 메시지를 전송합니다.
-	if channel.Type != discordgo.ChannelTypeGuildVoice {
-		sendMessage(session, interaction, "음성 채널에서만 사용할 수 있는 명령어입니다.")
-		return
-	}
-	guild, err := session.State.Guild(channel.GuildID)
-	// 서버 정보를 불러오는 중 오류가 발생하면 에러 메시지를 전송합니다.
+	log.Println("GetRandomUserByChannel | Received count:", count)
+	guild, err := session.State.Guild(guildId)
+	// 채널 정보를 불러오는 중 오류가 발생하면 에러 메시지를 전송합니다.
 	if err != nil {
 		logErrorAndSendMessage(session, interaction, "서버 정보를 불러오는 중 오류가 발생했습니다.", err)
 		return
 	}
-	var members []*discordgo.Member                 // 음성 채널에 있는 사용자 목록을 저장합니다.
-	memberMap := make(map[string]*discordgo.Member) // 사용자 ID를 키로 사용자 정보를 저장합니다.
-	for _, member := range guild.Members {
-		memberMap[member.User.ID] = member
-	}
-
-	// 음성 채널에 있는 사용자 목록을 불러옵니다.
-	for _, vs := range guild.VoiceStates {
-		if vs.ChannelID == channel.ID {
-			if member, ok := memberMap[vs.UserID]; ok {
-				members = append(members, member)
+	// 채널에 참여중인 사용자를 불러옵니다.
+	var members []string
+	for _, voiceState := range guild.VoiceStates {
+		if voiceState.ChannelID == interaction.ChannelID {
+			member, err := session.GuildMember(guildId, voiceState.UserID)
+			if err != nil {
+				log.Printf("Error getting member: %v", err)
+				continue
+			}
+			username := member.Nick
+			if username == "" {
+				username = member.User.GlobalName
 			}
+			members = append(members, username)
 		}
 	}
-	log.Printf("Members: %+v", members)
-	// 음성 채널에 사용자가 없으면 에러 메시지를 전송합니다.
-	if len(members) == 0 {
-		sendMessage(session, interaction, "음성 채널에 사용자가 없습니다.")
+	// 채널에 참여중인 사용자가 없거나 요청한 사용자 수가 채널에 참여중인 사용자 수보다 많을 경우 에러 메시지를 전송합니다.
+	log.Printf("GetRandomUserByChannel | Current member arr: %+v", members)
+	if len(members) == 0 || count > len(members) {
+		logErrorAndSendMessage(session, interaction, "채널에 참여중인 사용자가 부족합니다.", nil)
 		return
 	}
-
-	// 음성 채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.
-	randomMember := members[util.PickRandomNumber(len(members))]
-	// 사용자의 닉네임이 없으면 사용자 이름을 사용합니다.
-	username := randomMember.Nick
-	if username == "" {
-		username = randomMember.User.GlobalName
+	// 모든 사용자를 선택할 경우 모든 사용자를 선택했다는 메시지를 전송합니다.
+	if count == len(members) {
+		sendMessage(session, interaction, "모든 사용자가 선택되었습니다.")
+		return
 	}
-	sendMessage(session, interaction, fmt.Sprintf("랜덤으로 선택된 사람은 %s입니다!", username))
+	// 랜덤으로 사용자를 선택합니다.
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	r.Shuffle(len(members), func(i, j int) {
+		members[i], members[j] = members[j], members[i]
+	})
+	members = members[:count]
+	// 선택된 사용자를 메시지로 전송합니다.
+	sendMessage(session, interaction, fmt.Sprintf("선택된 사람은 %v 입니다!", strings.Join(members, ", ")))
+	return
 }
 
 // retrospectiveToString 회고 목록을 문자열로 변환합니다.

internal/util/util.go

@@ -3,7 +3,6 @@ package util
 import (
 	"fmt"
 	"io"
-	"math/rand"
 	"os"
 	"strings"
 	"time"
@@ -56,13 +55,3 @@ func LoadFile(filePath string) ([]byte, error) {
 	}
 	return arr, nil
 }
-
-// PickRandomNumber 0부터 upperBound-1 사이의 랜덤한 숫자를 반환 (upperBound는 포함하지 않음) upperBound가 0 이하일 경우 0을 반환
-func PickRandomNumber(upperBound int) int {
-	if upperBound <= 0 {
-		return 0
-	}
-	source := rand.NewSource(time.Now().UnixNano())
-	r := rand.New(source)
-	return r.Intn(upperBound)
-}

internal/util/util_test.go

@@ -2,7 +2,6 @@ package util
 
 import (
 	"os"
-	"strconv"
 	"testing"
 	"time"
 )
@@ -109,31 +108,3 @@ func TestLoadFile_InvalidFile(t *testing.T) {
 		t.Errorf("expected error, but nil")
 	}
 }
-
-func TestPickRandomNumber_ValidInput(t *testing.T) {
-	testCases := []int{1, 10, 100, 1000, 10000}
-	for _, tc := range testCases {
-		t.Run("UpperBound: "+strconv.Itoa(tc), func(t *testing.T) {
-			// Act
-			actual := PickRandomNumber(tc)
-			// Assert
-			if actual < 0 || actual >= tc {
-				t.Errorf("expected 0 <= actual < %d, but got %d", tc, actual)
-			}
-		})
-	}
-}
-
-func TestPickRandomNumber_InputLessThenOrEqualZero(t *testing.T) {
-	testCases := []int{0, -1, -10, -100, -1000}
-	for _, tc := range testCases {
-		t.Run("UpperBound: "+strconv.Itoa(tc), func(t *testing.T) {
-			// Act
-			actual := PickRandomNumber(tc)
-			// Assert
-			if actual != 0 {
-				t.Errorf("expected 0, but got %d", actual)
-			}
-		})
-	}
-}