/
fmt.go
54 lines (46 loc) · 963 Bytes
/
fmt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
// yamlfmt formats YAML. It reads from standard input or any files given on the command line.
package main
import (
"flag"
"fmt"
"io"
"log"
"os"
"gopkg.in/yaml.v3"
)
var (
strict = flag.Bool("strict", false, "be strict when parsing")
indent = flag.Uint("indent", 2, "default indent")
)
func main() {
flag.Parse()
decoders := []*yaml.Decoder{}
if flag.NArg() == 0 {
decoders = append(decoders, yaml.NewDecoder(os.Stdin))
}
for _, f := range flag.Args() {
r, err := os.Open(f)
if err != nil {
log.Fatal(err)
}
decoders = append(decoders, yaml.NewDecoder(r))
}
for _, d := range decoders {
in := yaml.Node{}
err := d.Decode(&in)
for err == nil {
e := yaml.NewEncoder(os.Stdout)
e.SetIndent(int(*indent))
if err := e.Encode(&in); err != nil {
log.Fatal(err)
}
e.Close()
if err = d.Decode(&in); err == nil {
fmt.Println("---\n")
}
}
if err != nil && err != io.EOF {
log.Fatal(err)
}
}
}