56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package config
|
|
|
|
import (
|
|
"git.haelnorr.com/h/golib/env"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type DBConfig struct {
|
|
User string // ENV DB_USER: Database user for authentication (required)
|
|
Password string // ENV DB_PASSWORD: Database password for authentication (required)
|
|
Host string // ENV DB_HOST: Database host address (required)
|
|
Port uint16 // ENV DB_PORT: Database port (default: 5432)
|
|
DB string // ENV DB_NAME: Database name to connect to (required)
|
|
SSL string // ENV DB_SSL: SSL mode for connection (default: disable)
|
|
}
|
|
|
|
func setupDB() (*DBConfig, error) {
|
|
cfg := &DBConfig{
|
|
User: env.String("DB_USER", ""),
|
|
Password: env.String("DB_PASSWORD", ""),
|
|
Host: env.String("DB_HOST", ""),
|
|
Port: env.UInt16("DB_PORT", 5432),
|
|
DB: env.String("DB_NAME", ""),
|
|
SSL: env.String("DB_SSL", "disable"),
|
|
}
|
|
|
|
// Validate SSL mode
|
|
validSSLModes := map[string]bool{
|
|
"disable": true,
|
|
"require": true,
|
|
"verify-ca": true,
|
|
"verify-full": true,
|
|
"allow": true,
|
|
"prefer": true,
|
|
}
|
|
if !validSSLModes[cfg.SSL] {
|
|
return nil, errors.Errorf("Invalid DB_SSL value: %s. Must be one of: disable, allow, prefer, require, verify-ca, verify-full", cfg.SSL)
|
|
}
|
|
|
|
// Check required fields
|
|
if cfg.User == "" {
|
|
return nil, errors.New("Envar not set: DB_USER")
|
|
}
|
|
if cfg.Password == "" {
|
|
return nil, errors.New("Envar not set: DB_PASSWORD")
|
|
}
|
|
if cfg.Host == "" {
|
|
return nil, errors.New("Envar not set: DB_HOST")
|
|
}
|
|
if cfg.DB == "" {
|
|
return nil, errors.New("Envar not set: DB_NAME")
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|