-
-
Notifications
You must be signed in to change notification settings - Fork 15
/
cells.lua
152 lines (126 loc) · 3.83 KB
/
cells.lua
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
local M = {}
--- Get the line type (cell separator, magic commands, empty, others)
---@param line string | number 1-indexed
---@return string "cell separator: markdown" | "cell separator: code" | "magic commands" | "empty" | "others"
function M.line_type(line)
if type(line) == "number" then
line = vim.api.nvim_buf_get_lines(0, line - 1, line, false)[1]
end
if line == nil then
return "empty"
elseif vim.startswith(line, "# %% [md]") or vim.startswith(line, "# %% [markdown]") then
return "cell separator: markdown"
elseif vim.fn.trim(line) == "# %%" then
return "cell separator: code"
elseif vim.startswith(line, "# ---") then
return "metadata"
elseif vim.startswith(line, "# %") then
return "magic command"
elseif vim.fn.trim(line) == "" then
return "empty"
end
return "others" -- code
end
local line_types_changedtick_per_buf = {}
local line_types_per_buf = {}
--- Get the line types of the entire buffer
--- Similar to `line_type` but returns a table of line types, and it returns more types
--- It also caches the results so repeated calls are fast
--- i.e. cell content: code, cell content: markdown, cell content: header
--- Useful for highlighting
---@return table
function M.line_types_entire_buf(bufnr)
bufnr = bufnr or vim.api.nvim_get_current_buf()
local changedtick = vim.api.nvim_buf_get_changedtick(bufnr)
if line_types_changedtick_per_buf[bufnr] == changedtick then
return line_types_per_buf[bufnr]
end
local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
local current_cell_type = "header"
local line_types = {}
for i, line in ipairs(lines) do
local line_type = M.line_type(line)
if line_type == "others" or line_type == "empty" then
line_types[i] = "cell content: " .. current_cell_type
elseif vim.startswith(line_type, "cell separator: markdown") then
current_cell_type = "markdown"
line_types[i] = line_type
elseif vim.startswith(line_type, "cell separator: code") then
current_cell_type = "code"
line_types[i] = line_type
else
line_types[i] = line_type
end
end
line_types_changedtick_per_buf[bufnr] = changedtick
line_types_per_buf[bufnr] = line_types
return line_types
end
--- Check if the line is a cell separator
---@param line string | number 1-indexed
---@return boolean
function M.is_line_separator(line)
local line_type = M.line_type(line)
if vim.startswith(line_type, "cell separator:") then
return true
end
return false
end
--- Get the current cell separator row
---@param row number | nil 1-indexed
---@return number | nil row 1-indexed
function M.current_cell_separator(row)
row = row or vim.api.nvim_win_get_cursor(0)[1]
if M.is_line_separator(row) then
return row
end
row = row - 1
while row > 0 do
if M.is_line_separator(row) then
return row
end
row = row - 1
end
return nil
end
--- Get the previous cell separator row
---@param row number | nil 1-indexed
---@return number | nil row 1-indexed
function M.previous_cell_separator(row)
row = row or vim.api.nvim_win_get_cursor(0)[1]
local found_separator
if M.is_line_separator(row) then
found_separator = row
end
row = row - 1
while row > 0 do
if M.is_line_separator(row) then
if found_separator ~= nil then
return row
else
found_separator = row
end
end
row = row - 1
end
if found_separator ~= nil then
return found_separator
end
return nil
end
--- Get the next cell separator row
---@param row number | nil 1-indexed
---@return number | nil row 1-indexed
function M.next_cell_separator(row)
row = row or vim.api.nvim_win_get_cursor(0)[1]
row = row + 1
local num_lines = vim.api.nvim_buf_line_count(0)
while row <= num_lines do
if M.is_line_separator(row) then
return row
end
row = row + 1
end
return nil
end
return M