muffin/goMuffin
1
1
Fork 0
Code Issues 1 Pull requests Projects Releases 3 Packages Wiki Activity Actions

fix: disallow mentions

Browse source
This commit is contained in:
Siwoo Jeon 2025-05-11 12:09:21 +09:00
parent a95bf28efa
commit 9cc5dc1c47
Signed by: migan
GPG key ID: 036E9A8C5E8E48DA
1 changed files with 27 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
handler/messageCreate.go
View file

@ -72,8 +72,8 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
if command == "" { if command == "" {
s.ChannelTyping(m.ChannelID) s.ChannelTyping(m.ChannelID)
var datas []databases.Text var data []databases.Text
var learnDatas []databases.Learn var learnData []databases.Learn
var filter bson.D var filter bson.D
ch := make(chan int) ch := make(chan int)
@ -103,21 +103,21 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
defer cur.Close(context.TODO()) defer cur.Close(context.TODO())
cur.All(context.TODO(), &datas) cur.All(context.TODO(), &data)
ch <- 1 ch <- 1
}() }()
go func() { go func() {
cur, err := databases.Learns.Find(context.TODO(), bson.D{{Key: "command", Value: content}}) cur, err := databases.Learns.Find(context.TODO(), bson.D{{Key: "command", Value: content}})
if err != nil { if err != nil {
if err == mongo.ErrNilDocument { if err == mongo.ErrNilDocument {
learnDatas = []databases.Learn{} learnData = []databases.Learn{}
} }
log.Fatalln(err) log.Fatalln(err)
} }
defer cur.Close(context.TODO()) defer cur.Close(context.TODO())
cur.All(context.TODO(), &learnDatas) cur.All(context.TODO(), &learnData)
ch <- 1 ch <- 1
}() }()
@ -126,16 +126,34 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
} }
close(ch) close(ch)
if x > 2 && len(learnDatas) != 0 { if x > 2 && len(learnData) != 0 {
data := learnDatas[rand.Intn(len(learnDatas))] data := learnData[rand.Intn(len(learnData))]
user, _ := s.User(data.UserId) user, _ := s.User(data.UserId)
result := resultParser(data.Result, s, m) result := resultParser(data.Result, s, m)
s.ChannelMessageSendReply(m.ChannelID, fmt.Sprintf("%s\n%s", result, utils.InlineCode(fmt.Sprintf("%s님이 알려주셨어요.", user.Username))), m.Reference()) // s.ChannelMessageSendReply(m.ChannelID, fmt.Sprintf("%s\n%s", result, utils.InlineCode(fmt.Sprintf("%s님이 알려주셨어요.", user.Username))), m.Reference())
s.ChannelMessageSendComplex(m.ChannelID, &discordgo.MessageSend{
Reference: m.Reference(),
Content: fmt.Sprintf("%s\n%s", result, utils.InlineCode(fmt.Sprintf("%s님이 알려주셨어요.", user.Username))),
AllowedMentions: &discordgo.MessageAllowedMentions{
Roles: []string{},
Parse: []discordgo.AllowedMentionType{},
Users: []string{},
},
})
return return
} }
s.ChannelMessageSendReply(m.ChannelID, datas[rand.Intn(len(datas))].Text, m.Reference()) // s.ChannelMessageSendReply(m.ChannelID, data[rand.Intn(len(data))].Text, m.Reference())
s.ChannelMessageSendComplex(m.ChannelID, &discordgo.MessageSend{
Reference: m.Reference(),
Content: data[rand.Intn(len(data))].Text,
AllowedMentions: &discordgo.MessageAllowedMentions{
Roles: []string{},
Parse: []discordgo.AllowedMentionType{},
Users: []string{},
},
})
return return
} }