New module tty.go
This commit is contained in:
parent
db726467c6
commit
a78c42a161
306
tty.go
Normal file
306
tty.go
Normal file
@ -0,0 +1,306 @@
|
||||
// tty.go
|
||||
package text
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type ctxFmt struct {
|
||||
isTTY bool
|
||||
ansiLen int
|
||||
}
|
||||
|
||||
func translateColor(name string) (c int) {
|
||||
//fmt.Println("TranslateColor:", name)
|
||||
if len(name) == 0 {
|
||||
return -1
|
||||
}
|
||||
|
||||
norm := strings.ToLower(name)
|
||||
switch {
|
||||
case strings.HasPrefix("black", norm):
|
||||
c = 30
|
||||
case strings.HasPrefix("blue", norm):
|
||||
c = 34
|
||||
case strings.HasPrefix("green", norm):
|
||||
c = 32
|
||||
case strings.HasPrefix("red", norm):
|
||||
c = 31
|
||||
case strings.HasPrefix("white", norm):
|
||||
c = 37
|
||||
default:
|
||||
c = -1
|
||||
}
|
||||
if c > 0 && name[0] >= 'A' && name[0] <= 'Z' {
|
||||
c += 10
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func extractArgs(spec string) (result string) {
|
||||
if len(spec) < 2 || spec[0] != '(' || spec[len(spec)-1] != ')' {
|
||||
return
|
||||
}
|
||||
|
||||
result = spec[1 : len(spec)-1]
|
||||
return
|
||||
}
|
||||
|
||||
func parseColor(fg bool, spec string) (result string) {
|
||||
var intro string
|
||||
if spec = extractArgs(spec); len(spec) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
if spec[0] >= '0' && spec[0] <= '9' {
|
||||
if fg {
|
||||
intro = "38;"
|
||||
} else {
|
||||
intro = "48;"
|
||||
}
|
||||
parts := strings.SplitN(spec, ",", 3)
|
||||
if len(parts) == 3 {
|
||||
result = intro + "2;" + strings.Join(parts, ";")
|
||||
} else if len(parts) == 1 {
|
||||
result = intro + "5;" + parts[0]
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
if c := translateColor(spec); c >= 0 {
|
||||
result = strconv.Itoa(c)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func advanceToTerms(s string, start int, terms string) (offset int) {
|
||||
for offset = start; offset < len(s) && strings.IndexByte(terms, s[offset]) < 0; offset++ {
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (ctx *ctxFmt) putAnsiString(sb *strings.Builder, code string) {
|
||||
if ctx.isTTY {
|
||||
a, _ := sb.WriteString("\x1b[")
|
||||
b, _ := sb.WriteString(code)
|
||||
sb.WriteByte('m')
|
||||
ctx.ansiLen += a + b + 1
|
||||
}
|
||||
}
|
||||
|
||||
func (ctx *ctxFmt) putAnsiCode(sb *strings.Builder, code int) {
|
||||
if ctx.isTTY {
|
||||
a, _ := sb.WriteString("\x1b[")
|
||||
b, _ := sb.WriteString(strconv.Itoa(code))
|
||||
sb.WriteByte('m')
|
||||
ctx.ansiLen += a + b + 1
|
||||
}
|
||||
}
|
||||
|
||||
func (ctx *ctxFmt) putAnsiReset(sb *strings.Builder) {
|
||||
if ctx.isTTY {
|
||||
a, _ := sb.WriteString("\x1b[0m")
|
||||
ctx.ansiLen += a
|
||||
}
|
||||
}
|
||||
|
||||
type caseMode uint8
|
||||
|
||||
const (
|
||||
normal = caseMode(iota)
|
||||
upper
|
||||
lower
|
||||
title
|
||||
firstOnly
|
||||
)
|
||||
|
||||
type alignMode uint8
|
||||
|
||||
const (
|
||||
noAlign = alignMode(iota)
|
||||
alignLeft
|
||||
alignRight
|
||||
)
|
||||
|
||||
func (ctx *ctxFmt) putText(sb *strings.Builder, text string, start int, mode caseMode, align alignMode, fieldSize int, filler byte) (offset int) {
|
||||
if t := text[start+1:]; len(t) > 0 {
|
||||
if align == alignRight {
|
||||
for k := 0; k < fieldSize-len(t); k++ {
|
||||
sb.WriteByte(filler)
|
||||
}
|
||||
}
|
||||
if mode == upper {
|
||||
sb.WriteString(strings.ToUpper(t))
|
||||
} else if mode == lower {
|
||||
sb.WriteString(strings.ToLower(t))
|
||||
} else if mode == title {
|
||||
sb.WriteString(strings.Title(t))
|
||||
} else if mode == firstOnly {
|
||||
if t[0] >= 'a' && t[0] <= 'z' {
|
||||
sb.WriteByte(t[0] - 'a' + 'A')
|
||||
} else {
|
||||
sb.WriteByte(t[0])
|
||||
}
|
||||
sb.WriteString(strings.ToLower(t[1:]))
|
||||
} else {
|
||||
sb.WriteString(t)
|
||||
}
|
||||
if align == alignLeft {
|
||||
for k := 0; k < fieldSize-len(t); k++ {
|
||||
sb.WriteByte(filler)
|
||||
}
|
||||
}
|
||||
offset = start + len(t)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (ctx *ctxFmt) Handle(varSpec string, flags ScannerFlag) (value string, err error) {
|
||||
var sb strings.Builder
|
||||
var mode caseMode = normal
|
||||
var align alignMode = noAlign
|
||||
var fieldSize int
|
||||
var filler byte = ' '
|
||||
|
||||
//fmt.Println("TTY:", varSpec)
|
||||
if len(varSpec) == 0 {
|
||||
ctx.putAnsiReset(&sb)
|
||||
} else {
|
||||
for i := 0; i < len(varSpec); i++ {
|
||||
if j := advanceToTerms(varSpec, i, ";-"); j > i+1 {
|
||||
if c := translateColor(varSpec[i:j]); c >= 0 {
|
||||
ctx.putAnsiCode(&sb, c)
|
||||
i = j
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
ch := varSpec[i]
|
||||
switch ch {
|
||||
case '.':
|
||||
ctx.putAnsiReset(&sb)
|
||||
case 'b':
|
||||
ctx.putAnsiString(&sb, "1")
|
||||
case 'B':
|
||||
ctx.putAnsiString(&sb, "22")
|
||||
case 'i':
|
||||
ctx.putAnsiString(&sb, "3")
|
||||
case 'I':
|
||||
ctx.putAnsiString(&sb, "23")
|
||||
case 'u':
|
||||
ctx.putAnsiString(&sb, "4")
|
||||
case 'U':
|
||||
ctx.putAnsiString(&sb, "24")
|
||||
case 'c', 'C':
|
||||
j := advanceToTerms(varSpec, i, ")")
|
||||
if c := parseColor(ch == 'c', varSpec[i+1:j+1]); len(c) > 0 {
|
||||
ctx.putAnsiString(&sb, c)
|
||||
//sb.WriteString("\x1b[" + c + "m")
|
||||
i = j
|
||||
}
|
||||
case '\'':
|
||||
mode = upper
|
||||
case ',':
|
||||
mode = lower
|
||||
case '"':
|
||||
mode = title
|
||||
case '^':
|
||||
mode = firstOnly
|
||||
case '<', '>':
|
||||
j := advanceToTerms(varSpec, i, ")")
|
||||
if arg := extractArgs(varSpec[i+1 : j+1]); len(arg) > 0 {
|
||||
var err1 error
|
||||
if arg[0] <= '0' || arg[0] > '9' {
|
||||
filler = arg[0]
|
||||
arg = arg[1:]
|
||||
}
|
||||
if fieldSize, err1 = strconv.Atoi(arg); err1 == nil {
|
||||
if ch == '<' {
|
||||
align = alignLeft
|
||||
} else {
|
||||
align = alignRight
|
||||
}
|
||||
}
|
||||
}
|
||||
i = j
|
||||
case '-':
|
||||
i = ctx.putText(&sb, varSpec, i, mode, align, fieldSize, filler)
|
||||
ctx.putAnsiReset(&sb)
|
||||
}
|
||||
}
|
||||
}
|
||||
return sb.String(), nil
|
||||
}
|
||||
|
||||
func IsTty(w io.Writer) (isTty bool) {
|
||||
if fh, ok := w.(*os.File); ok {
|
||||
if fi, err := fh.Stat(); err == nil {
|
||||
isTty = (fi.Mode() & os.ModeCharDevice) != 0
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func ToTty(source string) (result string, cleanLen int, err error) {
|
||||
ctx := ctxFmt{true, 0}
|
||||
if result, err = ScanExt('#', &ctx, source); err == nil {
|
||||
cleanLen = len(result) - ctx.ansiLen
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func ToTtyStream(w io.Writer, source string) (result string, cleanLen int, err error) {
|
||||
if IsTty(w) {
|
||||
return ToTty(source)
|
||||
} else {
|
||||
return source, len(source), nil
|
||||
}
|
||||
}
|
||||
|
||||
func Sprintf(templ string, args ...any) string {
|
||||
templ, _, _ = ToTty(templ)
|
||||
return fmt.Sprintf(templ, args...)
|
||||
}
|
||||
|
||||
func Printf(templ string, args ...any) (int, error) {
|
||||
return Fprintf(os.Stdout, templ, args...)
|
||||
}
|
||||
|
||||
func Fprintf(w io.Writer, templ string, args ...any) (int, error) {
|
||||
templ, _, _ = ToTtyStream(w, templ)
|
||||
return fmt.Fprintf(w, templ, args...)
|
||||
|
||||
}
|
||||
|
||||
// func DebugSprintf() {
|
||||
// list := []string{
|
||||
// "#{<(.20)-Param}: #{>(06)-123}",
|
||||
// "ciao #{c(r)i<(12)-GIAN Carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(r)i>(12)-GIAN Carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(red);i}Mario#. #{u;b}Come# #{GREEN;b}Stai#{.}?",
|
||||
// "ciao #{c(red)i-Mario}. #{ub-Come} #{GREEN;b-Stai}?",
|
||||
// "ciao #{c(r)i-Mario}. #{ub-Come} #{Gr;b-Stai}?",
|
||||
// "ciao #{c(r)i-Mario}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(r)i'-gian carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(r)i\"-gian carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(r)i,-GIAN Carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// "ciao #{c(r)i^-GIAN Carlo}. #{ub-Come} #{c(G)b-Stai}?",
|
||||
// }
|
||||
|
||||
// fmt.Println("--- Sprintf() ---")
|
||||
// for i, s := range list {
|
||||
// x := Sprintf(s)
|
||||
// fmt.Printf("--- Test nr %d: %q\n", i+1, s)
|
||||
// fmt.Println(x)
|
||||
// }
|
||||
// fmt.Println("\n--- Sprintf() ---")
|
||||
// for i, s := range list {
|
||||
// fmt.Printf("--- Test nr %d: %q\n", i+1, s)
|
||||
// Printf(s)
|
||||
// fmt.Println()
|
||||
// }
|
||||
// }
|
||||
Loading…
Reference in New Issue
Block a user