100 lines
2.3 KiB
Go
100 lines
2.3 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
|
|
configurator "github.com/OpenCHAMI/configurator/pkg"
|
|
"github.com/OpenCHAMI/configurator/pkg/client"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Jwks struct {
|
|
Uri string `yaml:"uri"`
|
|
Retries int `yaml:"retries,omitempty"`
|
|
}
|
|
|
|
type Server struct {
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
Jwks Jwks `yaml:"jwks,omitempty"`
|
|
}
|
|
|
|
type Config struct {
|
|
Version string `yaml:"version,omitempty"`
|
|
Server Server `yaml:"server,omitempty"`
|
|
SmdClient client.SmdClient `yaml:"smd,omitempty"`
|
|
AccessToken string `yaml:"access-token,omitempty"`
|
|
Targets map[string]configurator.Target `yaml:"targets,omitempty"`
|
|
PluginDirs []string `yaml:"plugins,omitempty"`
|
|
CertPath string `yaml:"cacert,omitempty"`
|
|
}
|
|
|
|
// Creates a new config with default parameters.
|
|
func New() Config {
|
|
return Config{
|
|
Version: "",
|
|
SmdClient: client.SmdClient{Host: "http://127.0.0.1:27779"},
|
|
Targets: map[string]configurator.Target{},
|
|
PluginDirs: []string{},
|
|
Server: Server{
|
|
Host: "127.0.0.1:3334",
|
|
Jwks: Jwks{
|
|
Uri: "",
|
|
Retries: 5,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func Load(path string) Config {
|
|
var c Config = New()
|
|
file, err := os.ReadFile(path)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to read config file")
|
|
return c
|
|
}
|
|
err = yaml.Unmarshal(file, &c)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to unmarshal config")
|
|
return c
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (config *Config) Save(path string) {
|
|
path = filepath.Clean(path)
|
|
if path == "" || path == "." {
|
|
path = "config.yaml"
|
|
}
|
|
data, err := yaml.Marshal(config)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to marshal config")
|
|
return
|
|
}
|
|
err = os.WriteFile(path, data, os.ModePerm)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to write default config file")
|
|
return
|
|
}
|
|
}
|
|
|
|
func SaveDefault(path string) {
|
|
path = filepath.Clean(path)
|
|
if path == "" || path == "." {
|
|
path = "config.yaml"
|
|
}
|
|
var c = New()
|
|
data, err := yaml.Marshal(c)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to marshal config")
|
|
return
|
|
}
|
|
err = os.WriteFile(path, data, os.ModePerm)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to write default config file")
|
|
return
|
|
}
|
|
}
|