-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
160 lines (140 loc) · 3.76 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
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package main
import (
"io"
"os"
"github.com/alecthomas/kong"
"github.com/rs/zerolog"
)
var CLI struct {
Input []string `arg:"" optional:"" type:"existingfile" help:"Input file(s)."`
Output string `short:"o" help:"Output file. Defaults to stdout."`
Json bool `short:"j" help:"Force JSON output."`
Yaml bool `short:"y" help:"Force YAML output."`
ArrayStrategy string `short:"m" help:"Set merge strategy for arrays. One of 'replace', 'extend', or 'merge'. Defaults to 'replace'"`
ValueFiles []string `short:"v" help:"Load template context values from a YAML or JSON file." type:"existingfile"`
Debug bool `help:"Set log level to debug."`
LogLevel string `help:"Set the log level." enum:"trace,debug,info,warn,error" default:"warn"`
LogFormat string `help:"Set the log format." enum:"json,text" default:"text"`
}
func setupLogger() zerolog.Logger {
var lvl zerolog.Level
if CLI.Debug {
lvl = zerolog.DebugLevel
} else {
switch CLI.LogLevel {
case "trace":
lvl = zerolog.TraceLevel
case "debug":
lvl = zerolog.DebugLevel
case "info":
lvl = zerolog.InfoLevel
case "warn":
lvl = zerolog.WarnLevel
case "error":
lvl = zerolog.ErrorLevel
}
}
var writer io.Writer
switch CLI.LogFormat {
case "json":
writer = os.Stdout
case "text":
writer = zerolog.ConsoleWriter{Out: os.Stdout}
}
return zerolog.New(writer).Level(lvl).With().Timestamp().Logger()
}
func main() {
kong.Parse(&CLI,
kong.Name("jam"),
kong.Description("A tool for merging and templating YAML and JSON files."),
kong.UsageOnError(),
kong.ConfigureHelp(kong.HelpOptions{
Compact: true,
Summary: true,
}),
)
if len(CLI.Input) == 0 {
CLI.Input = append(CLI.Input, "-")
}
logger := setupLogger()
tplCtx := make(map[string]any)
for _, file := range CLI.ValueFiles {
parser := GuessParser(file, nil)
f, err := os.Open(file)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to open file")
}
data, err := io.ReadAll(f)
f.Close()
if err != nil {
logger.Fatal().Err(err).Msg("Failed to read file")
}
var value any
err = parser.Unmarshal(data, &value)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to parse file")
}
tplCtx = Merge(tplCtx, value, arrayStrategy(ReplaceStrategy)).(map[string]any)
}
var outputParser Parser
var merged any
for _, file := range CLI.Input {
parser := GuessParser(file, YAML)
var f *os.File
if file == "-" {
f = os.Stdin
logger.Debug().Msg("Processing stdin")
} else {
logger.Debug().Str("path", file).Msg("Processing file")
var err error
f, err = os.Open(file)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to open file")
}
if outputParser == nil {
outputParser = parser
}
}
data, err := io.ReadAll(f)
f.Close()
if err != nil {
logger.Fatal().Err(err).Msg("Failed to read file")
}
rendered, err := RenderTemplate(data, tplCtx)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to render template")
}
var value any
err = parser.Unmarshal(rendered, &value)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to parse file")
}
merged = Merge(merged, value, arrayStrategy(CLI.ArrayStrategy))
}
if CLI.Json {
outputParser = JSON
} else if CLI.Yaml {
outputParser = YAML
} else if CLI.Output != "" {
guess := GuessParser(CLI.Output, nil)
if guess != nil {
outputParser = guess
}
} else if outputParser == nil {
outputParser = YAML
}
var f *os.File
if CLI.Output == "" {
f = os.Stdout
} else {
var err error
f, err = os.Create(CLI.Output)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to create output file")
}
}
err := WriteFile(f, outputParser, merged)
if err != nil {
logger.Fatal().Err(err).Msg("Failed to write output")
}
}