-
Notifications
You must be signed in to change notification settings - Fork 51
/
color.go
120 lines (90 loc) · 2.93 KB
/
color.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package terminal
import (
"os"
"regexp"
"golang.org/x/crypto/ssh/terminal"
"github.com/IBM-Cloud/ibm-cloud-cli-sdk/bluemix"
"github.com/fatih/color"
colorable "github.com/mattn/go-colorable"
)
var (
Output = colorable.NewColorableStdout()
ErrOutput = colorable.NewColorableStderr()
TerminalSupportsColors = isTerminal()
UserAskedForColors = ""
)
func init() {
InitColorSupport()
}
func InitColorSupport() {
color.NoColor = !ColorsEnabled()
}
func ColorsEnabled() bool {
return userDidNotDisableColor() &&
(userEnabledColors() || TerminalSupportsColors)
}
func userEnabledColors() bool {
return UserAskedForColors == "true" || bluemix.EnvColor.Get() == "true"
}
func userDidNotDisableColor() bool {
colorEnv := bluemix.EnvColor.Get()
return colorEnv != "false" && (UserAskedForColors != "false" || colorEnv == "true")
}
func Colorize(message string, color *color.Color) string {
return color.Sprint(message)
}
var decolorizerRegex = regexp.MustCompile(`\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]`)
func Decolorize(message string) string {
return string(decolorizerRegex.ReplaceAll([]byte(message), []byte("")))
}
func HeaderColor(message string) string {
return Colorize(message, color.New(color.Bold))
}
func CommandColor(message string) string {
return Colorize(message, color.New(color.FgYellow, color.Bold))
}
func StoppedColor(message string) string {
return Colorize(message, color.New(color.FgWhite, color.Bold))
}
func AdvisoryColor(message string) string {
return Colorize(message, color.New(color.FgYellow, color.Bold))
}
func CrashedColor(message string) string {
return Colorize(message, color.New(color.FgRed, color.Bold))
}
func FailureColor(message string) string {
return Colorize(message, color.New(color.FgRed, color.Bold))
}
func SuccessColor(message string) string {
return Colorize(message, color.New(color.FgGreen, color.Bold))
}
func EntityNameColor(message string) string {
return Colorize(message, color.New(color.FgCyan, color.Bold))
}
func PromptColor(message string) string {
return Colorize(message, color.New(color.FgCyan, color.Bold))
}
func TableContentHeaderColor(message string) string {
return Colorize(message, color.New(color.FgCyan, color.Bold))
}
func WarningColor(message string) string {
return Colorize(message, color.New(color.FgMagenta, color.Bold))
}
func LogStdoutColor(message string) string {
return Colorize(message, color.New(color.FgWhite, color.Bold))
}
func LogStderrColor(message string) string {
return Colorize(message, color.New(color.FgRed, color.Bold))
}
func LogHealthHeaderColor(message string) string {
return Colorize(message, color.New(color.FgWhite, color.Bold))
}
func LogAppHeaderColor(message string) string {
return Colorize(message, color.New(color.FgYellow, color.Bold))
}
func LogSysHeaderColor(message string) string {
return Colorize(message, color.New(color.FgCyan, color.Bold))
}
func isTerminal() bool {
return terminal.IsTerminal(int(os.Stdout.Fd()))
}