mirror of
https://github.com/containers/podman.git
synced 2025-09-12 09:54:43 +08:00

* Combine cobra.Command helper functions into validate package from registry and common packages * Introduce ChoiceValue for flags Signed-off-by: Jhon Honce <jhonce@redhat.com>
101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/containers/libpod/pkg/domain/entities"
|
|
"github.com/containers/libpod/pkg/domain/infra"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// DefaultAPIAddress is the default address of the REST socket
|
|
const DefaultAPIAddress = "unix:/run/podman/podman.sock"
|
|
|
|
// DefaultVarlinkAddress is the default address of the varlink socket
|
|
const DefaultVarlinkAddress = "unix:/run/podman/io.podman"
|
|
|
|
type CliCommand struct {
|
|
Mode []entities.EngineMode
|
|
Command *cobra.Command
|
|
Parent *cobra.Command
|
|
}
|
|
|
|
const ExecErrorCodeGeneric = 125
|
|
|
|
var (
|
|
cliCtx context.Context
|
|
containerEngine entities.ContainerEngine
|
|
exitCode = ExecErrorCodeGeneric
|
|
imageEngine entities.ImageEngine
|
|
|
|
// Commands holds the cobra.Commands to present to the user, including
|
|
// parent if not a child of "root"
|
|
Commands []CliCommand
|
|
)
|
|
|
|
func SetExitCode(code int) {
|
|
exitCode = code
|
|
}
|
|
|
|
func GetExitCode() int {
|
|
return exitCode
|
|
}
|
|
|
|
func ImageEngine() entities.ImageEngine {
|
|
return imageEngine
|
|
}
|
|
|
|
// NewImageEngine is a wrapper for building an ImageEngine to be used for PreRunE functions
|
|
func NewImageEngine(cmd *cobra.Command, args []string) (entities.ImageEngine, error) {
|
|
if imageEngine == nil {
|
|
podmanOptions.FlagSet = cmd.Flags()
|
|
engine, err := infra.NewImageEngine(&podmanOptions)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
imageEngine = engine
|
|
}
|
|
return imageEngine, nil
|
|
}
|
|
|
|
func ContainerEngine() entities.ContainerEngine {
|
|
return containerEngine
|
|
}
|
|
|
|
// NewContainerEngine is a wrapper for building an ContainerEngine to be used for PreRunE functions
|
|
func NewContainerEngine(cmd *cobra.Command, args []string) (entities.ContainerEngine, error) {
|
|
if containerEngine == nil {
|
|
podmanOptions.FlagSet = cmd.Flags()
|
|
engine, err := infra.NewContainerEngine(&podmanOptions)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
containerEngine = engine
|
|
}
|
|
return containerEngine, nil
|
|
}
|
|
|
|
type PodmanOptionsKey struct{}
|
|
|
|
func Context() context.Context {
|
|
if cliCtx == nil {
|
|
cliCtx = ContextWithOptions(context.Background())
|
|
}
|
|
return cliCtx
|
|
}
|
|
|
|
func ContextWithOptions(ctx context.Context) context.Context {
|
|
cliCtx = context.WithValue(ctx, PodmanOptionsKey{}, podmanOptions)
|
|
return cliCtx
|
|
}
|
|
|
|
// GetContextWithOptions deprecated, use NewContextWithOptions()
|
|
func GetContextWithOptions() context.Context {
|
|
return ContextWithOptions(context.Background())
|
|
}
|
|
|
|
// GetContext deprecated, use Context()
|
|
func GetContext() context.Context {
|
|
return Context()
|
|
}
|