101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package configurator
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Options struct {
|
|
JwksUri string `yaml:"jwks-uri"`
|
|
JwksRetries int `yaml:"jwks-retries"`
|
|
}
|
|
|
|
type Server struct {
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
}
|
|
type Config struct {
|
|
Version string `yaml:"version"`
|
|
SmdHost string `yaml:"smd-host"`
|
|
SmdPort int `yaml:"smd-port"`
|
|
AccessToken string `yaml:"access-token"`
|
|
TemplatePaths map[string]string `yaml:"templates"`
|
|
Server Server `yaml:"server"`
|
|
Options Options `yaml:"options"`
|
|
}
|
|
|
|
func NewConfig() Config {
|
|
return Config{
|
|
Version: "",
|
|
SmdHost: "http://127.0.0.1",
|
|
SmdPort: 27779,
|
|
TemplatePaths: map[string]string{
|
|
"dnsmasq": "templates/dhcp/dnsmasq.conf",
|
|
"syslog": "templates/syslog/",
|
|
"ansible": "templates/ansible",
|
|
"powerman": "templates/powerman",
|
|
"conman": "templates/conman",
|
|
},
|
|
Server: Server{
|
|
Host: "127.0.0.1",
|
|
Port: 3334,
|
|
},
|
|
Options: Options{
|
|
JwksUri: "",
|
|
JwksRetries: 5,
|
|
},
|
|
}
|
|
}
|
|
|
|
func LoadConfig(path string) Config {
|
|
var c Config = NewConfig()
|
|
file, err := os.ReadFile(path)
|
|
if err != nil {
|
|
log.Printf("failed to read config file: %v\n", err)
|
|
return c
|
|
}
|
|
err = yaml.Unmarshal(file, &c)
|
|
if err != nil {
|
|
log.Fatalf("failed to unmarshal config: %v\n", err)
|
|
return c
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (config *Config) SaveConfig(path string) {
|
|
path = filepath.Clean(path)
|
|
if path == "" || path == "." {
|
|
path = "config.yaml"
|
|
}
|
|
data, err := yaml.Marshal(config)
|
|
if err != nil {
|
|
log.Printf("failed to marshal config: %v\n", err)
|
|
return
|
|
}
|
|
err = os.WriteFile(path, data, os.ModePerm)
|
|
if err != nil {
|
|
log.Printf("failed to write default config file: %v\n", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func SaveDefaultConfig(path string) {
|
|
path = filepath.Clean(path)
|
|
if path == "" || path == "." {
|
|
path = "config.yaml"
|
|
}
|
|
var c = NewConfig()
|
|
data, err := yaml.Marshal(c)
|
|
if err != nil {
|
|
log.Printf("failed to marshal config: %v\n", err)
|
|
return
|
|
}
|
|
err = os.WriteFile(path, data, os.ModePerm)
|
|
if err != nil {
|
|
log.Printf("failed to write default config file: %v\n", err)
|
|
return
|
|
}
|
|
}
|