Rewrote generators to use plugin system with default plugins

internal/generator/plugins/conman/conman.go

@@ -0,0 +1,46 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/OpenCHAMI/configurator/internal/generator"
+	"github.com/OpenCHAMI/configurator/internal/util"
+	"github.com/nikolalohinski/gonja/v2"
+	"github.com/nikolalohinski/gonja/v2/exec"
+)
+
+type Conman struct{}
+
+func (g *Conman) GetName() string {
+	return "conman"
+}
+
+func (g *Conman) GetGroups() []string {
+	return []string{"conman"}
+}
+
+func (g *Conman) Generate(config *configurator.Config, opts ...util.Option) ([]byte, error) {
+	params := generator.GetParams(opts...)
+	var (
+		template = params["template"].(string)
+		path     = config.TemplatePaths[template]
+	)
+	data := exec.NewContext(map[string]any{})
+
+	t, err := gonja.FromFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read template from file: %v", err)
+	}
+	output := "# ========== GENERATED BY OCHAMI CONFIGURATOR ==========\n"
+	output += "# ======================================================"
+	b := bytes.Buffer{}
+	if err = t.Execute(&b, data); err != nil {
+		return nil, fmt.Errorf("failed to execute: %v", err)
+	}
+
+	return b.Bytes(), nil
+}
+
+var Generator Conman

internal/generator/plugins/coredhcp/coredhcp.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/OpenCHAMI/configurator/internal/util"
+)
+
+type CoreDhcp struct{}
+
+func (g *CoreDhcp) GetName() string {
+	return "coredhcp"
+}
+
+func (g *CoreDhcp) GetGroups() []string {
+	return []string{"coredhcp"}
+}
+
+func (g *CoreDhcp) Generate(config *configurator.Config, opts ...util.Option) ([]byte, error) {
+	return nil, nil
+}
+
+var Generator CoreDhcp

internal/generator/plugins/dnsmasq/dnsmasq.go

@@ -0,0 +1,89 @@
+package main
+
+import (
+	"fmt"
+
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/OpenCHAMI/configurator/internal/generator"
+	"github.com/OpenCHAMI/configurator/internal/util"
+)
+
+type DnsMasq struct{}
+
+func TestGenerateDnsMasq() {
+	var (
+		g      = DnsMasq{}
+		config = &configurator.Config{}
+		client = configurator.SmdClient{}
+	)
+	g.Generate(
+		config,
+		generator.WithTemplate("dnsmasq"),
+		generator.WithClient(client),
+	)
+}
+
+func (g *DnsMasq) GetName() string {
+	return "dnsmasq"
+}
+
+func (g *DnsMasq) GetGroups() []string {
+	return []string{"dnsmasq"}
+}
+
+func (g *DnsMasq) Generate(config *configurator.Config, opts ...util.Option) ([]byte, error) {
+	// make sure we have a valid config first
+	if config == nil {
+		return nil, fmt.Errorf("invalid config (config is nil)")
+	}
+
+	// set all the defaults for variables
+	var (
+		params                                    = generator.GetParams(opts...)
+		template                                  = params["template"].(string) // required param
+		path                                      = config.TemplatePaths[template]
+		eths     []configurator.EthernetInterface = nil
+		err      error                            = nil
+	)
+
+	// if we have a client, try making the request for the ethernet interfaces
+	if client, ok := params["client"].(configurator.SmdClient); ok {
+		eths, err = client.FetchEthernetInterfaces(opts...)
+		if err != nil {
+			return nil, fmt.Errorf("failed to fetch ethernet interfaces with client: %v", err)
+		}
+	}
+
+	// check if we have the required params first
+	if eths == nil {
+		return nil, fmt.Errorf("invalid ethernet interfaces (variable is nil)")
+	}
+	if len(eths) <= 0 {
+		return nil, fmt.Errorf("no ethernet interfaces found")
+	}
+
+	// print message if verbose param found
+	if verbose, ok := params["verbose"].(bool); ok {
+		if verbose {
+			fmt.Printf("path: %s\neth count: %v\n", path, len(eths))
+		}
+	}
+
+	// format output to write to config file
+	output := "# ========== GENERATED BY OCHAMI CONFIGURATOR ==========\n"
+	for _, eth := range eths {
+		if eth.Type == "NodeBMC" {
+			output += "dhcp-host=" + eth.MacAddress + "," + eth.ComponentId + "," + eth.IpAddresses[0].IpAddress + "\n"
+		} else {
+			output += "dhcp-host=" + eth.MacAddress + "," + eth.ComponentId + "," + eth.IpAddresses[0].IpAddress + "\n"
+		}
+	}
+	output += "# ======================================================"
+
+	// apply template substitutions and return output as byte array
+	return generator.ApplyTemplate(path, generator.Mappings{
+		"hosts": output,
+	})
+}
+
+var Generator DnsMasq

internal/generator/plugins/powerman/powerman.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/OpenCHAMI/configurator/internal/util"
+)
+
+type Powerman struct{}
+
+func (g *Powerman) GetName() string {
+	return "powerman"
+}
+
+func (g *Powerman) GetGroups() []string {
+	return []string{"powerman"}
+}
+
+func (g *Powerman) Generate(config *configurator.Config, opts ...util.Option) ([]byte, error) {
+	return nil, nil
+}
+
+var Generator Powerman

internal/generator/plugins/syslog/syslog.go

@@ -0,0 +1,22 @@
+package main
+
+import (
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/OpenCHAMI/configurator/internal/util"
+)
+
+type Syslog struct{}
+
+func (g *Syslog) GetName() string {
+	return "syslog"
+}
+
+func (g *Syslog) GetGroups() []string {
+	return []string{"log"}
+}
+
+func (g *Syslog) Generate(config *configurator.Config, opts ...util.Option) ([]byte, error) {
+	return nil, nil
+}
+
+var Generator Syslog

internal/generator/plugins/warewulf/warewulf.go

@@ -0,0 +1,44 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+
+	configurator "github.com/OpenCHAMI/configurator/internal"
+	"github.com/nikolalohinski/gonja/v2"
+	"github.com/nikolalohinski/gonja/v2/exec"
+)
+
+type Warewulf struct{}
+
+func (g *Warewulf) GetName() string {
+	return "warewulf"
+}
+
+func (g *Warewulf) GetGroups() []string {
+	return []string{"warewulf"}
+}
+
+func (g *Warewulf) Generate(config *configurator.Config, template string) ([]byte, error) {
+	var (
+		path = config.TemplatePaths[template]
+	)
+
+	t, err := gonja.FromFile(path)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read template from file: %v", err)
+	}
+	output := "# ========== GENERATED BY OCHAMI CONFIGURATOR ==========\n"
+
+	output += "# ======================================================"
+	data := exec.NewContext(map[string]any{
+		"hosts": output,
+	})
+	b := bytes.Buffer{}
+	if err = t.Execute(&b, data); err != nil {
+		return nil, fmt.Errorf("failed to execute: %v", err)
+	}
+	return nil, nil
+}
+
+var Generator Warewulf