Merge pull request #1 from kaduartur/fix/sleep_time

[BUG] Fix lines cleaning

Author: kaduartur

pkg/spinner/spinner.go

@@ -25,6 +25,7 @@ type Spinner struct {
 	stopOnce  sync.Once
 	Output    io.Writer
 	NoTty     bool
+	timer     *time.Timer
 }
 
 // NewSpinner creates a Spinner with default template and FrameRate
@@ -34,6 +35,7 @@ func New(title string) *Spinner {
 		Template:  template.Default,
 		FrameRate: DefaultFrameRate,
 		runChan:   make(chan struct{}),
+		timer:     time.NewTimer(DefaultFrameRate),
 	}
 
 	var stdout interface{} = syscall.Stdout
@@ -76,9 +78,11 @@ func (s *Spinner) SetTemplate(template template.Template) *Spinner {
 
 // Stop stops the spinner execution
 func (s *Spinner) Stop() {
-	// prevent multiple calls
 	s.stopOnce.Do(func() {
 		close(s.runChan)
+		if !s.timer.Stop() { // stop the timer when the spinner was stopped, preventing the next line from being cleared
+			<-s.timer.C
+		}
 		s.clearLine()
 	})
 }
@@ -105,7 +109,9 @@ func (s *Spinner) animate() {
 		case !s.NoTty:
 			fmt.Print(out)
 		}
-		time.Sleep(s.FrameRate)
+
+		s.timer.Reset(s.FrameRate)
+		<-s.timer.C // Wait for timer
 		s.clearLine()
 	}
 }