From fbade728fed18a084bc5601a62628d3dec0c631c Mon Sep 17 00:00:00 2001
From: Thomas Vigouroux <tomvig38@gmail.com>
Date: Mon, 25 May 2020 10:26:29 +0200
Subject: feat: add syntax-based folding

---
 lua/nvim-treesitter/fold.lua | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 lua/nvim-treesitter/fold.lua

(limited to 'lua')

diff --git a/lua/nvim-treesitter/fold.lua b/lua/nvim-treesitter/fold.lua
new file mode 100644
index 000000000..6a39461a7
--- /dev/null
+++ b/lua/nvim-treesitter/fold.lua
@@ -0,0 +1,29 @@
+local api = vim.api
+local parsers = require'nvim-treesitter.parsers'
+
+local M = {}
+
+function M.get_fold_indic(lnum)
+  if not parsers.has_parser() or not lnum then return '0' end
+
+  local function smallest_multiline_containing(node, level)
+    for index = 0,(node:named_child_count() -1) do
+      local child = node:named_child(index)
+      local start, _, stop, _ = child:range()
+
+      if start ~= stop and start <= (lnum -1) and stop >= (lnum -1) then
+        return smallest_multiline_containing(child, level + 1)
+      end
+    end
+
+    return node, level
+  end
+
+  local parser = parsers.get_parser()
+
+  local multiline_here, level = smallest_multiline_containing(parser:parse():root(), 0)
+
+  return tostring(level)
+end
+
+return M
-- 
cgit v1.2.3-70-g09d2