fix: Try to fix treesitter on python fold

1 files changed, 14 insertions, 4 deletions

diff --git a/lua/ache/plugins/treesitter.lua b/lua/ache/plugins/treesitter.lua
index 550ad54..139ed50 100644
--- a/lua/ache/plugins/treesitter.lua
+++ b/lua/ache/plugins/treesitter.lua
@@ -10,7 +10,8 @@
 return {
 	"nvim-treesitter/nvim-treesitter",
 	enabled = true,
-	event = { "BufReadPre", "BufNewFile" },
+	lazy = false,
+	-- event = { "BufReadPre", "BufNewFile" },
 	build = ":TSUpdate",
 	-- dependencies = {
 	-- 	"windwp/nvim-ts-autotag",
@@ -110,8 +111,17 @@ return {
 		--   -- filetype = "zu", -- if filetype does not match the parser name
 		-- }
 
-		vim.wo.foldmethod = "expr"
-		vim.wo.foldexpr = "v:lua.vim.treesitter.foldexpr()"
-		vim.wo.foldlevel = 9
+		-- vim.wo.foldmethod = "expr"
+		-- vim.wo.foldexpr = "v:lua.vim.treesitter.foldexpr()"
+		-- vim.wo.foldlevel = 9
+
+		vim.wo[0][0].foldexpr = "v:lua.vim.treesitter.foldexpr()"
+		vim.wo[0][0].foldmethod = "expr"
+		vim.wo[0][0].foldlevel = 9
+		vim.bo.indentexpr = "v:lua.require'nvim-treesitter'.indentexpr()"
+
+		vim.keymap.set("n", "<leader>z", function()
+			vim.cmd([[set foldexpr=nvim_treesitter#foldexpr()]])
+		end, { desc = "Refresh folding from Treesitter" })
 	end,
 }