Add auto close

1 files changed, 16 insertions, 0 deletions

diff --git a/lua/plugins/nvim-tree.lua b/lua/plugins/nvim-tree.lua
index a1731ff..df1bc23 100644
--- a/lua/plugins/nvim-tree.lua
+++ b/lua/plugins/nvim-tree.lua
@@ -79,6 +79,22 @@ nvim_tree.setup({
   },
 })
 
+vim.api.nvim_create_autocmd("QuitPre", {
+  callback = function()
+    local invalid_win = {}
+    local wins = vim.api.nvim_list_wins()
+    for _, w in ipairs(wins) do
+      local bufname = vim.api.nvim_buf_get_name(vim.api.nvim_win_get_buf(w))
+      if bufname:match("NvimTree_") ~= nil then
+        table.insert(invalid_win, w)
+      end
+    end
+    if #invalid_win == #wins - 1 then
+      -- Should quit, so we close all invalid windows.
+      for _, w in ipairs(invalid_win) do vim.api.nvim_win_close(w, true) end
+    end
+  end
+})
 -- Highlight Groups
 vim.api.nvim_command("highlight NvimTreeNormal guibg=none")
 --vim.api.nvim_command("highlight NvimTreeSymlink ")