STACKITGit/cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

check for tty before creating colorables

Browse source
This commit is contained in:
nate smith 2020-09-01 11:18:34 -05:00
parent 734959c39d
commit 49ab3ec5bf
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pkg/iostreams/iostreams.go
View file

@ -103,12 +103,20 @@ func (s *IOStreams) TerminalWidth() int {
}
func System() *IOStreams {
return &IOStreams{
stdoutIsTTY := isTerminal(os.Stdout)
stderrIsTTY := isTerminal(os.Stderr)
io := &IOStreams{
In: os.Stdin,
Out: colorable.NewColorable(os.Stdout),
ErrOut: colorable.NewColorable(os.Stderr),
colorEnabled: os.Getenv("NO_COLOR") == "" && isTerminal(os.Stdout),
colorEnabled: os.Getenv("NO_COLOR") == "" && stdoutIsTTY,
}
// prevent duplicate isTerminal queries now that we know the answer
io.SetStdoutTTY(stdoutIsTTY)
io.SetStderrTTY(stderrIsTTY)
return io
}
func Test() (*IOStreams, *bytes.Buffer, *bytes.Buffer, *bytes.Buffer) {