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>
106 lines
2.8 KiB
Go
106 lines
2.8 KiB
Go
package network
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/system/pruner"
|
|
"github.com/docker/cli/internal/prompt"
|
|
"github.com/docker/cli/opts"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func init() {
|
|
// Register the prune command to run as part of "docker system prune"
|
|
if err := pruner.Register(pruner.TypeNetwork, pruneFn); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
type pruneOptions struct {
|
|
force bool
|
|
filter opts.FilterOpt
|
|
}
|
|
|
|
// newPruneCommand returns a new cobra prune command for networks
|
|
func newPruneCommand(dockerCLI command.Cli) *cobra.Command {
|
|
options := pruneOptions{filter: opts.NewFilterOpt()}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "prune [OPTIONS]",
|
|
Short: "Remove all unused networks",
|
|
Args: cli.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
output, err := runPrune(cmd.Context(), dockerCLI, options)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if output != "" {
|
|
_, _ = fmt.Fprintln(dockerCLI.Out(), output)
|
|
}
|
|
return nil
|
|
},
|
|
Annotations: map[string]string{"version": "1.25"},
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
flags.BoolVarP(&options.force, "force", "f", false, "Do not prompt for confirmation")
|
|
flags.Var(&options.filter, "filter", `Provide filter values (e.g. "until=<timestamp>")`)
|
|
|
|
return cmd
|
|
}
|
|
|
|
const warning = `WARNING! This will remove all custom networks not used by at least one container.
|
|
Are you sure you want to continue?`
|
|
|
|
func runPrune(ctx context.Context, dockerCli command.Cli, options pruneOptions) (output string, err error) {
|
|
pruneFilters := command.PruneFilters(dockerCli, options.filter.Value())
|
|
|
|
if !options.force {
|
|
r, err := prompt.Confirm(ctx, dockerCli.In(), dockerCli.Out(), warning)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if !r {
|
|
return "", cancelledErr{errors.New("network prune has been cancelled")}
|
|
}
|
|
}
|
|
|
|
report, err := dockerCli.Client().NetworksPrune(ctx, pruneFilters)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(report.NetworksDeleted) > 0 {
|
|
output = "Deleted Networks:\n"
|
|
for _, id := range report.NetworksDeleted {
|
|
output += id + "\n"
|
|
}
|
|
}
|
|
|
|
return output, nil
|
|
}
|
|
|
|
type cancelledErr struct{ error }
|
|
|
|
func (cancelledErr) Cancelled() {}
|
|
|
|
// pruneFn calls the Network Prune API for use in "docker system prune"
|
|
// and returns the amount of space reclaimed and a detailed output string.
|
|
func pruneFn(ctx context.Context, dockerCLI command.Cli, options pruner.PruneOptions) (uint64, string, error) {
|
|
if !options.Confirmed {
|
|
// Dry-run: perform validation and produce confirmation before pruning.
|
|
confirmMsg := "all networks not used by at least one container"
|
|
return 0, confirmMsg, cancelledErr{errors.New("network prune has been cancelled")}
|
|
}
|
|
output, err := runPrune(ctx, dockerCLI, pruneOptions{
|
|
force: true,
|
|
filter: options.Filter,
|
|
})
|
|
return 0, output, err
|
|
}
|