base code | coloring format v1 (foreground and/or background) | only _real_ issue for now, text cannot contain the character '$' due to regexp mechanism

This commit is contained in:
xdrm-brackets 2019-01-24 14:38:35 +01:00
commit 2b9e25fa39
2 changed files with 167 additions and 0 deletions

82
colors.go Normal file
View File

@ -0,0 +1,82 @@
package clifmt
import (
"fmt"
"io"
"strconv"
)
type terminalColor uint32
var colorMap = make(map[string]terminalColor)
func init() {
colorMap["red"] = 0xff0000
colorMap["green"] = 0x00ff00
colorMap["blue"] = 0x0000ff
}
// fromName returns the integer value of a color name
// from the built-in color map ; it is case insensitive
func fromName(s string) (terminalColor, error) {
value, ok := colorMap[s]
if !ok {
return 0, fmt.Errorf("unknown color name '%'", s)
}
return value, nil
}
// fromHex returns the integer value associated with
// an hexadecimal string (full-sized or short version)
// the format is 'abc' or 'abcdef'
func fromHex(s string) (terminalColor, error) {
if len(s) != 3 && len(s) != 6 {
return 0, fmt.Errorf("expect a size of 3 or 6 (remove the '#' prefix)")
}
// short version
input := s
if len(s) == 3 {
input = fmt.Sprintf("%c%c%c%c%c%c", s[0], s[0], s[1], s[1], s[2], s[2])
}
n, err := strconv.ParseUint(input, 16, 32)
if err != nil {
return 0, err
}
return terminalColor(n), nil
}
// parseColor tries to parse a color string (can be a name or an hexa value)
func parseColor(s string) (terminalColor, error) {
// (0) ...
if len(s) < 1 {
return 0, io.ErrUnexpectedEOF
}
// (1) hexa
if s[0] == '#' {
return fromHex(s[1:])
}
// (2) name
return fromName(s)
}
// Red component of the color
func (c terminalColor) Red() uint8 {
return uint8((c >> 16) & 0xff)
}
// Green component of the color
func (c terminalColor) Green() uint8 {
return uint8((c >> 8) & 0xff)
}
// Blue component of the color
func (c terminalColor) Blue() uint8 {
return uint8(c & 0xff)
}

85
printer.go Normal file
View File

@ -0,0 +1,85 @@
package clifmt
import (
"fmt"
"regexp"
)
// extractor helps extract features from the coloring format defined as follows :
//
// - [Color] -> [a-z] # named color
// - [Color] -> #[0-9a-f]{3} # hexa color (shortcode)
// - [Color] -> #[0-9a-f]{6} # hexa color (full-sized)
// - [Text] -> ANY
// - [Format] -> ${Text}(Color:Color) # foreground, background colors
// - [Format] -> ${Text}(Color) # foreground color only
// - [Format] -> ${Text}(:Color) # background color only
var extractor = regexp.MustCompile(`(?m)\${([^$]+)}\(((?:[a-z]+|#(?:[0-9a-f]{3}|[0-9a-f]{6})))?(?:\:((?:[a-z]+|#(?:[0-9a-f]{3}|[0-9a-f]{6}))))?\)`)
func Printf(format string, a ...interface{}) error {
// 1. Pre-process format with 'fmt'
input := fmt.Sprintf(format, a...)
output := ""
cursor := int(0)
// 2. extract color format matches
for _, match := range extractor.FindAllStringSubmatchIndex(input, -1) {
// (1) add gap between input start OR previous match
output += input[cursor:match[0]]
cursor = match[1]
// (2) extract features
var (
err = error(nil)
text = ""
sForeground = ""
sBackground = ""
foreground = terminalColor(0)
background = terminalColor(0)
)
if match[3]-match[2] > 0 {
text = input[match[2]:match[3]]
}
if match[5]-match[4] > 0 {
sForeground = input[match[4]:match[5]]
foreground, err = parseColor(sForeground)
if err != nil {
return err
}
}
if match[7]-match[6] > 0 {
sBackground = input[match[6]:match[7]]
background, err = parseColor(sBackground)
if err != nil {
return err
}
}
// (3) replace text with colorized text
if len(sForeground) > 0 {
text = colorize(text, true, foreground)
}
if len(sBackground) > 0 {
text = colorize(text, false, background)
}
output += text
}
// 3. Add end of input
if cursor < len(input)-1 {
output += input[cursor:]
}
// 3. print final output
fmt.Print(output)
return nil
}
func colorize(text string, foregound bool, color terminalColor) string {
if foregound {
return fmt.Sprintf("\033[38;2;%d;%d;%dm%s\033[0m", color.Red(), color.Green(), color.Blue(), text)
}
return fmt.Sprintf("\033[48;2;%d;%d;%dm%s\033[0m", color.Red(), color.Green(), color.Blue(), text)
}