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

feat: rewrite chatbot

Browse source
This commit is contained in:
Siwoo Jeon 2025-05-26 22:46:37 +09:00
parent 3fcc446cce
commit ec410b9f81
Signed by: migan
GPG key ID: 036E9A8C5E8E48DA
6 changed files with 148 additions and 77 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

92
chatbot/chatbot.go Normal file
View file

@ -0,0 +1,92 @@
package chatbot
import (
"context"
"fmt"
"log"
"math/rand"
"git.wh64.net/muffin/goMuffin/databases"
"git.wh64.net/muffin/goMuffin/utils"
"github.com/bwmarrin/discordgo"
"go.mongodb.org/mongo-driver/v2/bson"
)
type Chatbot struct {
Mode ChatbotMode
s *discordgo.Session
}
var ChatBot *Chatbot
func New(s *discordgo.Session) {
ChatBot = &Chatbot{
Mode: ChatbotDefault,
s: s,
}
}
func (c *Chatbot) SetMode(mode ChatbotMode) *Chatbot {
c.Mode = mode
return c
}
func getDefaultResponse(s *discordgo.Session, question string) string {
var data []databases.Text
var learnData []databases.Learn
var result string
ch := make(chan int)
x := rand.Intn(10)
// 머핀 데이터
go func() {
cur, err := databases.Database.Texts.Find(context.TODO(), bson.D{{Key: "persona", Value: "muffin"}})
if err != nil {
log.Fatalln(err)
}
defer cur.Close(context.TODO())
cur.All(context.TODO(), &data)
ch <- 1
}()
// 지식 데이터
go func() {
cur, err := databases.Database.Learns.Find(context.TODO(), bson.D{{Key: "command", Value: question}})
if err != nil {
log.Fatalln(err)
}
defer cur.Close(context.TODO())
cur.All(context.TODO(), &learnData)
ch <- 1
}()
for range 2 {
<-ch
}
close(ch)
if x > 2 && len(learnData) != 0 {
data := learnData[rand.Intn(len(learnData))]
user, _ := s.User(data.UserId)
result =
fmt.Sprintf("%s\n%s", data.Result, utils.InlineCode(fmt.Sprintf("%s님이 알려주셨어요.", user.Username)))
} else {
result = data[rand.Intn(len(data))].Text
}
return result
}
func (c *Chatbot) GetResponse(question string) string {
switch c.Mode {
case ChatbotDefault:
return getDefaultResponse(c.s, question)
default:
return ""
}
}

8
chatbot/enum.go Normal file
View file

@ -0,0 +1,8 @@
package chatbot
type ChatbotMode int
const (
ChatbotAI ChatbotMode = iota
ChatbotDefault
)

28
chatbot/parser.go Normal file
View file

@ -0,0 +1,28 @@
package chatbot
import (
"strings"
"git.wh64.net/muffin/goMuffin/configs"
"git.wh64.net/muffin/goMuffin/utils"
"github.com/bwmarrin/discordgo"
)
func ParseResult(content string, s *discordgo.Session, m *discordgo.MessageCreate) string {
result := content
userCreatedAt, _ := discordgo.SnowflakeTimestamp(m.Author.ID)
result = strings.ReplaceAll(result, "{user.name}", m.Author.Username)
result = strings.ReplaceAll(result, "{user.mention}", m.Author.Mention())
result = strings.ReplaceAll(result, "{user.globalName}", m.Author.GlobalName)
result = strings.ReplaceAll(result, "{user.id}", m.Author.ID)
result = strings.ReplaceAll(result, "{user.createdAt}", utils.Time(&userCreatedAt, utils.RelativeTime))
result = strings.ReplaceAll(result, "{user.joinedAt}", utils.Time(&m.Member.JoinedAt, utils.RelativeTime))
result = strings.ReplaceAll(result, "{muffin.version}", configs.MUFFIN_VERSION)
result = strings.ReplaceAll(result, "{muffin.updatedAt}", utils.Time(configs.UpdatedAt, utils.RelativeTime))
result = strings.ReplaceAll(result, "{muffin.startedAt}", utils.Time(configs.StartedAt, utils.RelativeTime))
result = strings.ReplaceAll(result, "{muffin.name}", s.State.User.Username)
result = strings.ReplaceAll(result, "{muffin.id}", s.State.User.ID)
return result
}

82
handler/messageCreate.go
View file

@ -4,16 +4,15 @@ import (
"context"
"fmt"
"log"
"math/rand"
"strings"
"time"
"git.wh64.net/muffin/goMuffin/chatbot"
"git.wh64.net/muffin/goMuffin/commands"
"git.wh64.net/muffin/goMuffin/configs"
"git.wh64.net/muffin/goMuffin/databases"
"git.wh64.net/muffin/goMuffin/utils"
"github.com/bwmarrin/discordgo"
"go.mongodb.org/mongo-driver/v2/bson"
)
func argParser(content string) (args []string) {
@ -71,83 +70,12 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
if command == "" {
s.ChannelTyping(m.ChannelID)
var data []databases.Text
var learnData []databases.Learn
var filter bson.D
ch := make(chan int)
x := rand.Intn(10)
channel, _ := s.Channel(m.ChannelID)
if channel.NSFW {
filter = bson.D{{}}
if _, err := databases.Database.Texts.InsertOne(context.TODO(), databases.InsertText{
Text: content,
Persona: fmt.Sprintf("user:%s", m.Author.Username),
CreatedAt: time.Now(),
}); err != nil {
log.Fatalln(err)
}
} else {
filter = bson.D{{Key: "persona", Value: "muffin"}}
}
go func() {
cur, err := databases.Database.Texts.Find(context.TODO(), filter)
if err != nil {
log.Fatalln(err)
}
defer cur.Close(context.TODO())
cur.All(context.TODO(), &data)
ch <- 1
}()
go func() {
cur, err := databases.Database.Learns.Find(context.TODO(), bson.D{{Key: "command", Value: content}})
if err != nil {
log.Fatalln(err)
}
defer cur.Close(context.TODO())
cur.All(context.TODO(), &learnData)
ch <- 1
}()
for range 2 {
<-ch
}
close(ch)
if x > 2 && len(learnData) != 0 {
data := learnData[rand.Intn(len(learnData))]
user, _ := s.User(data.UserId)
result := resultParser(data.Result, s, m)
utils.NewMessageSender(m).
SetContent(fmt.Sprintf("%s\n%s", result, utils.InlineCode(fmt.Sprintf("%s님이 알려주셨어요.", user.Username)))).
SetAllowedMentions(discordgo.MessageAllowedMentions{
Roles: []string{},
Parse: []discordgo.AllowedMentionType{},
Users: []string{},
}).
SetReply(true).
Send()
return
}
utils.NewMessageSender(m).
SetContent(data[rand.Intn(len(data))].Text).
SetAllowedMentions(discordgo.MessageAllowedMentions{
Roles: []string{},
Parse: []discordgo.AllowedMentionType{},
Users: []string{},
}).
result := chatbot.ParseResult(chatbot.ChatBot.GetResponse(content), s, m)
err := utils.NewMessageSender(m).
SetContent(result).
SetReply(true).
Send()
fmt.Println(err)
return
}

12
main.go
View file

@ -2,6 +2,7 @@ package main
import (
"context"
"flag"
"fmt"
"log"
"os"
@ -9,6 +10,7 @@ import (
"syscall"
"time"
"git.wh64.net/muffin/goMuffin/chatbot"
"git.wh64.net/muffin/goMuffin/commands"
"git.wh64.net/muffin/goMuffin/components"
"git.wh64.net/muffin/goMuffin/configs"
@ -79,6 +81,10 @@ func main() {
return
}
isAI := flag.Bool("ai", false, "시작할 때 AI모드로 설정합니다.")
flag.Parse()
dg, _ := discordgo.New("Bot " + config.Bot.Token)
go dg.AddHandler(handler.MessageCreate)
@ -90,6 +96,12 @@ func main() {
log.Fatalln(err)
}
chatbot.New(dg)
if *isAI {
chatbot.ChatBot.SetMode(chatbot.ChatbotAI)
}
defer dg.Close()
// 봇의 상태메세지 변경

3
utils/messageBuilder.go
View file

@ -1,6 +1,8 @@
package utils
import (
"fmt"
"github.com/bwmarrin/discordgo"
)
@ -82,6 +84,7 @@ func (s *MessageSender) Send() error {
Flags: flags,
Reference: reference,
})
fmt.Println(err)
return err
case *InteractionCreate:
if s.Ephemeral {