66b9056500
- Prevent completion on "create" subcommands to prevent them from completing with local filenames - Add completion for "docker image save" - Add completion for "docker image tag" - Disable completion for "docker login" - Exclude "paused" containers for "docker container attach" and "docker container exec" Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
package checkpoint
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/completion"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type createOptions struct {
|
|
container string
|
|
checkpoint string
|
|
checkpointDir string
|
|
leaveRunning bool
|
|
}
|
|
|
|
func newCreateCommand(dockerCli command.Cli) *cobra.Command {
|
|
var opts createOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "create [OPTIONS] CONTAINER CHECKPOINT",
|
|
Short: "Create a checkpoint from a running container",
|
|
Args: cli.ExactArgs(2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.container = args[0]
|
|
opts.checkpoint = args[1]
|
|
return runCreate(dockerCli, opts)
|
|
},
|
|
ValidArgsFunction: completion.NoComplete,
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
flags.BoolVar(&opts.leaveRunning, "leave-running", false, "Leave the container running after checkpoint")
|
|
flags.StringVarP(&opts.checkpointDir, "checkpoint-dir", "", "", "Use a custom checkpoint storage directory")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runCreate(dockerCli command.Cli, opts createOptions) error {
|
|
client := dockerCli.Client()
|
|
|
|
checkpointOpts := types.CheckpointCreateOptions{
|
|
CheckpointID: opts.checkpoint,
|
|
CheckpointDir: opts.checkpointDir,
|
|
Exit: !opts.leaveRunning,
|
|
}
|
|
|
|
err := client.CheckpointCreate(context.Background(), opts.container, checkpointOpts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Fprintf(dockerCli.Out(), "%s\n", opts.checkpoint)
|
|
return nil
|
|
}
|