1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
package table
import (
"strings"
"unicode/utf8"
runewidth "github.com/mattn/go-runewidth"
)
type ansiBlob []ansiSegment
func (a ansiBlob) Strip() string {
var output string
for _, segment := range a {
output += segment.value
}
return output
}
func (a ansiBlob) TrimSpace() ansiBlob {
return newANSI(strings.TrimSpace(a.String()))
}
func (a ansiBlob) Len() int {
var c int
for _, segment := range a {
c += runewidth.StringWidth(segment.value)
}
return c
}
func (a ansiBlob) String() string {
var output string
for _, segment := range a {
output += segment.style + segment.value
}
return output
}
func (a ansiBlob) ANSI() string {
var output string
for _, segment := range a {
output += segment.style
}
return simplifyANSI(output)
}
func simplifyANSI(input string) string {
parts := strings.Split(input, "\x1b[0m")
return parts[len(parts)-1]
}
func (a ansiBlob) Cut(index int) (ansiBlob, ansiBlob) {
var current int
var outputBefore string
var outputAfter string
var found bool
for _, segment := range a {
if found {
outputAfter += segment.style + segment.value
continue
}
if index < current+utf8.RuneCountInString(segment.value) {
localIndex := index - current
outputBefore += string([]rune(segment.value)[:localIndex])
outputAfter = segment.style + string([]rune(segment.value)[localIndex:])
found = true
continue
}
outputBefore += segment.style + segment.value
current += utf8.RuneCountInString(segment.value)
}
return newANSI(outputBefore), newANSI(outputAfter)
}
func (a ansiBlob) Words() []ansiBlob {
var output []ansiBlob
words := strings.Split(a.String(), " ")
var ansi string
for _, word := range words {
w := newANSI(word).TrimSpace()
ansi = simplifyANSI(ansi + w.ANSI())
if w.Len() == 0 {
continue
}
output = append(output, w)
}
return output
}
type ansiSegment struct {
value string
style string
}
func newANSI(input string) ansiBlob {
var output []ansiSegment
var current ansiSegment
inCSI := false
prev := rune(0)
for _, r := range input {
if inCSI {
current.style += string(r)
if r >= 0x40 && r <= 0x7E {
inCSI = false
}
} else if r == '[' && prev == 0x1b {
current.value = current.value[:utf8.RuneCountInString(current.value)-1]
if current.value != "" {
output = append(output, current)
current = ansiSegment{}
}
inCSI = true
current.style += "\x1b["
} else {
current.value = current.value + string(r)
}
prev = r
}
if current.value != "" || current.style != "" {
output = append(output, current)
}
return output
}
|
