74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package helper
|
|
|
|
import (
|
|
"path"
|
|
"strings"
|
|
|
|
"sigs.k8s.io/yaml"
|
|
)
|
|
|
|
// Node represents a node structure in the YAML file.
|
|
type Node struct {
|
|
Hostname string `yaml:"hostname"`
|
|
IPAddress string `yaml:"ipAddress"`
|
|
}
|
|
|
|
// Config represents the top-level structure of the YAML file.
|
|
type Config struct {
|
|
Nodes []Node `yaml:"nodes"`
|
|
}
|
|
|
|
func ExtractNode(cmd string) string {
|
|
sliceOut := strings.Split(string(cmd), " ")
|
|
nodeout := ""
|
|
|
|
for _, str := range sliceOut {
|
|
if strings.Contains(str, "--nodes=") {
|
|
nodeout = strings.ReplaceAll(str, "--nodes=", "")
|
|
}
|
|
|
|
}
|
|
return string(nodeout)
|
|
}
|
|
func ExtractSchematic(cmd string) string {
|
|
sliceOut := strings.Split(string(cmd), " ")
|
|
schematic := ""
|
|
|
|
for _, str := range sliceOut {
|
|
if strings.Contains(str, "--image=") {
|
|
cleanstring := strings.ReplaceAll(str, "--image=factory.talos.dev/installer/", "")
|
|
schemslice := strings.Split(string(cleanstring), ":")
|
|
schematic = schemslice[0]
|
|
}
|
|
}
|
|
return string(schematic)
|
|
}
|
|
|
|
// CreateIPHostnameMap reads the YAML configuration file and returns a map of ipAddress to hostname.
|
|
func CreateIPHostnameMap() (map[string]string, error) {
|
|
// Read the YAML file.
|
|
data, err := EnvSubst(path.Join(ClusterPath, "/talos/talconfig.yaml"), TalEnv)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Convert the string data to bytes.
|
|
dataBytes := []byte(data)
|
|
|
|
// Unmarshal the YAML file into a Config struct.
|
|
var config Config
|
|
err = yaml.Unmarshal(dataBytes, &config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Create the map of ipAddress to hostname.
|
|
ipHostnameMap := make(map[string]string)
|
|
for _, node := range config.Nodes {
|
|
ipHostnameMap[node.IPAddress] = node.Hostname
|
|
}
|
|
|
|
return ipHostnameMap, nil
|
|
}
|