Files
warewulf/internal/app/wwctl/node/list/main.go
Xu Yang 16183e603c add multiple formats support
Signed-off-by: Xu Yang <xyang@ciq.com>
2024-03-19 01:48:08 -06:00

53 lines
1.5 KiB
Go

package list
import (
"strings"
"github.com/spf13/cobra"
"github.com/warewulf/warewulf/internal/app/wwctl/helper"
apinode "github.com/warewulf/warewulf/internal/pkg/api/node"
"github.com/warewulf/warewulf/internal/pkg/api/routes/wwapiv1"
"github.com/warewulf/warewulf/internal/pkg/wwlog"
)
func CobraRunE(vars *variables) func(cmd *cobra.Command, args []string) (err error) {
return func(cmd *cobra.Command, args []string) (err error) {
if len(args) > 0 && strings.Contains(args[0], ",") {
args = strings.FieldsFunc(args[0], func(r rune) bool { return r == ',' })
}
req := wwapiv1.GetNodeList{
Nodes: args,
Type: wwapiv1.GetNodeList_Simple,
}
if vars.showAll {
req.Type = wwapiv1.GetNodeList_All
} else if vars.showIpmi {
req.Type = wwapiv1.GetNodeList_Ipmi
} else if vars.showNet {
req.Type = wwapiv1.GetNodeList_Network
} else if vars.showLong {
req.Type = wwapiv1.GetNodeList_Long
} else if vars.showFullAll {
req.Type = wwapiv1.GetNodeList_FullAll
} else if vars.showYaml {
req.Type = wwapiv1.GetNodeList_YAML
} else if vars.showJson {
req.Type = wwapiv1.GetNodeList_JSON
}
nodeInfo, err := apinode.NodeList(&req)
if len(nodeInfo.Output) > 0 {
if req.Type == wwapiv1.GetNodeList_YAML || req.Type == wwapiv1.GetNodeList_JSON {
wwlog.Info(nodeInfo.Output[0])
} else {
ph := helper.NewPrintHelper(strings.Split(nodeInfo.Output[0], ":=:"))
for _, val := range nodeInfo.Output[1:] {
ph.Append(strings.Split(val, ":=:"))
}
ph.Render()
}
}
return
}
}