Use Treesitter for folding (#523)

Author: WhiteBlackGoose

ftplugin/org.vim

@@ -1,10 +1,6 @@
 lua require('orgmode.config'):setup_mappings('org')
 lua require('orgmode.config'):setup_mappings('text_objects')
 
-function! OrgmodeFoldExpr()
-  return luaeval('require("orgmode.org.indent").foldexpr()')
-endfunction
-
 function! OrgmodeFoldText()
   return luaeval('require("orgmode.org.indent").foldtext()')
 endfunction
@@ -20,7 +16,7 @@ endfunction
 setlocal nomodeline
 setlocal fillchars+=fold:\ 
 setlocal foldmethod=expr
-setlocal foldexpr=OrgmodeFoldExpr()
+setlocal foldexpr=nvim_treesitter#foldexpr()
 setlocal foldtext=OrgmodeFoldText()
 setlocal formatexpr=OrgmodeFormatExpr()
 setlocal foldlevel=0

lua/orgmode/utils/treesitter.lua

@@ -39,8 +39,10 @@ function M.get_node_at_cursor(cursor, winnr, ignore_injected_langs)
   winnr = winnr or 0
   local buf = vim.api.nvim_win_get_buf(winnr)
   -- TODO: Use only this function when 0.8 is released
-  if vim.treesitter.get_node_at_pos then
-    return vim.treesitter.get_node_at_pos(buf, cursor[1] - 1, cursor[2], {
+  if vim.treesitter.get_node then
+    return vim.treesitter.get_node({
+      bufrn = buf,
+      pos = { cursor[1] - 1, cursor[2] },
       ignore_injections = ignore_injected_langs,
     })
   end

queries/org/folds.scm

@@ -0,0 +1,8 @@
+([
+  (section)
+  (table)
+  (drawer)
+  (property_drawer)
+  (block)
+  ] @fold
+  (#trim! @fold))