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>
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package network
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/completion"
|
|
"github.com/moby/moby/api/types/container"
|
|
"github.com/moby/moby/client"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type disconnectOptions struct {
|
|
network string
|
|
container string
|
|
force bool
|
|
}
|
|
|
|
func newDisconnectCommand(dockerCLI command.Cli) *cobra.Command {
|
|
opts := disconnectOptions{}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "disconnect [OPTIONS] NETWORK CONTAINER",
|
|
Short: "Disconnect a container from a network",
|
|
Args: cli.ExactArgs(2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.network = args[0]
|
|
opts.container = args[1]
|
|
return runDisconnect(cmd.Context(), dockerCLI.Client(), opts)
|
|
},
|
|
ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
if len(args) == 0 {
|
|
return completion.NetworkNames(dockerCLI)(cmd, args, toComplete)
|
|
}
|
|
network := args[0]
|
|
return completion.ContainerNames(dockerCLI, true, isConnected(network))(cmd, args, toComplete)
|
|
},
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
flags.BoolVarP(&opts.force, "force", "f", false, "Force the container to disconnect from a network")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runDisconnect(ctx context.Context, apiClient client.NetworkAPIClient, opts disconnectOptions) error {
|
|
return apiClient.NetworkDisconnect(ctx, opts.network, opts.container, opts.force)
|
|
}
|
|
|
|
func isConnected(network string) func(container.Summary) bool {
|
|
return func(ctr container.Summary) bool {
|
|
if ctr.NetworkSettings == nil {
|
|
return false
|
|
}
|
|
_, ok := ctr.NetworkSettings.Networks[network]
|
|
return ok
|
|
}
|
|
}
|
|
|
|
func not(fn func(container.Summary) bool) func(container.Summary) bool {
|
|
return func(ctr container.Summary) bool {
|
|
ok := fn(ctr)
|
|
return !ok
|
|
}
|
|
}
|