github.com/panekj/cli@v0.0.0-20230304125325-467dd2f3797e/cli/command/image/remove.go (about)

     1  package image
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  	"strings"
     7  
     8  	"github.com/docker/cli/cli"
     9  	"github.com/docker/cli/cli/command"
    10  	"github.com/docker/docker/api/types"
    11  	apiclient "github.com/docker/docker/client"
    12  	"github.com/pkg/errors"
    13  	"github.com/spf13/cobra"
    14  )
    15  
    16  type removeOptions struct {
    17  	force   bool
    18  	noPrune bool
    19  }
    20  
    21  // NewRemoveCommand creates a new `docker remove` command
    22  func NewRemoveCommand(dockerCli command.Cli) *cobra.Command {
    23  	var opts removeOptions
    24  
    25  	cmd := &cobra.Command{
    26  		Use:   "rmi [OPTIONS] IMAGE [IMAGE...]",
    27  		Short: "Remove one or more images",
    28  		Args:  cli.RequiresMinArgs(1),
    29  		RunE: func(cmd *cobra.Command, args []string) error {
    30  			return runRemove(dockerCli, opts, args)
    31  		},
    32  		Annotations: map[string]string{
    33  			"aliases": "docker image rm, docker image remove, docker rmi",
    34  		},
    35  	}
    36  
    37  	flags := cmd.Flags()
    38  
    39  	flags.BoolVarP(&opts.force, "force", "f", false, "Force removal of the image")
    40  	flags.BoolVar(&opts.noPrune, "no-prune", false, "Do not delete untagged parents")
    41  
    42  	return cmd
    43  }
    44  
    45  func newRemoveCommand(dockerCli command.Cli) *cobra.Command {
    46  	cmd := *NewRemoveCommand(dockerCli)
    47  	cmd.Aliases = []string{"rmi", "remove"}
    48  	cmd.Use = "rm [OPTIONS] IMAGE [IMAGE...]"
    49  	return &cmd
    50  }
    51  
    52  func runRemove(dockerCli command.Cli, opts removeOptions, images []string) error {
    53  	client := dockerCli.Client()
    54  	ctx := context.Background()
    55  
    56  	options := types.ImageRemoveOptions{
    57  		Force:         opts.force,
    58  		PruneChildren: !opts.noPrune,
    59  	}
    60  
    61  	var errs []string
    62  	fatalErr := false
    63  	for _, img := range images {
    64  		dels, err := client.ImageRemove(ctx, img, options)
    65  		if err != nil {
    66  			if !apiclient.IsErrNotFound(err) {
    67  				fatalErr = true
    68  			}
    69  			errs = append(errs, err.Error())
    70  		} else {
    71  			for _, del := range dels {
    72  				if del.Deleted != "" {
    73  					fmt.Fprintf(dockerCli.Out(), "Deleted: %s\n", del.Deleted)
    74  				} else {
    75  					fmt.Fprintf(dockerCli.Out(), "Untagged: %s\n", del.Untagged)
    76  				}
    77  			}
    78  		}
    79  	}
    80  
    81  	if len(errs) > 0 {
    82  		msg := strings.Join(errs, "\n")
    83  		if !opts.force || fatalErr {
    84  			return errors.New(msg)
    85  		}
    86  		fmt.Fprintln(dockerCli.Err(), msg)
    87  	}
    88  	return nil
    89  }