117 lines
3.2 KiB
Go
117 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.wh64.net/muffin/goMuffin/commands"
|
|
"git.wh64.net/muffin/goMuffin/components"
|
|
"git.wh64.net/muffin/goMuffin/configs"
|
|
"git.wh64.net/muffin/goMuffin/databases"
|
|
"git.wh64.net/muffin/goMuffin/handler"
|
|
"git.wh64.net/muffin/goMuffin/scripts"
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/devproje/commando"
|
|
"github.com/devproje/commando/types"
|
|
)
|
|
|
|
func main() {
|
|
command := commando.NewCommando(os.Args[1:])
|
|
config := configs.Config
|
|
|
|
if len(os.Args) > 1 {
|
|
command.Root("db-migrate", "봇의 데이터를 MariaDB에서 MongoDB로 옮깁니다.", scripts.DBMigrate)
|
|
command.Root("delete-all-commands", "봇의 모든 슬래시 커맨드를 삭제합니다.", scripts.DeleteAllCommands,
|
|
types.OptionData{
|
|
Name: "id",
|
|
Desc: "봇의 디스코드 아이디",
|
|
Type: types.STRING,
|
|
},
|
|
types.OptionData{
|
|
Name: "isYes",
|
|
Short: []string{"y"},
|
|
Type: types.BOOLEAN,
|
|
},
|
|
)
|
|
|
|
command.Root("export", "머핀봇의 데이터를 추출합니다.", scripts.ExportData,
|
|
types.OptionData{
|
|
Name: "type",
|
|
Desc: "파일형식을 지정합니다. (json, txt(txt는 머핀 데이터만 적용))",
|
|
Type: types.STRING,
|
|
},
|
|
types.OptionData{
|
|
Name: "export-path",
|
|
Desc: "데이터를 저장할 위치를 지정합니다.",
|
|
Type: types.STRING,
|
|
},
|
|
types.OptionData{
|
|
Name: "refined",
|
|
Desc: "머핀 데이터를 있는 그대로 추출할 지, 가려내서 추출할 지를 지정합니다.",
|
|
Type: types.BOOLEAN,
|
|
},
|
|
)
|
|
|
|
err := command.Execute()
|
|
if err != nil {
|
|
_, _ = fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
return
|
|
}
|
|
|
|
dg, err := discordgo.New("Bot " + config.Bot.Token)
|
|
if err != nil {
|
|
log.Println("[goMuffin] 봇의 세션을 만들수가 없어요.")
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
go commands.Discommand.LoadCommand(commands.HelpCommand)
|
|
go commands.Discommand.LoadCommand(commands.DataLengthCommand)
|
|
go commands.Discommand.LoadCommand(commands.LearnCommand)
|
|
go commands.Discommand.LoadCommand(commands.LearnedDataListCommand)
|
|
go commands.Discommand.LoadCommand(commands.InformationCommand)
|
|
go commands.Discommand.LoadCommand(commands.DeleteLearnedDataCommand)
|
|
|
|
go commands.Discommand.LoadComponent(components.DeleteLearnedDataComponent)
|
|
|
|
go dg.AddHandler(handler.MessageCreate)
|
|
go dg.AddHandler(handler.InteractionCreate)
|
|
|
|
dg.Open()
|
|
defer dg.Close()
|
|
|
|
go func() {
|
|
for {
|
|
dg.UpdateCustomStatus("ㅅ살려주세요..!")
|
|
time.Sleep(time.Minute * 10)
|
|
}
|
|
}()
|
|
|
|
for _, cmd := range commands.Discommand.Commands {
|
|
if cmd.Name == commands.HelpCommand.Name {
|
|
// 극한의 성능 똥망 코드 탄생!
|
|
// 무려 똑같은 걸 반복해서 돌리는!
|
|
for _, a := range commands.Discommand.Commands {
|
|
cmd.Options[0].Choices = append(cmd.Options[0].Choices, &discordgo.ApplicationCommandOptionChoice{
|
|
Name: a.Name,
|
|
Value: a.Name,
|
|
})
|
|
}
|
|
}
|
|
|
|
go dg.ApplicationCommandCreate(dg.State.User.ID, "", cmd.ApplicationCommand)
|
|
}
|
|
|
|
defer databases.Database.Client.Disconnect(context.TODO())
|
|
|
|
log.Println("[goMuffin] 봇이 실행되고 있어요. 버전:", configs.MUFFIN_VERSION)
|
|
sc := make(chan os.Signal, 1)
|
|
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
|
|
<-sc
|
|
}
|