1berry/docker-cli
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve completion of service scale args

Browse Source 
Signed-off-by: albers <github@albersweb.de>
This commit is contained in:
albers 2025-03-25 21:38:21 +00:00
parent b8034c0ed7
commit ee275d5733
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cli/command/service/scale.go
View File

@ -29,9 +29,7 @@ func newScaleCommand(dockerCli command.Cli) *cobra.Command {
RunE: func(cmd *cobra.Command, args []string) error {
return runScale(cmd.Context(), dockerCli, options, args)
},
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
return CompletionFn(dockerCli)(cmd, args, toComplete)
},
ValidArgsFunction: completeScaleArgs(dockerCli),
}
flags := cmd.Flags()
@ -117,3 +115,19 @@ func runServiceScale(ctx context.Context, apiClient client.ServiceAPIClient, ser
}
return response.Warnings, nil
}
// completeScaleArgs returns a completion function for the args of the scale command.
// It completes service names followed by "=", suppressing the trailing space.
func completeScaleArgs(dockerCli command.Cli) func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
return func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
// reuse the existing logic for configurable completion of service names and IDs.
completions, directive := CompletionFn(dockerCli)(cmd, args, toComplete)
if directive == cobra.ShellCompDirectiveError {
return completions, directive
}
for i, v := range completions {
completions[i] = v + "="
}
return completions, directive | cobra.ShellCompDirectiveNoSpace
}
}