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.

88 lines
1.6 KiB

package style
import (
"errors"
"fmt"
"strings"
"sync"
)
//nolint:gochecknoglobals
var (
styles = map[string]Style{
"error": {Foreground: White, Background: Red},
"info": {Foreground: Green},
"comment": {Foreground: Yellow},
"question": {Foreground: Black, Background: Cyan},
}
stylesMu sync.Mutex
empty = Style{}
)
var (
ErrNotFound = errors.New("console: style not found")
ErrDuplicateStyle = errors.New("console: Register called twice")
)
func Empty() Style {
return empty
}
func Find(name string) (Style, error) {
if st, has := styles[name]; has {
return st, nil
}
return empty, ErrNotFound
}
func Register(name string, style Style) error {
stylesMu.Lock()
defer stylesMu.Unlock()
if _, has := styles[name]; has {
return fmt.Errorf("%w for style %s", ErrDuplicateStyle, name)
}
styles[name] = style
return nil
}
func MustRegister(name string, style Style) {
if err := Register(name, style); err != nil {
panic(err)
}
}
type Style struct {
Background Color
Foreground Color
Options []Option
}
func (s Style) Apply(msg string) string {
return s.Set(ActionSet) + msg + s.Set(ActionUnset)
}
func (s Style) Set(action int) string {
style := make([]string, 0, len(s.Options))
if s.Foreground != "" {
style = append(style, "3"+s.Foreground.Apply(action))
}
if s.Background != "" {
style = append(style, "4"+s.Background.Apply(action))
}
for _, opt := range s.Options {
style = append(style, opt.Apply(action))
}
if len(style) == 0 {
return ""
}
return "\033[" + strings.Join(style, ";") + "m"
}