yq/cmd/evalute_sequence_command.go

65 lines
1.7 KiB
Go
Raw Normal View History

2020-11-06 03:37:01 +00:00
package cmd
import (
"os"
"github.com/mikefarah/yq/v4/pkg/yqlib"
"github.com/spf13/cobra"
)
func createEvaluateSequenceCommand() *cobra.Command {
var cmdEvalSequence = &cobra.Command{
2020-11-13 03:07:11 +00:00
Use: "eval [expression] [yaml_file1]...",
Aliases: []string{"e"},
2020-11-06 03:37:01 +00:00
Short: "Apply expression to each document in each yaml file given in sequence",
Example: `
yq es '.a.b | length' file1.yml file2.yml
yq es < sample.yaml
yq es -n '{"a": "b"}'
`,
Long: "Evaluate Sequence:\nIterate over each yaml document, apply the expression and print the results, in sequence.",
RunE: evaluateSequence,
}
return cmdEvalSequence
}
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
stat, _ := os.Stdin.Stat()
pipingStdIn := (stat.Mode() & os.ModeCharDevice) == 0
2020-11-13 02:19:54 +00:00
out := cmd.OutOrStdout()
fileInfo, _ := os.Stdout.Stat()
if forceColor || (!forceNoColor && (fileInfo.Mode()&os.ModeCharDevice) != 0) {
colorsEnabled = true
}
2020-11-13 02:35:59 +00:00
printer := yqlib.NewPrinter(out, outputToJSON, unwrapScalar, colorsEnabled, indent, !noDocSeparators)
2020-11-13 02:19:54 +00:00
2020-11-06 03:37:01 +00:00
switch len(args) {
case 0:
if pipingStdIn {
2020-11-13 02:19:54 +00:00
err = yqlib.EvaluateFileStreamsSequence("", []string{"-"}, printer)
2020-11-06 03:37:01 +00:00
} else {
cmd.Println(cmd.UsageString())
return nil
}
case 1:
if nullInput {
2020-11-13 02:19:54 +00:00
err = yqlib.EvaluateAllFileStreams(args[0], []string{}, printer)
2020-11-06 03:37:01 +00:00
} else {
2020-11-13 02:19:54 +00:00
err = yqlib.EvaluateFileStreamsSequence("", []string{args[0]}, printer)
2020-11-06 03:37:01 +00:00
}
2020-11-13 02:19:54 +00:00
default:
2020-11-13 03:07:11 +00:00
err = yqlib.EvaluateFileStreamsSequence(args[0], args[1:], printer)
2020-11-06 03:37:01 +00:00
}
2020-11-13 02:19:54 +00:00
cmd.SilenceUsage = true
return err
2020-11-06 03:37:01 +00:00
}