nvim-tree.lua/lua/tree.lua

local api = vim.api

local fs_update = require 'lib/fs_update'
local create_file = fs_update.create_file
local rename_file = fs_update.rename_file
local remove_file = fs_update.remove_file

local fs = require 'lib/fs'
local check_dir_access = fs.check_dir_access
local is_dir = fs.is_dir
local get_cwd = fs.get_cwd

local state = require 'lib/state'
local get_tree = state.get_tree
local init_tree = state.init_tree
local open_dir = state.open_dir
local refresh_tree = state.refresh_tree
local set_root_path = state.set_root_path
local find_file = state.find_file

local winutils = require 'lib/winutils'
local update_view = winutils.update_view
local is_win_open = winutils.is_win_open
local close = winutils.close
local open = winutils.open
local set_mappings = winutils.set_mappings
local replace_tree = winutils.replace_tree
local get_win = winutils.get_win

local git = require 'lib/git'
local refresh_git = git.refresh_git
local force_refresh_git = git.force_refresh_git

require 'lib/colors'.init_colors()

init_tree()

local function toggle()
    if is_win_open() == true then
        local wins = api.nvim_list_wins()
        if #wins > 1 then close() end
    else
        open()
        update_view()
        set_mappings()
    end
end

local MOVE_TO = 'l'
if api.nvim_call_function('exists', { 'g:lua_tree_side' }) == 1 then
    if api.nvim_get_var('lua_tree_side') == 'right' then
         MOVE_TO = 'h'
    end
end

local function create_new_buf(open_type, bufname)
    if open_type == 'edit' or open_type == 'split' then
        api.nvim_command('wincmd '..MOVE_TO..' | '..open_type..' '..bufname)
    elseif open_type == 'vsplit' then
        local windows = api.nvim_list_wins();
        api.nvim_command(#windows..'wincmd '..MOVE_TO..' | vsplit '..bufname)
    elseif open_type == 'tabnew' then
        api.nvim_command('tabnew '..bufname)
    end
end

local function open_file(open_type)
    local tree_index = api.nvim_win_get_cursor(0)[1]
    local tree = get_tree()
    local node = tree[tree_index]

    if node.name == '..' then
        api.nvim_command('cd ..')

        local new_path = get_cwd()
        if new_path ~= '/' then
            new_path = new_path .. '/'
        end

        set_root_path(new_path)
        force_refresh_git()
        init_tree(new_path)
        update_view()

    elseif open_type == 'chdir' then
        if node.dir == false or check_dir_access(node.path .. node.name) == false then return end

        api.nvim_command('cd ' .. node.path .. node.name)
        local new_path = get_cwd() .. '/'
        set_root_path(new_path)
        force_refresh_git()
        init_tree(new_path)
        update_view()

    elseif node.link == true then
        local link_to_dir = is_dir(node.linkto)
        if link_to_dir == true and check_dir_access(node.linkto) == false then return end

        if link_to_dir == true then
            api.nvim_command('cd ' .. node.linkto)
            local new_path = get_cwd() .. '/'
            set_root_path(new_path)
            force_refresh_git()
            init_tree(new_path)
            update_view()
        else
            create_new_buf(open_type, node.link_to);
        end

    elseif node.dir == true then
        if check_dir_access(node.path .. node.name) == false then return end
        open_dir(tree_index)
        update_view(true)
    else
        create_new_buf(open_type, node.path .. node.name);
    end
end

local function edit_file(edit_type)
    local tree = get_tree()
    local tree_index = api.nvim_win_get_cursor(0)[1]
    local node = tree[tree_index]

    if edit_type == 'create' then
        if node.dir == true then
            create_file(node.path .. node.name .. '/')
        else
            create_file(node.path)
        end
    elseif edit_type == 'remove' then
        remove_file(node.name, node.path)
    elseif edit_type == 'rename' then
        rename_file(node.name, node.path)
    end
end

local function refresh()
    if refresh_git() == true then
        refresh_tree()
        update_view()
    end
end

local function check_windows_and_close()
    local wins = api.nvim_list_wins()

    if #wins == 1 and is_win_open() then
        api.nvim_command('q!')
    end
end

local function check_buffer_and_open()
    local bufname = api.nvim_buf_get_name(0)
    if bufname == '' then
        toggle()
    elseif is_dir(bufname) then
        api.nvim_command('cd ' .. bufname)

        local new_path = get_cwd()
        if new_path ~= '/' then
            new_path = new_path .. '/'
        end
        set_root_path(new_path)
        init_tree()

        toggle()
    end
end

local function find()
    local line = find_file(api.nvim_buf_get_name(0))
    if not line then return end

    update_view()

    local win = get_win()
    if win then 
        api.nvim_win_set_cursor(win, { line, 0 })
    end

end

return {
    toggle = toggle;
    open_file = open_file;
    edit_file = edit_file;
    refresh = refresh;
    check_windows_and_close = check_windows_and_close;
    check_buffer_and_open = check_buffer_and_open;
    replace_tree = replace_tree;
    find = find;
}