-
-
Notifications
You must be signed in to change notification settings - Fork 684
/
cli.go
159 lines (139 loc) · 4.28 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
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
package app
import (
"fmt"
"os"
"strings"
"github.com/docker-slim/docker-slim/pkg/app"
"github.com/docker-slim/docker-slim/pkg/app/master/commands"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/build"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/containerize"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/convert"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/debug"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/dockerclipm"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/edit"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/help"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/install"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/lint"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/probe"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/profile"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/run"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/server"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/update"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/version"
"github.com/docker-slim/docker-slim/pkg/app/master/commands/xray"
"github.com/docker-slim/docker-slim/pkg/system"
v "github.com/docker-slim/docker-slim/pkg/version"
log "github.com/sirupsen/logrus"
"github.com/urfave/cli/v2"
)
// DockerSlim app CLI constants
const (
AppName = "docker-slim"
AppUsage = "optimize and secure your Docker containers!"
)
func registerCommands() {
//registering commands explicitly instead of relying on init()
//also get to control the order of the commands in the interactive prompt
xray.RegisterCommand()
lint.RegisterCommand()
build.RegisterCommand()
profile.RegisterCommand()
version.RegisterCommand()
help.RegisterCommand()
update.RegisterCommand()
install.RegisterCommand()
edit.RegisterCommand()
probe.RegisterCommand()
convert.RegisterCommand()
run.RegisterCommand()
server.RegisterCommand()
debug.RegisterCommand()
containerize.RegisterCommand()
dockerclipm.RegisterCommand()
}
func newCLI() *cli.App {
registerCommands()
cliApp := cli.NewApp()
cliApp.Version = v.Current()
cliApp.Name = AppName
cliApp.Usage = AppUsage
cliApp.CommandNotFound = func(ctx *cli.Context, command string) {
fmt.Printf("unknown command - %v \n\n", command)
cli.ShowAppHelp(ctx)
}
cliApp.Flags = commands.GlobalFlags()
cliApp.Before = func(ctx *cli.Context) error {
if ctx.Bool(commands.FlagNoColor) {
app.NoColor()
}
if ctx.Bool(commands.FlagDebug) {
log.SetLevel(log.DebugLevel)
} else {
if ctx.Bool(commands.FlagVerbose) {
log.SetLevel(log.InfoLevel)
} else {
logLevel := log.WarnLevel
logLevelName := ctx.String(commands.FlagLogLevel)
switch logLevelName {
case "trace":
logLevel = log.TraceLevel
case "debug":
logLevel = log.DebugLevel
case "info":
logLevel = log.InfoLevel
case "warn":
logLevel = log.WarnLevel
case "error":
logLevel = log.ErrorLevel
case "fatal":
logLevel = log.FatalLevel
case "panic":
logLevel = log.PanicLevel
default:
log.Fatalf("unknown log-level %q", logLevelName)
}
log.SetLevel(logLevel)
}
}
if path := ctx.String(commands.FlagLog); path != "" {
f, err := os.Create(path)
if err != nil {
return err
}
log.SetOutput(f)
}
logFormat := ctx.String(commands.FlagLogFormat)
switch logFormat {
case "text":
log.SetFormatter(&log.TextFormatter{DisableColors: true})
case "json":
log.SetFormatter(new(log.JSONFormatter))
default:
log.Fatalf("unknown log-format %q", logFormat)
}
log.Debugf("sysinfo => %#v", system.GetSystemInfo())
//tmp hack
if !strings.Contains(strings.Join(os.Args, " "), " docker-cli-plugin-metadata") {
app.ShowCommunityInfo()
}
return nil
}
cliApp.After = func(ctx *cli.Context) error {
//tmp hack
if !strings.Contains(strings.Join(os.Args, " "), " docker-cli-plugin-metadata") {
app.ShowCommunityInfo()
}
return nil
}
cliApp.Action = func(ctx *cli.Context) error {
gcvalues, err := commands.GlobalFlagValues(ctx)
if err != nil {
return err
}
ia := commands.NewInteractiveApp(cliApp, gcvalues)
ia.Run()
return nil
}
cliApp.Commands = commands.CLI
return cliApp
}