mirror of
https://github.com/Luzifer/discord-community.git
synced 2024-11-08 23:20:01 +00:00
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"os"
|
|
"text/template"
|
|
|
|
korvike "github.com/Luzifer/korvike/functions"
|
|
"github.com/pkg/errors"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type (
|
|
configFile struct {
|
|
BotToken string `yaml:"bot_token"`
|
|
GuildID string `yaml:"guild_id"`
|
|
StoreLocation string `yaml:"store_location"`
|
|
|
|
ModuleConfigs []moduleConfig `yaml:"module_configs"`
|
|
}
|
|
|
|
moduleConfig struct {
|
|
ID string `yaml:"id"`
|
|
Type string `yaml:"type"`
|
|
Attributes moduleAttributeStore `yaml:"attributes"`
|
|
}
|
|
)
|
|
|
|
func newConfigFromFile(filename string) (*configFile, error) {
|
|
f, err := os.Open(filename)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "opening config file")
|
|
}
|
|
defer f.Close()
|
|
|
|
configContent, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "reading config file")
|
|
}
|
|
|
|
tpl, err := template.New("config").Funcs(korvike.GetFunctionMap()).Parse(string(configContent))
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "parsing config file template")
|
|
}
|
|
|
|
renderedConfig := new(bytes.Buffer)
|
|
if err = tpl.Execute(renderedConfig, nil); err != nil {
|
|
return nil, errors.Wrap(err, "rendering config template")
|
|
}
|
|
|
|
var (
|
|
decoder = yaml.NewDecoder(renderedConfig)
|
|
tmp configFile
|
|
)
|
|
|
|
decoder.SetStrict(true)
|
|
return &tmp, errors.Wrap(decoder.Decode(&tmp), "decoding config")
|
|
}
|