1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
-- Taken from: https://github.com/desdic/telescope-rooter.nvim/blob/69423216c75a5f1f1477bbf8faf6b0dc8af04099/lua/telescope/_extensions/rooter.lua
local has_telescope, telescope = pcall(require, "telescope")
if not has_telescope then
error("This extension requires telescope.nvim")
return
end
local has_plenary, plenary = pcall(require, "plenary")
if not has_plenary then
error("This extension requires plenary")
return
end
local log = plenary.log.new({ plugin = "telescope_rooter", level = "info" })
-- TODO: expose this function
local toggle = function(_)
vim.g["Telescope#rooter#enabled"] = not vim.g["Telescope#rooter#enabled"]
print("Telescope#rooter#enabled=" .. vim.inspect(vim.g["Telescope#rooter#enabled"]))
end
local config = { patterns = { ".git" }, enable = true, debug = false }
-- default enabled
vim.g["Telescope#rooter#enabled"] = vim.F.if_nil(config.enable, true)
-- redefine log if debug enabled
if vim.F.if_nil(config.debug, false) then log = plenary.log.new({ plugin = "telescope_rooter", level = "debug" }) end
local group = vim.api.nvim_create_augroup("TelescopeRooter", { clear = true })
vim.api.nvim_create_autocmd({ "DirChangedPre" }, {
callback = function()
if vim.g["Telescope#rooter#enabled"] ~= true then return end
if vim.g["Telescope#rooter#oldpwd"] == nil then
vim.g["Telescope#rooter#oldpwd"] = vim.loop.cwd()
log.debug("before " .. vim.inspect(vim.loop.cwd()))
end
end,
group = group,
})
vim.api.nvim_create_autocmd({ "BufEnter", "BufWinEnter" }, {
callback = function()
if vim.g["Telescope#rooter#enabled"] ~= true then return end
vim.schedule(function()
if vim.bo.filetype == "TelescopePrompt" then
local rootdir = vim.fs.dirname(vim.fs.find(config.patterns, { upward = true })[1])
if rootdir ~= nil then
vim.api.nvim_set_current_dir(rootdir)
log.debug("changing dir to " .. rootdir)
end
end
end)
end,
group = group,
})
vim.api.nvim_create_autocmd({ "BufWinLeave" }, {
callback = function()
if vim.g["Telescope#rooter#enabled"] ~= true then return end
vim.schedule(function()
if vim.bo.filetype ~= "TelescopePrompt" then
if vim.g["Telescope#rooter#oldpwd"] ~= nil then
log.debug("restoring " .. vim.g["Telescope#rooter#oldpwd"])
vim.api.nvim_set_current_dir(vim.g["Telescope#rooter#oldpwd"])
vim.g["Telescope#rooter#oldpwd"] = nil
end
end
end)
end,
group = group,
})
|
