474 lines
13 KiB
Lua
474 lines
13 KiB
Lua
local luv = vim.loop
|
|
local api = vim.api
|
|
|
|
local lib = require'nvim-tree.lib'
|
|
local config = require'nvim-tree.config'
|
|
local colors = require'nvim-tree.colors'
|
|
local renderer = require'nvim-tree.renderer'
|
|
local fs = require'nvim-tree.fs'
|
|
local view = require'nvim-tree.view'
|
|
local utils = require'nvim-tree.utils'
|
|
|
|
local _config = {
|
|
is_windows = vim.fn.has('win32') == 1 or vim.fn.has('win32unix') == 1,
|
|
is_macos = vim.fn.has('mac') == 1 or vim.fn.has('macunix') == 1,
|
|
is_unix = vim.fn.has('unix') == 1,
|
|
}
|
|
|
|
local M = {}
|
|
|
|
function M.focus()
|
|
if not view.win_open() then
|
|
lib.open()
|
|
end
|
|
view.focus();
|
|
end
|
|
|
|
function M.toggle(find_file)
|
|
if view.win_open() then
|
|
view.close()
|
|
else
|
|
if _config.update_focused_file.enable or find_file then
|
|
M.find_file(true)
|
|
end
|
|
if not view.win_open() then
|
|
lib.open()
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.close()
|
|
if view.win_open() then
|
|
view.close()
|
|
return true
|
|
end
|
|
end
|
|
|
|
function M.open()
|
|
if not view.win_open() then
|
|
lib.open()
|
|
else
|
|
lib.set_target_win()
|
|
end
|
|
end
|
|
|
|
function M.tab_change()
|
|
vim.schedule(function()
|
|
if not view.win_open() and view.win_open({ any_tabpage = true }) then
|
|
local bufname = vim.api.nvim_buf_get_name(0)
|
|
if bufname:match("Neogit") ~= nil or bufname:match("--graph") ~= nil then
|
|
return
|
|
end
|
|
view.open({ focus_tree = false })
|
|
end
|
|
end)
|
|
end
|
|
|
|
local function go_to(mode)
|
|
local icon_state = config.get_icon_state()
|
|
local flags = mode == 'prev_git_item' and 'b' or ''
|
|
local icons = table.concat(vim.tbl_values(icon_state.icons.git_icons), '\\|')
|
|
return function()
|
|
return icon_state.show_git_icon and vim.fn.search(icons, flags)
|
|
end
|
|
end
|
|
|
|
local keypress_funcs = {
|
|
create = fs.create,
|
|
remove = fs.remove,
|
|
rename = fs.rename(false),
|
|
full_rename = fs.rename(true),
|
|
copy = fs.copy,
|
|
copy_name = fs.copy_filename,
|
|
copy_path = fs.copy_path,
|
|
copy_absolute_path = fs.copy_absolute_path,
|
|
cut = fs.cut,
|
|
paste = fs.paste,
|
|
close_node = lib.close_node,
|
|
parent_node = lib.parent_node,
|
|
toggle_ignored = lib.toggle_ignored,
|
|
toggle_dotfiles = lib.toggle_dotfiles,
|
|
toggle_help = lib.toggle_help,
|
|
refresh = lib.refresh_tree,
|
|
first_sibling = function(node) lib.sibling(node, -math.huge) end,
|
|
last_sibling = function(node) lib.sibling(node, math.huge) end,
|
|
prev_sibling = function(node) lib.sibling(node, -1) end,
|
|
next_sibling = function(node) lib.sibling(node, 1) end,
|
|
prev_git_item = go_to('prev_git_item'),
|
|
next_git_item = go_to('next_git_item'),
|
|
dir_up = lib.dir_up,
|
|
close = function() M.close() end,
|
|
preview = function(node)
|
|
if node.entries ~= nil or node.name == '..' then return end
|
|
return lib.open_file('preview', node.absolute_path)
|
|
end,
|
|
system_open = function(node)
|
|
if not _config.system_open.cmd then
|
|
if _config.is_windows then
|
|
_config.system_open = {
|
|
cmd = "cmd",
|
|
args = {'/c', 'start', '""'}
|
|
}
|
|
elseif _config.is_macos then
|
|
_config.system_open.cmd = 'open'
|
|
elseif _config.is_unix then
|
|
_config.system_open.cmd = 'xdg-open'
|
|
else
|
|
require'nvim-tree.utils'.echo_warning("Cannot open file with system application. Unrecognized platform.")
|
|
return
|
|
end
|
|
end
|
|
|
|
local process = {
|
|
cmd = _config.system_open.cmd,
|
|
args = _config.system_open.args,
|
|
errors = '\n',
|
|
stderr = luv.new_pipe(false)
|
|
}
|
|
table.insert(process.args, node.link_to or node.absolute_path)
|
|
process.handle, process.pid = luv.spawn(process.cmd,
|
|
{ args = process.args, stdio = { nil, nil, process.stderr }, detached = true },
|
|
function(code)
|
|
process.stderr:read_stop()
|
|
process.stderr:close()
|
|
process.handle:close()
|
|
if code ~= 0 then
|
|
process.errors = process.errors .. string.format('NvimTree system_open: return code %d.', code)
|
|
error(process.errors)
|
|
end
|
|
end
|
|
)
|
|
table.remove(process.args)
|
|
if not process.handle then
|
|
error("\n" .. process.pid .. "\nNvimTree system_open: failed to spawn process using '" .. process.cmd .. "'.")
|
|
return
|
|
end
|
|
luv.read_start(process.stderr,
|
|
function(err, data)
|
|
if err then return end
|
|
if data then process.errors = process.errors .. data end
|
|
end
|
|
)
|
|
luv.unref(process.handle)
|
|
end,
|
|
}
|
|
|
|
function M.on_keypress(mode)
|
|
if view.is_help_ui() and mode ~= 'toggle_help' then return end
|
|
local node = lib.get_node_at_cursor()
|
|
if not node then return end
|
|
|
|
if keypress_funcs[mode] then
|
|
return keypress_funcs[mode](node)
|
|
end
|
|
|
|
if node.name == ".." then
|
|
return lib.change_dir("..")
|
|
elseif mode == "cd" and node.entries ~= nil then
|
|
return lib.change_dir(lib.get_last_group_node(node).absolute_path)
|
|
elseif mode == "cd" then
|
|
return
|
|
end
|
|
|
|
if node.link_to and not node.entries then
|
|
lib.open_file(mode, node.link_to)
|
|
elseif node.entries ~= nil then
|
|
lib.unroll_dir(node)
|
|
else
|
|
lib.open_file(mode, node.absolute_path)
|
|
end
|
|
end
|
|
|
|
function M.refresh()
|
|
lib.refresh_tree()
|
|
end
|
|
|
|
function M.print_clipboard()
|
|
fs.print_clipboard()
|
|
end
|
|
|
|
function M.hijack_current_window()
|
|
local View = require'nvim-tree.view'.View
|
|
if not View.bufnr then
|
|
View.bufnr = api.nvim_get_current_buf()
|
|
end
|
|
local current_tab = api.nvim_get_current_tabpage()
|
|
if not View.tabpages then
|
|
View.tabpages = {
|
|
[current_tab] = { winnr = api.nvim_get_current_win() }
|
|
}
|
|
else
|
|
View.tabpages[current_tab] = { winnr = api.nvim_get_current_win() }
|
|
end
|
|
end
|
|
|
|
function M.on_enter(opts)
|
|
local bufnr = api.nvim_get_current_buf()
|
|
local bufname = api.nvim_buf_get_name(bufnr)
|
|
local buftype = api.nvim_buf_get_option(bufnr, 'filetype')
|
|
local ft_ignore = _config.ignore_ft_on_setup
|
|
|
|
local stats = luv.fs_stat(bufname)
|
|
local is_dir = stats and stats.type == 'directory'
|
|
if is_dir then
|
|
lib.Tree.cwd = vim.fn.expand(bufname)
|
|
end
|
|
|
|
local netrw_disabled = opts.disable_netrw or opts.hijack_netrw
|
|
|
|
local lines = not is_dir and api.nvim_buf_get_lines(bufnr, 0, -1, false) or {}
|
|
local buf_has_content = #lines > 1 or (#lines == 1 and lines[1] ~= "")
|
|
|
|
local should_open = _config.open_on_setup
|
|
and ((is_dir and netrw_disabled) or (bufname == "" and not buf_has_content))
|
|
and not vim.tbl_contains(ft_ignore, buftype)
|
|
|
|
if should_open then
|
|
M.hijack_current_window()
|
|
end
|
|
|
|
lib.init(should_open, should_open)
|
|
end
|
|
|
|
local function is_file_readable(fname)
|
|
local stat = luv.fs_stat(fname)
|
|
return stat and stat.type == "file" and luv.fs_access(fname, 'R')
|
|
end
|
|
|
|
local function update_base_dir_with_filepath(filepath, bufnr)
|
|
if not _config.update_focused_file.update_cwd then
|
|
return
|
|
end
|
|
|
|
local ft = api.nvim_buf_get_option(bufnr, 'filetype') or ""
|
|
for _, value in pairs(_config.update_focused_file.ignore_list) do
|
|
if vim.fn.stridx(filepath, value) ~= -1 or vim.fn.stridx(ft, value) ~= -1 then
|
|
return
|
|
end
|
|
end
|
|
|
|
if not vim.startswith(filepath, lib.Tree.cwd or vim.loop.cwd()) then
|
|
lib.change_dir(vim.fn.fnamemodify(filepath, ':p:h'))
|
|
end
|
|
end
|
|
|
|
function M.find_file(with_open)
|
|
local bufname = vim.fn.bufname()
|
|
local bufnr = api.nvim_get_current_buf()
|
|
local filepath = vim.fn.fnamemodify(bufname, ':p')
|
|
if not is_file_readable(filepath) then
|
|
return
|
|
end
|
|
|
|
if with_open then
|
|
M.open()
|
|
view.focus()
|
|
end
|
|
|
|
update_base_dir_with_filepath(filepath, bufnr)
|
|
lib.set_index_and_redraw(filepath)
|
|
end
|
|
|
|
function M.resize(size)
|
|
view.View.width = size
|
|
view.View.height = size
|
|
view.resize()
|
|
end
|
|
|
|
function M.on_leave()
|
|
vim.defer_fn(function()
|
|
if not view.win_open() then
|
|
return
|
|
end
|
|
|
|
local windows = api.nvim_list_wins()
|
|
local curtab = api.nvim_get_current_tabpage()
|
|
local wins_in_tabpage = vim.tbl_filter(function(w)
|
|
return api.nvim_win_get_tabpage(w) == curtab
|
|
end, windows)
|
|
if #windows == 1 then
|
|
api.nvim_command(':silent qa!')
|
|
elseif #wins_in_tabpage == 1 then
|
|
api.nvim_command(':tabclose')
|
|
end
|
|
end, 50)
|
|
end
|
|
|
|
function M.open_on_directory()
|
|
local should_proceed = _config.update_to_buf_dir.auto_open or view.win_open()
|
|
if not _config.update_to_buf_dir.enable or not should_proceed then
|
|
return
|
|
end
|
|
local buf = api.nvim_get_current_buf()
|
|
local bufname = api.nvim_buf_get_name(buf)
|
|
if vim.fn.isdirectory(bufname) ~= 1 or bufname == lib.Tree.cwd then
|
|
return
|
|
end
|
|
|
|
view.close()
|
|
|
|
lib.change_dir(bufname)
|
|
M.hijack_current_window()
|
|
|
|
view.open()
|
|
view.focus()
|
|
view.replace_window()
|
|
|
|
lib.set_index_and_redraw(bufname)
|
|
vim.api.nvim_buf_delete(buf, { force = true })
|
|
end
|
|
|
|
function M.reset_highlight()
|
|
colors.setup()
|
|
renderer.render_hl(view.View.bufnr)
|
|
end
|
|
|
|
local prev_line
|
|
function M.place_cursor_on_node()
|
|
local l = vim.api.nvim_win_get_cursor(0)[1]
|
|
if l == prev_line then
|
|
return
|
|
end
|
|
prev_line = l
|
|
|
|
local node = lib.get_node_at_cursor()
|
|
if not node or node.name == ".." then
|
|
return
|
|
end
|
|
|
|
local line = api.nvim_get_current_line()
|
|
local cursor = api.nvim_win_get_cursor(0)
|
|
local idx = vim.fn.stridx(line, node.name)
|
|
api.nvim_win_set_cursor(0, {cursor[1], idx})
|
|
end
|
|
|
|
local function manage_netrw(disable_netrw, hijack_netrw)
|
|
if hijack_netrw then
|
|
vim.cmd "silent! autocmd! FileExplorer *"
|
|
end
|
|
if disable_netrw then
|
|
vim.g.loaded_netrw = 1
|
|
vim.g.loaded_netrwPlugin = 1
|
|
end
|
|
end
|
|
|
|
local function setup_vim_commands()
|
|
vim.cmd [[
|
|
command! NvimTreeOpen lua require'nvim-tree'.open()
|
|
command! NvimTreeClose lua require'nvim-tree'.close()
|
|
command! NvimTreeToggle lua require'nvim-tree'.toggle(false)
|
|
command! NvimTreeFocus lua require'nvim-tree'.focus()
|
|
command! NvimTreeRefresh lua require'nvim-tree'.refresh()
|
|
command! NvimTreeClipboard lua require'nvim-tree'.print_clipboard()
|
|
command! NvimTreeFindFile lua require'nvim-tree'.find_file(true)
|
|
command! NvimTreeFindFileToggle lua require'nvim-tree'.toggle(true)
|
|
command! -nargs=1 NvimTreeResize lua require'nvim-tree'.resize(<args>)
|
|
]]
|
|
end
|
|
|
|
function M.change_dir(name)
|
|
lib.change_dir(name)
|
|
|
|
if _config.update_focused_file.enable then
|
|
M.find_file(false)
|
|
end
|
|
end
|
|
|
|
local function setup_autocommands(opts)
|
|
vim.cmd "augroup NvimTree"
|
|
vim.cmd [[
|
|
""" reset highlights when colorscheme is changed
|
|
au ColorScheme * lua require'nvim-tree'.reset_highlight()
|
|
|
|
au BufWritePost * lua require'nvim-tree'.refresh()
|
|
au User FugitiveChanged,NeogitStatusRefreshed lua require'nvim-tree'.refresh()
|
|
]]
|
|
|
|
if opts.auto_close then
|
|
vim.cmd "au WinClosed * lua require'nvim-tree'.on_leave()"
|
|
end
|
|
if opts.open_on_tab then
|
|
vim.cmd "au TabEnter * lua require'nvim-tree'.tab_change()"
|
|
end
|
|
if opts.hijack_cursor then
|
|
vim.cmd "au CursorMoved NvimTree lua require'nvim-tree'.place_cursor_on_node()"
|
|
end
|
|
if opts.update_cwd then
|
|
vim.cmd "au DirChanged * lua require'nvim-tree'.change_dir(vim.loop.cwd())"
|
|
end
|
|
if opts.update_focused_file.enable then
|
|
vim.cmd "au BufEnter * lua require'nvim-tree'.find_file(false)"
|
|
end
|
|
|
|
vim.cmd "augroup end"
|
|
end
|
|
|
|
local DEFAULT_OPTS = {
|
|
disable_netrw = true,
|
|
hijack_netrw = true,
|
|
open_on_setup = false,
|
|
open_on_tab = false,
|
|
update_to_buf_dir = {
|
|
enable = true,
|
|
auto_open = true,
|
|
},
|
|
auto_close = false,
|
|
hijack_cursor = false,
|
|
update_cwd = false,
|
|
update_focused_file = {
|
|
enable = false,
|
|
update_cwd = false,
|
|
ignore_list = {}
|
|
},
|
|
ignore_ft_on_setup = {},
|
|
system_open = {
|
|
cmd = nil,
|
|
args = {}
|
|
},
|
|
diagnostics = {
|
|
enable = false,
|
|
icons = {
|
|
hint = "",
|
|
info = "",
|
|
warning = "",
|
|
error = "",
|
|
}
|
|
},
|
|
}
|
|
|
|
function M.setup(conf)
|
|
local opts = vim.tbl_deep_extend('force', DEFAULT_OPTS, conf or {})
|
|
|
|
manage_netrw(opts.disable_netrw, opts.hijack_netrw)
|
|
|
|
_config.update_focused_file = opts.update_focused_file
|
|
_config.system_open = opts.system_open
|
|
_config.open_on_setup = opts.open_on_setup
|
|
_config.ignore_ft_on_setup = opts.ignore_ft_on_setup
|
|
if type(opts.update_to_buf_dir) == "boolean" then
|
|
utils.echo_warning("update_to_buf_dir is now a table, see :help nvim-tree.update_to_buf_dir")
|
|
_config.update_to_buf_dir = {
|
|
enable = opts.update_to_buf_dir,
|
|
auto_open = opts.update_to_buf_dir,
|
|
}
|
|
else
|
|
_config.update_to_buf_dir = opts.update_to_buf_dir
|
|
end
|
|
|
|
if opts.lsp_diagnostics ~= nil then
|
|
utils.echo_warning("setup.lsp_diagnostics has been removed, see :help nvim-tree.diagnostics")
|
|
end
|
|
|
|
require'nvim-tree.colors'.setup()
|
|
require'nvim-tree.view'.setup(opts.view or {})
|
|
require'nvim-tree.diagnostics'.setup(opts)
|
|
|
|
setup_autocommands(opts)
|
|
setup_vim_commands()
|
|
|
|
-- scheduling to make sure current buffer has initialized before running buffer checks for auto open
|
|
vim.schedule(function() M.on_enter(opts) end)
|
|
end
|
|
|
|
return M
|