Add inspect_tree function

Akmadan23 · Nov 4, 2023 · 1330951 · 1330951
1 parent 2a6a2a9
commit 1330951
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/config/nvim/lua/functions.lua b/config/nvim/lua/functions.lua
@@ -503,4 +503,22 @@ M.inspect_object = function()
     end)
 end
 
+local trees = {}
+
+--- Toggle layer for vim.treesitter.inspect_tree
+M.inspect_tree = function()
+    local winnr = vim.api.nvim_get_current_win()
+
+    for i, t in ipairs(trees) do
+        if vim.tbl_contains(t, winnr) then
+            vim.api.nvim_win_close(t[2], false)
+            table.remove(trees, i)
+            return
+        end
+    end
+
+    vim.treesitter.inspect_tree()
+    table.insert(trees, { winnr, vim.api.nvim_get_current_win() })
+end
+
 return M