@@ -8,15 +8,19 @@ import (
8
8
9
9
func newSetStackOrchestratorCommand (dockerCli command.Cli ) * cobra.Command {
10
10
cmd := & cobra.Command {
11
- Use : "stack-orchestrator CONTEXT ORCHESTRATOR" ,
11
+ Use : "stack-orchestrator [ CONTEXT] ORCHESTRATOR" ,
12
12
Short : "Set the default stack orchestrator of a context" ,
13
- Args : cli .ExactArgs ( 2 ),
13
+ Args : cli .RequiresRangeArgs ( 1 , 2 ),
14
14
RunE : func (cmd * cobra.Command , args []string ) error {
15
- orchestrator , err := command .NormalizeOrchestrator (args [1 ])
15
+ name , err := contextNameFromArgsOrCurrent (args [:len (args )- 1 ], dockerCli )
16
+ if err != nil {
17
+ return err
18
+ }
19
+ orchestrator , err := command .NormalizeOrchestrator (args [len (args )- 1 ])
16
20
if err != nil {
17
21
return err
18
22
}
19
- return setContextLevelOption (dockerCli , args [ 0 ] , func (dc * command.DockerContext ) {
23
+ return setContextLevelOption (dockerCli , name , func (dc * command.DockerContext ) {
20
24
dc .StackOrchestrator = orchestrator
21
25
})
22
26
},
@@ -26,12 +30,16 @@ func newSetStackOrchestratorCommand(dockerCli command.Cli) *cobra.Command {
26
30
27
31
func newSetDescriptionCommand (dockerCli command.Cli ) * cobra.Command {
28
32
cmd := & cobra.Command {
29
- Use : "description CONTEXT DESCRIPTION" ,
33
+ Use : "description [ CONTEXT] DESCRIPTION" ,
30
34
Short : "Set description of a context" ,
31
35
Args : cli .ExactArgs (2 ),
32
36
RunE : func (cmd * cobra.Command , args []string ) error {
33
- return setContextLevelOption (dockerCli , args [0 ], func (dc * command.DockerContext ) {
34
- dc .Description = args [1 ]
37
+ name , err := contextNameFromArgsOrCurrent (args [:len (args )- 1 ], dockerCli )
38
+ if err != nil {
39
+ return err
40
+ }
41
+ return setContextLevelOption (dockerCli , name , func (dc * command.DockerContext ) {
42
+ dc .Description = args [len (args )- 1 ]
35
43
})
36
44
},
37
45
}
0 commit comments