forked from Mirrors/bubbletea
fix(output): reuse termenv output (#715)
If the passed io.Writer is a termenv.Output, use it instead of creating a new termenv.Output.
This commit is contained in:
parent
c56884c0e2
commit
29254a08f1
|
@ -28,7 +28,11 @@ 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))
|
||||
if o, ok := output.(*termenv.Output); ok {
|
||||
p.output = o
|
||||
} else {
|
||||
p.output = termenv.NewOutput(output, termenv.WithColorCache(true))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue