feat: Add save
This commit is contained in:
parent
7b965942c0
commit
6ea5198fd2
4 changed files with 50 additions and 9 deletions
|
@ -10,10 +10,19 @@ import (
|
|||
|
||||
var MUFFIN_VERSION = "0.0.0-gopher_canary.250323b"
|
||||
|
||||
type botConfig struct {
|
||||
Token string
|
||||
Prefix string
|
||||
}
|
||||
|
||||
type trainConfig struct {
|
||||
UserID string
|
||||
}
|
||||
|
||||
// MuffinConfig for Muffin bot
|
||||
type MuffinConfig struct {
|
||||
Token string
|
||||
Prefix string
|
||||
Bot botConfig
|
||||
Train trainConfig
|
||||
DatabaseURL string
|
||||
}
|
||||
|
||||
|
@ -23,15 +32,18 @@ func loadConfig() *MuffinConfig {
|
|||
fmt.Println("[goMuffin] 봇의 설절파일을 불러올 수가 없어요.")
|
||||
log.Fatalln(err)
|
||||
}
|
||||
config := MuffinConfig{}
|
||||
config := MuffinConfig{Bot: botConfig{}, Train: trainConfig{}}
|
||||
setConfig(&config)
|
||||
|
||||
return &config
|
||||
}
|
||||
|
||||
func setConfig(config *MuffinConfig) {
|
||||
config.Prefix = os.Getenv("BOT_PREFIX")
|
||||
config.Token = os.Getenv("BOT_TOKEN")
|
||||
config.Bot.Prefix = os.Getenv("BOT_PREFIX")
|
||||
config.Bot.Token = os.Getenv("BOT_TOKEN")
|
||||
|
||||
config.Train.UserID = os.Getenv("TRAIN_USER_ID")
|
||||
|
||||
config.DatabaseURL = os.Getenv("DATABASE_URL")
|
||||
}
|
||||
|
||||
|
|
|
@ -1,15 +1,23 @@
|
|||
package databases
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"go.mongodb.org/mongo-driver/v2/bson"
|
||||
"go.mongodb.org/mongo-driver/v2/mongo"
|
||||
)
|
||||
|
||||
type InsertText struct {
|
||||
Text string
|
||||
Persona string
|
||||
CreatedAt time.Time `bson:"created_at"`
|
||||
}
|
||||
|
||||
type Text struct {
|
||||
Id bson.ObjectID `bson:"_id"`
|
||||
Text string
|
||||
Persona string
|
||||
CreatedAt bson.DateTime `bson:"created_at"`
|
||||
CreatedAt time.Time `bson:"created_at"`
|
||||
}
|
||||
|
||||
var Texts *mongo.Collection = Client.Database("muffin_ai_test").Collection("text")
|
||||
|
|
|
@ -5,6 +5,7 @@ import (
|
|||
"log"
|
||||
"math/rand"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.wh64.net/muffin/goMuffin/commands"
|
||||
"git.wh64.net/muffin/goMuffin/configs"
|
||||
|
@ -20,10 +21,20 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|||
return
|
||||
}
|
||||
|
||||
if strings.HasPrefix(m.Content, config.Prefix) {
|
||||
content := strings.TrimPrefix(m.Content, config.Prefix)
|
||||
if strings.HasPrefix(m.Content, config.Bot.Prefix) {
|
||||
content := strings.TrimPrefix(m.Content, config.Bot.Prefix)
|
||||
command := commands.Discommand.Aliases[content]
|
||||
|
||||
if m.Author.ID == config.Train.UserID {
|
||||
if _, err := databases.Texts.InsertOne(context.TODO(), databases.InsertText{
|
||||
Text: content,
|
||||
Persona: "muffin",
|
||||
CreatedAt: time.Now(),
|
||||
}); err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
}
|
||||
|
||||
if command == "" {
|
||||
s.ChannelTyping(m.ChannelID)
|
||||
|
||||
|
@ -33,6 +44,14 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|||
channel, _ := s.Channel(m.ChannelID)
|
||||
if channel.NSFW {
|
||||
filter = bson.D{{}}
|
||||
|
||||
if _, err := databases.Texts.InsertOne(context.TODO(), databases.InsertText{
|
||||
Text: content,
|
||||
Persona: "user:" + m.Author.Username,
|
||||
CreatedAt: time.Now(),
|
||||
}); err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
} else {
|
||||
filter = bson.D{{Key: "persona", Value: "muffin"}}
|
||||
}
|
||||
|
@ -46,9 +65,11 @@ func MessageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|||
cur.All(context.TODO(), &datas)
|
||||
|
||||
s.ChannelMessageSendReply(m.ChannelID, datas[rand.Intn(len(datas))].Text, m.Reference())
|
||||
return
|
||||
}
|
||||
|
||||
commands.Discommand.MessageRun(command, s, m)
|
||||
return
|
||||
} else {
|
||||
return
|
||||
}
|
||||
|
|
2
main.go
2
main.go
|
@ -17,7 +17,7 @@ import (
|
|||
func main() {
|
||||
config := configs.Config
|
||||
|
||||
dg, err := discordgo.New("Bot " + config.Token)
|
||||
dg, err := discordgo.New("Bot " + config.Bot.Token)
|
||||
if err != nil {
|
||||
fmt.Println("[goMuffin] 봇의 세션을 만들수가 없어요.")
|
||||
log.Fatalln(err)
|
||||
|
|
Loading…
Reference in a new issue