70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package env
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/caarlos0/env/v8"
|
|
"github.com/containrrr/shoutrrr"
|
|
"github.com/go-playground/validator/v10"
|
|
)
|
|
|
|
type Config struct {
|
|
TimeZone string `env:"TZ" envDefault:"Etc/UTC" validate:"timezone"`
|
|
Port int `env:"PORT" envDefault:"8080" validate:"min=1024,max=49151"`
|
|
LogLevel string `env:"LOG_LEVEL" envDefault:"info" validate:"oneof=debug info warn error panic fatal"`
|
|
Version string `env:"VERSION" envDefault:"v0.0.0"`
|
|
SwaggerHost string `env:"SWAGGER_HOST"`
|
|
DB_Host string `env:"DB_HOST"`
|
|
DB_User string `env:"DB_USER"`
|
|
DB_Password string `env:"DB_PASSWORD"`
|
|
DB_Database string `env:"DB_DATABASE"`
|
|
}
|
|
|
|
var errParse = errors.New("error parsing environment variables")
|
|
|
|
func Parse() (*Config, error) {
|
|
cfg := &Config{}
|
|
if err := env.Parse(cfg); err != nil {
|
|
return cfg, err
|
|
}
|
|
if err := validateContent(cfg); err != nil {
|
|
return cfg, err
|
|
}
|
|
setAllDefaultEnvs(cfg)
|
|
return cfg, nil
|
|
}
|
|
|
|
func newEnvValidator() *validator.Validate {
|
|
validate := validator.New()
|
|
validate.RegisterValidation(`shoutrrr`, func(fl validator.FieldLevel) bool {
|
|
value := fl.Field().Interface().(string)
|
|
_, err := shoutrrr.CreateSender(value)
|
|
return err == nil
|
|
})
|
|
|
|
return validate
|
|
}
|
|
|
|
func validateContent(cfg *Config) error {
|
|
validate := newEnvValidator()
|
|
err := validate.Struct(cfg)
|
|
if err != nil {
|
|
if _, ok := err.(*validator.InvalidValidationError); ok {
|
|
return err
|
|
} else {
|
|
for _, err := range err.(validator.ValidationErrors) {
|
|
return err
|
|
}
|
|
}
|
|
return errParse
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func setAllDefaultEnvs(cfg *Config) {
|
|
os.Setenv("TZ", cfg.TimeZone)
|
|
os.Setenv("PORT", fmt.Sprintf("%d", cfg.Port))
|
|
os.Setenv("LOG_LEVEL", cfg.LogLevel)
|
|
}
|