0adaf6be3b
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>
66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package context
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/context/store"
|
|
"github.com/moby/moby/client"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func newUseCommand(dockerCLI command.Cli) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "use CONTEXT",
|
|
Short: "Set the current docker context",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
name := args[0]
|
|
return runUse(dockerCLI, name)
|
|
},
|
|
ValidArgsFunction: completeContextNames(dockerCLI, 1, false),
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
return cmd
|
|
}
|
|
|
|
// RunUse set the current Docker context
|
|
//
|
|
// Deprecated: this function was for internal use and will be removed in the next release.
|
|
func RunUse(dockerCLI command.Cli, name string) error {
|
|
return runUse(dockerCLI, name)
|
|
}
|
|
|
|
// runUse set the current Docker context
|
|
func runUse(dockerCLI command.Cli, name string) error {
|
|
// configValue uses an empty string for "default"
|
|
var configValue string
|
|
if name != command.DefaultContextName {
|
|
if err := store.ValidateContextName(name); err != nil {
|
|
return err
|
|
}
|
|
if _, err := dockerCLI.ContextStore().GetMetadata(name); err != nil {
|
|
return err
|
|
}
|
|
configValue = name
|
|
}
|
|
dockerConfig := dockerCLI.ConfigFile()
|
|
// Avoid updating the config-file if nothing changed. This also prevents
|
|
// creating the file and config-directory if the default is used and
|
|
// no config-file existed yet.
|
|
if dockerConfig.CurrentContext != configValue {
|
|
dockerConfig.CurrentContext = configValue
|
|
if err := dockerConfig.Save(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
_, _ = fmt.Fprintln(dockerCLI.Out(), name)
|
|
_, _ = fmt.Fprintf(dockerCLI.Err(), "Current context is now %q\n", name)
|
|
if name != command.DefaultContextName && os.Getenv(client.EnvOverrideHost) != "" {
|
|
_, _ = fmt.Fprintf(dockerCLI.Err(), "Warning: %[1]s environment variable overrides the active context. "+
|
|
"To use %[2]q, either set the global --context flag, or unset %[1]s environment variable.\n", client.EnvOverrideHost, name)
|
|
}
|
|
return nil
|
|
}
|