forked from mirrors/homebox
80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/ardanlabs/conf/v2"
|
|
"github.com/ardanlabs/conf/v2/yaml"
|
|
|
|
"os"
|
|
)
|
|
|
|
const (
|
|
ModeDevelopment = "development"
|
|
ModeProduction = "production"
|
|
)
|
|
|
|
type Config struct {
|
|
Mode string `yaml:"mode" conf:"default:development"` // development or production
|
|
Web WebConfig `yaml:"web"`
|
|
Database Database `yaml:"database"`
|
|
Log LoggerConf `yaml:"logger"`
|
|
Mailer MailerConf `yaml:"mailer"`
|
|
Seed Seed `yaml:"seed"`
|
|
Swagger SwaggerConf `yaml:"swagger"`
|
|
}
|
|
|
|
type SwaggerConf struct {
|
|
Host string `yaml:"host" conf:"default:localhost:7745"`
|
|
Scheme string `yaml:"scheme" conf:"default:http"`
|
|
}
|
|
|
|
type WebConfig struct {
|
|
Port string `yaml:"port" conf:"default:7745"`
|
|
Host string `yaml:"host"`
|
|
}
|
|
|
|
// NewConfig parses the CLI/Config file and returns a Config struct. If the file argument is an empty string, the
|
|
// file is not read. If the file is not empty, the file is read and the Config struct is returned.
|
|
func NewConfig(file string) (*Config, error) {
|
|
var cfg Config
|
|
|
|
const prefix = "API"
|
|
|
|
help, err := func() (string, error) {
|
|
if _, err := os.Stat(file); errors.Is(err, os.ErrNotExist) {
|
|
return conf.Parse(prefix, &cfg)
|
|
} else {
|
|
yamlData, err := os.ReadFile(file)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return conf.Parse(prefix, &cfg, yaml.WithData(yamlData))
|
|
}
|
|
}()
|
|
|
|
if err != nil {
|
|
if errors.Is(err, conf.ErrHelpWanted) {
|
|
fmt.Println(help)
|
|
os.Exit(0)
|
|
}
|
|
return &cfg, fmt.Errorf("parsing config: %w", err)
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|
|
|
|
// Print prints the configuration to stdout as a json indented string
|
|
// This is useful for debugging. If the marshaller errors out, it will panic.
|
|
func (c *Config) Print() {
|
|
res, err := json.MarshalIndent(c, "", " ")
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
fmt.Println(string(res))
|
|
|
|
}
|