120 lines
3.1 KiB
Lua
120 lines
3.1 KiB
Lua
local api = vim.api
|
|
|
|
local function get_padding(depth)
|
|
local str = ""
|
|
|
|
while 0 < depth do
|
|
str = str .. " "
|
|
depth = depth - 1
|
|
end
|
|
|
|
return str
|
|
end
|
|
|
|
local function default_icons(_, isdir, open)
|
|
if isdir == true then
|
|
if open == true then return " " end
|
|
return " "
|
|
end
|
|
|
|
return ""
|
|
end
|
|
|
|
local function dev_icons(pathname, isdir, open)
|
|
if isdir == true then return default_icons(pathname, isdir, open) end
|
|
|
|
local icon = api.nvim_call_function('WebDevIconsGetFileTypeSymbol', { pathname, isdir })
|
|
if icon == "" then return "" end
|
|
return icon .. " "
|
|
end
|
|
|
|
local function get_icon_func_gen()
|
|
if api.nvim_call_function('exists', { "WebDevIconsGetFileTypeSymbol" }) == 0 then
|
|
return dev_icons
|
|
else
|
|
return default_icons
|
|
end
|
|
end
|
|
|
|
local get_icon = get_icon_func_gen()
|
|
|
|
local function format_tree(tree)
|
|
local dirs = {}
|
|
|
|
for i, node in pairs(tree) do
|
|
local padding = get_padding(node.depth)
|
|
local icon = ""
|
|
if node.icon == true then
|
|
icon = get_icon(node.path .. node.name, node.dir, node.open)
|
|
end
|
|
dirs[i] = padding .. icon .. node.name
|
|
end
|
|
|
|
return dirs
|
|
end
|
|
|
|
local HIGHLIGHT_GROUPS = {
|
|
['^.*%.md$'] = 'MarkdownFile';
|
|
['^LICENSE$'] = 'LicenseFile';
|
|
['^.*%.vim$'] = 'VimFile';
|
|
['^.*%.c$'] = 'CFile';
|
|
['^.*%.cpp$'] = 'CFile';
|
|
['^.*%.cxx$'] = 'CFile';
|
|
['^.*%.h$'] = 'CFile';
|
|
['^.*%.hpp$'] = 'CFile';
|
|
['^.*%.py$'] = 'PythonFile';
|
|
['^.*%.lua$'] = 'LuaFile';
|
|
['^.*%.rs$'] = 'RustFile';
|
|
['^.*%.sh$'] = 'ShellFile';
|
|
['^.*%.zsh$'] = 'ShellFile';
|
|
['^.*%.csh$'] = 'ShellFile';
|
|
['^.*%.rs$'] = 'RustFile';
|
|
['^.*%.js$'] = 'JavascriptFile';
|
|
['^.*%.json$'] = 'JsonFile';
|
|
['^.*%.toml$'] = 'TomlFile';
|
|
['^.*%.yml$'] = 'YamlFile';
|
|
['^.*%.gitignore$'] = 'GitignoreFile';
|
|
['^.*%.ts$'] = 'TypescriptFile';
|
|
['^.*%.jsx$'] = 'ReactFile';
|
|
['^.*%.tsx$'] = 'ReactFile';
|
|
['^.*%.html?$'] = 'HtmlFile';
|
|
['^.*%.png$'] = 'ImageFile';
|
|
['^.*%.jpe?g$'] = 'ImageFile';
|
|
}
|
|
|
|
local function highlight_line(buffer)
|
|
local function highlight(group, line, from, to)
|
|
vim.api.nvim_buf_add_highlight(buffer, -1, group, line, from, to)
|
|
end
|
|
return function(line, node)
|
|
local text_start = node.depth * 2 + 4
|
|
if node.dir then
|
|
if node.name ~= '..' then
|
|
highlight('LuaTreeFolderIcon', line, 0, text_start)
|
|
highlight('LuaTreeFolderName', line, text_start, -1)
|
|
else
|
|
highlight('LuaTreeFolderName', line, 0, -1)
|
|
end
|
|
else
|
|
for k, v in pairs(HIGHLIGHT_GROUPS) do
|
|
if string.match(node.name, k) ~= nil then
|
|
highlight('LuaTree' .. v, line, 0, text_start)
|
|
break
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
local function highlight_buffer(buffer, tree)
|
|
local highlight = highlight_line(buffer)
|
|
for i, node in pairs(tree) do
|
|
highlight(i - 1, node)
|
|
end
|
|
end
|
|
|
|
return {
|
|
format_tree = format_tree;
|
|
highlight_buffer = highlight_buffer;
|
|
}
|