You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.0 KiB
86 lines
2.0 KiB
5 years ago
|
package color
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"os"
|
||
|
|
||
|
"github.com/fatih/color"
|
||
|
"github.com/mattn/go-isatty"
|
||
|
)
|
||
|
|
||
|
// global colors.
|
||
|
var (
|
||
|
// Check if we stderr, stdout are dumb terminals, we do not apply
|
||
|
// ansi coloring on dumb terminals.
|
||
|
IsTerminal = func() bool {
|
||
|
return isatty.IsTerminal(os.Stdout.Fd()) && isatty.IsTerminal(os.Stderr.Fd())
|
||
|
}
|
||
|
|
||
|
Bold = func() func(a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.Bold).SprintFunc()
|
||
|
}
|
||
|
return fmt.Sprint
|
||
|
}()
|
||
|
Red = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgRed).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
Blue = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgBlue).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
Yellow = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgYellow).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
CyanBold = func() func(a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
color.New(color.FgCyan, color.Bold).SprintFunc()
|
||
|
}
|
||
|
return fmt.Sprint
|
||
|
}()
|
||
|
YellowBold = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgYellow, color.Bold).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
BgYellow = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.BgYellow).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
Black = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgBlack).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
FgRed = func() func(a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgRed).SprintFunc()
|
||
|
}
|
||
|
return fmt.Sprint
|
||
|
}()
|
||
|
BgRed = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.BgRed).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
FgWhite = func() func(format string, a ...interface{}) string {
|
||
|
if IsTerminal() {
|
||
|
return color.New(color.FgWhite).SprintfFunc()
|
||
|
}
|
||
|
return fmt.Sprintf
|
||
|
}()
|
||
|
)
|