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