mirror of
https://github.com/walles/moar.git
synced 2024-09-22 01:18:48 +03:00
235 lines
5.0 KiB
Go
235 lines
5.0 KiB
Go
package m
|
||
|
||
import (
|
||
"log"
|
||
"regexp"
|
||
"strings"
|
||
|
||
"github.com/gdamore/tcell"
|
||
)
|
||
|
||
const _TabSize = 4
|
||
|
||
// Token is a rune with a style to be written to a cell on screen
|
||
type Token struct {
|
||
Rune rune
|
||
Style tcell.Style
|
||
}
|
||
|
||
// TokensFromString turns a (formatted) string into a series of tokens,
|
||
// and an unformatted string
|
||
func TokensFromString(logger *log.Logger, s string) ([]Token, *string) {
|
||
var tokens []Token
|
||
|
||
styleBrokenUtf8 := tcell.StyleDefault.Background(7).Foreground(1)
|
||
|
||
for _, styledString := range _StyledStringsFromString(logger, s) {
|
||
for _, token := range _TokensFromStyledString(styledString) {
|
||
switch token.Rune {
|
||
|
||
case '\x09': // TAB
|
||
for {
|
||
tokens = append(tokens, Token{
|
||
Rune: ' ',
|
||
Style: styledString.Style,
|
||
})
|
||
|
||
if (len(tokens))%_TabSize == 0 {
|
||
// We arrived at the next tab stop
|
||
break
|
||
}
|
||
}
|
||
|
||
case '<27>': // Go's broken-UTF8 marker
|
||
tokens = append(tokens, Token{
|
||
Rune: '?',
|
||
Style: styleBrokenUtf8,
|
||
})
|
||
|
||
case '\x08': // Backspace
|
||
tokens = append(tokens, Token{
|
||
Rune: '<',
|
||
Style: styleBrokenUtf8,
|
||
})
|
||
|
||
default:
|
||
tokens = append(tokens, token)
|
||
}
|
||
}
|
||
}
|
||
|
||
plainString := ""
|
||
for _, token := range tokens {
|
||
plainString += string(token.Rune)
|
||
}
|
||
return tokens, &plainString
|
||
}
|
||
|
||
func _TokensFromStyledString(styledString _StyledString) []Token {
|
||
tokens := make([]Token, 0, len(styledString.String)+1)
|
||
oneBack := '\x00'
|
||
twoBack := '\x00'
|
||
for _, char := range []rune(styledString.String) {
|
||
if oneBack == '\x08' && twoBack != '\x00' {
|
||
// Something-Backspace-Something
|
||
|
||
replacement := (*Token)(nil)
|
||
|
||
if char == twoBack {
|
||
replacement = &Token{
|
||
Rune: twoBack,
|
||
Style: styledString.Style.Bold(true),
|
||
}
|
||
}
|
||
|
||
if twoBack == '_' {
|
||
replacement = &Token{
|
||
Rune: char,
|
||
Style: styledString.Style.Underline(true),
|
||
}
|
||
}
|
||
|
||
// FIXME: Man page formatting fails, if I do (in bash)...
|
||
// "man printf|hexdump -C|grep -10 leading| grep --color 08"
|
||
// ... I get...
|
||
// "000003e0 20 20 20 20 20 20 20 20 2b 08 2b 08 6f 08 6f 20 | +.+.o.o |"
|
||
// ... wich "less" renders as a bold "o". We should as well.
|
||
//
|
||
// I don't get the logic though, the sequence is:
|
||
// plus-backspace-plus-backspace-o-backspace-o
|
||
//
|
||
// Maybe the interpretation should be:
|
||
// "Make a bold +, then erase that and replace it with a bold o"?
|
||
|
||
if replacement != nil {
|
||
tokens = append(tokens[0:len(tokens)-2], *replacement)
|
||
|
||
twoBack = oneBack
|
||
oneBack = char
|
||
continue
|
||
}
|
||
|
||
// No match, just keep going
|
||
}
|
||
|
||
tokens = append(tokens, Token{
|
||
Rune: char,
|
||
Style: styledString.Style,
|
||
})
|
||
|
||
twoBack = oneBack
|
||
oneBack = char
|
||
}
|
||
|
||
return tokens
|
||
}
|
||
|
||
type _StyledString struct {
|
||
String string
|
||
Style tcell.Style
|
||
}
|
||
|
||
func _StyledStringsFromString(logger *log.Logger, s string) []_StyledString {
|
||
// This function was inspired by the
|
||
// https://golang.org/pkg/regexp/#Regexp.Split source code
|
||
|
||
pattern := regexp.MustCompile("\x1b\\[([0-9;]*m)")
|
||
|
||
matches := pattern.FindAllStringIndex(s, -1)
|
||
styledStrings := make([]_StyledString, 0, len(matches)+1)
|
||
|
||
style := tcell.StyleDefault
|
||
|
||
beg := 0
|
||
end := 0
|
||
for _, match := range matches {
|
||
end = match[0]
|
||
|
||
if end > beg {
|
||
// Found non-zero length string
|
||
styledStrings = append(styledStrings, _StyledString{
|
||
String: s[beg:end],
|
||
Style: style,
|
||
})
|
||
}
|
||
|
||
matchedPart := s[match[0]:match[1]]
|
||
style = _UpdateStyle(logger, style, matchedPart)
|
||
|
||
beg = match[1]
|
||
}
|
||
|
||
if end != len(s) {
|
||
styledStrings = append(styledStrings, _StyledString{
|
||
String: s[beg:],
|
||
Style: style,
|
||
})
|
||
}
|
||
|
||
return styledStrings
|
||
}
|
||
|
||
// _UpdateStyle parses a string of the form "ESC[33m" into changes to style
|
||
func _UpdateStyle(logger *log.Logger, style tcell.Style, escapeSequence string) tcell.Style {
|
||
for _, number := range strings.Split(escapeSequence[2:len(escapeSequence)-1], ";") {
|
||
switch number {
|
||
case "", "0", "00":
|
||
style = tcell.StyleDefault
|
||
|
||
case "1":
|
||
style = style.Bold(true)
|
||
|
||
case "7":
|
||
style = style.Reverse(true)
|
||
|
||
case "27":
|
||
style = style.Reverse(false)
|
||
|
||
// Foreground colors
|
||
case "30":
|
||
style = style.Foreground(0)
|
||
case "31":
|
||
style = style.Foreground(1)
|
||
case "32":
|
||
style = style.Foreground(2)
|
||
case "33":
|
||
style = style.Foreground(3)
|
||
case "34":
|
||
style = style.Foreground(4)
|
||
case "35":
|
||
style = style.Foreground(5)
|
||
case "36":
|
||
style = style.Foreground(6)
|
||
case "37":
|
||
style = style.Foreground(7)
|
||
case "39":
|
||
style = style.Foreground(tcell.ColorDefault)
|
||
|
||
// Background colors
|
||
case "40":
|
||
style = style.Background(0)
|
||
case "41":
|
||
style = style.Background(1)
|
||
case "42":
|
||
style = style.Background(2)
|
||
case "43":
|
||
style = style.Background(3)
|
||
case "44":
|
||
style = style.Background(4)
|
||
case "45":
|
||
style = style.Background(5)
|
||
case "46":
|
||
style = style.Background(6)
|
||
case "47":
|
||
style = style.Background(7)
|
||
case "49":
|
||
style = style.Background(tcell.ColorDefault)
|
||
|
||
default:
|
||
logger.Printf("Unrecognized ANSI SGI code <%s>", number)
|
||
}
|
||
}
|
||
|
||
return style
|
||
}
|