37 lines
1 KiB
Lua
37 lines
1 KiB
Lua
local M = {}
|
|
|
|
M.setup = function()
|
|
-- Array of file names indicating root directory. Modify to your liking.
|
|
local root_names = { ".git", "Makefile", ".svn", ".hg" }
|
|
|
|
-- Cache to use for speed up (at cost of possibly outdated results)
|
|
local root_cache = {}
|
|
|
|
local set_root = function()
|
|
-- Get directory path to start search from
|
|
local path = vim.api.nvim_buf_get_name(0)
|
|
if path == "" then
|
|
return
|
|
end
|
|
path = vim.fs.dirname(path)
|
|
|
|
-- Try cache and resort to searching upward for root directory
|
|
local root = root_cache[path]
|
|
if root == nil then
|
|
local root_file = vim.fs.find(root_names, { path = path, upward = true })[1]
|
|
if root_file == nil then
|
|
return
|
|
end
|
|
root = vim.fs.dirname(root_file)
|
|
root_cache[path] = root
|
|
end
|
|
|
|
-- Set current directory
|
|
vim.fn.chdir(root)
|
|
end
|
|
|
|
local root_augroup = vim.api.nvim_create_augroup("MyAutoRoot", {})
|
|
vim.api.nvim_create_autocmd("BufEnter", { group = root_augroup, callback = set_root })
|
|
end
|
|
|
|
return M
|