Merge pull request #98 from thin-edge/feat-container-group-prune

feat(container-group): prune images during container-group finalize hook

Author: reubenmiller

cli/container_group/finalize.go

@@ -4,10 +4,14 @@ Copyright © 2024 thin-edge.io <[email protected]>
 package container_group
 
 import (
+	"context"
 	"log/slog"
 
+	"github.com/docker/docker/api/types/filters"
+	"github.com/docker/go-units"
 	"github.com/spf13/cobra"
 	"github.com/thin-edge/tedge-container-plugin/pkg/cli"
+	"github.com/thin-edge/tedge-container-plugin/pkg/container"
 )
 
 func NewFinalizeCommand(ctx cli.Cli) *cobra.Command {
@@ -16,6 +20,25 @@ func NewFinalizeCommand(ctx cli.Cli) *cobra.Command {
 		Short: "Finalize container install/remove operation",
 		RunE: func(cmd *cobra.Command, args []string) error {
 			slog.Info("Executing", "cmd", cmd.CalledAs(), "args", args)
+
+			pruneImages := ctx.GetBool("container.pruneImages")
+			if !pruneImages {
+				return nil
+			}
+			cli, err := container.NewContainerClient()
+			if err != nil {
+				return err
+			}
+			slog.Info("Pruning images")
+			ctx := context.Background()
+			resp, err := cli.Client.ImagesPrune(ctx, filters.Args{})
+			if err != nil {
+				return err
+			}
+			for _, image := range resp.ImagesDeleted {
+				slog.Info("Deleted image.", "deleted", image.Deleted, "untagged", image.Untagged)
+			}
+			slog.Info("Reclaimed space.", "size", units.HumanSizeWithPrecision(float64(resp.SpaceReclaimed), 3))
 			return nil
 		},
 	}