Properly sync nvim-tree across tabs

Author: rwblokzijl

lua/nvim-tree.lua

@@ -88,6 +88,7 @@ end
 
 function M.open(cwd)
   cwd = cwd ~= "" and cwd or nil
+  vim.g.nvim_tree_tab_open = 1
   if view.is_visible() then
     lib.set_target_win()
     view.focus()
@@ -120,7 +121,7 @@ function M.open_replacing_current_buffer(cwd)
 end
 
 function M.tab_change()
-  if view.is_visible { any_tabpage = true } then
+  if vim.g.nvim_tree_tab_open == 1 then
     local bufname = api.nvim_buf_get_name(0)
     local ft = api.nvim_buf_get_option(0, "ft")
     for _, filter in ipairs(M.config.ignore_buf_on_tab_change) do
@@ -130,6 +131,33 @@ function M.tab_change()
     end
     view.open { focus_tree = false }
     require("nvim-tree.renderer").draw()
+  else
+    view.close()
+  end
+end
+
+function M.tab_win_closed(winnr)
+  local tabnr = vim.api.nvim_win_get_tabpage(winnr)
+  local bufnr = vim.api.nvim_win_get_buf(winnr)
+  local buf_info = vim.fn.getbufinfo(bufnr)[1]
+  local tab_wins = vim.tbl_filter(function(w) return w~=winnr end, vim.api.nvim_tabpage_list_wins(tabnr))
+  local tab_bufs = vim.tbl_map(vim.api.nvim_win_get_buf, tab_wins)
+  if buf_info.name:match(".*NvimTree_%d*$") then -- close buffer was nvim tree
+    if not vim.tbl_isempty(tab_bufs) then -- and was not the last window (closed automatically)
+      vim.g.nvim_tree_tab_open = 0 -- NOTE: this also catches a lot of cases where nvim tree was already closed.
+    end
+  else -- closed buffer was normal buffer
+    if #tab_bufs == 1 then                                    -- if there is only 1 buffer left
+      local last_buf_info = vim.fn.getbufinfo(tab_bufs[1])[1]
+      if last_buf_info.name:match(".*NvimTree_%d*$") then     -- and that buffer is nvim tree
+        vim.schedule(function ()
+          if #vim.api.nvim_list_wins() == 1 then
+            vim.cmd "quit"
+          end
+          vim.api.nvim_win_close(tab_wins[1], true)           -- then close that window
+        end)
+      end
+    end
   end
 end
 
@@ -360,6 +388,11 @@ local function setup_autocommands(opts)
 
   if opts.open_on_tab then
     create_nvim_tree_autocmd("TabEnter", { callback = vim.schedule_wrap(M.tab_change) })
+    create_nvim_tree_autocmd("WinClosed", { callback = function ()
+      local winnr = tonumber(vim.fn.expand("<amatch>"))
+      vim.schedule_wrap(M.tab_win_closed(winnr))
+    end
+    })
   end
   if opts.hijack_cursor then
     create_nvim_tree_autocmd("CursorMoved", { pattern = "NvimTree_*", callback = M.place_cursor_on_node })
@@ -717,6 +750,8 @@ function M.setup(conf)
 
   manage_netrw(opts.disable_netrw, opts.hijack_netrw)
 
+  vim.g.nvim_tree_tab_open = 0
+
   M.config = opts
   require("nvim-tree.log").setup(opts)
 

lua/nvim-tree/view.lua

@@ -189,6 +189,7 @@ function M.close()
   if not M.is_visible() then
     return
   end
+  vim.g.nvim_tree_tab_open = 0
   save_tab_state()
   switch_buf_if_last_buf()
   local tree_win = M.get_winnr()