88aacc14f0
We used to send the ANSI sequence for "bright black" when we wanted gray, but this color turns out to not be visible in some popular color schemes. Instead, when we detect a 256-color terminal, switch to displaying a color sequence for gray that is consistent and does not depend on terminal color scheme.
123 lines
2.2 KiB
Go
123 lines
2.2 KiB
Go
package iostreams
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/mgutz/ansi"
|
|
)
|
|
|
|
var (
|
|
magenta = ansi.ColorFunc("magenta")
|
|
cyan = ansi.ColorFunc("cyan")
|
|
red = ansi.ColorFunc("red")
|
|
yellow = ansi.ColorFunc("yellow")
|
|
blue = ansi.ColorFunc("blue")
|
|
green = ansi.ColorFunc("green")
|
|
gray = ansi.ColorFunc("black+h")
|
|
bold = ansi.ColorFunc("default+b")
|
|
|
|
gray256 = func(t string) string {
|
|
return fmt.Sprintf("\x1b[%d;5;%dm%s\x1b[m", 38, 242, t)
|
|
}
|
|
)
|
|
|
|
func EnvColorDisabled() bool {
|
|
return os.Getenv("NO_COLOR") != "" || os.Getenv("CLICOLOR") == "0"
|
|
}
|
|
|
|
func EnvColorForced() bool {
|
|
return os.Getenv("CLICOLOR_FORCE") != "" && os.Getenv("CLICOLOR_FORCE") != "0"
|
|
}
|
|
|
|
func Is256ColorSupported() bool {
|
|
term := os.Getenv("TERM")
|
|
colorterm := os.Getenv("COLORTERM")
|
|
|
|
return strings.Contains(term, "256") ||
|
|
strings.Contains(term, "24bit") ||
|
|
strings.Contains(term, "truecolor") ||
|
|
strings.Contains(colorterm, "256") ||
|
|
strings.Contains(colorterm, "24bit") ||
|
|
strings.Contains(colorterm, "truecolor")
|
|
}
|
|
|
|
func NewColorScheme(enabled, is256enabled bool) *ColorScheme {
|
|
return &ColorScheme{
|
|
enabled: enabled,
|
|
is256enabled: is256enabled,
|
|
}
|
|
}
|
|
|
|
type ColorScheme struct {
|
|
enabled bool
|
|
is256enabled bool
|
|
}
|
|
|
|
func (c *ColorScheme) Bold(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return bold(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Red(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return red(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Yellow(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return yellow(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Green(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return green(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Gray(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
if c.is256enabled {
|
|
return gray256(t)
|
|
}
|
|
return gray(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Magenta(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return magenta(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Cyan(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return cyan(t)
|
|
}
|
|
|
|
func (c *ColorScheme) Blue(t string) string {
|
|
if !c.enabled {
|
|
return t
|
|
}
|
|
return blue(t)
|
|
}
|
|
|
|
func (c *ColorScheme) SuccessIcon() string {
|
|
return c.Green("✓")
|
|
}
|
|
|
|
func (c *ColorScheme) WarningIcon() string {
|
|
return c.Yellow("!")
|
|
}
|