Fix the timer display

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

pkg/console/console.go

@@ -15,6 +15,7 @@
 package console
 
 import (
+	"context"
 	"fmt"
 	"os/exec"
 	"time"
@@ -54,33 +55,37 @@ func (s StandardConsole) Run(cmd string, opts ...func(cmd *exec.Cmd)) (string, e
 	for _, o := range opts {
 		o(c)
 	}
-	displayProgress(s.logger, 10*time.Second, fmt.Sprintf("Still running command '%s'", cmd))
 
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+	// Start the timer log
+	go displayProgress(ctx, s.logger, 10*time.Second, fmt.Sprintf("Still running command '%s'", cmd))
+
+	// Run the command
 	out, err := c.CombinedOutput()
+
+	// Stop the timer log
+	cancel()
+
 	if err != nil {
 		return string(out), fmt.Errorf("failed to run %s: %v", cmd, err)
 	}
 
 	return string(out), err
 }
 
-func displayProgress(log logger.Interface, tick time.Duration, message string) {
+func displayProgress(ctx context.Context, log logger.Interface, tick time.Duration, message string) {
 	ticker := time.NewTicker(tick)
-	done := make(chan bool)
-
-	go func() {
-		for {
-			select {
-			case <-done:
-				ticker.Stop()
-				return
-			case <-ticker.C:
-				log.Info(message)
-			}
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ctx.Done():
+			return
+		case <-ticker.C:
+			log.Info(message)
 		}
-	}()
-
-	return
+	}
 }
 
 func (s StandardConsole) Start(cmd *exec.Cmd, opts ...func(cmd *exec.Cmd)) error {