252 lines
9.1 KiB
Lua
252 lines
9.1 KiB
Lua
local M = {}
|
|
|
|
local function highlight_todo_items()
|
|
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
|
vim.api.nvim_buf_clear_namespace(0, -1, 0, -1)
|
|
for i, line in ipairs(lines) do
|
|
if line:match("^ @") then
|
|
if line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDoneTopic", i - 1, 8, idx - 2)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDoneTopic", i - 1, idx + 3, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDoneHighlight", i - 1, idx - 1, idx + 3)
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoPendingTopic", i - 1, 8, idx - 2)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoPendingTopic", i - 1, idx + 3, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoPendingHighlight", i - 1, idx - 1, idx + 3)
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportantTopic", i - 1, 8, idx - 2)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportantTopic", i - 1, idx + 3, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportantHighlight", i - 1, idx - 1, idx + 3)
|
|
else
|
|
vim.api.nvim_buf_add_highlight(0, -1, "Identifier", i - 1, 0, -1)
|
|
end
|
|
elseif line:match("^ #") then
|
|
vim.api.nvim_buf_add_highlight(0, -1, "Topic", i - 1, 4, -1)
|
|
elseif line:match("^*") then
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TopicHead", i - 1, 0, -1)
|
|
else
|
|
if line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDone", i - 1, 0, idx - 1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDone", i - 1, idx + 2, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDoneHighlight", i - 1, idx - 1, idx + 2)
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDone", i - 1, 0, idx - 1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoPending", i - 1, idx + 2, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoPendingHighlight", i - 1, idx - 1, idx + 2)
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoDone", i - 1, 0, idx - 1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportant", i - 1, idx + 2, -1)
|
|
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportantHighlight", i - 1, idx - 1, idx + 2)
|
|
else
|
|
vim.api.nvim_buf_add_highlight(0, -1, "Identifier", i - 1, 0, -1)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function right(val)
|
|
local current_pos = vim.api.nvim_win_get_cursor(0)
|
|
local new_pos
|
|
if val == nil then
|
|
new_pos = { current_pos[1], current_pos[2] + 1 }
|
|
elseif val == -1 then
|
|
local line = vim.api.nvim_buf_get_lines(0, current_pos[1] - 1, current_pos[1], false)[1]
|
|
local end_of_line = #line
|
|
new_pos = { current_pos[1], end_of_line }
|
|
else
|
|
new_pos = { current_pos[1], val }
|
|
end
|
|
vim.api.nvim_win_set_cursor(0, new_pos)
|
|
end
|
|
|
|
local function replace_task_symbols()
|
|
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
|
for i, line in ipairs(lines) do
|
|
if line:match("%[x%]") then
|
|
local idx = string.find(line, "%[x%]")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
right()
|
|
elseif line:match("%[%-%]") then
|
|
local idx = string.find(line, "%[%-%]")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
right()
|
|
elseif line:match("%[!%]") then
|
|
local idx = string.find(line, "%[!%]")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
right()
|
|
end
|
|
end
|
|
end
|
|
|
|
local function handle_inp()
|
|
local lines = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
|
for i, line in ipairs(lines) do
|
|
if line:match("^%s*#") then
|
|
if not line:match("^ # ") then
|
|
local idx = string.find(line, "#")
|
|
if not line:match("^%s*# ") then
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { " # " })
|
|
else
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { " #" })
|
|
end
|
|
right(-1)
|
|
end
|
|
elseif line:match("^%s*%*") then
|
|
if not line:match("^* ") then
|
|
local idx = string.find(line, "*")
|
|
if not line:match("^%s*%* ") then
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { "* " })
|
|
else
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { "*" })
|
|
end
|
|
right(-1)
|
|
end
|
|
elseif line:match("^%s*%@") then
|
|
if not line:match("^ @") then
|
|
local idx = string.find(line, "@")
|
|
if not line:match("^ @ ") then
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { " @ " })
|
|
else
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { " @" })
|
|
end
|
|
right(-1)
|
|
end
|
|
if not line:match("[]") then
|
|
local current_pos = vim.api.nvim_win_get_cursor(0)
|
|
local line2 = vim.api.nvim_buf_get_lines(0, current_pos[1] - 1, current_pos[1], false)[1]
|
|
local end_of_line = #line2
|
|
if string.sub(line2, -2) == " " then
|
|
vim.api.nvim_buf_set_text(0, i - 1, end_of_line, i - 1, end_of_line, { "" })
|
|
elseif string.sub(line2, -2) == " " then
|
|
vim.api.nvim_buf_set_text(0, i - 1, end_of_line, i - 1, end_of_line, { " " })
|
|
else
|
|
vim.api.nvim_buf_set_text(0, i - 1, end_of_line, i - 1, end_of_line, { " " })
|
|
end
|
|
right(end_of_line)
|
|
end
|
|
elseif line:match("^%s*;") then
|
|
if not line:match("^ ;[]") then
|
|
local idx = string.find(line, ";")
|
|
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, idx + 0, { " ; " })
|
|
right(-1)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.setup()
|
|
vim.api.nvim_set_hl(0, "TodoDoneHighlight", { fg = "#00FF00", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoPendingHighlight", { fg = "#FFFF00", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoImportantHighlight", { fg = "#FF0000", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoDoneTopic", { fg = "#000000", bg = "#00FF00", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoPendingTopic", { fg = "#000000", bg = "#FFFF00", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoImportantTopic", { fg = "#000000", bg = "#FF0000", bold = true })
|
|
vim.api.nvim_set_hl(0, "TodoDone", { link = "Comment" })
|
|
vim.api.nvim_set_hl(0, "TodoPending", { link = "Identifier" })
|
|
vim.api.nvim_set_hl(0, "TodoImportant", { link = "Error" })
|
|
vim.api.nvim_set_hl(0, "Topic", { fg = "#000000", bg = "#FFFFFF", bold = true })
|
|
vim.api.nvim_set_hl(0, "TopicHead", { fg = "#000000", bg = "#FFFFFF", bold = true })
|
|
vim.api.nvim_create_autocmd({ "BufRead", "BufNewFile" }, {
|
|
pattern = "*.udi",
|
|
callback = function()
|
|
vim.bo.filetype = "udi"
|
|
vim.opt_local.conceallevel = 2
|
|
local buf = vim.api.nvim_get_current_buf()
|
|
vim.api.nvim_buf_set_keymap(
|
|
buf,
|
|
"n",
|
|
"<leader>t",
|
|
":lua require('udivim').toggle_todo()<CR>",
|
|
{ noremap = true, silent = true }
|
|
)
|
|
vim.api.nvim_buf_set_keymap(
|
|
buf,
|
|
"n",
|
|
"",
|
|
":lua require('udivim').toggle_imp()<CR>",
|
|
{ noremap = true, silent = true }
|
|
)
|
|
end,
|
|
})
|
|
vim.api.nvim_create_autocmd("FileType", {
|
|
pattern = "udi",
|
|
callback = function()
|
|
vim.bo.tabstop = 4
|
|
vim.bo.shiftwidth = 4
|
|
vim.bo.expandtab = true
|
|
vim.opt_local.foldmethod = "expr"
|
|
vim.opt_local.foldexpr = "v:lua.require('udivim').fold()"
|
|
vim.opt_local.foldlevel = 99
|
|
replace_task_symbols()
|
|
highlight_todo_items()
|
|
end,
|
|
})
|
|
vim.api.nvim_create_autocmd("TextChangedI", {
|
|
pattern = "*.udi",
|
|
callback = function()
|
|
local current_content = vim.api.nvim_buf_get_lines(0, 0, -1, false)
|
|
local prev_content = vim.b.prev_content or {}
|
|
local current_text = table.concat(current_content, "\n")
|
|
local prev_text = table.concat(prev_content, "\n")
|
|
if #current_text > #prev_text then
|
|
handle_inp()
|
|
end
|
|
replace_task_symbols()
|
|
highlight_todo_items()
|
|
vim.b.prev_content = current_content
|
|
end,
|
|
})
|
|
end
|
|
|
|
function M.fold()
|
|
local lnum = vim.v.lnum
|
|
local line = vim.fn.getline(vim.v.lnum)
|
|
local indent_level = vim.fn.indent(vim.v.lnum)
|
|
if line:match("^%s*$") then
|
|
indent_level = vim.fn.indent(lnum - 1)
|
|
end
|
|
return math.floor(indent_level / 4)
|
|
end
|
|
|
|
function M.toggle_todo()
|
|
local line = vim.api.nvim_get_current_line()
|
|
local pos = vim.api.nvim_win_get_cursor(0)
|
|
local i = pos[1]
|
|
if line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
end
|
|
highlight_todo_items()
|
|
end
|
|
|
|
function M.toggle_imp()
|
|
local line = vim.api.nvim_get_current_line()
|
|
local pos = vim.api.nvim_win_get_cursor(0)
|
|
local i = pos[1]
|
|
if line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
elseif line:match("") then
|
|
local idx = string.find(line, "")
|
|
vim.api.nvim_buf_set_text(0, i - 1, idx - 1, i - 1, idx + 2, { "" })
|
|
end
|
|
highlight_todo_items()
|
|
end
|
|
|
|
return M
|