manifest: support nerdctl manifest annotate command

Signed-off-by: ChengyuZhu6 <[email protected]>

Author: ChengyuZhu6

cmd/nerdctl/manifest/manifest.go

@@ -35,6 +35,7 @@ func Command() *cobra.Command {
 	cmd.AddCommand(
 		InspectCommand(),
 		CreateCommand(),
+		AnnotateCommand(),
 	)
 
 	return cmd

cmd/nerdctl/manifest/manifest_annotate.go

@@ -0,0 +1,98 @@
+/*
+   Copyright The containerd Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package manifest
+
+import (
+	"github.com/spf13/cobra"
+
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/completion"
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/helpers"
+	"github.com/containerd/nerdctl/v2/pkg/api/types"
+	"github.com/containerd/nerdctl/v2/pkg/cmd/manifest"
+)
+
+func AnnotateCommand() *cobra.Command {
+	var cmd = &cobra.Command{
+		Use:               "annotate INDEX/MANIFESTLIST MANIFEST",
+		Short:             "Add additional information to a local image manifest",
+		Args:              cobra.ExactArgs(2),
+		RunE:              annotateAction,
+		ValidArgsFunction: annotateShellComplete,
+		SilenceUsage:      true,
+		SilenceErrors:     true,
+	}
+	cmd.Flags().String("os", "", "Set operating system")
+	cmd.Flags().String("arch", "", "Set architecture")
+	cmd.Flags().String("os-version", "", "Set operating system version")
+	cmd.Flags().String("variant", "", "Set operating system feature")
+	cmd.Flags().StringArray("os-features", []string{}, "Set architecture variant")
+	return cmd
+}
+
+func processAnnotateFlags(cmd *cobra.Command) (types.ManifestAnnotateOptions, error) {
+	globalOptions, err := helpers.ProcessRootCmdFlags(cmd)
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+
+	os, err := cmd.Flags().GetString("os")
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+	arch, err := cmd.Flags().GetString("arch")
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+	osVersion, err := cmd.Flags().GetString("os-version")
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+	variant, err := cmd.Flags().GetString("variant")
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+	osFeatures, err := cmd.Flags().GetStringArray("os-features")
+	if err != nil {
+		return types.ManifestAnnotateOptions{}, err
+	}
+
+	return types.ManifestAnnotateOptions{
+		Stdout:     cmd.OutOrStdout(),
+		GOptions:   globalOptions,
+		Os:         os,
+		Arch:       arch,
+		OsVersion:  osVersion,
+		Variant:    variant,
+		OsFeatures: osFeatures,
+	}, nil
+}
+
+func annotateAction(cmd *cobra.Command, args []string) error {
+	annotateOptions, err := processAnnotateFlags(cmd)
+	if err != nil {
+		return err
+	}
+
+	listRef := args[0]
+	manifestRef := args[1]
+
+	return manifest.Annotate(cmd.Context(), listRef, manifestRef, annotateOptions)
+}
+
+func annotateShellComplete(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+	return completion.ImageNames(cmd)
+}

pkg/api/types/manifest_types.go

@@ -18,6 +18,17 @@ package types
 
 import "io"
 
+// ManifestAnnotateOptions specifies options for `nerdctl manifest annotate`.
+type ManifestAnnotateOptions struct {
+	Stdout     io.Writer
+	GOptions   GlobalCommandOptions
+	Os         string
+	Arch       string
+	OsVersion  string
+	Variant    string
+	OsFeatures []string
+}
+
 // ManifestCreateOptions specifies options for `nerdctl manifest create`.
 type ManifestCreateOptions struct {
 	Stdout   io.Writer

pkg/cmd/manifest/annotate.go

@@ -0,0 +1,90 @@
+/*
+   Copyright The containerd Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package manifest
+
+import (
+	"context"
+	"errors"
+	"fmt"
+
+	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
+
+	"github.com/containerd/nerdctl/v2/pkg/api/types"
+	"github.com/containerd/nerdctl/v2/pkg/manifeststore"
+	"github.com/containerd/nerdctl/v2/pkg/referenceutil"
+	"github.com/containerd/nerdctl/v2/pkg/store"
+)
+
+func Annotate(ctx context.Context, listRef string, manifestRef string, options types.ManifestAnnotateOptions) error {
+	parsedListRef, err := referenceutil.Parse(listRef)
+	if err != nil {
+		return fmt.Errorf("failed to parse list reference: %w", err)
+	}
+
+	parsedManifestRef, err := referenceutil.Parse(manifestRef)
+	if err != nil {
+		return fmt.Errorf("failed to parse manifest reference: %w", err)
+	}
+
+	manifestStore, err := manifeststore.NewStore(options.GOptions.DataRoot)
+	if err != nil {
+		return fmt.Errorf("failed to create manifest store: %w", err)
+	}
+
+	imageManifest, err := manifestStore.Get(parsedListRef, parsedManifestRef)
+	if err != nil {
+		if errors.Is(err, store.ErrNotFound) {
+			return fmt.Errorf("manifest for image %s does not exist in %s", manifestRef, listRef)
+		}
+		return fmt.Errorf("failed to get manifest: %w", err)
+	}
+
+	if imageManifest.Descriptor.Platform == nil {
+		imageManifest.Descriptor.Platform = new(ocispec.Platform)
+	}
+
+	if options.Os != "" {
+		imageManifest.Descriptor.Platform.OS = options.Os
+	}
+
+	if options.Arch != "" {
+		imageManifest.Descriptor.Platform.Architecture = options.Arch
+	}
+
+	if options.Variant != "" {
+		imageManifest.Descriptor.Platform.Variant = options.Variant
+	}
+
+	if options.OsVersion != "" {
+		imageManifest.Descriptor.Platform.OSVersion = options.OsVersion
+	}
+
+	for _, osFeature := range options.OsFeatures {
+		imageManifest.Descriptor.Platform.OSFeatures = appendIfUnique(imageManifest.Descriptor.Platform.OSFeatures, osFeature)
+	}
+
+	return manifestStore.Save(parsedListRef, parsedManifestRef, imageManifest)
+}
+
+func appendIfUnique(list []string, str string) []string {
+	for _, s := range list {
+		if s == str {
+			return list
+		}
+	}
+	return append(list, str)
+}

pkg/manifeststore/manifeststore.go

@@ -28,6 +28,7 @@ import (
 )
 
 type Store interface {
+	Get(listRef *referenceutil.ImageReference, manifestRef *referenceutil.ImageReference) (*manifesttypes.DockerManifestEntry, error)
 	// GetList returns all the local manifests for a index or manifest list
 	GetList(listRef *referenceutil.ImageReference) ([]*manifesttypes.DockerManifestEntry, error)
 	// Save saves a manifest as part of a index or local manifest list
@@ -47,6 +48,19 @@ func NewStore(dataRoot string) (Store, error) {
 	return &manifestStore{store: st}, nil
 }
 
+func (s *manifestStore) Get(listRef *referenceutil.ImageReference, manifestRef *referenceutil.ImageReference) (*manifesttypes.DockerManifestEntry, error) {
+	var manifest *manifesttypes.DockerManifestEntry
+	err := s.store.WithLock(func() error {
+		listPath := makeFilesafeName(listRef.String())
+		manifestPath := makeFilesafeName(manifestRef.String())
+
+		var err error
+		manifest, err = s.getManifestFromPath(listPath, manifestPath)
+		return err
+	})
+	return manifest, err
+}
+
 func (s *manifestStore) GetList(listRef *referenceutil.ImageReference) ([]*manifesttypes.DockerManifestEntry, error) {
 	listPath := makeFilesafeName(listRef.String())