fix: lock renderer during stop/kill

charmbracelet · Oct 9, 2022 · b9bd01b · b9bd01b
1 parent e119cd8
commit b9bd01b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/standard_renderer.go b/standard_renderer.go
@@ -76,7 +76,12 @@ func (r *standardRenderer) start() {
 
 // stop permanently halts the renderer, rendering the final frame.
 func (r *standardRenderer) stop() {
+	// flush locks the mutex
 	r.flush()
+
+	r.mtx.Lock()
+	defer r.mtx.Unlock()
+
 	r.out.ClearLine()
 	r.once.Do(func() {
 		close(r.done)
@@ -91,6 +96,9 @@ func (r *standardRenderer) stop() {
 
 // kill halts the renderer. The final frame will not be rendered.
 func (r *standardRenderer) kill() {
+	r.mtx.Lock()
+	defer r.mtx.Unlock()
+
 	r.out.ClearLine()
 	r.once.Do(func() {
 		close(r.done)