fix(nvim): rewrite fold plugin and fix text folding while editing

1 files changed, 11 insertions, 12 deletions

diff --git a/.config/nvim/plugin/50-fold.lua b/.config/nvim/plugin/50-fold.lua
index 27cb5e5..f887614 100644
--- a/.config/nvim/plugin/50-fold.lua
+++ b/.config/nvim/plugin/50-fold.lua
@@ -1,27 +1,26 @@
 --
--- fold plugin
+-- 50-fold.lua
 --
-
--- Initialize folds with foldmethod = expr if a treesitter parser is available, then switch back to manual
--- This is to have the folds calculated and closed when editing the file initially, but with foldmethod set to manual
--- Without foldmethod set to manual folds are created while writing in insert mode which is annoying
+-- * Creates an autocommand to fold with treesitter automatically and close all folds on 'FileType'.
+-- * Sets 'foldlevel' to 99 so that new text will not fold while editing.
 --
--- Skip floating windows, because they often are documentation (and thus should be visible by default)
-local function init_folds()
+
+local function fold_with_treesitter()
 	if vim.api.nvim_win_get_config(0).relative ~= "" or vim.treesitter.get_parser(0) == nil then
 		return
 	end
 	vim.opt_local.foldexpr = "v:lua.vim.treesitter.foldexpr()"
 	vim.opt_local.foldmethod = "expr"
-	vim.cmd.normal({ "zX", bang = true })
-	vim.opt_local.foldmethod = "manual"
+	vim.cmd("silent! %foldclose!")
 end
 
--- Initial fold setup
+----------------------------------------------------------------------------------------------------
+
 vim.api.nvim_create_autocmd("FileType", {
 	desc = "Initialize folds",
 	group = vim.g.dotfiles.augroup,
-	callback = init_folds,
+	callback = fold_with_treesitter,
 })
 
-vim.api.nvim_create_user_command("FoldInit", init_folds, { desc = "(Re)initialize folds" })
+vim.opt.foldenable = true
+vim.opt.foldlevel = 99