67 lines
1.4 KiB
Lua
67 lines
1.4 KiB
Lua
|
Config = {
|
||
|
Suffixes = {
|
||
|
'.c',
|
||
|
'.cpp',
|
||
|
'.h',
|
||
|
'.hpp',
|
||
|
'.inl',
|
||
|
}
|
||
|
}
|
||
|
|
||
|
local function setup(config)
|
||
|
if config then
|
||
|
for key, value in pairs(config) do
|
||
|
Config[key] = config[key] or value
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
local function file_exists(path)
|
||
|
return vim.loop.fs_stat(path) ~= nil
|
||
|
end
|
||
|
|
||
|
local function path_pivot(path, suffix)
|
||
|
local start, stop = path:find(suffix)
|
||
|
if start and stop == #path then
|
||
|
return path:sub(0, start - 1)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
local function revolve(back)
|
||
|
local path = vim.api.nvim_buf_get_name(vim.api.nvim_get_current_buf())
|
||
|
|
||
|
local suffix_index = nil
|
||
|
local prefix = nil
|
||
|
for i, s in ipairs(Config.Suffixes) do
|
||
|
prefix = path_pivot(path, s)
|
||
|
if prefix then
|
||
|
suffix_index = i
|
||
|
break
|
||
|
end
|
||
|
end
|
||
|
|
||
|
if prefix then
|
||
|
local start = 1
|
||
|
local stop = #Config.Suffixes - 1
|
||
|
local step = 1
|
||
|
if back then
|
||
|
start, stop = stop, start
|
||
|
step = -1
|
||
|
end
|
||
|
|
||
|
for i = start, stop, step do
|
||
|
local j = (i + suffix_index - 1) % #Config.Suffixes + 1
|
||
|
local next_path = prefix .. Config.Suffixes[j]
|
||
|
if path ~= next_path and file_exists(next_path) then
|
||
|
vim.cmd(string.format('e %s', next_path))
|
||
|
break
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
return {
|
||
|
revolve = revolve,
|
||
|
setup = setup,
|
||
|
}
|