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
|
--
-- Custom highlighting plugin (in `after/` to that the autocommand in `treesitter/init.lua` can be set first in order)
--
-- Returns an iterator over the ascendants of `node`, starting at the root node
local function node_ascendants(node)
local root = node:tree():root()
local current = root
return function()
if current:equal(node) then
return nil
end
current = assert(current:child_with_descendant(node))
return current
end
end
-- Returns true if `node` or any of its parent has the type `type`
local function node_within_type(node, type)
local next_ascendant = node_ascendants(node)
local next = next_ascendant()
while next do
if next:type() == type then
return true
end
next = next_ascendant()
end
return false
end
-- Returns an iterator over the positions (`{ col1, col2, row }`) of matches of pattern in lines
local function pmatches(lines, pattern)
local col1, col2, row = nil, nil, 1
return function()
while row < #lines do
col1, col2 = string.find(lines[row], pattern, (col2 or 0) + 1)
if col1 then
return { col1 = col1, col2 = col2, row = row }
else
col1, col2, row = 0, 0, row + 1
end
end
return nil
end
end
-- Set extmarks are the positions (`{ col1, col2, row }`) given by `next_pos()`
-- `opts`: {
-- ns: highlight namespace
-- predicate: is passed row and col, if true then set the mark
-- hl_group: highlight group
-- }
local function set_extmarks(next_pos, opts)
for pos in next_pos do
local marks = vim.api.nvim_buf_get_extmarks(
0,
opts.ns,
{ pos.row - 1, pos.col1 - 1 },
{ pos.row - 1, pos.col2 - 1 },
{}
)
if #marks == 0 and opts.predicate(pos.row - 1, pos.col1 - 1) then
vim.api.nvim_buf_set_extmark(
0,
opts.ns,
pos.row - 1,
pos.col1 - 1,
{ end_col = pos.col2, hl_group = opts.hl_group }
)
end
pos = next_pos()
end
end
-- TODO a better way to do this:
-- For code, only check in "comment" node ranges
-- For non-code, do it per type. For Markdown, only check "paragraph" nodes
local function hl_todo_predicate(row, col)
if vim.o.filetype == "markdown" or vim.treesitter.get_parser() == nil then
return true
end
local node = assert(vim.treesitter.get_node({ pos = { row, col } }))
return node_within_type(node, "comment")
end
local patterns = { todo = "TODO" }
local hl_ns = vim.api.nvim_create_namespace("extmarks")
vim.api.nvim_set_hl(hl_ns, "dotfiles.Todo", { bg = "Yellow", fg = "Black", bold = true })
local extmarks_todo_opts = { ns = hl_ns, hl_group = "dotfiles.Todo", predicate = hl_todo_predicate }
-- Initialize extmarks
vim.api.nvim_create_autocmd("FileType", {
desc = "Initialize extmarks",
group = vim.g.dotfiles.augroup,
callback = function()
vim.api.nvim_win_set_hl_ns(vim.api.nvim_get_current_win(), hl_ns)
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
set_extmarks(pmatches(lines, patterns.todo), extmarks_todo_opts)
end,
})
|
