mirror of
https://github.com/mikefarah/yq.git
synced 2024-11-15 07:38:14 +00:00
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/kylelemons/godebug/diff"
|
|
"github.com/mikefarah/yq/v3/pkg/yqlib"
|
|
errors "github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// turn off for unit tests :(
|
|
var forceOsExit = true
|
|
|
|
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().StringVarP(&defaultValue, "defaultValue", "D", "", "default value printed when there are no results")
|
|
cmdCompare.PersistentFlags().BoolVarP(&stripComments, "stripComments", "", false, "strip comments out before comparing")
|
|
cmdCompare.PersistentFlags().BoolVarP(&explodeAnchors, "explodeAnchors", "X", false, "explode anchors")
|
|
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
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
diffString := diff.Diff(strings.TrimSuffix(dataBufferA.String(), "\n"), strings.TrimSuffix(dataBufferB.String(), "\n"))
|
|
|
|
if len(diffString) > 1 {
|
|
cmd.Print(diffString)
|
|
cmd.Print("\n")
|
|
if forceOsExit {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
return nil
|
|
}
|