ctr/commands/images/push: don't show progress if it is debug mode

If user sets debug mode, the command push should only show the debug log
information. If the stdout is with flush by the progress status, it is
hard to see the debug log.

Signed-off-by: Wei Fu <fuweid89@gmail.com>

cmd/ctr/commands/images/push.go

@@ -63,16 +63,19 @@ var pushCommand = cli.Command{
 		var (
 			ref   = context.Args().First()
 			local = context.Args().Get(1)
+			debug = context.GlobalBool("debug")
 			desc  ocispec.Descriptor
 		)
 		if ref == "" {
 			return errors.New("please provide a remote image reference to push")
 		}
+
 		client, ctx, cancel, err := commands.NewClient(context)
 		if err != nil {
 			return err
 		}
 		defer cancel()
+
 		if manifest := context.String("manifest"); manifest != "" {
 			desc.Digest, err = digest.Parse(manifest)
 			if err != nil {
@@ -98,7 +101,12 @@ var pushCommand = cli.Command{
 
 		eg, ctx := errgroup.WithContext(ctx)
 
+		// used to notify the progress writer
+		doneCh := make(chan struct{})
+
 		eg.Go(func() error {
+			defer close(doneCh)
+
 			log.G(ctx).WithField("image", ref).WithField("digest", desc.Digest).Debug("pushing")
 
 			jobHandler := images.HandlerFunc(func(ctx gocontext.Context, desc ocispec.Descriptor) ([]ocispec.Descriptor, error) {
@@ -112,18 +120,16 @@ var pushCommand = cli.Command{
 			)
 		})
 
-		errs := make(chan error)
-		go func() {
-			defer close(errs)
-			errs <- eg.Wait()
-		}()
-
+		// don't show progress if debug mode is set
+		if !debug {
+			eg.Go(func() error {
 				var (
 					ticker = time.NewTicker(100 * time.Millisecond)
 					fw     = progress.NewWriter(os.Stdout)
 					start  = time.Now()
 					done   bool
 				)
+
 				defer ticker.Stop()
 
 				for {
@@ -140,15 +146,15 @@ var pushCommand = cli.Command{
 							fw.Flush()
 							return nil
 						}
-			case err := <-errs:
-				if err != nil {
-					return err
-				}
+					case <-doneCh:
 						done = true
 					case <-ctx.Done():
 						done = true // allow ui to update once more
 					}
 				}
+			})
+		}
+		return eg.Wait()
 	},
 }