0adaf6be3b
This replaces the visitAll recursive function with a test that verifies that the option is set for all commands and subcommands, so that it doesn't have to be modified at runtime. We currently still have to loop over all functions for the setValidateArgs call, but that can be looked at separately. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
87 lines
2.6 KiB
Go
87 lines
2.6 KiB
Go
package stack
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/formatter"
|
|
"github.com/docker/cli/cli/command/service"
|
|
flagsHelper "github.com/docker/cli/cli/flags"
|
|
cliopts "github.com/docker/cli/opts"
|
|
"github.com/fvbommel/sortorder"
|
|
"github.com/moby/moby/api/types/swarm"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// serviceListOptions holds docker stack services options
|
|
type serviceListOptions = struct {
|
|
quiet bool
|
|
format string
|
|
filter cliopts.FilterOpt
|
|
namespace string
|
|
}
|
|
|
|
func newServicesCommand(dockerCLI command.Cli) *cobra.Command {
|
|
opts := serviceListOptions{filter: cliopts.NewFilterOpt()}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "services [OPTIONS] STACK",
|
|
Short: "List the services in the stack",
|
|
Args: cli.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.namespace = args[0]
|
|
if err := validateStackName(opts.namespace); err != nil {
|
|
return err
|
|
}
|
|
return runServices(cmd.Context(), dockerCLI, opts)
|
|
},
|
|
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
return completeNames(dockerCLI)(cmd, args, toComplete)
|
|
},
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
flags := cmd.Flags()
|
|
flags.BoolVarP(&opts.quiet, "quiet", "q", false, "Only display IDs")
|
|
flags.StringVar(&opts.format, "format", "", flagsHelper.FormatHelp)
|
|
flags.VarP(&opts.filter, "filter", "f", "Filter output based on conditions provided")
|
|
return cmd
|
|
}
|
|
|
|
// runServices performs a stack services against the specified swarm cluster
|
|
func runServices(ctx context.Context, dockerCLI command.Cli, opts serviceListOptions) error {
|
|
services, err := getServices(ctx, dockerCLI.Client(), opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return formatWrite(dockerCLI, services, opts)
|
|
}
|
|
|
|
func formatWrite(dockerCLI command.Cli, services []swarm.Service, opts serviceListOptions) error {
|
|
// if no services in the stack, print message and exit 0
|
|
if len(services) == 0 {
|
|
_, _ = fmt.Fprintln(dockerCLI.Err(), "Nothing found in stack:", opts.namespace)
|
|
return nil
|
|
}
|
|
sort.Slice(services, func(i, j int) bool {
|
|
return sortorder.NaturalLess(services[i].Spec.Name, services[j].Spec.Name)
|
|
})
|
|
|
|
f := opts.format
|
|
if len(f) == 0 {
|
|
if len(dockerCLI.ConfigFile().ServicesFormat) > 0 && !opts.quiet {
|
|
f = dockerCLI.ConfigFile().ServicesFormat
|
|
} else {
|
|
f = formatter.TableFormatKey
|
|
}
|
|
}
|
|
|
|
servicesCtx := formatter.Context{
|
|
Output: dockerCLI.Out(),
|
|
Format: service.NewListFormat(f, opts.quiet),
|
|
}
|
|
return service.ListFormatWrite(servicesCtx, services)
|
|
}
|