78c54646c3
This uses the DefaultShellCompDirective feature which was added in cobra to override the default (which would complete to use files for commands and flags). Note that we set "cobra.NoFileCompletions" for many commands, which is redundant with this change, so we could remove as well. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/idresolver"
|
|
"github.com/docker/cli/cli/command/task"
|
|
"github.com/docker/cli/opts"
|
|
"github.com/moby/moby/api/types/swarm"
|
|
"github.com/moby/moby/client"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type psOptions struct {
|
|
nodeIDs []string
|
|
noResolve bool
|
|
noTrunc bool
|
|
quiet bool
|
|
format string
|
|
filter opts.FilterOpt
|
|
}
|
|
|
|
func newPsCommand(dockerCLI command.Cli) *cobra.Command {
|
|
options := psOptions{filter: opts.NewFilterOpt()}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "ps [OPTIONS] [NODE...]",
|
|
Short: "List tasks running on one or more nodes, defaults to current node",
|
|
Args: cli.RequiresMinArgs(0),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
options.nodeIDs = []string{"self"}
|
|
|
|
if len(args) != 0 {
|
|
options.nodeIDs = args
|
|
}
|
|
|
|
return runPs(cmd.Context(), dockerCLI, options)
|
|
},
|
|
ValidArgsFunction: completeNodeNames(dockerCLI),
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
flags := cmd.Flags()
|
|
flags.BoolVar(&options.noTrunc, "no-trunc", false, "Do not truncate output")
|
|
flags.BoolVar(&options.noResolve, "no-resolve", false, "Do not map IDs to Names")
|
|
flags.VarP(&options.filter, "filter", "f", "Filter output based on conditions provided")
|
|
flags.StringVar(&options.format, "format", "", "Pretty-print tasks using a Go template")
|
|
flags.BoolVarP(&options.quiet, "quiet", "q", false, "Only display task IDs")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runPs(ctx context.Context, dockerCLI command.Cli, options psOptions) error {
|
|
apiClient := dockerCLI.Client()
|
|
|
|
var (
|
|
errs []error
|
|
tasks []swarm.Task
|
|
)
|
|
|
|
for _, nodeID := range options.nodeIDs {
|
|
nodeRef, err := Reference(ctx, apiClient, nodeID)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
continue
|
|
}
|
|
|
|
node, _, err := apiClient.NodeInspectWithRaw(ctx, nodeRef)
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
continue
|
|
}
|
|
|
|
filter := options.filter.Value()
|
|
filter.Add("node", node.ID)
|
|
|
|
nodeTasks, err := apiClient.TaskList(ctx, client.TaskListOptions{Filters: filter})
|
|
if err != nil {
|
|
errs = append(errs, err)
|
|
continue
|
|
}
|
|
|
|
tasks = append(tasks, nodeTasks...)
|
|
}
|
|
|
|
format := options.format
|
|
if len(format) == 0 {
|
|
format = task.DefaultFormat(dockerCLI.ConfigFile(), options.quiet)
|
|
}
|
|
|
|
if len(errs) == 0 || len(tasks) != 0 {
|
|
if err := task.Print(ctx, dockerCLI, tasks, idresolver.New(apiClient, options.noResolve), !options.noTrunc, options.quiet, format); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
|
|
return errors.Join(errs...)
|
|
}
|