57d28c5bc9
Before #1628, all default constructors for nodes and profiles attempted to construct objects with pre-populated empty fields (e.g., for pointers to member structs and for collections that default to nil). Changes to MergeNode in #1629 changed this behavior intentionally to simplify the generation of minimal json; but this had the unintended side-effect of breaking existing templates that may now encounter a nil pointer, particularly when accessing member structs. Introduced here, the Expand() methods on Node and Profile are now responsible for populating such fields, and Node.Expand() is now called during template processing. Signed-off-by: Jonathon Anderson <janderson@ciq.com>
96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package overlay
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
|
|
"github.com/warewulf/warewulf/internal/pkg/kernel"
|
|
"github.com/warewulf/warewulf/internal/pkg/node"
|
|
)
|
|
|
|
/*
|
|
struct which contains the variables to which are available in
|
|
the templates.
|
|
*/
|
|
type TemplateStruct struct {
|
|
Id string
|
|
Hostname string
|
|
BuildHost string
|
|
BuildTime string
|
|
BuildTimeUnix string
|
|
BuildSource string
|
|
Ipaddr string
|
|
Ipaddr6 string
|
|
Netmask string
|
|
Network string
|
|
NetworkCIDR string
|
|
Overlay string
|
|
Ipv6 bool
|
|
Dhcp warewulfconf.DHCPConf
|
|
Nfs warewulfconf.NFSConf
|
|
Ssh warewulfconf.SSHConf
|
|
Warewulf warewulfconf.WarewulfConf
|
|
Tftp warewulfconf.TFTPConf
|
|
Paths warewulfconf.BuildConfig
|
|
AllNodes []node.Node
|
|
node.Node
|
|
// backward compatiblity
|
|
Container string
|
|
ThisNode *node.Node
|
|
}
|
|
|
|
/*
|
|
Initialize an TemplateStruct with the given node.NodeInfo
|
|
*/
|
|
func InitStruct(overlayName string, nodeData node.Node, allNodes []node.Node) (TemplateStruct, error) {
|
|
var tstruct TemplateStruct
|
|
tstruct.Overlay = overlayName
|
|
hostname, _ := os.Hostname()
|
|
tstruct.BuildHost = hostname
|
|
controller := warewulfconf.Get()
|
|
tstruct.ThisNode = &nodeData
|
|
tstruct.ThisNode.Expand()
|
|
if tstruct.ThisNode.Kernel.Version == "" {
|
|
if kernel_ := kernel.FromNode(tstruct.ThisNode); kernel_ != nil {
|
|
tstruct.ThisNode.Kernel.Version = kernel_.Version()
|
|
}
|
|
}
|
|
tstruct.Nfs = *controller.NFS
|
|
tstruct.Ssh = *controller.SSH
|
|
tstruct.Dhcp = *controller.DHCP
|
|
tstruct.Tftp = *controller.TFTP
|
|
tstruct.Paths = *controller.Paths
|
|
tstruct.Warewulf = *controller.Warewulf
|
|
tstruct.Ipaddr = controller.Ipaddr
|
|
tstruct.Ipaddr6 = controller.Ipaddr6
|
|
tstruct.Netmask = controller.Netmask
|
|
tstruct.Network = controller.Network
|
|
// init some convenience vars
|
|
tstruct.Id = nodeData.Id()
|
|
tstruct.Hostname = nodeData.Id()
|
|
tstruct.Container = nodeData.ContainerName
|
|
// Backwards compatibility for templates using "Keys"
|
|
tstruct.AllNodes = allNodes
|
|
for i := range tstruct.AllNodes {
|
|
tstruct.AllNodes[i].Expand()
|
|
}
|
|
dt := time.Now()
|
|
tstruct.BuildTime = dt.Format("01-02-2006 15:04:05 MST")
|
|
tstruct.BuildTimeUnix = strconv.FormatInt(dt.Unix(), 10)
|
|
// tstruct.Node.Tags = map[string]string{}
|
|
var buf bytes.Buffer
|
|
enc := gob.NewEncoder(&buf)
|
|
dec := gob.NewDecoder(&buf)
|
|
if err := enc.Encode(nodeData); err != nil {
|
|
return tstruct, err
|
|
}
|
|
if err := dec.Decode(&tstruct); err != nil {
|
|
return tstruct, err
|
|
}
|
|
return tstruct, nil
|
|
}
|