45 lines
866 B
Go
45 lines
866 B
Go
package core
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
"github.com/bwmarrin/discordgo"
|
|
)
|
|
|
|
type ConfigRegister struct {
|
|
Token string `yaml:"token"`
|
|
Name string `yaml:"name"`
|
|
}
|
|
|
|
func (c *ConfigRegister) LoadConf() *ConfigRegister {
|
|
yamlFile, err := ioutil.ReadFile("config.yaml")
|
|
if err != nil {
|
|
log.Printf("yamlFile.Get err #%v ", err)
|
|
}
|
|
err = yaml.Unmarshal(yamlFile, c)
|
|
if err != nil {
|
|
log.Fatalf("Unmarshal: %v", err)
|
|
}
|
|
return c
|
|
}
|
|
|
|
var CommandRegister = make(
|
|
map[string]func(
|
|
s *discordgo.Session,
|
|
m *discordgo.MessageCreate))
|
|
|
|
func SetCommand(aliases []string, command func(
|
|
s *discordgo.Session,
|
|
m *discordgo.MessageCreate),
|
|
discord *discordgo.Session) {
|
|
for _, alias := range aliases {
|
|
CommandRegister[alias] = command
|
|
log.Printf("added %s to Register", alias)
|
|
}
|
|
discord.AddHandler(command)
|
|
}
|
|
|
|
var Config ConfigRegister
|