homebox/backend/internal/sys/config/conf.go

72 lines
1.8 KiB
Go
Raw Normal View History

2022-08-30 02:30:36 +00:00
package config
import (
"encoding/json"
"errors"
"fmt"
"os"
2022-08-30 02:30:36 +00:00
2023-01-28 21:03:51 +00:00
"github.com/ardanlabs/conf/v3"
2022-08-30 02:30:36 +00:00
)
const (
ModeDevelopment = "development"
ModeProduction = "production"
)
type Config struct {
Mode string `yaml:"mode" conf:"default:development"` // development or production
Web WebConfig `yaml:"web"`
Storage Storage `yaml:"storage"`
Log LoggerConf `yaml:"logger"`
Mailer MailerConf `yaml:"mailer"`
Demo bool `yaml:"demo"`
Debug DebugConf `yaml:"debug"`
Options Options `yaml:"options"`
}
type Options struct {
AllowRegistration bool `yaml:"disable_registration" conf:"default:true"`
AutoIncrementAssetID bool `yaml:"auto_increment_asset_id" conf:"default:true"`
}
type DebugConf struct {
Enabled bool `yaml:"enabled" conf:"default:false"`
Port string `yaml:"port" conf:"default:4000"`
2022-08-30 02:30:36 +00:00
}
type WebConfig struct {
Port string `yaml:"port" conf:"default:7745"`
Host string `yaml:"host"`
MaxUploadSize int64 `yaml:"max_file_upload" conf:"default:10"`
2022-08-30 02:30:36 +00:00
}
2022-10-12 21:03:42 +00:00
// New parses the CLI/Config file and returns a Config struct. If the file argument is an empty string, the
2022-08-30 02:30:36 +00:00
// file is not read. If the file is not empty, the file is read and the Config struct is returned.
2022-10-12 21:03:42 +00:00
func New() (*Config, error) {
2022-08-30 02:30:36 +00:00
var cfg Config
2022-09-14 17:35:23 +00:00
const prefix = "HBOX"
2022-08-30 02:30:36 +00:00
2022-10-12 21:03:42 +00:00
help, err := conf.Parse(prefix, &cfg)
2022-08-30 02:30:36 +00:00
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))
}