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>
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/cli/cli"
|
|
"github.com/docker/cli/cli/command"
|
|
"github.com/docker/cli/cli/command/formatter"
|
|
flagsHelper "github.com/docker/cli/cli/flags"
|
|
"github.com/moby/moby/client"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type diskUsageOptions struct {
|
|
verbose bool
|
|
format string
|
|
}
|
|
|
|
// newDiskUsageCommand creates a new cobra.Command for `docker df`
|
|
func newDiskUsageCommand(dockerCLI command.Cli) *cobra.Command {
|
|
var opts diskUsageOptions
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "df [OPTIONS]",
|
|
Short: "Show docker disk usage",
|
|
Args: cli.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
return runDiskUsage(cmd.Context(), dockerCLI, opts)
|
|
},
|
|
Annotations: map[string]string{"version": "1.25"},
|
|
ValidArgsFunction: cobra.NoFileCompletions,
|
|
DisableFlagsInUseLine: true,
|
|
}
|
|
|
|
flags := cmd.Flags()
|
|
|
|
flags.BoolVarP(&opts.verbose, "verbose", "v", false, "Show detailed information on space usage")
|
|
flags.StringVar(&opts.format, "format", "", flagsHelper.FormatHelp)
|
|
|
|
return cmd
|
|
}
|
|
|
|
func runDiskUsage(ctx context.Context, dockerCli command.Cli, opts diskUsageOptions) error {
|
|
// TODO expose types.DiskUsageOptions.Types as flag on the command-line and/or as separate commands (docker container df / docker container usage)
|
|
du, err := dockerCli.Client().DiskUsage(ctx, client.DiskUsageOptions{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
format := opts.format
|
|
if len(format) == 0 {
|
|
format = formatter.TableFormatKey
|
|
}
|
|
|
|
var bsz int64
|
|
for _, bc := range du.BuildCache {
|
|
if !bc.Shared {
|
|
bsz += bc.Size
|
|
}
|
|
}
|
|
|
|
duCtx := formatter.DiskUsageContext{
|
|
Context: formatter.Context{
|
|
Output: dockerCli.Out(),
|
|
Format: formatter.NewDiskUsageFormat(format, opts.verbose),
|
|
},
|
|
LayersSize: du.LayersSize,
|
|
BuilderSize: bsz,
|
|
BuildCache: du.BuildCache,
|
|
Images: du.Images,
|
|
Containers: du.Containers,
|
|
Volumes: du.Volumes,
|
|
Verbose: opts.verbose,
|
|
}
|
|
|
|
return duCtx.Write()
|
|
}
|