62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Config struct {
|
|
Port int `yaml:"port"`
|
|
LogLevel string `yaml:"log_level"`
|
|
DataDir string `yaml:"data_dir"`
|
|
SCSTConfig string `yaml:"scst_config"`
|
|
BaculaConfig string `yaml:"bacula_config"`
|
|
Security SecurityConfig `yaml:"security"`
|
|
}
|
|
|
|
type SecurityConfig struct {
|
|
RequireHTTPS bool `yaml:"require_https"`
|
|
AllowedUsers []string `yaml:"allowed_users"`
|
|
}
|
|
|
|
func Load() (*Config, error) {
|
|
cfg := &Config{
|
|
Port: 8080,
|
|
LogLevel: "info",
|
|
DataDir: "/var/lib/bams",
|
|
SCSTConfig: "/etc/scst.conf",
|
|
BaculaConfig: "/etc/bacula/bacula-sd.conf",
|
|
Security: SecurityConfig{
|
|
RequireHTTPS: false,
|
|
AllowedUsers: []string{},
|
|
},
|
|
}
|
|
|
|
// Load from environment or config file
|
|
if configPath := os.Getenv("BAMS_CONFIG"); configPath != "" {
|
|
data, err := os.ReadFile(configPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to read config file: %w", err)
|
|
}
|
|
if err := yaml.Unmarshal(data, cfg); err != nil {
|
|
return nil, fmt.Errorf("failed to parse config file: %w", err)
|
|
}
|
|
}
|
|
|
|
// Override with environment variables
|
|
if portStr := os.Getenv("BAMS_PORT"); portStr != "" {
|
|
if port, err := strconv.Atoi(portStr); err == nil {
|
|
cfg.Port = port
|
|
}
|
|
}
|
|
|
|
if logLevel := os.Getenv("BAMS_LOG_LEVEL"); logLevel != "" {
|
|
cfg.LogLevel = logLevel
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|