mirror of https://github.com/tiyn/tidy.nvim
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
1.8 KiB
81 lines
1.8 KiB
local M = {}
|
|
|
|
M.enabled = true
|
|
|
|
function M.toggle()
|
|
M.enabled = not M.enabled
|
|
|
|
if not M.enabled then
|
|
vim.notify("Tidy disabled on save", vim.log.levels.WARN, { title = "Tidy" })
|
|
else
|
|
vim.notify("Tidy enabled on save", vim.log.levels.INFO, { title = "Tidy" })
|
|
end
|
|
end
|
|
|
|
local function list_to_set(list)
|
|
local set = {}
|
|
|
|
for _, item in ipairs(list) do
|
|
set[item] = true
|
|
end
|
|
|
|
return set
|
|
end
|
|
|
|
local function is_excluded_ft(opts)
|
|
local ft = vim.api.nvim_buf_get_option(0, "filetype")
|
|
local ft_set = list_to_set(opts.filetype_exclude)
|
|
|
|
return ft_set[ft]
|
|
end
|
|
|
|
local function reset_cursor_pos(pos)
|
|
local num_rows = vim.api.nvim_buf_line_count(0)
|
|
|
|
--[[
|
|
if the row value in the original cursor
|
|
position tuple is greater than the
|
|
line count after empty line deletion
|
|
(meaning that the cursor was inside of
|
|
the group of empty lines at the end of
|
|
the file when they were deleted), set
|
|
the cursor row to the last line.
|
|
]]
|
|
if pos[1] > num_rows then
|
|
pos[1] = num_rows
|
|
end
|
|
|
|
vim.api.nvim_win_set_cursor(0, pos)
|
|
end
|
|
|
|
function M.setup(opts)
|
|
local defaults = {
|
|
filetype_exclude = {},
|
|
}
|
|
|
|
opts = vim.tbl_extend("force", defaults, opts or {})
|
|
|
|
local tidy_grp = vim.api.nvim_create_augroup("tidy", { clear = true })
|
|
|
|
vim.api.nvim_create_autocmd("BufWritePre", {
|
|
group = tidy_grp,
|
|
callback = function()
|
|
if not M.enabled or is_excluded_ft(opts) or not vim.tbl_isempty(vim.b.editorconfig) then
|
|
return false
|
|
end
|
|
|
|
local cursor_pos = vim.api.nvim_win_get_cursor(0)
|
|
|
|
-- delete trailing whitespace
|
|
vim.cmd([[:keepjumps keeppatterns %s/\s\+$//e]])
|
|
|
|
-- delete lines @ eof
|
|
vim.cmd([[:keepjumps keeppatterns silent! 0;/^\%(\n*.\)\@!/,$d_]])
|
|
|
|
reset_cursor_pos(cursor_pos)
|
|
end,
|
|
})
|
|
end
|
|
|
|
return M
|