feature: support to print snapshot info for image

Signed-off-by: ningmingxiao <ning.mingxiao@zte.com.cn>

Author: ningmingxiao

pkg/imageinspector/imageinspector.go

@@ -19,6 +19,8 @@ package imageinspector
 import (
 	"context"
 
+	"github.com/opencontainers/image-spec/identity"
+
 	containerd "github.com/containerd/containerd/v2/client"
 	"github.com/containerd/containerd/v2/core/images"
 	"github.com/containerd/containerd/v2/core/snapshots"
@@ -56,6 +58,19 @@ func Inspect(ctx context.Context, client *containerd.Client, image images.Image,
 	} else {
 		n.ImageConfigDesc = imageConfigDesc
 		n.ImageConfig = imageConfig
+		diffIDS := imageConfig.RootFS.DiffIDs
+		// chainIDS := calculateChainID(diffIDS)
+		chainIDS := identity.ChainIDs(diffIDS)
+		snapshots := make([]snapshots.Info, len(chainIDS))
+		for i, id := range chainIDS {
+			snapInfo, err := snapshotter.Stat(ctx, id.String())
+			if err == nil {
+				snapshots[i] = snapInfo
+			} else {
+				log.G(ctx).WithError(err).WithField("id", image.Name).Warnf("failed to get snapshot %s info", id.String())
+			}
+		}
+		n.Snapshots = snapshots
 	}
 	n.Size, err = imgutil.UnpackedImageSize(ctx, snapshotter, img)
 	if err != nil {

pkg/inspecttypes/native/image.go

@@ -20,6 +20,7 @@ import (
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 
 	"github.com/containerd/containerd/v2/core/images"
+	"github.com/containerd/containerd/v2/core/snapshots"
 )
 
 // Image corresponds to a containerd-native image object.
@@ -34,4 +35,5 @@ type Image struct {
 	ImageConfigDesc ocispec.Descriptor `json:"ImageConfigDesc"`
 	ImageConfig     ocispec.Image      `json:"ImageConfig"`
 	Size            int64              `json:"size"`
+	Snapshots       []snapshots.Info   `json:"Snapshots,omitempty"`
 }