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.
61 lines
1.5 KiB
61 lines
1.5 KiB
local M = {}
|
|
|
|
local function is_excluded_ft(opts)
|
|
if not opts or not opts.filetype_exclude then
|
|
return false
|
|
end
|
|
|
|
local ft = vim.api.nvim_buf_get_option(0, "filetype")
|
|
|
|
for _, entry in ipairs(opts.filetype_exclude) do
|
|
if entry == ft then
|
|
return true
|
|
end
|
|
end
|
|
|
|
return false
|
|
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 tidy_grp = vim.api.nvim_create_augroup("tidy", { clear = true })
|
|
|
|
vim.api.nvim_create_autocmd("BufWritePre", {
|
|
group = tidy_grp,
|
|
callback = function()
|
|
if is_excluded_ft(opts) then
|
|
return false
|
|
end
|
|
|
|
local cursor_pos = vim.api.nvim_win_get_cursor(0)
|
|
|
|
vim.cmd([[:keepjumps keeppatterns %s/\s\+$//e]])
|
|
|
|
-- delete all lines at end of buffer, see source 2
|
|
vim.cmd([[:keepjumps keeppatterns silent! 0;/^\%(\n*.\)\@!/,$d]])
|
|
|
|
reset_cursor_pos(cursor_pos)
|
|
end,
|
|
})
|
|
end
|
|
|
|
return M
|