45 lines
843 B
Go
45 lines
843 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,
|
|
message string))
|
|
|
|
func SetCommand(aliases []string, command func(
|
|
s *discordgo.Session,
|
|
m *discordgo.MessageCreate,
|
|
message string)) {
|
|
for _, alias := range aliases {
|
|
CommandRegister[alias] = command
|
|
log.Printf("added %s to Register", alias)
|
|
}
|
|
}
|
|
|
|
var Config ConfigRegister
|