canopeas/config/config.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"`

	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
}