fix(treesitter): attempt to call method 'type' (a nil value) (#254)

kevinhwang91 · Oct 23, 2024 · c96bb3b · c96bb3b
1 parent 203c9f4
commit c96bb3b
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/lua/ufo/provider/treesitter.lua b/lua/ufo/provider/treesitter.lua
@@ -174,13 +174,15 @@ function Treesitter.getFolds(bufnr)
         error('UfoFallbackException')
     end
     for _, node in ipairs(matches) do
-        local start, _, stop, stop_col = node:range()
-        if stop_col == 0 then
-            stop = stop - 1
-        end
-        if stop > start then
-            local type = node.type and node:type() or nil
-            table.insert(ranges, foldingrange.new(start, stop, nil, nil, type))
+        if node.range then
+            local start, _, stop, stop_col = node:range()
+            if stop_col == 0 then
+                stop = stop - 1
+            end
+            if stop > start then
+                local type = node.type and node:type() or nil
+                table.insert(ranges, foldingrange.new(start, stop, nil, nil, type))
+            end
         end
     end
     foldingrange.sortRanges(ranges)