wip - write in place

cmd/evaluate_all_command.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"fmt"
 	"os"
 
 	"github.com/mikefarah/yq/v4/pkg/yqlib"
@@ -39,7 +40,24 @@ func evaluateAll(cmd *cobra.Command, args []string) error {
 	if forceColor || (!forceNoColor && (fileInfo.Mode()&os.ModeCharDevice) != 0) {
 		colorsEnabled = true
 	}
+
+	if writeInplace && len(args) < 2 {
+		return fmt.Errorf("Write inplace flag only applicable when giving an expression and at least one file")
+	}
+
+	completedSuccessfully := false
+
+	if writeInplace {
+		writeInPlaceHandler := yqlib.NewWriteInPlaceHandler(args[1])
+		out, err = writeInPlaceHandler.CreateTempFile()
+		if err != nil {
+			return err
+		}
+		defer writeInPlaceHandler.FinishWriteInPlace(completedSuccessfully)
+	}
+
 	printer := yqlib.NewPrinter(out, outputToJSON, unwrapScalar, colorsEnabled, indent, !noDocSeparators)
+
 	allAtOnceEvaluator := yqlib.NewAllAtOnceEvaluator()
 	switch len(args) {
 	case 0:
@@ -59,6 +77,8 @@ func evaluateAll(cmd *cobra.Command, args []string) error {
 		err = allAtOnceEvaluator.EvaluateFiles(args[0], args[1:], printer)
 	}
 
+	completedSuccessfully = err == nil
+
 	cmd.SilenceUsage = true
 	return err
 }