78 lines
2 KiB
Go
78 lines
2 KiB
Go
|
package iostreams
|
||
|
|
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
// ColorScheme provides semantic color methods that respect whether color output is enabled.
|
||
|
|
type ColorScheme struct {
|
||
|
|
enabled bool
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewColorScheme creates a ColorScheme. When enabled is false, all methods return
|
||
|
|
// undecorated text.
|
||
|
|
func NewColorScheme(enabled bool) *ColorScheme {
|
||
|
|
return &ColorScheme{enabled: enabled}
|
||
|
|
}
|
||
|
|
|
||
|
|
// colorize wraps text in ANSI escape codes if color is enabled.
|
||
|
|
func (cs *ColorScheme) colorize(code string, text string) string {
|
||
|
|
if !cs.enabled {
|
||
|
|
return text
|
||
|
|
}
|
||
|
|
return fmt.Sprintf("\033[%sm%s\033[0m", code, text)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Bold renders text in bold.
|
||
|
|
func (cs *ColorScheme) Bold(s string) string {
|
||
|
|
return cs.colorize("1", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Red renders text in red.
|
||
|
|
func (cs *ColorScheme) Red(s string) string {
|
||
|
|
return cs.colorize("31", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Green renders text in green.
|
||
|
|
func (cs *ColorScheme) Green(s string) string {
|
||
|
|
return cs.colorize("32", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Yellow renders text in yellow.
|
||
|
|
func (cs *ColorScheme) Yellow(s string) string {
|
||
|
|
return cs.colorize("33", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Cyan renders text in cyan.
|
||
|
|
func (cs *ColorScheme) Cyan(s string) string {
|
||
|
|
return cs.colorize("36", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Magenta renders text in magenta.
|
||
|
|
func (cs *ColorScheme) Magenta(s string) string {
|
||
|
|
return cs.colorize("35", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Muted renders text in gray (dimmed).
|
||
|
|
func (cs *ColorScheme) Muted(s string) string {
|
||
|
|
return cs.colorize("90", s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SuccessIcon returns a green check mark if color is enabled, plain otherwise.
|
||
|
|
func (cs *ColorScheme) SuccessIcon() string {
|
||
|
|
return cs.Green("✓")
|
||
|
|
}
|
||
|
|
|
||
|
|
// WarningIcon returns a yellow exclamation mark if color is enabled, plain otherwise.
|
||
|
|
func (cs *ColorScheme) WarningIcon() string {
|
||
|
|
return cs.Yellow("!")
|
||
|
|
}
|
||
|
|
|
||
|
|
// FailureIcon returns a red X mark if color is enabled, plain otherwise.
|
||
|
|
func (cs *ColorScheme) FailureIcon() string {
|
||
|
|
return cs.Red("✗")
|
||
|
|
}
|
||
|
|
|
||
|
|
// SuccessIconWithColor returns the success icon followed by the message in green.
|
||
|
|
func (cs *ColorScheme) SuccessIconWithColor(msg string) string {
|
||
|
|
return cs.SuccessIcon() + " " + cs.Green(msg)
|
||
|
|
}
|