feat: print x/y progress to the stdout

print a counter to show progress while pushing images and embedded
cluster artifacts.

Author: ricardomaraschini

pkg/image/airgap.go

@@ -172,6 +172,7 @@ func TagAndPushImagesFromBundle(airgapBundle string, options imagetypes.PushImag
 		Registry:   options.Registry,
 		Tag:        imageutil.SanitizeTag(fmt.Sprintf("%s-%s-%s", airgap.Spec.ChannelID, airgap.Spec.UpdateCursor, airgap.Spec.VersionLabel)),
 		HTTPClient: orasretry.DefaultClient,
+		LogForUI:   options.LogForUI,
 	}
 	pushedArtifacts, err := PushEmbeddedClusterArtifacts(airgapBundle, pushEmbeddedArtifactsOpts)
 	if err != nil {
@@ -226,6 +227,8 @@ func PushImagesFromTempRegistry(airgapRootDir string, imageList []string, option
 		defer wc.Close()
 	}
 
+	totalImages := len(imageInfos)
+	var imageCounter int
 	for imageID, imageInfo := range imageInfos {
 		srcRef, err := tempRegistry.SrcRef(imageID)
 		if err != nil {
@@ -280,6 +283,10 @@ func PushImagesFromTempRegistry(airgapRootDir string, imageList []string, option
 				ReportWriter:      reportWriter,
 			},
 		}
+		imageCounter++
+		if pushImageOpts.LogForUI {
+			fmt.Printf("Pushing image %d/%d\n", imageCounter, totalImages)
+		}
 		if err := pushImage(pushImageOpts); err != nil {
 			return errors.Wrapf(err, "failed to push image %s", imageID)
 		}
@@ -699,6 +706,8 @@ func PushEmbeddedClusterArtifacts(airgapBundle string, opts imagetypes.PushEmbed
 	}
 	defer gzipReader.Close()
 
+	var artifacts []string
+
 	tarReader := tar.NewReader(gzipReader)
 	pushedArtifacts := make([]string, 0)
 	for {
@@ -718,52 +727,52 @@ func PushEmbeddedClusterArtifacts(airgapBundle string, opts imagetypes.PushEmbed
 			continue
 		}
 
-		if err := func() error {
-			dstFilePath := filepath.Join(tmpDir, header.Name)
-			if err := os.MkdirAll(filepath.Dir(dstFilePath), 0755); err != nil {
-				return errors.Wrap(err, "failed to create path")
-			}
-			defer os.RemoveAll(dstFilePath)
+		dstFilePath := filepath.Join(tmpDir, header.Name)
+		if err := os.MkdirAll(filepath.Dir(dstFilePath), 0755); err != nil {
+			return nil, errors.Wrap(err, "failed to create path")
+		}
 
-			dstFile, err := os.Create(dstFilePath)
-			if err != nil {
-				return errors.Wrap(err, "failed to create file")
-			}
-			defer dstFile.Close()
+		dstFile, err := os.Create(dstFilePath)
+		if err != nil {
+			return nil, errors.Wrap(err, "failed to create file")
+		}
 
-			if _, err := io.Copy(dstFile, tarReader); err != nil {
-				return errors.Wrap(err, "failed to copy file data")
-			}
+		if _, err := io.Copy(dstFile, tarReader); err != nil {
+			dstFile.Close()
+			return nil, errors.Wrap(err, "failed to copy file data")
+		}
 
-			// push each file as an oci artifact to the registry
-			name := filepath.Base(dstFilePath)
-			repository := filepath.Join("embedded-cluster", imageutil.SanitizeRepo(name))
-			artifactFile := imagetypes.OCIArtifactFile{
-				Name:      name,
-				Path:      dstFilePath,
-				MediaType: EmbeddedClusterMediaType,
-			}
+		dstFile.Close()
+		artifacts = append(artifacts, dstFilePath)
+	}
 
-			pushOCIArtifactOpts := imagetypes.PushOCIArtifactOptions{
-				Files:        []imagetypes.OCIArtifactFile{artifactFile},
-				ArtifactType: EmbeddedClusterArtifactType,
-				Registry:     opts.Registry,
-				Repository:   repository,
-				Tag:          opts.Tag,
-				HTTPClient:   opts.HTTPClient,
-			}
+	for i, dstFilePath := range artifacts {
+		name := filepath.Base(dstFilePath)
+		repository := filepath.Join("embedded-cluster", imageutil.SanitizeRepo(name))
+		artifactFile := imagetypes.OCIArtifactFile{
+			Name:      name,
+			Path:      dstFilePath,
+			MediaType: EmbeddedClusterMediaType,
+		}
 
-			artifact := fmt.Sprintf("%s:%s", filepath.Join(opts.Registry.Endpoint, opts.Registry.Namespace, repository), opts.Tag)
-			fmt.Printf("Pushing artifact %s\n", artifact)
-			if err := pushOCIArtifact(pushOCIArtifactOpts); err != nil {
-				return errors.Wrapf(err, "failed to push oci artifact %s", name)
-			}
-			pushedArtifacts = append(pushedArtifacts, artifact)
+		pushOCIArtifactOpts := imagetypes.PushOCIArtifactOptions{
+			Files:        []imagetypes.OCIArtifactFile{artifactFile},
+			ArtifactType: EmbeddedClusterArtifactType,
+			Registry:     opts.Registry,
+			Repository:   repository,
+			Tag:          opts.Tag,
+			HTTPClient:   opts.HTTPClient,
+		}
 
-			return nil
-		}(); err != nil {
-			return nil, err
+		if opts.LogForUI {
+			fmt.Printf("Pushing artifact %d/%d\n", i+1, len(artifacts))
+		}
+
+		artifact := fmt.Sprintf("%s:%s", filepath.Join(opts.Registry.Endpoint, opts.Registry.Namespace, repository), opts.Tag)
+		if err := pushOCIArtifact(pushOCIArtifactOpts); err != nil {
+			return nil, errors.Wrapf(err, "failed to push oci artifact %s", name)
 		}
+		pushedArtifacts = append(pushedArtifacts, artifact)
 	}
 
 	return pushedArtifacts, nil

pkg/image/types/types.go

@@ -87,6 +87,7 @@ type PushEmbeddedClusterArtifactsOptions struct {
 	Registry   dockerregistrytypes.RegistryOptions
 	Tag        string
 	HTTPClient *http.Client
+	LogForUI   bool
 }
 
 type PushOCIArtifactOptions struct {