mirror of
https://github.com/spf13/cobra
synced 2024-11-18 03:37:11 +00:00
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os/exec"
|
|
)
|
|
|
|
func init() {
|
|
// Mute commands.
|
|
addCmd.SetOut(new(bytes.Buffer))
|
|
addCmd.SetErr(new(bytes.Buffer))
|
|
initCmd.SetOut(new(bytes.Buffer))
|
|
initCmd.SetErr(new(bytes.Buffer))
|
|
}
|
|
|
|
// ensureLF converts any \r\n to \n
|
|
func ensureLF(content []byte) []byte {
|
|
return bytes.Replace(content, []byte("\r\n"), []byte("\n"), -1)
|
|
}
|
|
|
|
// compareFiles compares the content of files with pathA and pathB.
|
|
// If contents are equal, it returns nil.
|
|
// If not, it returns which files are not equal
|
|
// and diff (if system has diff command) between these files.
|
|
func compareFiles(pathA, pathB string) error {
|
|
contentA, err := ioutil.ReadFile(pathA)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
contentB, err := ioutil.ReadFile(pathB)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !bytes.Equal(ensureLF(contentA), ensureLF(contentB)) {
|
|
output := new(bytes.Buffer)
|
|
output.WriteString(fmt.Sprintf("%q and %q are not equal!\n\n", pathA, pathB))
|
|
|
|
diffPath, err := exec.LookPath("diff")
|
|
if err != nil {
|
|
// Don't execute diff if it can't be found.
|
|
return nil
|
|
}
|
|
diffCmd := exec.Command(diffPath, "-u", "--strip-trailing-cr", pathA, pathB)
|
|
diffCmd.Stdout = output
|
|
diffCmd.Stderr = output
|
|
|
|
output.WriteString("$ diff -u " + pathA + " " + pathB + "\n")
|
|
if err := diffCmd.Run(); err != nil {
|
|
output.WriteString("\n" + err.Error())
|
|
}
|
|
return errors.New(output.String())
|
|
}
|
|
return nil
|
|
}
|