Expose image 'DeleteTag' method in the 'daemon'

pkg/v1/daemon/image_test.go

@@ -82,6 +82,8 @@ type MockClient struct {
 	inspectBody []byte
 
 	tagErr error
+
+	imageRemoveErr error
 }
 
 func (m *MockClient) NegotiateAPIVersion(_ context.Context) {

pkg/v1/daemon/options.go

@@ -102,4 +102,5 @@ type Client interface {
 	ImageTag(context.Context, string, string) error
 	ImageInspectWithRaw(context.Context, string) (types.ImageInspect, []byte, error)
 	ImageHistory(context.Context, string) ([]api.HistoryResponseItem, error)
+	ImageRemove(context.Context, string, types.ImageRemoveOptions) ([]types.ImageDeleteResponseItem, error)
 }

pkg/v1/daemon/write.go

@@ -16,11 +16,11 @@ package daemon
 
 import (
 	"fmt"
-	"io"
-
+	"github.com/docker/docker/api/types"
 	"github.com/google/go-containerregistry/pkg/name"
 	v1 "github.com/google/go-containerregistry/pkg/v1"
 	"github.com/google/go-containerregistry/pkg/v1/tarball"
+	"io"
 )
 
 // Tag adds a tag to an already existent image.
@@ -76,3 +76,18 @@ func Write(tag name.Tag, img v1.Image, options ...Option) (string, error) {
 	}
 	return response, nil
 }
+
+// DeleteTag - Deletes given tag:
+//   - force - delete the image even if it is being used by stopped containers or has other tags
+//   - pruneChildren - deletes untagged parent images
+func DeleteTag(tag name.Tag, pruneChildren bool, force bool, options ...Option) error {
+	o, err := makeOptions(options...)
+	if err != nil {
+		return err
+	}
+	_, err = o.client.ImageRemove(o.ctx, tag.Name(), types.ImageRemoveOptions{PruneChildren: pruneChildren, Force: force})
+	if err != nil {
+		return fmt.Errorf("error deleting image %s: %w", tag.Name(), err)
+	}
+	return nil
+}

pkg/v1/daemon/write_test.go

@@ -61,6 +61,10 @@ func (m *MockClient) ImageTag(ctx context.Context, _, _ string) error {
 	return m.tagErr
 }
 
+func (m *MockClient) ImageRemove(context.Context, string, types.ImageRemoveOptions) ([]types.ImageDeleteResponseItem, error) {
+	return nil, m.imageRemoveErr
+}
+
 func TestWriteImage(t *testing.T) {
 	for _, tc := range []struct {
 		name         string
@@ -165,4 +169,7 @@ func TestWriteDefaultClient(t *testing.T) {
 	if _, err := Write(tag, empty.Image, WithContext(ctx)); err != nil {
 		t.Fatal(err)
 	}
+	if err := DeleteTag(tag, false, false, WithContext(ctx)); err != nil {
+		t.Fatal(err)
+	}
 }