Compare first cut

cmd/compare.go

@@ -0,0 +1,72 @@
+package cmd
+
+import (
+	"bufio"
+	"bytes"
+
+	"github.com/kylelemons/godebug/diff"
+	errors "github.com/pkg/errors"
+	"github.com/spf13/cobra"
+	yaml "gopkg.in/yaml.v3"
+)
+
+func createCompareCmd() *cobra.Command {
+	var cmdCompare = &cobra.Command{
+		Use:     "compare [yaml_file_a] [yaml_file_b]",
+		Aliases: []string{"x"},
+		Short:   "yq x data1.yml data2.yml",
+		Example: `
+yq x - data2.yml # reads from stdin
+`,
+		Long: "Compares two yaml files, prints the difference",
+		RunE: compareDocuments,
+	}
+	cmdCompare.PersistentFlags().StringVarP(&docIndex, "doc", "d", "0", "process document index number (0 based)")
+	cmdCompare.PersistentFlags().BoolVarP(&prettyPrint, "prettyPrint", "P", false, "pretty print (does not have an affect with json output)")
+	return cmdCompare
+}
+
+func compareDocuments(cmd *cobra.Command, args []string) error {
+	if len(args) != 2 {
+		return errors.New("Must provide at 2 yaml files")
+	}
+	if docIndex == "*" {
+		return errors.New("Document splat for compare not yet supported")
+	}
+
+	var _, docIndexInt, errorParsingDocIndex = parseDocumentIndex()
+	if errorParsingDocIndex != nil {
+		return errorParsingDocIndex
+	}
+
+	var dataBucketA yaml.Node
+	var dataBucketB yaml.Node
+	var errorReadingStream error
+	errorReadingStream = readData(args[0], docIndexInt, &dataBucketA)
+	if errorReadingStream != nil {
+		return errorReadingStream
+	}
+
+	errorReadingStream = readData(args[1], docIndexInt, &dataBucketB)
+	if errorReadingStream != nil {
+		return errorReadingStream
+	}
+
+	if prettyPrint {
+		updateStyleOfNode(&dataBucketA, 0)
+		updateStyleOfNode(&dataBucketB, 0)
+	}
+
+	if errorReadingStream != nil {
+		return errorReadingStream
+	}
+
+	var dataBufferA bytes.Buffer
+	printNode(&dataBucketA, bufio.NewWriter(&dataBufferA))
+
+	var dataBufferB bytes.Buffer
+	printNode(&dataBucketB, bufio.NewWriter(&dataBufferB))
+
+	cmd.Print(diff.Diff(dataBufferA.String(), dataBufferB.String()))
+	return nil
+}

cmd/root.go

@@ -44,6 +44,7 @@ func New() *cobra.Command {
 
 	rootCmd.AddCommand(
 		createReadCmd(),
+		createCompareCmd(),
 		createValidateCmd(),
 		createWriteCmd(),
 		createPrefixCmd(),

cmd/utils.go

@@ -71,8 +71,11 @@ func printValue(node *yaml.Node, cmd *cobra.Command) error {
 		cmd.Print(node.Value)
 		return nil
 	}
+	return printNode(node, cmd.OutOrStdout())
+}
 
-	bufferedWriter := bufio.NewWriter(cmd.OutOrStdout())
+func printNode(node *yaml.Node, writer io.Writer) error {
+	bufferedWriter := bufio.NewWriter(writer)
 	defer safelyFlush(bufferedWriter)
 
 	var encoder yqlib.Encoder
@@ -81,10 +84,7 @@ func printValue(node *yaml.Node, cmd *cobra.Command) error {
 	} else {
 		encoder = yqlib.NewYamlEncoder(bufferedWriter)
 	}
-	if err := encoder.Encode(node); err != nil {
-		return err
-	}
-	return nil
+	return encoder.Encode(node)
 }
 
 func setStyle(matchingNodes []*yqlib.NodeContext, style yaml.Style) {