-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
72 lines (59 loc) · 1.73 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
package main
import (
"flag"
"fmt"
"io"
log "log/slog"
"os"
"path"
"strings"
"github.com/dramich/aws-mocker/pkg/mock"
"github.com/dramich/aws-mocker/pkg/writer"
)
func main() {
var (
mockOpts mock.Options
logLevel string
)
flag.StringVar(&mockOpts.BaseDir, "dir", "", "Base directory for the module (required)")
flag.StringVar(&mockOpts.OutputDir, "output-dir", "", "Output directory for the generated file, if not provided will write to stdout")
flag.StringVar(&mockOpts.PackageName, "package-name", "awsmocked", "Name of the generated package")
flag.StringVar(&mockOpts.SearchPackages, "packages", "", "Comma separated list of packages to search (required)")
flag.BoolVar(&mockOpts.ClientDefault, "default-panic", false, "Add a panic for Operations that are not mocked")
flag.StringVar(&logLevel, "log-level", "info", "Set the log level [debug, info, warn, error]")
flag.Parse()
log.SetDefault(log.New(log.NewTextHandler(os.Stderr, &log.HandlerOptions{
Level: logLevelFromArg(logLevel),
})))
if mockOpts.SearchPackages == "" || mockOpts.BaseDir == "" {
fmt.Println("'packages' and 'dir' are required flags")
flag.Usage()
os.Exit(1)
}
var w io.Writer
if mockOpts.OutputDir == "" {
w = os.Stdout
} else {
w = writer.New(path.Join(mockOpts.OutputDir, mockOpts.PackageName+".go"))
}
mockOpts.Writer = w
if err := mock.Run(&mockOpts); err != nil {
log.Error(err.Error())
os.Exit(1)
}
}
func logLevelFromArg(arg string) log.Level {
switch strings.ToLower(arg) {
case "debug":
return log.LevelDebug
case "info":
return log.LevelInfo
case "warn":
return log.LevelWarn
case "error":
return log.LevelError
default:
log.Warn("Unable to parse log level, defaulting to 'Info'")
return log.LevelInfo
}
}