blob: 79bdb541b7c3ccb7c07cbddfd7c7e980467c9b6c (
plain) (
blame)
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
|
import streams
import unicode
import buffer/cell
import data/charset
import encoding/decoderstream
import utils/twtstr
const tabwidth = 8
proc renderPlainText*(text: string): FlexibleGrid =
var format = newFormat()
template add_format() =
if af:
af = false
result[result.high].addFormat(result[^1].str.len, format)
result.addLine()
var spaces = 0
var i = 0
var af = false
while i < text.len:
case text[i]
of '\n':
add_format
result.addLine()
of '\r': discard
of '\t':
add_format
for i in 0 ..< tabwidth:
result[^1].str &= ' '
spaces = 0
of ' ':
add_format
result[^1].str &= ' '
inc spaces
if spaces == 8:
spaces = 0
of '\e':
i = format.parseAnsiCode(text, i)
af = true
elif text[i] in Controls:
add_format
result[^1].str &= '^' & text[i].getControlLetter()
else:
add_format
result[^1].str &= text[i]
inc i
if result.len > 1 and result[^1].str.len == 0 and result[^1].formats.len == 0:
discard result.pop()
type StreamRenderer* = object
spaces: int
ansiparser: AnsiCodeParser
format: Format
af: bool
decoder: DecoderStream
proc newStreamRenderer*(stream: Stream): StreamRenderer =
result.format = newFormat()
result.ansiparser.state = PARSE_DONE
result.decoder = newDecoderStream(stream, CHARSET_UTF_8)
proc renderStream*(grid: var FlexibleGrid, renderer: var StreamRenderer, len: int) =
if len == 0: return
template add_format() =
if renderer.af:
renderer.af = false
grid[grid.high].addFormat(grid[^1].str.len, renderer.format)
if grid.len == 0: grid.addLine()
var i = 0
var buf = newSeq[Rune](len * 4)
let n = renderer.decoder.readData(addr buf[0], buf.len * sizeof(buf[0]))
while i < n div sizeof(buf[0]):
let r = buf[i]
if r.isAscii():
let c = cast[char](r)
if renderer.ansiparser.state != PARSE_DONE:
let cancel = renderer.ansiparser.parseAnsiCode(renderer.format, c)
if not cancel:
if renderer.ansiparser.state == PARSE_DONE:
renderer.af = true
inc i
continue
case c
of '\n':
add_format
grid.addLine()
of '\r': discard
of '\t':
add_format
for i in 0 ..< tabwidth - renderer.spaces:
grid[^1].str &= ' '
renderer.spaces = 0
of ' ':
add_format
grid[^1].str &= c
inc renderer.spaces
if renderer.spaces == 8:
renderer.spaces = 0
of '\e':
renderer.ansiparser.reset()
elif c in Controls:
add_format
grid[^1].str &= '^' & c.getControlLetter()
else:
add_format
grid[^1].str &= c
else:
add_format
grid[^1].str &= r
inc i
|