refactor: status, metadata and content handlers for manifest index commands

cmd/oras/internal/display/content/discard.go

@@ -17,14 +17,19 @@ package content
 
 import ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 
-type discardHandler struct{}
+type DiscardHandler struct{}
 
 // OnContentFetched implements ManifestFetchHandler.
-func (discardHandler) OnContentFetched(ocispec.Descriptor, []byte) error {
+func (DiscardHandler) OnContentFetched(ocispec.Descriptor, []byte) error {
+	return nil
+}
+
+// OnContentCreated implements ManifestIndexCreateHandler.
+func (DiscardHandler) OnContentCreated([]byte) error {
 	return nil
 }
 
 // NewDiscardHandler returns a new discard handler.
-func NewDiscardHandler() ManifestFetchHandler {
-	return discardHandler{}
+func NewDiscardHandler() DiscardHandler {
+	return DiscardHandler{}
 }

cmd/oras/internal/display/content/interface.go

@@ -24,3 +24,12 @@ type ManifestFetchHandler interface {
 	// OnContentFetched is called after the manifest content is fetched.
 	OnContentFetched(desc ocispec.Descriptor, content []byte) error
 }
+
+// ManifestIndexCreateHandler handles raw output for manifest index create events.
+type ManifestIndexCreateHandler interface {
+	// OnContentCreated is called after the index content is created.
+	OnContentCreated(content []byte) error
+}
+
+// ManifestIndexUpdateHandler handles raw output for manifest index update events.
+type ManifestIndexUpdateHandler ManifestIndexCreateHandler

cmd/oras/internal/display/content/manifest_fetch.go

@@ -46,6 +46,10 @@ func (h *manifestFetch) OnContentFetched(desc ocispec.Descriptor, manifest []byt
 
 // NewManifestFetchHandler creates a new handler.
 func NewManifestFetchHandler(out io.Writer, pretty bool, outputPath string) ManifestFetchHandler {
+	// ignore --pretty when output to a file
+	if outputPath != "" && outputPath != "-" {
+		pretty = false
+	}
 	return &manifestFetch{
 		pretty:     pretty,
 		stdout:     out,

cmd/oras/internal/display/content/manifest_index.go

@@ -0,0 +1,58 @@
+/*
+Copyright The ORAS 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 content
+
+import (
+	"fmt"
+	"io"
+	"os"
+
+	"oras.land/oras/cmd/oras/internal/output"
+)
+
+// manifestIndexCreate handles raw content output.
+type manifestIndexCreate struct {
+	pretty     bool
+	stdout     io.Writer
+	outputPath string
+}
+
+// OnContentCreated is called after index content is created.
+func (h *manifestIndexCreate) OnContentCreated(manifest []byte) error {
+	out := h.stdout
+	if h.outputPath != "" && h.outputPath != "-" {
+		f, err := os.Create(h.outputPath)
+		if err != nil {
+			return fmt.Errorf("failed to open %q: %w", h.outputPath, err)
+		}
+		defer f.Close()
+		out = f
+	}
+	return output.PrintJSON(out, manifest, h.pretty)
+}
+
+// NewManifestIndexCreateHandler creates a new handler.
+func NewManifestIndexCreateHandler(out io.Writer, pretty bool, outputPath string) ManifestIndexCreateHandler {
+	// ignore --pretty when output to a file
+	if outputPath != "" && outputPath != "-" {
+		pretty = false
+	}
+	return &manifestIndexCreate{
+		pretty:     pretty,
+		stdout:     out,
+		outputPath: outputPath,
+	}
+}

cmd/oras/internal/display/handler.go

@@ -174,9 +174,40 @@ func NewManifestPushHandler(printer *output.Printer) metadata.ManifestPushHandle
 	return text.NewManifestPushHandler(printer)
 }
 
-// NewManifestIndexCreateHandler returns an index create handler.
-func NewManifestIndexCreateHandler(printer *output.Printer) metadata.ManifestIndexCreateHandler {
-	return text.NewManifestIndexCreateHandler(printer)
+// NewManifestIndexCreateHandler returns status, metadata and content handlers for index create command.
+func NewManifestIndexCreateHandler(outputPath string, printer *output.Printer, pretty bool) (
+	status.ManifestIndexCreateHandler,
+	metadata.ManifestIndexCreateHandler,
+	content.ManifestIndexCreateHandler) {
+	statusHandler := status.NewTextManifestIndexCreateHandler(printer)
+	metadataHandler := text.NewManifestIndexCreateHandler(printer)
+	contentHandler := content.NewManifestIndexCreateHandler(printer, pretty, outputPath)
+	switch outputPath {
+	case "":
+		contentHandler = content.NewDiscardHandler()
+	case "-":
+		statusHandler = status.NewDiscardHandler()
+		metadataHandler = metadata.NewDiscardHandler()
+	}
+	return statusHandler, metadataHandler, contentHandler
+}
+
+// NewManifestIndexUpdateHandler returns status, metadata and content handlers for index update command.
+func NewManifestIndexUpdateHandler(outputPath string, printer *output.Printer, pretty bool) (
+	status.ManifestIndexUpdateHandler,
+	metadata.ManifestIndexUpdateHandler,
+	content.ManifestIndexUpdateHandler) {
+	statusHandler := status.NewTextManifestIndexUpdateHandler(printer)
+	metadataHandler := text.NewManifestIndexCreateHandler(printer)
+	contentHandler := content.NewManifestIndexCreateHandler(printer, pretty, outputPath)
+	switch outputPath {
+	case "":
+		contentHandler = content.NewDiscardHandler()
+	case "-":
+		statusHandler = status.NewDiscardHandler()
+		metadataHandler = metadata.NewDiscardHandler()
+	}
+	return statusHandler, metadataHandler, contentHandler
 }
 
 // NewCopyHandler returns copy handlers.

cmd/oras/internal/display/metadata/discard.go

@@ -15,16 +15,28 @@
 
 package metadata
 
-import ocispec "github.com/opencontainers/image-spec/specs-go/v1"
+import (
+	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
+)
 
-type discard struct{}
+type Discard struct{}
 
 // NewDiscardHandler creates a new handler that discards output for all events.
-func NewDiscardHandler() discard {
-	return discard{}
+func NewDiscardHandler() Discard {
+	return Discard{}
 }
 
 // OnFetched implements ManifestFetchHandler.
-func (discard) OnFetched(string, ocispec.Descriptor, []byte) error {
+func (Discard) OnFetched(string, ocispec.Descriptor, []byte) error {
+	return nil
+}
+
+// OnTagged implements ManifestIndexCreateHandler.
+func (Discard) OnTagged(ocispec.Descriptor, string) error {
+	return nil
+}
+
+// OnCompleted implements ManifestIndexCreateHandler.
+func (Discard) OnCompleted(ocispec.Descriptor) error {
 	return nil
 }

cmd/oras/internal/display/metadata/interface.go

@@ -81,8 +81,12 @@ type ManifestPushHandler interface {
 // ManifestIndexCreateHandler handles metadata output for index create events.
 type ManifestIndexCreateHandler interface {
 	TaggedHandler
+	OnCompleted(desc ocispec.Descriptor) error
 }
 
+// ManifestIndexUpdateHandler handles metadata output for index update events.
+type ManifestIndexUpdateHandler ManifestIndexCreateHandler
+
 // CopyHandler handles metadata output for cp events.
 type CopyHandler interface {
 	TaggedHandler

cmd/oras/internal/display/metadata/text/manifest_index_create.go

@@ -33,7 +33,12 @@ func NewManifestIndexCreateHandler(printer *output.Printer) metadata.ManifestInd
 	}
 }
 
-// OnTagged implements metadata.TaggedHandler.
+// OnTagged implements TaggedHandler.
 func (h *ManifestIndexCreateHandler) OnTagged(_ ocispec.Descriptor, tag string) error {
 	return h.printer.Println("Tagged", tag)
 }
+
+// OnCompleted implements ManifestIndexCreateHandler.
+func (h *ManifestIndexCreateHandler) OnCompleted(desc ocispec.Descriptor) error {
+	return h.printer.Println("Digest:", desc.Digest)
+}

cmd/oras/internal/display/status/discard.go

@@ -18,6 +18,7 @@
 import (
 	"context"
 
+	"github.com/opencontainers/go-digest"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 	"oras.land/oras-go/v2"
 )
@@ -88,3 +89,38 @@
 func (DiscardHandler) OnNodeSkipped(desc ocispec.Descriptor) error {
 	return nil
 }
+
+// OnFetching implements referenceFetchHandler.
+func (DiscardHandler) OnFetching(string) error {
+	return nil
+}
+
+// OnFetched implements referenceFetchHandler.
+func (DiscardHandler) OnFetched(string, ocispec.Descriptor) error {
+	return nil
+}
+
+// OnManifestRemoved implements ManifestIndexUpdateHandler.
+func (DiscardHandler) OnManifestRemoved(digest.Digest) error {
+	return nil
+}
+
+// OnManifestAdded implements ManifestIndexUpdateHandler.
+func (DiscardHandler) OnManifestAdded(string, ocispec.Descriptor) error {
+	return nil
+}
+
+// OnIndexMerged implements ManifestIndexUpdateHandler.
+func (DiscardHandler) OnIndexMerged(string, ocispec.Descriptor) error {
+	return nil
+}
+
+// OnIndexPacked implements ManifestIndexCreateHandler.
+func (DiscardHandler) OnIndexPacked(ocispec.Descriptor) error {
+	return nil
+}
+
+// OnIndexPushed implements ManifestIndexCreateHandler.
+func (DiscardHandler) OnIndexPushed(string) error {
+	return nil
+}

cmd/oras/internal/display/status/interface.go

@@ -17,6 +17,8 @@ package status
 
 import (
 	"context"
+
+	"github.com/opencontainers/go-digest"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 	"oras.land/oras-go/v2"
 )
@@ -61,3 +63,19 @@ type CopyHandler interface {
 	PostCopy(ctx context.Context, desc ocispec.Descriptor) error
 	OnMounted(ctx context.Context, desc ocispec.Descriptor) error
 }
+
+// ManifestIndexCreateHandler handles status output for manifest index create command.
+type ManifestIndexCreateHandler interface {
+	OnFetching(manifestRef string) error
+	OnFetched(manifestRef string, desc ocispec.Descriptor) error
+	OnIndexPacked(desc ocispec.Descriptor) error
+	OnIndexPushed(path string) error
+}
+
+// ManifestIndexUpdateHandler handles status output for manifest index update command.
+type ManifestIndexUpdateHandler interface {
+	ManifestIndexCreateHandler
+	OnManifestRemoved(digest digest.Digest) error
+	OnManifestAdded(manifestRef string, desc ocispec.Descriptor) error
+	OnIndexMerged(indexRef string, desc ocispec.Descriptor) error
+}

cmd/oras/internal/display/status/text.go

@@ -19,8 +19,11 @@
 	"context"
 	"sync"
 
+	"oras.land/oras/internal/contentutil"
+	"oras.land/oras/internal/descriptor"
 	"oras.land/oras/internal/graph"
 
+	"github.com/opencontainers/go-digest"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 	"oras.land/oras-go/v2"
 	"oras.land/oras-go/v2/content"
@@ -179,3 +182,93 @@
 	ch.committed.Store(desc.Digest.String(), desc.Annotations[ocispec.AnnotationTitle])
 	return ch.printer.PrintStatus(desc, copyPromptMounted)
 }
+
+// TextManifestIndexCreateHandler handles text status output for manifest index create events.
+type TextManifestIndexCreateHandler struct {
+	printer *output.Printer
+}
+
+// OnFetching implements ManifestIndexCreateHandler.
+func (mich TextManifestIndexCreateHandler) OnFetching(source string) error {
+	return mich.printer.Println(IndexPromptFetching, source)
+}
+
+// OnFetched implements ManifestIndexCreateHandler.
+func (mich TextManifestIndexCreateHandler) OnFetched(source string, _ ocispec.Descriptor) error {
+	return mich.printer.Println(IndexPromptFetched, source)
+}
+
+// OnIndexPacked implements ManifestIndexCreateHandler.
+func (mich TextManifestIndexCreateHandler) OnIndexPacked(desc ocispec.Descriptor) error {
+	return mich.printer.Println(IndexPromptPacked, descriptor.ShortDigest(desc), ocispec.MediaTypeImageIndex)
+}
+
+// OnIndexPushed implements ManifestIndexCreateHandler.
+func (mich TextManifestIndexCreateHandler) OnIndexPushed(path string) error {
+	return mich.printer.Println(IndexPromptPushed, path)
+}
+
+// NewTextManifestIndexCreateHandler returns a new handler for manifest index create command.
+func NewTextManifestIndexCreateHandler(printer *output.Printer) ManifestIndexCreateHandler {
+	tmich := TextManifestIndexCreateHandler{
+		printer: printer,
+	}
+	return &tmich
+}
+
+// TextManifestIndexUpdateHandler handles text status output for manifest index update events.
+type TextManifestIndexUpdateHandler struct {
+	printer *output.Printer
+}
+
+// OnFetching implements ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnFetching(ref string) error {
+	return miuh.printer.Println(IndexPromptFetching, ref)
+}
+
+// OnFetched implements ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnFetched(ref string, desc ocispec.Descriptor) error {
+	if contentutil.IsDigest(ref) {
+		return miuh.printer.Println(IndexPromptFetched, ref)
+	}
+	return miuh.printer.Println(IndexPromptFetched, desc.Digest, ref)
+}
+
+// OnManifestRemoved implements metadata.ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnManifestRemoved(digest digest.Digest) error {
+	return miuh.printer.Println("Removed", digest)
+}
+
+// OnManifestAdded implements metadata.ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnManifestAdded(ref string, desc ocispec.Descriptor) error {
+	if contentutil.IsDigest(ref) {
+		return miuh.printer.Println("Added", ref)
+	}
+	return miuh.printer.Println("Added", desc.Digest, ref)
+}
+
+// OnIndexMerged implements metadata.ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnIndexMerged(ref string, desc ocispec.Descriptor) error {
+	if contentutil.IsDigest(ref) {
+		return miuh.printer.Println("Merged", ref)
+	}
+	return miuh.printer.Println("Merged", desc.Digest, ref)
+}
+
+// OnIndexUpdated implements metadata.ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnIndexPacked(desc ocispec.Descriptor) error {
+	return miuh.printer.Println("Updated", desc.Digest)
+}
+
+// OnIndexPushed implements metadata.ManifestIndexUpdateHandler.
+func (miuh TextManifestIndexUpdateHandler) OnIndexPushed(indexRef string) error {
+	return miuh.printer.Println("Pushed", indexRef)
+}
+
+// NewTextManifestIndexUpdateHandler returns a new handler for manifest index create command.
+func NewTextManifestIndexUpdateHandler(printer *output.Printer) ManifestIndexUpdateHandler {
+	miuh := TextManifestIndexUpdateHandler{
+		printer: printer,
+	}
+	return &miuh
+}