package config import ( _ "embed" "io/ioutil" "os" "github.com/pelletier/go-toml" ) //go:embed canopeas.example.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"` ServerAPI struct { Endpoint string `toml:"endpoint"` Token string `toml:"token"` } `toml:"serverapi"` Twitter struct { ConsumerKey string `toml:"consumer_key"` ConsumerSecret string `toml:"consumer_secret"` AccessToken string `toml:"access_token"` AccessSecret string `toml:"access_secret"` } `toml:"twitter"` 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 }