Moved internal/ to pkg/ to make building external plugins possible

pkg/config.go

@@ -0,0 +1,125 @@
+package configurator
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"gopkg.in/yaml.v2"
+)
+
+type Options struct{}
+
+type Target struct {
+	Templates  []string `yaml:"templates,omitempty"`
+	FilePaths  []string `yaml:"files,omitempty"`
+	RunTargets []string `yaml:"targets,omitempty"`
+}
+
+type Jwks struct {
+	Uri     string `yaml:"uri"`
+	Retries int    `yaml:"retries"`
+}
+
+type Server struct {
+	Host string `yaml:"host"`
+	Port int    `yaml:"port"`
+	Jwks Jwks   `yaml:"jwks"`
+}
+
+type Config struct {
+	Version     string            `yaml:"version"`
+	Server      Server            `yaml:"server"`
+	SmdClient   SmdClient         `yaml:"smd"`
+	AccessToken string            `yaml:"access-token"`
+	Targets     map[string]Target `yaml:"targets"`
+	PluginDirs  []string          `yaml:"plugins"`
+	CertPath    string            `yaml:"ca-cert"`
+	Options     Options           `yaml:"options"`
+}
+
+// Creates a new config with default parameters.
+func NewConfig() Config {
+	return Config{
+		Version: "",
+		SmdClient: SmdClient{
+			Host: "http://127.0.0.1",
+			Port: 27779,
+		},
+		Targets: map[string]Target{
+			"dnsmasq": Target{
+				Templates: []string{},
+			},
+			"conman": Target{
+				Templates: []string{},
+			},
+			"warewulf": Target{
+				Templates: []string{
+					"templates/warewulf/defaults/node.jinja",
+					"templates/warewulf/defaults/provision.jinja",
+				},
+			},
+		},
+
+		PluginDirs: []string{},
+		Server: Server{
+			Host: "127.0.0.1",
+			Port: 3334,
+			Jwks: Jwks{
+				Uri:     "",
+				Retries: 5,
+			},
+		},
+		Options: Options{},
+	}
+}
+
+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
+	}
+}