Add config option to follow buffer directory

lua/tree.lua

@@ -153,6 +153,15 @@ function M.check_windows_and_close()
   end
 end
 
+function M.navigate_to_buffer_dir(bufname)
+  local new_path = get_cwd()
+  if new_path ~= '/' then
+    new_path = new_path .. '/'
+  end
+  set_root_path(new_path)
+  init_tree()
+end
+
 function M.check_buffer_and_open()
   local bufname = api.nvim_buf_get_name(0)
   if bufname == '' then
@@ -169,12 +178,7 @@ function M.check_buffer_and_open()
 
     M.toggle()
   else
-    local new_path = get_cwd()
+    M.navigate_to_buffer_dir()
-    if new_path ~= '/' then
-      new_path = new_path .. '/'
-    end
-    set_root_path(new_path)
-    init_tree()
   end
 end
 

plugin/tree.vim

@@ -23,6 +23,7 @@ augroup LuaTree
     au BufEnter * :LuaTreeFindFile
   endif
 
+  au BufEnter * lua require'tree'.navigate_to_buffer_dir()
   au ColorScheme * lua require'tree'.reset_highlight()
 augroup end