Refacto: rewrite everything
- The tree is created with libuv functions, which makes it blazingly fast. - The tree may now be faster than any other vim trees, it can handle directories with thousands of files without any latency at all (tested on 40K files, works flawlessly). - More solid logic for opening and closing the tree. - tree state is remembered (closing / opening a folder keeps opened subdirectories open) - detection of multiple git projects in the tree - more icon support - smart rendering - smart updates - ms windows support - gx replacement function running xdg-open on linux, open on macos
This commit is contained in:
kiyan42
183
lua/lib/renderer.lua
Normal file
183
lua/lib/renderer.lua
Normal file
@@ -0,0 +1,183 @@
|
||||
local colors = require'lib.colors'
|
||||
local config = require'lib.config'
|
||||
|
||||
local api = vim.api
|
||||
|
||||
local lines = {}
|
||||
local hl = {}
|
||||
local index = 0
|
||||
local namespace_id = api.nvim_create_namespace('LuaTreeHighlights')
|
||||
|
||||
local icon_state = config.get_icon_state()
|
||||
|
||||
local get_folder_icon = function() return "" end
|
||||
local set_folder_hl = function(index, depth, git_icon_len)
|
||||
table.insert(hl, {'LuaTreeFolderName', index, depth+git_icon_len, -1})
|
||||
end
|
||||
|
||||
if icon_state.show_folder_icon then
|
||||
get_folder_icon = function(open)
|
||||
if open then
|
||||
return " "
|
||||
else
|
||||
return " "
|
||||
end
|
||||
end
|
||||
set_folder_hl = function(index, depth, icon_len, name_len)
|
||||
table.insert(hl, {'LuaTreeFolderName', index, depth+icon_len, depth+icon_len+name_len})
|
||||
table.insert(hl, {'LuaTreeFolderIcon', index, depth, depth+icon_len})
|
||||
end
|
||||
end
|
||||
|
||||
local get_file_icon = function() return "" end
|
||||
if icon_state.show_file_icon then
|
||||
local web_devicons = require'nvim-web-devicons'
|
||||
|
||||
get_file_icon = function(fname, extension, index, depth)
|
||||
local icon, hl_group = web_devicons.get_icon(fname, extension)
|
||||
-- TODO: remove this hl_group and make this in nvim-web-devicons
|
||||
if #extension == 0 then
|
||||
hl_group = colors.hl_groups[fname]
|
||||
else
|
||||
hl_group = colors.hl_groups[extension]
|
||||
end
|
||||
if hl_group and icon then
|
||||
table.insert(hl, { 'LuaTree'..hl_group, index, depth, depth + #icon })
|
||||
return icon.." "
|
||||
else
|
||||
return icon_state.icons.default and icon_state.icons.default.." " or ""
|
||||
end
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
local get_git_icons = function() return "" end
|
||||
local git_icon_state = {}
|
||||
if icon_state.show_git_icon then
|
||||
|
||||
git_icon_state = {
|
||||
["M "] = { { icon = icon_state.icons.git_icons.staged, hl = "LuaTreeGitStaged" } },
|
||||
[" M"] = { { icon = icon_state.icons.git_icons.unstaged, hl = "LuaTreeGitDirty" } },
|
||||
["MM"] = {
|
||||
{ icon = icon_state.icons.git_icons.staged, hl = "LuaTreeGitStaged" },
|
||||
{ icon = icon_state.icons.git_icons.unstaged, hl = "LuaTreeGitDirty" }
|
||||
},
|
||||
["A "] = {
|
||||
{ icon = icon_state.icons.git_icons.staged, hl = "LuaTreeGitStaged" },
|
||||
{ icon = icon_state.icons.git_icons.untracked, hl = "LuaTreeGitNew" }
|
||||
},
|
||||
["AM"] = {
|
||||
{ icon = icon_state.icons.git_icons.staged, hl = "LuaTreeGitStaged" },
|
||||
{ icon = icon_state.icons.git_icons.untracked, hl = "LuaTreeGitNew" },
|
||||
{ icon = icon_state.icons.git_icons.unstaged, hl = "LuaTreeGitDirty" }
|
||||
},
|
||||
["??"] = { { icon = icon_state.icons.git_icons.untracked, hl = "LuaTreeGitNew" } },
|
||||
["R "] = { { icon = icon_state.icons.git_icons.renamed, hl = "LuaTreeGitRenamed" } },
|
||||
["UU"] = { { icon = icon_state.icons.git_icons.unmerged, hl = "LuaTreeGitMerge" } },
|
||||
dirty = { { icon = icon_state.icons.git_icons.unstaged, hl = "LuaTreeGitDirty" } },
|
||||
}
|
||||
|
||||
get_git_icons = function(node, index, depth, icon_len)
|
||||
local git_status = node.git_status
|
||||
if not git_status then return "" end
|
||||
|
||||
local icon = ""
|
||||
local icons = git_icon_state[git_status]
|
||||
for _, v in ipairs(icons) do
|
||||
table.insert(hl, { v.hl, index, depth+icon_len+#icon, depth+icon_len+#icon+#v.icon })
|
||||
icon = icon..v.icon.." "
|
||||
end
|
||||
|
||||
return icon
|
||||
end
|
||||
end
|
||||
|
||||
local picture = {
|
||||
jpg = true,
|
||||
jpeg = true,
|
||||
png = true,
|
||||
gif = true,
|
||||
}
|
||||
|
||||
local special = {
|
||||
["Cargo.toml"] = true,
|
||||
Makefile = true,
|
||||
["README.md"] = true,
|
||||
["readme.md"] = true,
|
||||
}
|
||||
|
||||
local function update_draw_data(tree, depth)
|
||||
if tree.cwd and tree.cwd ~= '/' then
|
||||
table.insert(lines, "..")
|
||||
table.insert(hl, {'LuaTreeFolderName', index, 0, 2})
|
||||
index = 1
|
||||
end
|
||||
|
||||
for _, node in ipairs(tree.entries) do
|
||||
local padding = string.rep(" ", depth)
|
||||
if node.entries then
|
||||
local icon = get_folder_icon(node.open)
|
||||
local git_icon = get_git_icons(node, index, depth+#node.name, #icon+1)
|
||||
set_folder_hl(index, depth, #icon, #node.name)
|
||||
index = index + 1
|
||||
if node.open then
|
||||
table.insert(lines, padding..icon..node.name.." "..git_icon)
|
||||
update_draw_data(node, depth + 2)
|
||||
else
|
||||
table.insert(lines, padding..icon..node.name.." "..git_icon)
|
||||
end
|
||||
elseif node.link_to then
|
||||
table.insert(hl, { 'LuaTreeSymlink', index, depth, -1 })
|
||||
table.insert(lines, padding..node.name.." ➛ "..node.link_to)
|
||||
index = index + 1
|
||||
|
||||
else
|
||||
local icon
|
||||
local git_icons
|
||||
if special[node.name] then
|
||||
icon = ""
|
||||
git_icons = get_git_icons(node, index, depth, 0)
|
||||
table.insert(hl, {'LuaTreeSpecialFile', index, depth+#git_icons, -1})
|
||||
else
|
||||
icon = get_file_icon(node.name, node.extension, index, depth)
|
||||
git_icons = get_git_icons(node, index, depth, #icon)
|
||||
end
|
||||
table.insert(lines, padding..icon..git_icons..node.name)
|
||||
if node.executable then
|
||||
table.insert(hl, {'LuaTreeExecFile', index, depth+#icon+#git_icons, -1 })
|
||||
elseif picture[node.extension] then
|
||||
table.insert(hl, {'LuaTreeImageFile', index, depth+#icon+#git_icons, -1 })
|
||||
end
|
||||
index = index + 1
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.draw(tree, reload)
|
||||
api.nvim_buf_set_option(tree.bufnr, 'modifiable', true)
|
||||
local cursor = api.nvim_win_get_cursor(tree.winnr)
|
||||
if reload then
|
||||
index = 0
|
||||
lines = {}
|
||||
hl = {}
|
||||
update_draw_data(tree, 0)
|
||||
end
|
||||
|
||||
api.nvim_buf_set_lines(tree.bufnr, 0, -1, false, lines)
|
||||
M.render_hl(tree.bufnr)
|
||||
if #lines > cursor[1] then
|
||||
api.nvim_win_set_cursor(tree.winnr, cursor)
|
||||
end
|
||||
api.nvim_buf_set_option(tree.bufnr, 'modifiable', false)
|
||||
end
|
||||
|
||||
function M.render_hl(bufnr)
|
||||
api.nvim_buf_clear_namespace(bufnr, namespace_id, 0, -1)
|
||||
for _, data in ipairs(hl) do
|
||||
api.nvim_buf_add_highlight(bufnr, namespace_id, data[1], data[2], data[3], data[4])
|
||||
end
|
||||
end
|
||||
|
||||
return M
|
||||
Reference in New Issue
Block a user