-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
65 lines (53 loc) · 1.47 KB
/
main.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
// A tool that takes a JSON or YAML on standard input and produces tj tree from it
package main
import (
"encoding/json"
"fmt"
"os"
"github.com/dottedmag/must"
"github.com/spf13/pflag"
)
func args() formatConfig {
var packageName, variableName, tjPrefix, tjPackage string
pflag.StringVar(&packageName, "package", "", "Generate package declaration (requires --variable)")
pflag.StringVar(&variableName, "variable", "", "Generate variable declaration")
pflag.StringVar(&tjPrefix, "tj-prefix", "tj", "Import prefix for to use for tj package")
pflag.StringVar(&tjPackage, "tj-package", "github.com/dottedmag/tj", "Import path for tj package")
pflag.Usage = func() {
fmt.Fprintf(os.Stderr, "Usage: %s [OPTION]... < INPUT.json > OUTPUT.go\n", os.Args[0])
pflag.PrintDefaults()
}
pflag.Parse()
if packageName != "" && variableName == "" {
panic("--package requires --variable")
}
return formatConfig{
w: os.Stdout,
packageName: packageName,
variableName: variableName,
tjPrefix: tjPrefix,
tjPackage: tjPackage,
}
}
func main() {
cfg := args()
var val any
must.OK(json.NewDecoder(os.Stdin).Decode(&val))
formatHeader(cfg)
format(cfg, val)
}
func keysToStrings(i interface{}) interface{} {
switch x := i.(type) {
case map[interface{}]interface{}:
m2 := map[string]interface{}{}
for k, v := range x {
m2[k.(string)] = keysToStrings(v)
}
return m2
case []interface{}:
for i, v := range x {
x[i] = keysToStrings(v)
}
}
return i
}