13ad67e56e
This commit adds the ability to configure the default IPMI port for a profile or per node. If the port is not set, we default to 623. Updated some areas to default to 'lan' for IPMI interface if it has not been set.
380 lines
10 KiB
Go
380 lines
10 KiB
Go
package set
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/hpcng/warewulf/internal/pkg/container"
|
|
"github.com/hpcng/warewulf/internal/pkg/node"
|
|
"github.com/hpcng/warewulf/internal/pkg/util"
|
|
"github.com/hpcng/warewulf/internal/pkg/warewulfd"
|
|
"github.com/hpcng/warewulf/internal/pkg/wwlog"
|
|
"github.com/manifoldco/promptui"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func CobraRunE(cmd *cobra.Command, args []string) error {
|
|
var err error
|
|
var SetProfiles []string
|
|
|
|
nodeDB, err := node.New()
|
|
if err != nil {
|
|
wwlog.Printf(wwlog.ERROR, "Could not open node configuration: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
nodes, err := nodeDB.FindAllNodes()
|
|
if err != nil {
|
|
wwlog.Printf(wwlog.ERROR, "Cloud not get nodeList: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if !SetNodeAll {
|
|
if len(args) > 0 {
|
|
nodes = node.FilterByName(nodes, args)
|
|
} else {
|
|
cmd.Usage()
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
if len(nodes) == 0 {
|
|
fmt.Printf("No nodes found\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if SetContainer != "" {
|
|
if container.ValidSource(SetContainer) == true {
|
|
imageFile := container.ImageFile(SetContainer)
|
|
if util.IsFile(imageFile) == false {
|
|
wwlog.Printf(wwlog.ERROR, "Container has not been built: %s\n", SetContainer)
|
|
if SetForce == false {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
} else {
|
|
wwlog.Printf(wwlog.ERROR, "Container does not exist: %s\n", SetContainer)
|
|
if SetForce == false {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|
|
|
|
if SetProfile != "" {
|
|
profiles, _ := nodeDB.FindAllProfiles()
|
|
for _, r := range strings.Split(SetProfile, ",") {
|
|
var match bool
|
|
|
|
for _, p := range profiles {
|
|
if p.Id.Get() == r || SetForce {
|
|
match = true
|
|
SetProfiles = append(SetProfiles, r)
|
|
}
|
|
}
|
|
|
|
if !match {
|
|
wwlog.Printf(wwlog.WARN, "Requested profile is undefined: %s\n", r)
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(SetAddProfile) > 0 {
|
|
profiles, _ := nodeDB.FindAllProfiles()
|
|
for _, r := range SetAddProfile {
|
|
var match bool
|
|
|
|
for _, p := range profiles {
|
|
if p.Id.Get() == r || SetForce {
|
|
match = true
|
|
}
|
|
}
|
|
|
|
if !match {
|
|
wwlog.Printf(wwlog.WARN, "Requested profile is undefined: %s\n", r)
|
|
SetAddProfile = util.SliceRemoveElement(SetAddProfile, r)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, n := range nodes {
|
|
wwlog.Printf(wwlog.VERBOSE, "Evaluating node: %s\n", n.Id.Get())
|
|
|
|
if SetComment != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting comment to: %s\n", n.Id.Get(), SetComment)
|
|
n.Comment.Set(SetComment)
|
|
}
|
|
|
|
if SetContainer != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting container name to: %s\n", n.Id.Get(), SetContainer)
|
|
n.ContainerName.Set(SetContainer)
|
|
}
|
|
|
|
if SetInit != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting init command to: %s\n", n.Id.Get(), SetInit)
|
|
n.Init.Set(SetInit)
|
|
}
|
|
|
|
if SetRoot != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting root to: %s\n", n.Id.Get(), SetRoot)
|
|
n.Root.Set(SetRoot)
|
|
}
|
|
|
|
if SetKernel != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting kernel to: %s\n", n.Id.Get(), SetKernel)
|
|
n.KernelVersion.Set(SetKernel)
|
|
}
|
|
|
|
if SetKernelArgs != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting kernel args to: %s\n", n.Id.Get(), SetKernelArgs)
|
|
n.KernelArgs.Set(SetKernelArgs)
|
|
}
|
|
|
|
if SetClusterName != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting cluster name to: %s\n", n.Id.Get(), SetClusterName)
|
|
n.ClusterName.Set(SetClusterName)
|
|
}
|
|
|
|
if SetIpxe != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting iPXE template to: %s\n", n.Id.Get(), SetIpxe)
|
|
n.Ipxe.Set(SetIpxe)
|
|
}
|
|
|
|
if SetRuntimeOverlay != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting runtime overlay to: %s\n", n.Id.Get(), SetRuntimeOverlay)
|
|
n.RuntimeOverlay.Set(SetRuntimeOverlay)
|
|
}
|
|
|
|
if SetSystemOverlay != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting system overlay to: %s\n", n.Id.Get(), SetSystemOverlay)
|
|
n.SystemOverlay.Set(SetSystemOverlay)
|
|
}
|
|
|
|
if SetIpmiIpaddr != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI IP address to: %s\n", n.Id.Get(), SetIpmiIpaddr)
|
|
n.IpmiIpaddr.Set(SetIpmiIpaddr)
|
|
}
|
|
|
|
if SetIpmiNetmask != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI netmask to: %s\n", n.Id.Get(), SetIpmiNetmask)
|
|
n.IpmiNetmask.Set(SetIpmiNetmask)
|
|
}
|
|
|
|
if SetIpmiPort != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI port to: %s\n", n.Id.Get(), SetIpmiPort)
|
|
n.IpmiPort.Set(SetIpmiPort)
|
|
}
|
|
|
|
if SetIpmiGateway != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI gateway to: %s\n", n.Id.Get(), SetIpmiGateway)
|
|
n.IpmiGateway.Set(SetIpmiGateway)
|
|
}
|
|
|
|
if SetIpmiUsername != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI IP username to: %s\n", n.Id.Get(), SetIpmiUsername)
|
|
n.IpmiUserName.Set(SetIpmiUsername)
|
|
}
|
|
|
|
if SetIpmiPassword != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI IP password to: %s\n", n.Id.Get(), SetIpmiPassword)
|
|
n.IpmiPassword.Set(SetIpmiPassword)
|
|
}
|
|
|
|
if SetIpmiInterface != "" {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting IPMI IP interface to: %s\n", n.Id.Get(), SetIpmiInterface)
|
|
n.IpmiInterface.Set(SetIpmiInterface)
|
|
}
|
|
|
|
if SetDiscoverable == true {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting node to discoverable\n", n.Id.Get())
|
|
n.Discoverable.SetB(true)
|
|
}
|
|
|
|
if SetUndiscoverable == true {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting node to undiscoverable\n", n.Id.Get())
|
|
n.Discoverable.SetB(false)
|
|
}
|
|
|
|
if len(SetProfiles) > 0 {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Setting profiles to: %s\n", n.Id.Get(), strings.Join(SetProfiles, ","))
|
|
n.Profiles = SetProfiles
|
|
}
|
|
|
|
if len(SetAddProfile) > 0 {
|
|
for _, p := range SetAddProfile {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, adding profile '%s'\n", n.Id.Get(), p)
|
|
n.Profiles = util.SliceAddUniqueElement(n.Profiles, p)
|
|
}
|
|
}
|
|
|
|
if len(SetDelProfile) > 0 {
|
|
for _, p := range SetDelProfile {
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, deleting profile '%s'\n", n.Id.Get(), p)
|
|
n.Profiles = util.SliceRemoveElement(n.Profiles, p)
|
|
}
|
|
}
|
|
|
|
if SetNetDevDel == true {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
wwlog.Printf(wwlog.ERROR, "Network Device doesn't exist: %s\n", SetNetDev)
|
|
os.Exit(1)
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Deleting network device: %s\n", n.Id.Get(), SetNetDev)
|
|
delete(n.NetDevs, SetNetDev)
|
|
}
|
|
if SetIpaddr != "" {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting Ipaddr to: %s\n", n.Id.Get(), SetNetDev, SetIpaddr)
|
|
n.NetDevs[SetNetDev].Ipaddr.Set(SetIpaddr)
|
|
}
|
|
if SetNetmask != "" {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting netmask to: %s\n", n.Id.Get(), SetNetDev, SetNetmask)
|
|
n.NetDevs[SetNetDev].Netmask.Set(SetNetmask)
|
|
}
|
|
if SetGateway != "" {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting gateway to: %s\n", n.Id.Get(), SetNetDev, SetGateway)
|
|
n.NetDevs[SetNetDev].Gateway.Set(SetGateway)
|
|
}
|
|
|
|
if SetHwaddr != "" {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting HW address to: %s\n", n.Id.Get(), SetNetDev, SetHwaddr)
|
|
n.NetDevs[SetNetDev].Hwaddr.Set(SetHwaddr)
|
|
}
|
|
|
|
if SetType != "" {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting Type %s\n", n.Id.Get(), SetNetDev, SetType)
|
|
n.NetDevs[SetNetDev].Type.Set(SetType)
|
|
}
|
|
|
|
if SetNetDevDefault == true {
|
|
if SetNetDev == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--netdev' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.NetDevs[SetNetDev]; !ok {
|
|
var nd node.NetDevEntry
|
|
n.NetDevs[SetNetDev] = &nd
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting device as default\n", n.Id.Get(), SetNetDev)
|
|
for _, dev := range n.NetDevs {
|
|
// First clear all other devices that might be configured as default
|
|
dev.Default.SetB(false)
|
|
}
|
|
n.NetDevs[SetNetDev].Default.SetB(true)
|
|
}
|
|
|
|
if SetValue != "" {
|
|
if SetKey == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--key/-k' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.Keys[SetKey]; !ok {
|
|
var nd node.Entry
|
|
n.Keys[SetKey] = &nd
|
|
}
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s:%s, Setting Value %s\n", n.Id.Get(), SetKey, SetValue)
|
|
n.Keys[SetKey].Set(SetValue)
|
|
}
|
|
|
|
if SetKeyDel == true {
|
|
if SetKey == "" {
|
|
wwlog.Printf(wwlog.ERROR, "You must include the '--key/-k' option\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
if _, ok := n.Keys[SetKey]; !ok {
|
|
wwlog.Printf(wwlog.ERROR, "Custom parameter doesn't exist: %s\n", SetKey)
|
|
os.Exit(1)
|
|
}
|
|
|
|
wwlog.Printf(wwlog.VERBOSE, "Node: %s, Deleting custom parameter: %s\n", n.Id.Get(), SetNetDev)
|
|
delete(n.Keys, SetKey)
|
|
}
|
|
|
|
err := nodeDB.NodeUpdate(n)
|
|
if err != nil {
|
|
wwlog.Printf(wwlog.ERROR, "%s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
if SetYes == true {
|
|
nodeDB.Persist()
|
|
warewulfd.DaemonReload()
|
|
} else {
|
|
q := fmt.Sprintf("Are you sure you want to modify %d nodes(s)", len(nodes))
|
|
|
|
prompt := promptui.Prompt{
|
|
Label: q,
|
|
IsConfirm: true,
|
|
}
|
|
|
|
result, _ := prompt.Run()
|
|
|
|
if result == "y" || result == "yes" {
|
|
nodeDB.Persist()
|
|
warewulfd.DaemonReload()
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|