mirror of https://github.com/edouardparis/lntop
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
134 lines
2.9 KiB
Go
134 lines
2.9 KiB
Go
package color
|
|
|
|
import "github.com/gookit/color"
|
|
|
|
type Color color.Color
|
|
|
|
var (
|
|
yellow = SprintFunc(color.New(color.FgYellow))
|
|
yellowBold = SprintFunc(color.New(color.FgYellow, color.Bold))
|
|
green = SprintFunc(color.New(color.FgGreen))
|
|
greenBold = SprintFunc(color.New(color.FgGreen, color.Bold))
|
|
greenBg = SprintFunc(color.New(color.FgBlack, color.BgGreen))
|
|
magentaBg = SprintFunc(color.New(color.FgBlack, color.BgMagenta))
|
|
red = SprintFunc(color.New(color.FgRed))
|
|
redBold = SprintFunc(color.New(color.FgRed, color.Bold))
|
|
cyan = SprintFunc(color.New(color.FgCyan))
|
|
cyanBold = SprintFunc(color.New(color.FgCyan, color.Bold))
|
|
cyanBg = SprintFunc(color.New(color.BgCyan, color.FgBlack))
|
|
white = SprintFunc(color.New())
|
|
whiteBold = SprintFunc(color.New(color.Bold))
|
|
blackBg = SprintFunc(color.New(color.BgBlack, color.FgWhite))
|
|
black = SprintFunc(color.New(color.FgBlack))
|
|
)
|
|
|
|
func SprintFunc(c color.Style) func(args ...interface{}) string {
|
|
return func(args ...interface{}) string {
|
|
return c.Sprint(args...)
|
|
}
|
|
}
|
|
|
|
type Option func(*options)
|
|
|
|
type options struct {
|
|
bold bool
|
|
bg bool
|
|
}
|
|
|
|
func newOptions(opts []Option) options {
|
|
options := options{}
|
|
for i := range opts {
|
|
if opts[i] == nil {
|
|
continue
|
|
}
|
|
opts[i](&options)
|
|
}
|
|
return options
|
|
}
|
|
|
|
func Bold(o *options) { o.bold = true }
|
|
func Background(o *options) { o.bg = true }
|
|
|
|
func Yellow(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bold {
|
|
return yellowBold
|
|
}
|
|
return yellow
|
|
}
|
|
|
|
func Green(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bold {
|
|
return greenBold
|
|
}
|
|
|
|
if options.bg {
|
|
return greenBg
|
|
}
|
|
|
|
return green
|
|
}
|
|
|
|
func Red(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bold {
|
|
return redBold
|
|
}
|
|
return red
|
|
}
|
|
|
|
func White(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bold {
|
|
return whiteBold
|
|
}
|
|
return white
|
|
}
|
|
|
|
func Cyan(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bold {
|
|
return cyanBold
|
|
}
|
|
if options.bg {
|
|
return cyanBg
|
|
}
|
|
return cyan
|
|
}
|
|
|
|
func Black(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bg {
|
|
return blackBg
|
|
}
|
|
return black
|
|
}
|
|
|
|
func Magenta(opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
if options.bg {
|
|
return magentaBg
|
|
}
|
|
return magentaBg
|
|
}
|
|
|
|
func HSL256(h, s, l float64, opts ...Option) func(a ...interface{}) string {
|
|
options := newOptions(opts)
|
|
val := color.HSL(h, s, l).C256().Value()
|
|
c := color.S256(val)
|
|
if options.bg {
|
|
fg := color.White.C256().Value()
|
|
if l > 0.5 {
|
|
fg = color.Black.C256().Value()
|
|
}
|
|
c = color.S256(fg, val)
|
|
}
|
|
if options.bold {
|
|
c.AddOpts(color.Bold)
|
|
}
|
|
return func(a ...interface{}) string {
|
|
return c.Sprint(a...)
|
|
}
|
|
}
|