-
Notifications
You must be signed in to change notification settings - Fork 0
/
cli.go
120 lines (96 loc) · 2.35 KB
/
cli.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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package main
import (
"encoding/json"
"errors"
"flag"
"fmt"
"io"
"os"
"github.com/ryanolee/go-chaff"
)
var (
version = "UNKNOWN"
commit = "UNKNOWN"
buildDate = "UNKNOWN"
)
func main() {
// String Flags
path := flag.String("file", "", "Specify a file path to read the JSON Schema from")
output := flag.String("output", "", "Specify file path to write generated output to.")
// Bool Flags
formatted := flag.Bool("format", false, "Format JSON output.")
showHelp := flag.Bool("help", false, "Print out help.")
verbose := flag.Bool("verbose", false, "Print out detailed error information.")
showVersion := flag.Bool("version", false, "Print out cli version information.")
flag.Parse()
if *showHelp {
fmt.Println("CLI too for generating random JSON data matching given JSON schema\nUsage: go-chaff [flags]")
flag.PrintDefaults()
os.Exit(0)
}
if *showVersion {
fmt.Printf("Version: %s\nCommit: %s\nBuilt On: %s\n", version, commit, buildDate)
os.Exit(0)
}
var generator chaff.RootGenerator;
var err error
if *path != "" {
generator, err = chaff.ParseSchemaFileWithDefaults(*path)
checkErr(err)
} else if hasStdin() {
stdin := readStdin()
generator, err = chaff.ParseSchemaWithDefaults(stdin)
checkErr(err)
} else {
checkErr(fmt.Errorf("no schema specified! (On Stdin or through the --file flag)"))
}
if *verbose {
for key, value := range generator.Metadata.Errors {
fmt.Printf("[%s]%e", key, value)
}
}
result := generator.GenerateWithDefaults()
checkErr(err)
var res []byte
if *formatted {
res, err = json.MarshalIndent(result, "", " ")
} else {
res, err = json.Marshal(result)
}
checkErr(err)
if *output != "" {
writeFile(res, *output)
} else {
fmt.Print(string(res))
}
}
func writeFile(data []byte, filepath string){
_, err := os.Stat(filepath)
if !errors.Is(err, os.ErrNotExist) {
checkErr(fmt.Errorf("file '%s' already exists", filepath ))
}
file, err := os.Create(filepath)
checkErr(err)
_, err = file.Write(data)
checkErr(err)
err = file.Close()
checkErr(err)
}
func readStdin() []byte{
if hasStdin() {
stdin, err := io.ReadAll(os.Stdin)
checkErr(err)
return stdin
}
return nil
}
func hasStdin() bool {
stat, _ := os.Stdin.Stat()
return (stat.Mode() & os.ModeCharDevice) == 0
}
func checkErr(err error){
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}