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
|
-- mod-version:2 -- lite-xl 2.0
local style = require "core.style"
local config = require "core.config"
local DocView = require "core.docview"
-- TODO: replace with `doc:get_indent_info()` when 2.1 releases
local function get_indent_info(doc)
if doc.get_indent_info then
return doc:get_indent_info()
end
return config.tab_type, config.indent_size
end
local function get_line_spaces(doc, idx, dir)
local _, indent_size = get_indent_info(doc)
local text = doc.lines[idx]
if not text or #text == 1 then
return -1
end
local s, e = text:find("^%s*")
if e == #text then
return get_line_spaces(doc, idx + dir, dir)
end
local n = 0
for _,b in pairs({text:byte(s, e)}) do
n = n + (b == 9 and indent_size or 1)
end
return n
end
local function get_line_indent_guide_spaces(doc, idx)
if doc.lines[idx]:find("^%s*\n") then
return math.max(
get_line_spaces(doc, idx - 1, -1),
get_line_spaces(doc, idx + 1, 1))
end
return get_line_spaces(doc, idx)
end
local docview_update = DocView.update
function DocView:update()
docview_update(self)
local function get_indent(idx)
if idx < 1 or idx > #self.doc.lines then return -1 end
if not self.lineguide_indents[idx] then
self.lineguide_indents[idx] = get_line_indent_guide_spaces(self.doc, idx)
end
return self.lineguide_indents[idx]
end
self.lineguide_indents = {}
self.lineguide_indent_active = {}
local minline, maxline = self:get_visible_line_range()
for i = minline, maxline do
self.lineguide_indents[i] = get_line_indent_guide_spaces(self.doc, i)
end
local line = self.doc:get_selection()
local lvl = get_indent(line)
local top, bottom
local _, indent_size = get_indent_info(self.doc)
-- check if we're the header or the footer of a block
if get_indent(line + 1) > lvl and get_indent(line + 1) <= lvl + indent_size then
top = true
lvl = get_indent(line + 1)
elseif get_indent(line - 1) > lvl and get_indent(line - 1) <= lvl + indent_size then
bottom = true
lvl = get_indent(line - 1)
end
self.lineguide_indent_active[line] = lvl
-- check if the lines before the current are part of the block
local i = line - 1
if i > 0 and not top then
repeat
if get_indent(i) <= lvl - indent_size then break end
self.lineguide_indent_active[i] = lvl
i = i - 1
until i < minline
end
-- check if the lines after the current are part of the block
i = line + 1
if i <= #self.doc.lines and not bottom then
repeat
if get_indent(i) <= lvl - indent_size then break end
self.lineguide_indent_active[i] = lvl
i = i + 1
until i > maxline
end
end
local draw_line_text = DocView.draw_line_text
function DocView:draw_line_text(idx, x, y)
local spaces = self.lineguide_indents[idx] or -1
local _, indent_size = get_indent_info(self.doc)
local w = math.max(1, SCALE)
local h = self:get_line_height()
local font = self:get_font()
local space_sz = font:get_width(" ")
for i = 0, spaces - 1, indent_size do
local color = style.guide or style.selection
local active_lvl = self.lineguide_indent_active[idx] or -1
if i < active_lvl and i + indent_size >= active_lvl then
color = style.accent
end
local sw = space_sz * i
renderer.draw_rect(math.ceil(x + sw), y, w, h, color)
end
draw_line_text(self, idx, x, y)
end
|
