From 9b32aaf5814d1c238b024867484a023ebd8f6d50 Mon Sep 17 00:00:00 2001
From: Siwoo Jeon <me@migan.co.kr>
Date: Sat, 5 Apr 2025 17:03:16 +0900
Subject: [PATCH] fix: Finding nil data

---
 commands/deleteLearnedData.go | 37 ++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 16 deletions(-)

diff --git a/commands/deleteLearnedData.go b/commands/deleteLearnedData.go
index 465861e..8e6b4c6 100644
--- a/commands/deleteLearnedData.go
+++ b/commands/deleteLearnedData.go
@@ -10,7 +10,6 @@ import (
 	"git.wh64.net/muffin/goMuffin/utils"
 	"github.com/bwmarrin/discordgo"
 	"go.mongodb.org/mongo-driver/v2/bson"
-	"go.mongodb.org/mongo-driver/v2/mongo"
 )
 
 var DeleteLearnedDataCommand *Command = &Command{
@@ -79,23 +78,11 @@ func deleteLearnedDataRun(c *Command, s *discordgo.Session, m any, args *[]strin
 	cur, err := databases.Learns.Find(context.TODO(), bson.M{"user_id": userId, "command": command})
 	if err != nil {
 		embed := &discordgo.MessageEmbed{
-			Title: "❌ 오류",
-			Color: utils.EmbedFail,
-		}
-		if err == mongo.ErrNoDocuments {
-			embed.Description = "해당 하는 지식ㅇ을 찾을 수 없어요."
-			switch m := m.(type) {
-			case *discordgo.MessageCreate:
-				s.ChannelMessageSendEmbedReply(m.ChannelID, embed, m.Reference())
-			case *utils.InteractionCreate:
-				m.EditReply(&discordgo.WebhookEdit{
-					Embeds: &[]*discordgo.MessageEmbed{embed},
-				})
-			}
-			return
+			Title:       "❌ 오류",
+			Description: "데이터를 가져오는데 실패했어요.",
+			Color:       utils.EmbedFail,
 		}
 
-		embed.Description = "데이터를 가져오는데 실패했어요."
 		switch m := m.(type) {
 		case *discordgo.MessageCreate:
 			s.ChannelMessageSendEmbedReply(m.ChannelID, embed, m.Reference())
@@ -109,6 +96,24 @@ func deleteLearnedDataRun(c *Command, s *discordgo.Session, m any, args *[]strin
 
 	cur.All(context.TODO(), &datas)
 
+	if len(datas) < 1 {
+		embed := &discordgo.MessageEmbed{
+			Title:       "❌ 오류",
+			Description: "해당 하는 지식ㅇ을 찾을 수 없어요.",
+			Color:       utils.EmbedFail,
+		}
+
+		switch m := m.(type) {
+		case *discordgo.MessageCreate:
+			s.ChannelMessageSendEmbedReply(m.ChannelID, embed, m.Reference())
+		case *utils.InteractionCreate:
+			m.EditReply(&discordgo.WebhookEdit{
+				Embeds: &[]*discordgo.MessageEmbed{embed},
+			})
+		}
+		return
+	}
+
 	for i := range len(datas) {
 		data := datas[i]