package main import ( "bytes" "encoding/json" "flag" "fmt" "io/ioutil" "os" "github.com/kisom/goutils/lib" ) func prettify(file string, validateOnly bool) error { var in []byte var err error if file == "-" { in, err = ioutil.ReadAll(os.Stdin) } else { in, err = ioutil.ReadFile(file) } if err != nil { lib.Warn(err, "ReadFile") return err } var buf = &bytes.Buffer{} err = json.Indent(buf, in, "", " ") if err != nil { lib.Warn(err, "%s", file) return err } if validateOnly { return nil } if file == "-" { _, err = os.Stdout.Write(buf.Bytes()) } else { err = ioutil.WriteFile(file, buf.Bytes(), 0644) } if err != nil { lib.Warn(err, "WriteFile") } return err } func compact(file string, validateOnly bool) error { var in []byte var err error if file == "-" { in, err = ioutil.ReadAll(os.Stdin) } else { in, err = ioutil.ReadFile(file) } if err != nil { lib.Warn(err, "ReadFile") return err } var buf = &bytes.Buffer{} err = json.Compact(buf, in) if err != nil { lib.Warn(err, "%s", file) return err } if validateOnly { return nil } if file == "-" { _, err = os.Stdout.Write(buf.Bytes()) } else { err = ioutil.WriteFile(file, buf.Bytes(), 0644) } if err != nil { lib.Warn(err, "WriteFile") } return err } func usage() { progname := lib.ProgName() fmt.Printf(`Usage: %s [-h] files... %s is used to lint and prettify (or compact) JSON files. The files will be updated in-place. Flags: -c Compact files. -h Print this help message. -n Don't prettify; only perform validation. `, progname, progname) } func init() { flag.Usage = usage } func main() { var shouldCompact, validateOnly bool flag.BoolVar(&shouldCompact, "c", false, "Compact files instead of prettifying.") flag.BoolVar(&validateOnly, "n", false, "Don't write changes; only perform validation.") flag.Parse() action := prettify if shouldCompact { action = compact } var errCount int for _, fileName := range flag.Args() { err := action(fileName, validateOnly) if err != nil { errCount++ } } if errCount > 0 { lib.Errx(lib.ExitFailure, "Not all files succeeded.") } }