mirror of
https://github.com/podman-container-tools/podman.git
synced 2026-08-26 18:27:53 +00:00
b5d1d89a37
Allow automatic generation for shell completion scripts with the internal cobra functions (requires v1.0.0+). This should replace the handwritten completion scripts and even adds support for fish. With this approach it is less likley that completions and code are out of sync. We can now create the scripts with - podman completion bash - podman completion zsh - podman completion fish To test the completion run: source <(podman completion bash) The same works for podman-remote and podman --remote and it will complete your remote containers/images with the correct endpoints values from --url/--connection. The completion logic is written in go and provided by the cobra library. The completion functions lives in `cmd/podman/completion/completion.go`. The unit test at cmd/podman/shell_completion_test.go checks if each command and flag has an autocompletion function set. This prevents that commands and flags have no shell completion set. This commit does not replace the current autocompletion scripts. Closes #6440 Signed-off-by: Paul Holzinger <paul.holzinger@web.de>
110 lines
3.5 KiB
Go
110 lines
3.5 KiB
Go
package containers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/containers/common/pkg/completion"
|
|
"github.com/containers/podman/v2/cmd/podman/common"
|
|
"github.com/containers/podman/v2/cmd/podman/registry"
|
|
"github.com/containers/podman/v2/cmd/podman/utils"
|
|
"github.com/containers/podman/v2/cmd/podman/validate"
|
|
"github.com/containers/podman/v2/libpod/define"
|
|
"github.com/containers/podman/v2/pkg/domain/entities"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
restartDescription = fmt.Sprintf(`Restarts one or more running containers. The container ID or name can be used.
|
|
|
|
A timeout before forcibly stopping can be set, but defaults to %d seconds.`, containerConfig.Engine.StopTimeout)
|
|
|
|
restartCommand = &cobra.Command{
|
|
Use: "restart [options] CONTAINER [CONTAINER...]",
|
|
Short: "Restart one or more containers",
|
|
Long: restartDescription,
|
|
RunE: restart,
|
|
Args: func(cmd *cobra.Command, args []string) error {
|
|
return validate.CheckAllLatestAndCIDFile(cmd, args, false, false)
|
|
},
|
|
ValidArgsFunction: common.AutocompleteContainers,
|
|
Example: `podman restart ctrID
|
|
podman restart --latest
|
|
podman restart ctrID1 ctrID2`,
|
|
}
|
|
|
|
containerRestartCommand = &cobra.Command{
|
|
Use: restartCommand.Use,
|
|
Short: restartCommand.Short,
|
|
Long: restartCommand.Long,
|
|
RunE: restartCommand.RunE,
|
|
ValidArgsFunction: restartCommand.ValidArgsFunction,
|
|
Example: `podman container restart ctrID
|
|
podman container restart --latest
|
|
podman container restart ctrID1 ctrID2`,
|
|
}
|
|
)
|
|
|
|
var (
|
|
restartOptions = entities.RestartOptions{}
|
|
restartTimeout uint
|
|
)
|
|
|
|
func restartFlags(cmd *cobra.Command) {
|
|
flags := cmd.Flags()
|
|
|
|
flags.BoolVarP(&restartOptions.All, "all", "a", false, "Restart all non-running containers")
|
|
flags.BoolVar(&restartOptions.Running, "running", false, "Restart only running containers when --all is used")
|
|
|
|
timeFlagName := "time"
|
|
flags.UintVarP(&restartTimeout, timeFlagName, "t", containerConfig.Engine.StopTimeout, "Seconds to wait for stop before killing the container")
|
|
_ = cmd.RegisterFlagCompletionFunc(timeFlagName, completion.AutocompleteNone)
|
|
|
|
flags.SetNormalizeFunc(utils.AliasFlags)
|
|
}
|
|
|
|
func init() {
|
|
registry.Commands = append(registry.Commands, registry.CliCommand{
|
|
Mode: []entities.EngineMode{entities.ABIMode, entities.TunnelMode},
|
|
Command: restartCommand,
|
|
})
|
|
restartFlags(restartCommand)
|
|
validate.AddLatestFlag(restartCommand, &restartOptions.Latest)
|
|
|
|
registry.Commands = append(registry.Commands, registry.CliCommand{
|
|
Mode: []entities.EngineMode{entities.ABIMode, entities.TunnelMode},
|
|
Command: containerRestartCommand,
|
|
Parent: containerCmd,
|
|
})
|
|
restartFlags(containerRestartCommand)
|
|
validate.AddLatestFlag(containerRestartCommand, &restartOptions.Latest)
|
|
}
|
|
|
|
func restart(cmd *cobra.Command, args []string) error {
|
|
var (
|
|
errs utils.OutputErrors
|
|
)
|
|
if len(args) < 1 && !restartOptions.Latest && !restartOptions.All {
|
|
return errors.Wrapf(define.ErrInvalidArg, "you must provide at least one container name or ID")
|
|
}
|
|
if len(args) > 0 && restartOptions.Latest {
|
|
return errors.Wrapf(define.ErrInvalidArg, "--latest and containers cannot be used together")
|
|
}
|
|
|
|
if cmd.Flag("time").Changed {
|
|
restartOptions.Timeout = &restartTimeout
|
|
}
|
|
responses, err := registry.ContainerEngine().ContainerRestart(context.Background(), args, restartOptions)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, r := range responses {
|
|
if r.Err == nil {
|
|
fmt.Println(r.Id)
|
|
} else {
|
|
errs = append(errs, r.Err)
|
|
}
|
|
}
|
|
return errs.PrintErrors()
|
|
}
|