add edit command for direct edit of node entries

2022-09-15 15:32:29 +02:00
parent 8c0710e412
commit 7087f3e4a1
10 changed files with 467 additions and 162 deletions
--- a/internal/app/wwctl/node/edit/main.go
+++ b/internal/app/wwctl/node/edit/main.go
@@ -0,0 +1,103 @@
+package edit
+
+import (
+	"crypto/sha256"
+	"encoding/hex"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+
+	apinode "github.com/hpcng/warewulf/internal/pkg/api/node"
+	"github.com/hpcng/warewulf/internal/pkg/api/routes/wwapiv1"
+	apiutil "github.com/hpcng/warewulf/internal/pkg/api/util"
+	"github.com/hpcng/warewulf/internal/pkg/node"
+	"github.com/hpcng/warewulf/internal/pkg/util"
+	"github.com/hpcng/warewulf/internal/pkg/wwlog"
+	"github.com/spf13/cobra"
+	"gopkg.in/yaml.v2"
+)
+
+func CobraRunE(cmd *cobra.Command, args []string) error {
+	editor := os.Getenv("EDITOR")
+	if editor == "" {
+		editor = "/bin/vi"
+	}
+	if len(args) == 0 {
+		args = append(args, ".*")
+	}
+	filterList := wwapiv1.NodeList{
+		Output: args,
+	}
+	nodeListMsg := apinode.FilteredNodes(&filterList)
+	nodeMap := make(map[string]*node.NodeConf)
+	yaml.Unmarshal([]byte(nodeListMsg.NodeConfMapYaml), nodeMap)
+	file, err := ioutil.TempFile("/tmp", "ww4NodeEdit*.yaml")
+	if err != nil {
+		wwlog.Error("Could not create temp file:%s \n", err)
+	}
+	defer os.Remove(file.Name())
+	for {
+		file.Truncate(0)
+		file.Seek(0, 0)
+		file.WriteString(nodeListMsg.NodeConfMapYaml)
+		file.Seek(0, 0)
+		hasher := sha256.New()
+		if _, err := io.Copy(hasher, file); err != nil {
+			wwlog.Error("Problems getting checksum of file %s\n", err)
+		}
+		sum1 := hex.EncodeToString(hasher.Sum(nil))
+		err = util.ExecInteractive(editor, file.Name())
+		if err != nil {
+			wwlog.Error("Editor process existed with non-zero\n")
+			os.Exit(1)
+		}
+		file.Seek(0, 0)
+		hasher.Reset()
+		if _, err := io.Copy(hasher, file); err != nil {
+			wwlog.Error("Problems getting checksum of file %s\n", err)
+		}
+		sum2 := hex.EncodeToString(hasher.Sum(nil))
+		wwlog.Debug("Hashes are before %s and after %s\n", sum1, sum2)
+		if sum1 != sum2 {
+			wwlog.Debug("Nodes were modified")
+			modifiedNodeMap := make(map[string]*node.NodeConf)
+			file.Seek(0, 0)
+			// ignore error as only may occurs under strange circumstances
+			buffer, _ := io.ReadAll(file)
+			err = yaml.Unmarshal(buffer, modifiedNodeMap)
+			if err == nil {
+				nodeList := make([]string, len(nodeMap))
+				i := 0
+				for key := range nodeMap {
+					nodeList[i] = key
+					i++
+				}
+				yes := apiutil.ConfirmationPrompt(fmt.Sprintf("Are you sure you want to modify %d nodes", len(modifiedNodeMap)))
+				if !yes {
+					break
+				}
+				err = apinode.NodeDelete(&wwapiv1.NodeDeleteParameter{NodeNames: nodeList, Force: true})
+				if err != nil {
+					wwlog.Verbose("Problem deleting nodes before modification %s")
+				}
+				buffer, _ = yaml.Marshal(modifiedNodeMap)
+				err = apinode.NodeAddFromYaml(&wwapiv1.NodeYaml{NodeConfMapYaml: string(buffer)})
+				if err != nil {
+					wwlog.Error("Got following problem when writing back yaml: %s", err)
+					os.Exit(1)
+				}
+				break
+			} else {
+				yes := apiutil.ConfirmationPrompt(fmt.Sprintf("Got following error on parsing: %s, Retry", err))
+				if !yes {
+					break
+				}
+			}
+		} else {
+			break
+		}
+	}
+
+	return nil
+}