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>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package checkpoint
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/completion"
|
|
"github.com/docker/cli/cli/command/formatter"
|
|
"github.com/moby/moby/api/types/checkpoint"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type listOptions struct {
|
|
checkpointDir string
|
|
}
|
|
|
|
func newListCommand(dockerCLI command.Cli) *cobra.Command {
|
|
var opts listOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "ls [OPTIONS] CONTAINER",
|
|
Aliases: []string{"list"},
|
|
Short: "List checkpoints for a container",
|
|
Args: cli.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return runList(cmd.Context(), dockerCLI, args[0], opts)
|
|
},
|
|
ValidArgsFunction: completion.ContainerNames(dockerCLI, false),
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
flags.StringVar(&opts.checkpointDir, "checkpoint-dir", "", "Use a custom checkpoint storage directory")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runList(ctx context.Context, dockerCli command.Cli, container string, opts listOptions) error {
|
|
checkpoints, err := dockerCli.Client().CheckpointList(ctx, container, checkpoint.ListOptions{
|
|
CheckpointDir: opts.checkpointDir,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
cpCtx := formatter.Context{
|
|
Output: dockerCli.Out(),
|
|
Format: newFormat(formatter.TableFormatKey),
|
|
}
|
|
return formatWrite(cpCtx, checkpoints)
|
|
}
|