93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/pelletier/go-toml"
|
|
)
|
|
|
|
var defaultConfig = []byte("")
|
|
|
|
type Config struct {
|
|
Token string `toml:"token"`
|
|
Prefix string `toml:"prefix"`
|
|
MCM struct {
|
|
Token string `toml:"token"`
|
|
URL string `toml:"url"`
|
|
} `toml:"mcm"`
|
|
Server struct {
|
|
Address string `toml:"address"`
|
|
Port int `toml:"port"`
|
|
} `toml:"server"`
|
|
DBPath string `toml:"db_path"`
|
|
MCPath string `toml:"mc_path"`
|
|
ImgurClientID string `toml:"imgur_client_id"`
|
|
|
|
StaffRoles []string `toml:"staff_roles"`
|
|
Echoes []Echo `toml:"echoes"`
|
|
MessageRoles []MessageRole `toml:"message_roles"`
|
|
RegisterRole string `toml:"register_role"`
|
|
RegisteredChannel string `toml:"registered_channel"`
|
|
LeaveChannel string `toml:"leave_channel"`
|
|
FiredRole string `toml:"fired_role"`
|
|
MemeRate string `toml:"meme_rate"`
|
|
Insult struct {
|
|
Targets []string `toml:"targets"`
|
|
Comparisons []string `toml:"comparisons"`
|
|
Adjectives []string `toml:"adjectives"`
|
|
Nouns []string `toml:"nouns"`
|
|
} `toml:"insult"`
|
|
Compliment struct {
|
|
Verbs []string `toml:"verbs"`
|
|
Nouns []string `toml:"nouns"`
|
|
MinorThings []string `toml:"minor_things"`
|
|
} `toml:"compliment"`
|
|
}
|
|
|
|
type MessageRole struct {
|
|
ChannelID string `toml:"channel_id"`
|
|
MessageID string `toml:"message_id"`
|
|
RoleID string `toml:"role_id"`
|
|
Emoji string `toml:"emoji"`
|
|
}
|
|
|
|
type Echo struct {
|
|
Name string `toml:"name"`
|
|
Aliases []string `toml:"aliases"`
|
|
Message string `toml:"message"`
|
|
Help string `toml:"help"`
|
|
}
|
|
|
|
func Load(configPath string) (*Config, error) {
|
|
var err error
|
|
var configContent []byte
|
|
if len(configPath) == 0 {
|
|
configPath = "sedbot.toml"
|
|
}
|
|
|
|
configContent, err = ioutil.ReadFile(configPath)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
if err = ioutil.WriteFile(configPath, defaultConfig, os.ModePerm); err != nil {
|
|
return nil, err
|
|
}
|
|
configContent = defaultConfig
|
|
} else {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
var cfg Config
|
|
tree, err := toml.LoadBytes(configContent)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := tree.Unmarshal(&cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|