Fix everything and have fun

This commit is contained in:
2025-02-10 18:22:44 +03:00
parent c8902137d0
commit 623951310e
2 changed files with 202 additions and 23 deletions

View File

@@ -2,31 +2,140 @@ 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
vim.api.nvim_buf_add_highlight(0, -1, "TodoDoneHighlight", i - 1, 0, 3)
vim.api.nvim_buf_add_highlight(0, -1, "TodoDone", i - 1, 3, -1)
elseif line:match("^● ") then
vim.api.nvim_buf_add_highlight(0, -1, "TodoPendingHighlight", i - 1, 0, 3)
vim.api.nvim_buf_add_highlight(0, -1, "TodoPending", i - 1, 3, -1)
elseif line:match("^⚠ ") then
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportantHighlight", i - 1, 0, 3)
vim.api.nvim_buf_add_highlight(0, -1, "TodoImportant", i - 1, 3, -1)
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
vim.api.nvim_buf_add_highlight(0, -1, "Normal", i - 1, 0, -1)
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
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, 1, { "" })
elseif line:match("^- ") then
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, 1, { "" })
elseif line:match("^! ") then
vim.api.nvim_buf_set_text(0, i - 1, 0, i - 1, 1, { "" })
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
@@ -35,19 +144,45 @@ 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,
@@ -55,24 +190,62 @@ function M.setup()
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()
if line:match("^✔ ") then
vim.api.nvim_set_current_line("" .. line:sub(4))
elseif line:match("^● ") then
vim.api.nvim_set_current_line("" .. line:sub(4))
elseif line:match("^⚠ ") then
vim.api.nvim_set_current_line("" .. line:sub(4))
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
vim.api.nvim_set_keymap("n", "<leader>t", ":lua require('udivim').toggle_todo()<CR>", { noremap = true, silent = true })
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