mirror of
https://github.com/mikefarah/yq.git
synced 2024-11-13 22:38:04 +00:00
83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"strings"
|
|
|
|
"github.com/kylelemons/godebug/diff"
|
|
"github.com/mikefarah/yq/v3/pkg/yqlib"
|
|
errors "github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func createCompareCmd() *cobra.Command {
|
|
var cmdCompare = &cobra.Command{
|
|
Use: "compare [yaml_file_a] [yaml_file_b]",
|
|
Aliases: []string{"x"},
|
|
Short: "yq x [--prettyPrint/-P] dataA.yaml dataB.yaml 'b.e(name==fr*).value'",
|
|
Example: `
|
|
yq x - data2.yml # reads from stdin
|
|
yq x -pp dataA.yaml dataB.yaml '**' # compare paths
|
|
yq x -d1 dataA.yaml dataB.yaml 'a.b.c'
|
|
`,
|
|
Long: "Deeply compares two yaml files, prints the difference. Use with prettyPrint flag to ignore formatting differences.",
|
|
RunE: compareDocuments,
|
|
}
|
|
cmdCompare.PersistentFlags().StringVarP(&docIndex, "doc", "d", "0", "process document index number (0 based, * for all documents)")
|
|
cmdCompare.PersistentFlags().StringVarP(&printMode, "printMode", "p", "v", "print mode (v (values, default), p (paths), pv (path and value pairs)")
|
|
cmdCompare.PersistentFlags().BoolVarP(&prettyPrint, "prettyPrint", "P", false, "pretty print (does not have an affect with json output)")
|
|
cmdCompare.PersistentFlags().StringVarP(&defaultValue, "defaultValue", "D", "", "default value printed when there are no results")
|
|
return cmdCompare
|
|
}
|
|
|
|
func compareDocuments(cmd *cobra.Command, args []string) error {
|
|
var path = ""
|
|
|
|
if len(args) < 2 {
|
|
return errors.New("Must provide at 2 yaml files")
|
|
} else if len(args) > 2 {
|
|
path = args[2]
|
|
}
|
|
|
|
var updateAll, docIndexInt, errorParsingDocIndex = parseDocumentIndex()
|
|
if errorParsingDocIndex != nil {
|
|
return errorParsingDocIndex
|
|
}
|
|
|
|
var matchingNodesA []*yqlib.NodeContext
|
|
var matchingNodesB []*yqlib.NodeContext
|
|
var errorDoingThings error
|
|
|
|
matchingNodesA, errorDoingThings = readYamlFile(args[0], path, updateAll, docIndexInt)
|
|
|
|
if errorDoingThings != nil {
|
|
return errorDoingThings
|
|
}
|
|
|
|
matchingNodesB, errorDoingThings = readYamlFile(args[1], path, updateAll, docIndexInt)
|
|
if errorDoingThings != nil {
|
|
return errorDoingThings
|
|
}
|
|
|
|
if prettyPrint {
|
|
setStyle(matchingNodesA, 0)
|
|
setStyle(matchingNodesB, 0)
|
|
}
|
|
|
|
var dataBufferA bytes.Buffer
|
|
var dataBufferB bytes.Buffer
|
|
errorDoingThings = printResults(matchingNodesA, bufio.NewWriter(&dataBufferA))
|
|
if errorDoingThings != nil {
|
|
return errorDoingThings
|
|
}
|
|
errorDoingThings = printResults(matchingNodesB, bufio.NewWriter(&dataBufferB))
|
|
if errorDoingThings != nil {
|
|
return errorDoingThings
|
|
}
|
|
|
|
cmd.Print(diff.Diff(strings.TrimSuffix(dataBufferA.String(), "\n"), strings.TrimSuffix(dataBufferB.String(), "\n")))
|
|
cmd.Print("\n")
|
|
return nil
|
|
}
|