91 lines
1.6 KiB
Go
91 lines
1.6 KiB
Go
|
package color
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strconv"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Interface guard
|
||
|
|
var _ Color = &Basic{}
|
||
|
|
|
||
|
|
// Basic defines a custom color object which is defined by SGR parameters.
|
||
|
|
type Basic struct {
|
||
|
|
Attrs []BasicAttribute
|
||
|
|
}
|
||
|
|
|
||
|
|
// BasicAttribute defines a single SGR Code
|
||
|
|
type BasicAttribute int
|
||
|
|
|
||
|
|
// Logger attributes
|
||
|
|
const (
|
||
|
|
Reset BasicAttribute = iota
|
||
|
|
Bold
|
||
|
|
Faint
|
||
|
|
Italic
|
||
|
|
Underline
|
||
|
|
BlinkSlow
|
||
|
|
BlinkRapid
|
||
|
|
ReverseVideo
|
||
|
|
Concealed
|
||
|
|
CrossedOut
|
||
|
|
)
|
||
|
|
|
||
|
|
// New returns a new Color with attrs Attributes
|
||
|
|
func New(attrs ...BasicAttribute) *Basic {
|
||
|
|
return &Basic{Attrs: attrs}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add adds new Attributes to a Color
|
||
|
|
func (b *Basic) Add(attrs ...BasicAttribute) {
|
||
|
|
for _, attr := range attrs {
|
||
|
|
has := false
|
||
|
|
for _, att := range b.Attrs {
|
||
|
|
if attr == att {
|
||
|
|
has = true
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if !has {
|
||
|
|
b.Attrs = append(b.Attrs, attr)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// String returns a string representation of the sum of a Color's Attributes
|
||
|
|
func (b *Basic) String() string {
|
||
|
|
attrs := 0
|
||
|
|
for _, attr := range b.Attrs {
|
||
|
|
attrs += int(attr)
|
||
|
|
}
|
||
|
|
return fmt.Sprintf("%d", attrs)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Basic) sequence() string {
|
||
|
|
format := make([]string, len(b.Attrs))
|
||
|
|
for i, v := range b.Attrs {
|
||
|
|
format[i] = strconv.Itoa(int(v))
|
||
|
|
}
|
||
|
|
|
||
|
|
return strings.Join(format, ";")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Basic) wrap(s string) string {
|
||
|
|
return b.format() + s + b.unformat()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Basic) format() string {
|
||
|
|
return fmt.Sprintf("%s[%sm", escape, b.sequence())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Basic) unformat() string {
|
||
|
|
return fmt.Sprintf("%s[%dm", escape, Reset)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Basic) Format(text string) string {
|
||
|
|
if len(b.Attrs) > 0 {
|
||
|
|
return b.wrap(text)
|
||
|
|
}
|
||
|
|
return text
|
||
|
|
}
|