clifmt/printer.go

73 lines
2.8 KiB
Go

package clifmt
import (
"fmt"
"git.xdrm.io/go/clifmt/internal/color"
tbold "git.xdrm.io/go/clifmt/internal/syntax/bold"
tcolor "git.xdrm.io/go/clifmt/internal/syntax/color"
thyperlink "git.xdrm.io/go/clifmt/internal/syntax/hyperlink"
titalic "git.xdrm.io/go/clifmt/internal/syntax/italic"
tunderline "git.xdrm.io/go/clifmt/internal/syntax/underline"
"git.xdrm.io/go/clifmt/internal/transform"
"strings"
)
var ErrInvalidFormat = fmt.Errorf("invalid format")
var theme = color.DefaultTheme()
var (
dollarToken = `e4d097183ab04e49f25cb7b0956fb9eb25b90c0316a32cb5afcbcdd9a6692e8d2974919035789d5632b10d799db5b3e5bf8539592c904497f5c356f117ef37382`
asteriskToken = `253c3cd0a904d28abc3e601e3557d59ea69da2616079ceef4987d58d55c9820c83026be92a917ee19a298e613ea0b393cc70d4e55dc614a9afc6a020d8f08f37`
underscoreToken = `2b08e24b7833e90c74ed8e6c27b7b3cd5fe949e0f18b28af813d5f2df863d55f97b0ed7f8fbb26a152eda55ac073331ce11ac10702caca5b3ea4a29f722840b9`
squareBracketToken = `51b06edd58f36003844941916cd3b313979fece55824d89ba02af052a229b2673aafffa541b703472c1a21d8e6a1bb3e844d236fb0e8bf5d62902b24042f4fb5`
)
// Sprintf returns a terminal-colorized output following the coloring format
func Sprintf(format string, a ...interface{}) (string, error) {
// 1. Pre-process format with 'fmt'
formatted := fmt.Sprintf(format, a...)
if strings.Contains(formatted, "%!") { // error
return "", ErrInvalidFormat
}
// 2. Protect escaped characters with tokens
formatted = strings.Replace(formatted, "\\$", dollarToken, -1)
formatted = strings.Replace(formatted, "\\*", asteriskToken, -1)
formatted = strings.Replace(formatted, "\\_", underscoreToken, -1)
formatted = strings.Replace(formatted, "\\[", squareBracketToken, -1)
// 3. create transformation registry
reg := transform.Registry{Transformers: make([]transform.Transformer, 0, 10)}
reg.Transformers = append(reg.Transformers, tcolor.Export)
reg.Transformers = append(reg.Transformers, tbold.Export)
reg.Transformers = append(reg.Transformers, titalic.Export)
reg.Transformers = append(reg.Transformers, tunderline.Export)
reg.Transformers = append(reg.Transformers, thyperlink.Export)
transformed, err := reg.Transform(formatted)
if err != nil {
return "", err
}
// 5. Restore token-protected characters
transformed = strings.Replace(transformed, dollarToken, "$", -1)
transformed = strings.Replace(transformed, asteriskToken, "*", -1)
transformed = strings.Replace(transformed, underscoreToken, "_", -1)
transformed = strings.Replace(transformed, squareBracketToken, "[", -1)
// 6. return final output
return transformed, nil
}
// Printf prints a terminal-colorized output following the coloring format
func Printf(format string, a ...interface{}) error {
s, err := Sprintf(format, a...)
if err != nil {
return err
}
fmt.Print(s)
return nil
}