nvim-tree.lua/lua/lib/tree.lua

local api = vim.api
local luv = vim.loop

local renderer = require'lib.renderer'
local config = require'lib.config'
local pops = require'lib.populate'
local populate = pops.populate
local refresh_entries = pops.refresh_entries
local update_git = pops.update_git_status

local M = {}

M.Tree = {
  entries = {},
  buf_name = 'LuaTree',
  cwd = nil,
  win_width =  vim.g.lua_tree_width or 30,
  loaded = false,
  side = 'H',
  bufnr = nil,
  winnr = nil,
  buf_options = {
    'sidescroll=5', 'noswapfile', 'splitbelow',
    'noruler', 'noshowmode', 'noshowcmd'
  },
  win_options = {
    relativenumber = false,
    number =  false,
    list = false,
    winfixwidth = true,
    winfixheight = true,
    foldenable = false,
    spell = false,
    foldmethod = 'manual',
    foldcolumn = '0'
  }
}

if vim.g.lua_tree_side == 'right' then
  M.Tree.side = 'L'
end

function M.init(with_open, with_render)
  M.Tree.cwd = luv.cwd()
  populate(M.Tree.entries, M.Tree.cwd, M.Tree)

  local stat = luv.fs_stat(M.Tree.cwd)
  M.Tree.last_modified = stat.mtime.sec

  if with_open then
    M.open()
  end

  if with_render then
    renderer.draw(M.Tree, true)
    M.Tree.loaded = true
  end
end

local function get_node_at_line(line)
  local index = 2
  local function iter(entries)
    for _, node in ipairs(entries) do
      if index == line then
        return node
      end
      index = index + 1
      if node.open == true then
        local child = iter(node.entries)
        if child ~= nil then return child end
      end
    end
  end
  return iter
end

function M.get_node_at_cursor()
  local cursor = api.nvim_win_get_cursor(M.Tree.winnr)
  local line = cursor[1]
  if line == 1 and M.Tree.cwd ~= "/" then
    return { name = ".." }
  end

  if M.Tree.cwd == "/" then
    line = line + 1
  end
  return get_node_at_line(line)(M.Tree.entries)
end

function M.unroll_dir(node)
  node.open = not node.open
  if #node.entries > 0 then
    renderer.draw(M.Tree, true)
  else
    populate(node.entries, node.absolute_path)
    renderer.draw(M.Tree, true)
  end
end

local function refresh_git(node)
  update_git(node.entries, node.absolute_path or node.cwd)
  for _, entry in pairs(node.entries) do
    if entry.entries ~= nil then
      refresh_git(entry)
    end
  end
end

-- TODO update only entries where directory has changed
local function refresh_nodes(node)
  refresh_entries(node.entries, node.absolute_path or node.cwd)
  for _, entry in ipairs(node.entries) do
    if entry.entries and entry.open then
      refresh_nodes(entry)
    end
  end
end

function M.refresh_tree()
  local stat = luv.fs_stat(M.Tree.cwd)
  -- if stat.mtime.sec ~= M.Tree.last_modified then
    refresh_nodes(M.Tree)
  -- end
  if config.get_icon_state().show_git_icon then
    refresh_git(M.Tree)
  end
  if M.Tree.winnr ~= nil then
    renderer.draw(M.Tree, true)
  end
end

function M.set_index_and_redraw(fname)
  local i
  if M.Tree.cwd == '/' then
    i = 0
  else
    i = 1
  end
  local reload = false

  local function iter(entries)
    for _, entry in ipairs(entries) do
      i = i + 1
      if entry.absolute_path == fname then
        return i
      end

      if fname:match(entry.match_path..'/') ~= nil then
        if #entry.entries == 0 then
          reload = true
          populate(entry.entries, entry.absolute_path)
        end
        if entry.open == false then
          reload = true
          entry.open = true
        end
        if iter(entry.entries) ~= nil then
          return i
        end
      elseif entry.open == true then
        iter(entry.entries)
      end
    end
  end

  local index = iter(M.Tree.entries)
  renderer.draw(M.Tree, reload)
  if index then
    api.nvim_win_set_cursor(M.Tree.winnr, {index, 0})
  end

  return index
end

function M.open_file(mode, filename)
  if vim.g.lua_tree_side == 'right' then
    api.nvim_command('noautocmd wincmd h')
  else
    api.nvim_command('noautocmd wincmd l')
  end
  api.nvim_command(string.format("%s %s", mode, filename))
end

function M.change_dir(foldername)
  api.nvim_command('cd '..foldername)
  M.Tree.entries = {}
  M.init(false, M.Tree.bufnr ~= nil)
end

local function set_mappings()
  local buf = M.Tree.bufnr
  local bindings = config.get_bindings()

  local mappings = {
    ['<2-LeftMouse>'] = 'on_keypress("edit")';
    ['<2-RightMouse>'] = 'on_keypress("cd")';
    [bindings.cd] = 'on_keypress("cd")';
    [bindings.edit] = 'on_keypress("edit")';
    [bindings.edit_vsplit] = 'on_keypress("vsplit")';
    [bindings.edit_split] = 'on_keypress("split")';
    [bindings.edit_tab] = 'on_keypress("tabnew")';
    [bindings.create] = 'on_keypress("create")';
    [bindings.remove] = 'on_keypress("remove")';
    [bindings.rename] = 'on_keypress("rename")';
    gx = "xdg_open()";
  }

  for k,v in pairs(mappings) do
    api.nvim_buf_set_keymap(buf, 'n', k, ':lua require"tree".'..v..'<cr>', {
        nowait = true, noremap = true, silent = true
      })
  end
end

local function create_buf()
  local options = {
    bufhidden = 'delete';
    buftype = 'nofile';
    modifiable = false;
  }

  M.Tree.bufnr = api.nvim_create_buf(false, true)
  api.nvim_buf_set_name(M.Tree.bufnr, M.Tree.buf_name)
  api.nvim_buf_set_option(M.Tree.bufnr, 'filetype', M.Tree.buf_name)

  for opt, val in pairs(options) do
    api.nvim_buf_set_option(M.Tree.bufnr, opt, val)
  end

  for _, opt in pairs(M.Tree.buf_options) do
    api.nvim_command('setlocal '..opt)
  end

  if M.Tree.side == 'L' then
    api.nvim_command('setlocal nosplitright')
  else
    api.nvim_command('setlocal splitright')
  end

  set_mappings()
end

local function create_win()
  api.nvim_command("vsplit")
  api.nvim_command("wincmd "..M.Tree.side)
  api.nvim_command("vertical resize "..M.Tree.win_width)

  M.Tree.winnr = api.nvim_get_current_win()

  for opt, val in pairs(M.Tree.win_options) do
    api.nvim_win_set_option(M.Tree.winnr, opt, val)
  end
  api.nvim_command('setlocal winhighlight+=EndOfBuffer:LuaTreeEndOfBuffer,Normal:LuaTreeNormal,CursorLine:LuaTreeCursorLine,VertSplit:LuaTreeVertSplit')
end

function M.close()
  api.nvim_win_close(M.Tree.winnr, true)
  M.Tree.winnr = nil
  M.Tree.bufnr = nil
end

function M.open()
  create_buf()
  create_win()
  api.nvim_win_set_buf(M.Tree.winnr, M.Tree.bufnr)
  renderer.draw(M.Tree, not M.Tree.loaded)
  M.Tree.loaded = true
end

function M.win_open()
  for _, win in pairs(api.nvim_list_wins()) do
    if win == M.Tree.winnr then
      return true
    end
  end
  return false
end

return M