85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package configuration
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"go.dev.pztrn.name/metricator/internal/application"
|
|
"go.dev.pztrn.name/metricator/internal/logger"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
var (
|
|
ErrConfigurationFileDoesNotExist = errors.New("configuration file does not exist")
|
|
ErrConfigurationFilePathUndefined = errors.New("configuration file path wasn't provided")
|
|
)
|
|
|
|
// Config is an application's configuration.
|
|
type Config struct {
|
|
// Applications describes configuration for remote application's endpoints.
|
|
// Key is an application's name.
|
|
Applications map[string]*application.Config `yaml:"applications"`
|
|
// Logger is a logging configuration.
|
|
Logger *logger.Config `yaml:"logger"`
|
|
configPath string
|
|
}
|
|
|
|
// NewConfig returns new configuration.
|
|
func NewConfig() *Config {
|
|
// Fields are initialized when parsing YAML file.
|
|
// nolint:exhaustivestruct
|
|
c := &Config{}
|
|
c.initialize()
|
|
|
|
return c
|
|
}
|
|
|
|
// Initializes configuration.
|
|
func (c *Config) initialize() {
|
|
flag.StringVar(&c.configPath, "config", "", "Configuration file path.")
|
|
}
|
|
|
|
// Parse parses configuration.
|
|
func (c *Config) Parse() error {
|
|
if c.configPath == "" {
|
|
return ErrConfigurationFilePathUndefined
|
|
}
|
|
|
|
if strings.HasPrefix(c.configPath, "~") {
|
|
userDir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", "file path normalization: getting user's home directory", err)
|
|
}
|
|
|
|
c.configPath = strings.Replace(c.configPath, "~", userDir, 1)
|
|
}
|
|
|
|
cfgPath, err := filepath.Abs(c.configPath)
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", "file path normalization: getting absolute path", err)
|
|
}
|
|
|
|
c.configPath = cfgPath
|
|
|
|
if c.configPath == "" {
|
|
return fmt.Errorf("%s: %w", "file path normalization", ErrConfigurationFilePathUndefined)
|
|
}
|
|
|
|
fileData, err := ioutil.ReadFile(c.configPath)
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", "configuration file read", err)
|
|
}
|
|
|
|
err = yaml.Unmarshal(fileData, c)
|
|
if err != nil {
|
|
return fmt.Errorf("%s: %w", "configuration file parsing", err)
|
|
}
|
|
|
|
return nil
|
|
}
|