43 lines
1.2 KiB
Lua
43 lines
1.2 KiB
Lua
-- Smart folding: native treesitter for small files, indent for large files
|
|
local LARGE_FILE_THRESHOLD = 1000
|
|
|
|
local function setup_folding()
|
|
local line_count = vim.api.nvim_buf_line_count(0)
|
|
|
|
if line_count > LARGE_FILE_THRESHOLD then
|
|
-- Large files: indent-based folding (fast)
|
|
vim.opt_local.foldmethod = "indent"
|
|
else
|
|
-- Normal files: native treesitter folding
|
|
vim.opt_local.foldmethod = "expr"
|
|
vim.opt_local.foldexpr = "v:lua.vim.treesitter.foldexpr()"
|
|
end
|
|
end
|
|
|
|
vim.api.nvim_create_autocmd({ "BufReadPost", "FileType" }, {
|
|
group = vim.api.nvim_create_augroup("myconfig-smart-folding", { clear = true }),
|
|
callback = setup_folding,
|
|
})
|
|
|
|
-- Global fold settings
|
|
vim.opt.foldcolumn = "0"
|
|
vim.opt.foldtext = ""
|
|
vim.opt.foldlevel = 99
|
|
vim.opt.foldlevelstart = 99
|
|
vim.opt.foldnestmax = 3
|
|
|
|
-- Toggle all folds with <leader>z
|
|
vim.keymap.set("n", "<leader>z", function()
|
|
local any_fold_closed = false
|
|
for lnum = 1, vim.fn.line("$") do
|
|
if vim.fn.foldclosed(lnum) ~= -1 then
|
|
any_fold_closed = true
|
|
break
|
|
end
|
|
end
|
|
if any_fold_closed then
|
|
vim.cmd("normal! zR")
|
|
else
|
|
vim.cmd("normal! zM")
|
|
end
|
|
end, { noremap = true, silent = true, desc = "Toggle all folds" })
|