Feat/#18 사용자 랜덤 선택 기능 구현 (#19)

* feat: Added command

* feat: Implemented user random choice

* docs: Docs for `getRandomUserByChannel`

* feat: Implemented random number generator in util

* test: Tested `PickRandomNumber`

* feat: Modified `PickRandomNumber` logic

- upperBound가 0 이하일 경우 0을 반환하는 로직 추가

* feat: add exception handling to restrict command execution to voice channels

- Check the channel type to ensure the command runs only in voice channels
- Send an error message to the user if the command is executed in a text channel

Author: tidavid1

internal/handler/commands.go

@@ -11,6 +11,7 @@ const (
 	commandGetTodayRetrospectives     = "오늘-회고-보기"
 	commandGetScrumByDate             = "다짐-보기"
 	commandGetRetrospectivesByDate    = "회고-보기"
+	commandRandomUserPick             = "사용자-랜덤-선택"
 )
 
 const (
@@ -76,4 +77,8 @@ var commands = []*discordgo.ApplicationCommand{
 			},
 		},
 	},
+	{
+		Name:        commandRandomUserPick,
+		Description: "채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.",
+	},
 }

internal/handler/handlers.go

@@ -28,6 +28,7 @@ var (
 		commandGetTodayRetrospectives:     getTodayRetrospectives,
 		commandGetScrumByDate:             getScrumsByDate,
 		commandGetRetrospectivesByDate:    getRetrospectivesByDate,
+		commandRandomUserPick:             getRandomUserByChannel,
 	}
 )
 
@@ -180,6 +181,56 @@ func getScrumsByDate(session *discordgo.Session, interaction *discordgo.Interact
 	sendMessage(session, interaction, scrumsToString(date, scrums))
 }
 
+// getRandomUserByChannel 채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.
+func getRandomUserByChannel(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
+	channel, err := session.Channel(interaction.ChannelID)
+	// 채널 정보를 불러오는 중 오류가 발생하면 에러 메시지를 전송합니다.
+	if err != nil {
+		logErrorAndSendMessage(session, interaction, "채널을 불러오는 중 오류가 발생했습니다.", err)
+		return
+	}
+	// 음성 채널이 아니면 에러 메시지를 전송합니다.
+	if channel.Type != discordgo.ChannelTypeGuildVoice {
+		sendMessage(session, interaction, "음성 채널에서만 사용할 수 있는 명령어입니다.")
+		return
+	}
+	log.Printf("Channel type: %v", channel.Type)
+	guild, err := session.State.Guild(channel.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)
+			}
+		}
+	}
+	// 음성 채널에 사용자가 없으면 에러 메시지를 전송합니다.
+	if len(members) == 0 {
+		sendMessage(session, interaction, "음성 채널에 사용자가 없습니다.")
+		return
+	}
+
+	// 음성 채널에 있는 사용자 중 랜덤으로 한 명을 선택합니다.
+	randomMember := members[util.PickRandomNumber(len(members))]
+	// 사용자의 닉네임이 없으면 사용자 이름을 사용합니다.
+	username := randomMember.Nick
+	if username == "" {
+		username = randomMember.User.GlobalName
+	}
+	sendMessage(session, interaction, fmt.Sprintf("랜덤으로 선택된 사람은 %s입니다!", username))
+}
+
 // retrospectiveToString 회고 목록을 문자열로 변환합니다.
 func retrospectiveToString(date time.Time, retrospectives []*model.RetrospectiveDto) string {
 	var builder strings.Builder

internal/util/util.go

@@ -3,6 +3,7 @@ package util
 import (
 	"fmt"
 	"io"
+	"math/rand"
 	"os"
 	"strings"
 	"time"
@@ -55,3 +56,13 @@ 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 - 1)
+}

internal/util/util_test.go

@@ -108,3 +108,14 @@ func TestLoadFile_InvalidFile(t *testing.T) {
 		t.Errorf("expected error, but nil")
 	}
 }
+
+func TestPickRandomNumber(t *testing.T) {
+	// Arrange
+	upperBound := 10
+	// Act
+	actual := PickRandomNumber(upperBound)
+	// Assert
+	if actual < 0 || actual >= upperBound {
+		t.Errorf("expected 0 <= actual < %d, but got %d", upperBound, actual)
+	}
+}