diff options
Diffstat (limited to 'vendor/maunium.net/go/tview/util.go')
| -rw-r--r-- | vendor/maunium.net/go/tview/util.go | 605 |
1 files changed, 0 insertions, 605 deletions
diff --git a/vendor/maunium.net/go/tview/util.go b/vendor/maunium.net/go/tview/util.go deleted file mode 100644 index e408b18..0000000 --- a/vendor/maunium.net/go/tview/util.go +++ /dev/null @@ -1,605 +0,0 @@ -package tview - -import ( - "math" - "regexp" - "strconv" - "unicode" - - "maunium.net/go/tcell" - runewidth "github.com/mattn/go-runewidth" -) - -// Text alignment within a box. -const ( - AlignLeft = iota - AlignCenter - AlignRight -) - -// Common regular expressions. -var ( - colorPattern = regexp.MustCompile(`\[([a-zA-Z]+|#[0-9a-zA-Z]{6}|\-)?(:([a-zA-Z]+|#[0-9a-zA-Z]{6}|\-)?(:([lbdru]+|\-)?)?)?\]`) - regionPattern = regexp.MustCompile(`\["([a-zA-Z0-9_,;: \-\.]*)"\]`) - escapePattern = regexp.MustCompile(`\[([a-zA-Z0-9_,;: \-\."#]+)\[(\[*)\]`) - nonEscapePattern = regexp.MustCompile(`(\[[a-zA-Z0-9_,;: \-\."#]+\[*)\]`) - boundaryPattern = regexp.MustCompile(`(([[:punct:]]|\n)[ \t\f\r]*|(\s+))`) - spacePattern = regexp.MustCompile(`\s+`) -) - -// Positions of substrings in regular expressions. -const ( - colorForegroundPos = 1 - colorBackgroundPos = 3 - colorFlagPos = 5 -) - -// Predefined InputField acceptance functions. -var ( - // InputFieldInteger accepts integers. - InputFieldInteger func(text string, ch rune) bool - - // InputFieldFloat accepts floating-point numbers. - InputFieldFloat func(text string, ch rune) bool - - // InputFieldMaxLength returns an input field accept handler which accepts - // input strings up to a given length. Use it like this: - // - // inputField.SetAcceptanceFunc(InputFieldMaxLength(10)) // Accept up to 10 characters. - InputFieldMaxLength func(maxLength int) func(text string, ch rune) bool -) - -// Package initialization. -func init() { - // Initialize the predefined input field handlers. - InputFieldInteger = func(text string, ch rune) bool { - if text == "-" { - return true - } - _, err := strconv.Atoi(text) - return err == nil - } - InputFieldFloat = func(text string, ch rune) bool { - if text == "-" || text == "." || text == "-." { - return true - } - _, err := strconv.ParseFloat(text, 64) - return err == nil - } - InputFieldMaxLength = func(maxLength int) func(text string, ch rune) bool { - return func(text string, ch rune) bool { - return len([]rune(text)) <= maxLength - } - } -} - -// styleFromTag takes the given style, defined by a foreground color (fgColor), -// a background color (bgColor), and style attributes, and modifies it based on -// the substrings (tagSubstrings) extracted by the regular expression for color -// tags. The new colors and attributes are returned where empty strings mean -// "don't modify" and a dash ("-") means "reset to default". -func styleFromTag(fgColor, bgColor, attributes string, tagSubstrings []string) (newFgColor, newBgColor, newAttributes string) { - if tagSubstrings[colorForegroundPos] != "" { - color := tagSubstrings[colorForegroundPos] - if color == "-" { - fgColor = "-" - } else if color != "" { - fgColor = color - } - } - - if tagSubstrings[colorBackgroundPos-1] != "" { - color := tagSubstrings[colorBackgroundPos] - if color == "-" { - bgColor = "-" - } else if color != "" { - bgColor = color - } - } - - if tagSubstrings[colorFlagPos-1] != "" { - flags := tagSubstrings[colorFlagPos] - if flags == "-" { - attributes = "-" - } else if flags != "" { - attributes = flags - } - } - - return fgColor, bgColor, attributes -} - -// overlayStyle mixes a background color with a foreground color (fgColor), -// a (possibly new) background color (bgColor), and style attributes, and -// returns the resulting style. For a definition of the colors and attributes, -// see styleFromTag(). Reset instructions cause the corresponding part of the -// default style to be used. -func overlayStyle(background tcell.Color, defaultStyle tcell.Style, fgColor, bgColor, attributes string) tcell.Style { - defFg, defBg, defAttr := defaultStyle.Decompose() - style := defaultStyle.Background(background) - - style = style.Foreground(defFg) - if fgColor != "" { - style = style.Foreground(tcell.GetColor(fgColor)) - } - - if bgColor == "-" || bgColor == "" && defBg != tcell.ColorDefault { - style = style.Background(defBg) - } else if bgColor != "" { - style = style.Background(tcell.GetColor(bgColor)) - } - - if attributes == "-" { - style = style.Bold(defAttr&tcell.AttrBold > 0) - style = style.Blink(defAttr&tcell.AttrBlink > 0) - style = style.Reverse(defAttr&tcell.AttrReverse > 0) - style = style.Underline(defAttr&tcell.AttrUnderline > 0) - style = style.Dim(defAttr&tcell.AttrDim > 0) - } else if attributes != "" { - style = style.Normal() - for _, flag := range attributes { - switch flag { - case 'l': - style = style.Blink(true) - case 'b': - style = style.Bold(true) - case 'd': - style = style.Dim(true) - case 'r': - style = style.Reverse(true) - case 'u': - style = style.Underline(true) - } - } - } - - return style -} - -// decomposeString returns information about a string which may contain color -// tags. It returns the indices of the color tags (as returned by -// re.FindAllStringIndex()), the color tags themselves (as returned by -// re.FindAllStringSubmatch()), the indices of an escaped tags, the string -// stripped by any color tags and escaped, and the screen width of the stripped -// string. -func decomposeString(text string) (colorIndices [][]int, colors [][]string, escapeIndices [][]int, stripped string, width int) { - // Get positions of color and escape tags. - colorIndices = colorPattern.FindAllStringIndex(text, -1) - colors = colorPattern.FindAllStringSubmatch(text, -1) - escapeIndices = escapePattern.FindAllStringIndex(text, -1) - - // Because the color pattern detects empty tags, we need to filter them out. - for i := len(colorIndices) - 1; i >= 0; i-- { - if colorIndices[i][1]-colorIndices[i][0] == 2 { - colorIndices = append(colorIndices[:i], colorIndices[i+1:]...) - colors = append(colors[:i], colors[i+1:]...) - } - } - - // Remove the color tags from the original string. - var from int - buf := make([]byte, 0, len(text)) - for _, indices := range colorIndices { - buf = append(buf, []byte(text[from:indices[0]])...) - from = indices[1] - } - buf = append(buf, text[from:]...) - - // Escape string. - stripped = string(escapePattern.ReplaceAll(buf, []byte("[$1$2]"))) - - // Get the width of the stripped string. - width = runewidth.StringWidth(stripped) - - return -} - -// Print prints text onto the screen into the given box at (x,y,maxWidth,1), -// not exceeding that box. "align" is one of AlignLeft, AlignCenter, or -// AlignRight. The screen's background color will not be changed. -// -// You can change the colors and text styles mid-text by inserting a color tag. -// See the package description for details. -// -// Returns the number of actual bytes of the text printed (including color tags) -// and the actual width used for the printed runes. -func Print(screen tcell.Screen, text string, x, y, maxWidth, align int, color tcell.Color) (int, int) { - return printWithStyle(screen, text, x, y, maxWidth, align, tcell.StyleDefault.Foreground(color)) -} - -// printWithStyle works like Print() but it takes a style instead of just a -// foreground color. -func printWithStyle(screen tcell.Screen, text string, x, y, maxWidth, align int, style tcell.Style) (int, int) { - if maxWidth <= 0 || len(text) == 0 { - return 0, 0 - } - - // Decompose the text. - colorIndices, colors, escapeIndices, strippedText, strippedWidth := decomposeString(text) - - // We want to reduce all alignments to AlignLeft. - if align == AlignRight { - if strippedWidth <= maxWidth { - // There's enough space for the entire text. - return printWithStyle(screen, text, x+maxWidth-strippedWidth, y, maxWidth, AlignLeft, style) - } - // Trim characters off the beginning. - var ( - bytes, width, colorPos, escapePos, tagOffset int - foregroundColor, backgroundColor, attributes string - ) - _, originalBackground, _ := style.Decompose() - iterateString(strippedText, func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool { - // Update color/escape tag offset and style. - if colorPos < len(colorIndices) && textPos+tagOffset >= colorIndices[colorPos][0] && textPos+tagOffset < colorIndices[colorPos][1] { - foregroundColor, backgroundColor, attributes = styleFromTag(foregroundColor, backgroundColor, attributes, colors[colorPos]) - style = overlayStyle(originalBackground, style, foregroundColor, backgroundColor, attributes) - tagOffset += colorIndices[colorPos][1] - colorIndices[colorPos][0] - colorPos++ - } - if escapePos < len(escapeIndices) && textPos+tagOffset >= escapeIndices[escapePos][0] && textPos+tagOffset < escapeIndices[escapePos][1] { - tagOffset++ - escapePos++ - } - if strippedWidth-screenPos < maxWidth { - // We chopped off enough. - if escapePos > 0 && textPos+tagOffset-1 >= escapeIndices[escapePos-1][0] && textPos+tagOffset-1 < escapeIndices[escapePos-1][1] { - // Unescape open escape sequences. - escapeCharPos := escapeIndices[escapePos-1][1] - 2 - text = text[:escapeCharPos] + text[escapeCharPos+1:] - } - // Print and return. - bytes, width = printWithStyle(screen, text[textPos+tagOffset:], x, y, maxWidth, AlignLeft, style) - return true - } - return false - }) - return bytes, width - } else if align == AlignCenter { - if strippedWidth == maxWidth { - // Use the exact space. - return printWithStyle(screen, text, x, y, maxWidth, AlignLeft, style) - } else if strippedWidth < maxWidth { - // We have more space than we need. - half := (maxWidth - strippedWidth) / 2 - return printWithStyle(screen, text, x+half, y, maxWidth-half, AlignLeft, style) - } else { - // Chop off runes until we have a perfect fit. - var choppedLeft, choppedRight, leftIndex, rightIndex int - rightIndex = len(strippedText) - for rightIndex-1 > leftIndex && strippedWidth-choppedLeft-choppedRight > maxWidth { - if choppedLeft < choppedRight { - // Iterate on the left by one character. - iterateString(strippedText[leftIndex:], func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool { - choppedLeft += screenWidth - leftIndex += textWidth - return true - }) - } else { - // Iterate on the right by one character. - iterateStringReverse(strippedText[leftIndex:rightIndex], func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool { - choppedRight += screenWidth - rightIndex -= textWidth - return true - }) - } - } - - // Add tag offsets and determine start style. - var ( - colorPos, escapePos, tagOffset int - foregroundColor, backgroundColor, attributes string - ) - _, originalBackground, _ := style.Decompose() - for index := range strippedText { - // We only need the offset of the left index. - if index > leftIndex { - // We're done. - if escapePos > 0 && leftIndex+tagOffset-1 >= escapeIndices[escapePos-1][0] && leftIndex+tagOffset-1 < escapeIndices[escapePos-1][1] { - // Unescape open escape sequences. - escapeCharPos := escapeIndices[escapePos-1][1] - 2 - text = text[:escapeCharPos] + text[escapeCharPos+1:] - } - break - } - - // Update color/escape tag offset. - if colorPos < len(colorIndices) && index+tagOffset >= colorIndices[colorPos][0] && index+tagOffset < colorIndices[colorPos][1] { - if index <= leftIndex { - foregroundColor, backgroundColor, attributes = styleFromTag(foregroundColor, backgroundColor, attributes, colors[colorPos]) - style = overlayStyle(originalBackground, style, foregroundColor, backgroundColor, attributes) - } - tagOffset += colorIndices[colorPos][1] - colorIndices[colorPos][0] - colorPos++ - } - if escapePos < len(escapeIndices) && index+tagOffset >= escapeIndices[escapePos][0] && index+tagOffset < escapeIndices[escapePos][1] { - tagOffset++ - escapePos++ - } - } - return printWithStyle(screen, text[leftIndex+tagOffset:], x, y, maxWidth, AlignLeft, style) - } - } - - // Draw text. - var ( - drawn, drawnWidth, colorPos, escapePos, tagOffset int - foregroundColor, backgroundColor, attributes string - ) - iterateString(strippedText, func(main rune, comb []rune, textPos, length, screenPos, screenWidth int) bool { - // Only continue if there is still space. - if drawnWidth+screenWidth > maxWidth { - return true - } - - // Handle color tags. - if colorPos < len(colorIndices) && textPos+tagOffset >= colorIndices[colorPos][0] && textPos+tagOffset < colorIndices[colorPos][1] { - foregroundColor, backgroundColor, attributes = styleFromTag(foregroundColor, backgroundColor, attributes, colors[colorPos]) - tagOffset += colorIndices[colorPos][1] - colorIndices[colorPos][0] - colorPos++ - } - - // Handle scape tags. - if escapePos < len(escapeIndices) && textPos+tagOffset >= escapeIndices[escapePos][0] && textPos+tagOffset < escapeIndices[escapePos][1] { - if textPos+tagOffset == escapeIndices[escapePos][1]-2 { - tagOffset++ - escapePos++ - } - } - - // Print the rune sequence. - finalX := x + drawnWidth - _, _, finalStyle, _ := screen.GetContent(finalX, y) - _, background, _ := finalStyle.Decompose() - finalStyle = overlayStyle(background, style, foregroundColor, backgroundColor, attributes) - for offset := screenWidth - 1; offset >= 0; offset-- { - // To avoid undesired effects, we populate all cells. - if offset == 0 { - screen.SetContent(finalX+offset, y, main, comb, finalStyle) - } else { - screen.SetContent(finalX+offset, y, ' ', nil, finalStyle) - } - } - - // Advance. - drawn += length - drawnWidth += screenWidth - - return false - }) - - return drawn + tagOffset + len(escapeIndices), drawnWidth -} - -// PrintSimple prints white text to the screen at the given position. -func PrintSimple(screen tcell.Screen, text string, x, y int) { - Print(screen, text, x, y, math.MaxInt32, AlignLeft, Styles.PrimaryTextColor) -} - -// StringWidth returns the width of the given string needed to print it on -// screen. The text may contain color tags which are not counted. -func StringWidth(text string) int { - _, _, _, _, width := decomposeString(text) - return width -} - -// WordWrap splits a text such that each resulting line does not exceed the -// given screen width. Possible split points are after any punctuation or -// whitespace. Whitespace after split points will be dropped. -// -// This function considers color tags to have no width. -// -// Text is always split at newline characters ('\n'). -func WordWrap(text string, width int) (lines []string) { - colorTagIndices, _, escapeIndices, strippedText, _ := decomposeString(text) - - // Find candidate breakpoints. - breakpoints := boundaryPattern.FindAllStringSubmatchIndex(strippedText, -1) - // Results in one entry for each candidate. Each entry is an array a of - // indices into strippedText where a[6] < 0 for newline/punctuation matches - // and a[4] < 0 for whitespace matches. - - // Process stripped text one character at a time. - var ( - colorPos, escapePos, breakpointPos, tagOffset int - lastBreakpoint, lastContinuation, currentLineStart int - lineWidth, continuationWidth int - newlineBreakpoint bool - ) - unescape := func(substr string, startIndex int) string { - // A helper function to unescape escaped tags. - for index := escapePos; index >= 0; index-- { - if index < len(escapeIndices) && startIndex > escapeIndices[index][0] && startIndex < escapeIndices[index][1]-1 { - pos := escapeIndices[index][1] - 2 - startIndex - return substr[:pos] + substr[pos+1:] - } - } - return substr - } - iterateString(strippedText, func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool { - // Handle colour tags. - if colorPos < len(colorTagIndices) && textPos+tagOffset >= colorTagIndices[colorPos][0] && textPos+tagOffset < colorTagIndices[colorPos][1] { - tagOffset += colorTagIndices[colorPos][1] - colorTagIndices[colorPos][0] - colorPos++ - } - - // Handle escape tags. - if escapePos < len(escapeIndices) && textPos+tagOffset == escapeIndices[escapePos][1]-2 { - tagOffset++ - escapePos++ - } - - // Check if a break is warranted. - afterContinuation := lastContinuation > 0 && textPos+tagOffset >= lastContinuation - noBreakpoint := lastContinuation == 0 - beyondWidth := lineWidth > 0 && lineWidth > width - if beyondWidth && noBreakpoint { - // We need a hard break without a breakpoint. - lines = append(lines, unescape(text[currentLineStart:textPos+tagOffset], currentLineStart)) - currentLineStart = textPos + tagOffset - lineWidth = continuationWidth - } else if afterContinuation && (beyondWidth || newlineBreakpoint) { - // Break at last breakpoint or at newline. - lines = append(lines, unescape(text[currentLineStart:lastBreakpoint], currentLineStart)) - currentLineStart = lastContinuation - lineWidth = continuationWidth - lastBreakpoint, lastContinuation, newlineBreakpoint = 0, 0, false - } - - // Is this a breakpoint? - if breakpointPos < len(breakpoints) && textPos == breakpoints[breakpointPos][0] { - // Yes, it is. Set up breakpoint infos depending on its type. - lastBreakpoint = breakpoints[breakpointPos][0] + tagOffset - lastContinuation = breakpoints[breakpointPos][1] + tagOffset - newlineBreakpoint = main == '\n' - if breakpoints[breakpointPos][6] < 0 && !newlineBreakpoint { - lastBreakpoint++ // Don't skip punctuation. - } - breakpointPos++ - } - - // Once we hit the continuation point, we start buffering widths. - if textPos+tagOffset < lastContinuation { - continuationWidth = 0 - } - - lineWidth += screenWidth - continuationWidth += screenWidth - return false - }) - - // Flush the rest. - if currentLineStart < len(text) { - lines = append(lines, unescape(text[currentLineStart:], currentLineStart)) - } - - return -} - -// Escape escapes the given text such that color and/or region tags are not -// recognized and substituted by the print functions of this package. For -// example, to include a tag-like string in a box title or in a TextView: -// -// box.SetTitle(tview.Escape("[squarebrackets]")) -// fmt.Fprint(textView, tview.Escape(`["quoted"]`)) -func Escape(text string) string { - return nonEscapePattern.ReplaceAllString(text, "$1[]") -} - -// iterateString iterates through the given string one printed character at a -// time. For each such character, the callback function is called with the -// Unicode code points of the character (the first rune and any combining runes -// which may be nil if there aren't any), the starting position (in bytes) -// within the original string, its length in bytes, the screen position of the -// character, and the screen width of it. The iteration stops if the callback -// returns true. This function returns true if the iteration was stopped before -// the last character. -func iterateString(text string, callback func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool) bool { - var ( - runes []rune - lastZeroWidthJoiner bool - startIndex int - startPos int - pos int - ) - - // Helper function which invokes the callback. - flush := func(index int) bool { - var comb []rune - if len(runes) > 1 { - comb = runes[1:] - } - return callback(runes[0], comb, startIndex, index-startIndex, startPos, pos-startPos) - } - - for index, r := range text { - if unicode.In(r, unicode.Lm, unicode.M) || r == '\u200d' { - lastZeroWidthJoiner = r == '\u200d' - } else { - // We have a rune that's not a modifier. It could be the beginning of a - // new character. - if !lastZeroWidthJoiner { - if len(runes) > 0 { - // It is. Invoke callback. - if flush(index) { - return true // We're done. - } - // Reset rune store. - runes = runes[:0] - startIndex = index - startPos = pos - } - pos += runewidth.RuneWidth(r) - } else { - lastZeroWidthJoiner = false - } - } - runes = append(runes, r) - } - - // Flush any remaining runes. - if len(runes) > 0 { - flush(len(text)) - } - - return false -} - -// iterateStringReverse iterates through the given string in reverse, starting -// from the end of the string, one printed character at a time. For each such -// character, the callback function is called with the Unicode code points of -// the character (the first rune and any combining runes which may be nil if -// there aren't any), the starting position (in bytes) within the original -// string, its length in bytes, the screen position of the character, and the -// screen width of it. The iteration stops if the callback returns true. This -// function returns true if the iteration was stopped before the last character. -func iterateStringReverse(text string, callback func(main rune, comb []rune, textPos, textWidth, screenPos, screenWidth int) bool) bool { - type runePos struct { - r rune - pos int // The byte position of the rune in the original string. - width int // The screen width of the rune. - mod bool // Modifier or zero-width-joiner. - } - - // We use the following: - // len(text) >= number of runes in text. - - // Put all runes into a runePos slice in reverse. - runesReverse := make([]runePos, len(text)) - index := len(text) - 1 - for pos, ch := range text { - runesReverse[index].r = ch - runesReverse[index].pos = pos - runesReverse[index].width = runewidth.RuneWidth(ch) - runesReverse[index].mod = unicode.In(ch, unicode.Lm, unicode.M) || ch == '\u200d' - index-- - } - runesReverse = runesReverse[index+1:] - - // Parse reverse runes. - var screenWidth int - buffer := make([]rune, len(text)) // We fill this up from the back so it's forward again. - bufferPos := len(text) - stringWidth := runewidth.StringWidth(text) - for index, r := range runesReverse { - // Put this rune into the buffer. - bufferPos-- - buffer[bufferPos] = r.r - - // Do we need to flush the buffer? - if r.pos == 0 || !r.mod && runesReverse[index+1].r != '\u200d' { - // Yes, invoke callback. - var comb []rune - if len(text)-bufferPos > 1 { - comb = buffer[bufferPos+1:] - } - if callback(r.r, comb, r.pos, len(text)-r.pos, stringWidth-screenWidth, r.width) { - return true - } - screenWidth += r.width - bufferPos = len(text) - } - } - - return false -} |