feature: change worktree on all buffers

Change `update_current_buffer` function to update all currently open
buffers and de-list buffers from the previous worktree. Buffers from
files that don't exist on another worktree will only be de-listed.

Author: Kasama

lua/git-worktree/init.lua

@@ -478,34 +478,60 @@ M.update_current_buffer = function(prev_path)
         return false
     end
 
+    local buffers = vim.api.nvim_list_bufs()
+    local listed_buffers = vim.tbl_filter(function(bufn)
+        return vim.api.nvim_buf_is_valid(bufn) and vim.api.nvim_buf_get_option(bufn, 'buflisted')
+    end, buffers)
+
+    local changed = false
+    local current_buf_name = vim.api.nvim_buf_get_name(vim.api.nvim_get_current_buf())
+    local new_current_buf = 0
+
     local cwd = vim.loop.cwd()
-    local current_buf_name = vim.api.nvim_buf_get_name(0)
-    if not current_buf_name or current_buf_name == "" then
-        return false
-    end
+    for _, existing_buffer in ipairs(listed_buffers) do
+        local buf_name = vim.api.nvim_buf_get_name(existing_buffer)
+        if not buf_name or buf_name == "" then
+            goto continue
+        end
 
-    local name = Path:new(current_buf_name):absolute()
-    local start, fin = string.find(name, cwd..Path.path.sep, 1, true)
-    if start ~= nil then
-        return true
-    end
+        local name = Path:new(buf_name):absolute()
+        local start, fin = string.find(name, cwd .. Path.path.sep, 1, true)
+        if start ~= nil then
+            changed = true
+            goto continue
+        end
 
-    start, fin = string.find(name, prev_path, 1, true)
-    if start == nil then
-        return false
-    end
+        -- de-list buffer if it's related to the old worktree
+        vim.api.nvim_buf_set_option(existing_buffer, 'buflisted', false)
 
-    local local_name = name:sub(fin + 2)
+        start, fin = string.find(name, prev_path, 1, true)
+        if start == nil then
+            goto continue
+        end
 
-    local final_path = Path:new({cwd, local_name}):absolute()
+        local local_name = name:sub(fin + 2)
 
-    if not Path:new(final_path):exists() then
-        return false
+        local final_path = Path:new({ cwd, local_name }):absolute()
+
+        if not Path:new(final_path):exists() then
+            goto continue
+        end
+
+        local new_buffer = vim.fn.bufnr(final_path, true)
+        vim.api.nvim_buf_set_option(new_buffer, 'buflisted', true)
+        vim.api.nvim_set_current_buf(new_buffer)
+
+        if current_buf_name == buf_name then
+            new_current_buf = new_buffer
+        end
+
+        changed = true
+
+        ::continue::
     end
+    vim.api.nvim_set_current_buf(new_current_buf)
 
-    local bufnr = vim.fn.bufnr(final_path, true)
-    vim.api.nvim_set_current_buf(bufnr)
-    return true
+    return changed
 end
 
 M.on_tree_change = function(cb)