moar/m/ansiTokenizer.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 string into a series of tokens
func TokensFromString(logger *log.Logger, s string) []Token {
	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)
			}
		}
	}

	return tokens
}

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),
				}
			}

			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)

		// 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)

		default:
			logger.Printf("Unrecognized ANSI SGI code <%s>", number)
		}
	}

	return style
}