@@ -24,7 +24,7 @@ import (
24
24
"github.com/pkg/errors"
25
25
certutil "github.com/rancher/dynamiclistener/cert"
26
26
"github.com/sirupsen/logrus"
27
- "github.com/urfave/cli"
27
+ "github.com/urfave/cli/v2 "
28
28
)
29
29
30
30
func commandSetup (app * cli.Context , cfg * cmds.Server , sc * server.Config ) (string , error ) {
@@ -71,22 +71,22 @@ func check(app *cli.Context, cfg *cmds.Server) error {
71
71
return err
72
72
}
73
73
74
- if len (cmds .ServicesList ) == 0 {
74
+ if len (cmds .ServicesList . Value () ) == 0 {
75
75
// detecting if the command is being run on an agent or server based on presence of the server data-dir
76
76
_ , err := os .Stat (serverConfig .ControlConfig .DataDir )
77
77
if err != nil {
78
78
if ! os .IsNotExist (err ) {
79
79
return err
80
80
}
81
81
logrus .Infof ("Agent detected, checking agent certificates" )
82
- cmds .ServicesList = services .Agent
82
+ cmds .ServicesList = cli . NewStringSlice ( services .Agent ... )
83
83
} else {
84
84
logrus .Infof ("Server detected, checking agent and server certificates" )
85
- cmds .ServicesList = services .All
85
+ cmds .ServicesList = cli . NewStringSlice ( services .All ... )
86
86
}
87
87
}
88
88
89
- fileMap , err := services .FilesForServices (serverConfig .ControlConfig , cmds .ServicesList )
89
+ fileMap , err := services .FilesForServices (serverConfig .ControlConfig , cmds .ServicesList . Value () )
90
90
if err != nil {
91
91
return err
92
92
}
@@ -171,22 +171,22 @@ func rotate(app *cli.Context, cfg *cmds.Server) error {
171
171
return err
172
172
}
173
173
174
- if len (cmds .ServicesList ) == 0 {
174
+ if len (cmds .ServicesList . Value () ) == 0 {
175
175
// detecting if the command is being run on an agent or server based on presence of the server data-dir
176
176
_ , err := os .Stat (serverConfig .ControlConfig .DataDir )
177
177
if err != nil {
178
178
if ! os .IsNotExist (err ) {
179
179
return err
180
180
}
181
181
logrus .Infof ("Agent detected, rotating agent certificates" )
182
- cmds .ServicesList = services .Agent
182
+ cmds .ServicesList = cli . NewStringSlice ( services .Agent ... )
183
183
} else {
184
184
logrus .Infof ("Server detected, rotating agent and server certificates" )
185
- cmds .ServicesList = services .All
185
+ cmds .ServicesList = cli . NewStringSlice ( services .All ... )
186
186
}
187
187
}
188
188
189
- fileMap , err := services .FilesForServices (serverConfig .ControlConfig , cmds .ServicesList )
189
+ fileMap , err := services .FilesForServices (serverConfig .ControlConfig , cmds .ServicesList . Value () )
190
190
if err != nil {
191
191
return err
192
192
}
@@ -200,7 +200,7 @@ func rotate(app *cli.Context, cfg *cmds.Server) error {
200
200
201
201
// The dynamiclistener cache file can't be simply deleted, we need to create a trigger
202
202
// file to indicate that the cert needs to be regenerated on startup.
203
- for _ , service := range cmds .ServicesList {
203
+ for _ , service := range cmds .ServicesList . Value () {
204
204
if service == version .Program + services .ProgramServer {
205
205
dynamicListenerRegenFilePath := filepath .Join (serverConfig .ControlConfig .DataDir , "tls" , "dynamic-cert-regenerate" )
206
206
if err := os .WriteFile (dynamicListenerRegenFilePath , []byte {}, 0600 ); err != nil {
@@ -256,7 +256,7 @@ func backupCertificates(serverDataDir, agentDataDir string, fileMap map[string][
256
256
}
257
257
258
258
func validateCertConfig () error {
259
- for _ , s := range cmds .ServicesList {
259
+ for _ , s := range cmds .ServicesList . Value () {
260
260
if ! services .IsValid (s ) {
261
261
return errors .New ("service " + s + " is not recognized" )
262
262
}
0 commit comments