package cmd import ( "errors" "fmt" "os" "github.com/mikefarah/yq/v4/pkg/yqlib" "github.com/spf13/cobra" ) func createEvaluateSequenceCommand() *cobra.Command { var cmdEvalSequence = &cobra.Command{ Use: "eval [expression] [yaml_file1]...", Aliases: []string{"e"}, Short: "Apply the expression to each document in each yaml file in sequence", Example: ` # Reads field under the given path for each file yq e '.a.b' f1.yml f2.yml # Prints out the file yq e sample.yaml # Pipe from STDIN ## use '-' as a filename to pipe from STDIN cat file2.yml | yq e '.a.b' file1.yml - file3.yml # Creates a new yaml document ## Note that editing an empty file does not work. yq e -n '.a.b.c = "cat"' # Update a file inplace yq e '.a.b = "cool"' -i file.yaml `, Long: `yq is a portable command-line YAML processor (https://github.com/mikefarah/yq/) See https://mikefarah.gitbook.io/yq/ for detailed documentation and examples. ## Evaluate Sequence ## This command iterates over each yaml document from each given file, applies the expression and prints the result in sequence.`, RunE: evaluateSequence, } return cmdEvalSequence } func processExpression(expression string) string { var prettyPrintExp = `(... | (select(tag != "!!str"), select(tag == "!!str") | select(test("(?i)^(y|yes|n|no|on|off)$") | not)) ) style=""` if prettyPrint && expression == "" { return prettyPrintExp } else if prettyPrint { return fmt.Sprintf("%v | %v", expression, prettyPrintExp) } return expression } func evaluateSequence(cmd *cobra.Command, args []string) error { // 0 args, read std in // 1 arg, null input, process expression // 1 arg, read file in sequence // 2+ args, [0] = expression, file the rest var err error firstFileIndex, err := initCommand(cmd, args) if err != nil { return err } stat, _ := os.Stdin.Stat() pipingStdIn := (stat.Mode() & os.ModeCharDevice) == 0 out := cmd.OutOrStdout() if writeInplace { // only use colors if its forced colorsEnabled = forceColor writeInPlaceHandler := yqlib.NewWriteInPlaceHandler(args[firstFileIndex]) out, err = writeInPlaceHandler.CreateTempFile() if err != nil { return err } // need to indirectly call the function so that completedSuccessfully is // passed when we finish execution as opposed to now defer func() { writeInPlaceHandler.FinishWriteInPlace(completedSuccessfully) }() } format, err := yqlib.OutputFormatFromString(outputFormat) if err != nil { return err } printerWriter := configurePrinterWriter(format, out) printer := yqlib.NewPrinter(printerWriter, format, unwrapScalar, colorsEnabled, indent, !noDocSeparators) decoder, err := configureDecoder() if err != nil { return err } streamEvaluator := yqlib.NewStreamEvaluator() if frontMatter != "" { frontMatterHandler := yqlib.NewFrontMatterHandler(args[firstFileIndex]) err = frontMatterHandler.Split() if err != nil { return err } args[firstFileIndex] = frontMatterHandler.GetYamlFrontMatterFilename() if frontMatter == "process" { reader := frontMatterHandler.GetContentReader() printer.SetAppendix(reader) defer yqlib.SafelyCloseReader(reader) } defer frontMatterHandler.CleanUp() } switch len(args) { case 0: if pipingStdIn { err = streamEvaluator.EvaluateFiles(processExpression(""), []string{"-"}, printer, leadingContentPreProcessing, decoder) } else { cmd.Println(cmd.UsageString()) return nil } case 1: if nullInput { err = streamEvaluator.EvaluateNew(processExpression(args[0]), printer, "") } else { err = streamEvaluator.EvaluateFiles(processExpression(""), []string{args[0]}, printer, leadingContentPreProcessing, decoder) } default: err = streamEvaluator.EvaluateFiles(processExpression(args[0]), args[1:], printer, leadingContentPreProcessing, decoder) } completedSuccessfully = err == nil if err == nil && exitStatus && !printer.PrintedAnything() { return errors.New("no matches found") } return err }