status/core/config.go

package core

import (
	"fmt"
	"io"
	"os"
	"path"

	"github.com/pelletier/go-toml/v2"
)

type Config struct {
	_ConfigDirectory string `toml:"-"`
	SiteName         string
	IconSrc          string
	// db location
	Database          string
	UnixSocket        string
	BasicAuthUsername string
	BasicAuthPassword string
	Service           []ServiceConfig
	Matrix            MatrixConfig
}

func LoadConfigFromTOML(configPath string) (*Config, error) {
	file, err := os.Open(configPath)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	fileData, err := io.ReadAll(file)
	if err != nil {
		return nil, err
	}

	var cfg Config
	err = toml.Unmarshal([]byte(fileData), &cfg)
	if err != nil {
		return nil, err
	}
	return &cfg, nil
}

func LoadConfig() (*Config, error) {
	configLocations := []string{
		//"/etc/status/",
		"proto/",
		//"../",
	}
	errMsgPath := ""
	for _, directory := range configLocations {
		configPath := path.Join(directory, "config.toml")
		errMsgPath += fmt.Sprintf("\n- %v", configPath)
		config, err := LoadConfigFromTOML(configPath)
		if err == nil {
			config._ConfigDirectory = directory
			return config, nil
		}
		if os.IsNotExist(err) {
			continue
		} else {
			return nil, err
		}
	}
	return nil, fmt.Errorf("Cannot load config file:" + errMsgPath + "\nSuggestion: cp proto/config.toml.in proto/config.toml")
}