manifest: support nerdctl manifest create command

Support `nerdctl manifest create` command

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

Author: ChengyuZhu6

cmd/nerdctl/manifest/manifest.go

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

cmd/nerdctl/manifest/manifest_create.go

@@ -0,0 +1,87 @@
+/*
+   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 (
+	"fmt"
+
+	"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 CreateCommand() *cobra.Command {
+	var cmd = &cobra.Command{
+		Use:               "create INDEX/MANIFESTLIST MANIFEST [MANIFEST...]",
+		Short:             "Create a local index/manifest list for annotating and pushing to a registry",
+		Args:              cobra.MinimumNArgs(2),
+		RunE:              createAction,
+		ValidArgsFunction: createShellComplete,
+		SilenceUsage:      true,
+		SilenceErrors:     true,
+	}
+	cmd.Flags().Bool("amend", false, "Amend the existing index/manifest list")
+	cmd.Flags().Bool("insecure", false, "Allow communication with an insecure registry")
+	return cmd
+}
+
+func processCreateFlags(cmd *cobra.Command) (types.ManifestCreateOptions, error) {
+	globalOptions, err := helpers.ProcessRootCmdFlags(cmd)
+	if err != nil {
+		return types.ManifestCreateOptions{}, err
+	}
+	amend, err := cmd.Flags().GetBool("amend")
+	if err != nil {
+		return types.ManifestCreateOptions{}, err
+	}
+	insecure, err := cmd.Flags().GetBool("insecure")
+	if err != nil {
+		return types.ManifestCreateOptions{}, err
+	}
+	return types.ManifestCreateOptions{
+		Stdout:   cmd.OutOrStdout(),
+		GOptions: globalOptions,
+		Amend:    amend,
+		Insecure: insecure,
+	}, nil
+}
+
+func createAction(cmd *cobra.Command, args []string) error {
+	createOptions, err := processCreateFlags(cmd)
+	if err != nil {
+		return err
+	}
+
+	listRef := args[0]
+	manifestRefs := args[1:]
+
+	listRef, err = manifest.Create(cmd.Context(), listRef, manifestRefs, createOptions)
+	if err != nil {
+		return err
+	}
+
+	fmt.Fprintln(createOptions.Stdout, "Created manifest list", listRef)
+
+	return nil
+}
+
+func createShellComplete(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+	return completion.ImageNames(cmd)
+}

pkg/api/types/manifest_types.go

@@ -18,6 +18,16 @@ package types
 
 import "io"
 
+// ManifestCreateOptions specifies options for `nerdctl manifest create`.
+type ManifestCreateOptions struct {
+	Stdout   io.Writer
+	GOptions GlobalCommandOptions
+	// Amend an existing manifest list
+	Amend bool
+	// Allow communication with an insecure registry
+	Insecure bool
+}
+
 // ManifestInspectOptions specifies options for `nerdctl manifest inspect`.
 type ManifestInspectOptions struct {
 	Stdout   io.Writer

pkg/cmd/manifest/create.go

@@ -0,0 +1,86 @@
+/*
+   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"
+	"fmt"
+
+	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
+
+	"github.com/containerd/containerd/v2/core/images"
+
+	"github.com/containerd/nerdctl/v2/pkg/api/types"
+	"github.com/containerd/nerdctl/v2/pkg/manifeststore"
+	"github.com/containerd/nerdctl/v2/pkg/manifestutil"
+	"github.com/containerd/nerdctl/v2/pkg/referenceutil"
+)
+
+// Create creates a local manifest list/index
+func Create(ctx context.Context, listRef string, manifestRefs []string, options types.ManifestCreateOptions) (string, error) {
+	parsedListRef, err := referenceutil.Parse(listRef)
+	if err != nil {
+		return "", fmt.Errorf("failed to parse list reference: %w", err)
+	}
+
+	manifestStore, err := manifeststore.NewStore(options.GOptions.DataRoot)
+	if err != nil {
+		return "", fmt.Errorf("failed to create manifest store: %w", err)
+	}
+
+	existingManifests, err := manifestStore.GetList(parsedListRef)
+	if err == nil && len(existingManifests) > 0 && !options.Amend {
+		return "", fmt.Errorf("refusing to amend an existing manifest list with no --amend flag")
+	}
+
+	for _, manifestRef := range manifestRefs {
+		parsedRef, err := referenceutil.Parse(manifestRef)
+		if err != nil {
+			return "", fmt.Errorf("failed to parse manifest reference %s: %w", manifestRef, err)
+		}
+
+		manifest, desc, rawData, err := manifestutil.GetManifest(ctx, parsedRef, options.GOptions, options.Insecure)
+		if err != nil {
+			return "", fmt.Errorf("failed to fetch manifest %s: %w", manifestRef, err)
+		}
+
+		// Check if the manifest is manifest list
+		if desc.MediaType == images.MediaTypeDockerSchema2ManifestList || desc.MediaType == ocispec.MediaTypeImageIndex {
+			return "", fmt.Errorf("%s is a manifest list", manifestRef)
+		}
+
+		imageManifest, err := manifestutil.CreateManifestEntry(parsedRef, desc, rawData)
+		if err != nil {
+			return "", fmt.Errorf("failed to create manifest entry for %s: %w", manifestRef, err)
+		}
+
+		// Get platform information from config
+		if desc.MediaType == ocispec.MediaTypeImageManifest || desc.MediaType == images.MediaTypeDockerSchema2Manifest {
+			platform, err := manifestutil.GetPlatform(ctx, parsedRef.Domain, options.GOptions, options.Insecure, manifestRef, manifest)
+			if err != nil {
+				return "", fmt.Errorf("failed to extract platform for %s: %w", manifestRef, err)
+			}
+			imageManifest.Descriptor.Platform = platform
+		}
+
+		if err := manifestStore.Save(parsedListRef, parsedRef, &imageManifest); err != nil {
+			return "", fmt.Errorf("failed to store manifest %s: %w", manifestRef, err)
+		}
+	}
+
+	return listRef, nil
+}