fix: train

This commit is contained in:
Siwoo Jeon 2025-05-26 22:48:23 +09:00
parent e5090fa35d
commit b074e49909
Signed by: migan
GPG key ID: 036E9A8C5E8E48DA

View file

@ -59,16 +59,6 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
args := argParser(content)
command := commands.Discommand.Aliases[args[0]]
if m.Author.ID == config.Train.UserID {
if _, err := databases.Database.Texts.InsertOne(context.TODO(), databases.InsertText{
Text: content,
Persona: "muffin",
CreatedAt: time.Now(),
}); err != nil {
log.Fatalln(err)
}
}
if command == "" {
s.ChannelTyping(m.ChannelID)
@ -158,6 +148,15 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
commands.Discommand.MessageRun(command, s, m, args[1:])
return
} else {
if m.Author.ID == config.Train.UserID {
if _, err := databases.Database.Texts.InsertOne(context.TODO(), databases.InsertText{
Text: m.Content,
Persona: "muffin",
CreatedAt: time.Now(),
}); err != nil {
log.Fatalln(err)
}
}
return
}
}