config.go

package main

import (
	"fmt"
	"os"

	"gopkg.in/yaml.v3"
)

const configFile = "config.yaml" // Default config file name

type Configuration struct {
	User        string `yaml:"user"`
	Password    string `yaml:"password"`
	MongoHost   string `yaml:"mongoHost"`
	MongoPort   uint16 `yaml:"mongoPort"` // Only allow 0-65535
	SlackToken  string `yaml:"slackToken"`
	SlackChanID string `yaml:"slackChanID"`
}

func defaultConfig() *Configuration {
	return &Configuration{
		User:        "simple-wd",
		Password:    "",
		MongoHost:   "localhost",
		MongoPort:   27017,
		SlackToken:  "",
		SlackChanID: "",
	}
}

func NewConfig() *Configuration {
	newConfig := defaultConfig()

	data, err := os.ReadFile(configFile)
	if err != nil {
		fmt.Fprintf(
			os.Stderr,
			"Failed to read config file '%s': %s\nUsing default configuration values\n",
			configFile,
			err,
		)
		return newConfig
	}

	// yaml.Unmarshal applies the YAML config to the config object
	if err = yaml.Unmarshal(data, &newConfig); err != nil {
		fmt.Fprintf(
			os.Stderr,
			"Failed to parse YAML in config file '%s': %s\nUsing default configuration values\n",
			configFile,
			err,
		)
	}

	return newConfig
}