fix: do not create an output from an output

Signed-off-by: Carlos A Becker <caarlos0@users.noreply.github.com>
charmbracelet · Apr 12, 2023 · 3912fa0 · 3912fa0
1 parent 8514d90
commit 3912fa0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/options.go b/options.go
@@ -28,7 +28,12 @@ func WithContext(ctx context.Context) ProgramOption {
 // won't need to use this.
 func WithOutput(output io.Writer) ProgramOption {
 	return func(p *Program) {
-		p.output = termenv.NewOutput(output, termenv.WithColorCache(true))
+		switch o := output.(type) {
+		case *termenv.Output:
+			p.output = o
+		default:
+			p.output = termenv.NewOutput(o, termenv.WithColorCache(true))
+		}
 	}
 }