Merge pull request #43 from kyazdani42/feat/indent-markers

add indent markers

Author: kyazdani42

README.md

@@ -30,6 +30,7 @@ let g:lua_tree_ignore = [ '.git', 'node_modules', '.cache' ] "empty by default
 let g:lua_tree_auto_open = 1 "0 by default, opens the tree when typing `vim $DIR` or `vim`
 let g:lua_tree_auto_close = 1 "0 by default, closes the tree when it's the last window
 let g:lua_tree_follow = 1 "0 by default, this option allows the cursor to be updated when entering a buffer
+let g:lua_tree_indent_markers = 1 "0 by default, this option shows indent markers when folders are open
 let g:lua_tree_show_icons = {
     \ 'git': 1,
     \ 'folders': 0,
@@ -113,6 +114,7 @@ This plugin is very fast because it uses the `libuv` `scandir` and `scandir_next
 - Change directory with `.`
 - Add / Rename / delete files
 - Git integration
+- Indent markers
 - Mouse support
 - It's fast
 

doc/nvim-tree-lua.txt

@@ -121,6 +121,9 @@ Can be `0` or `1`. When `1`, will disable all keybindings by the plugin.
 |g:lua_tree_bindings| as well as default bindings will not take effect.
 Default is 0
 
+|g:lua_tree_indent_markers|                       *g:lua_tree_indent_markers*
+
+Can be `0` or `1`. When `1`, will display indent markers when folders are open
 
 ==============================================================================
 INFORMATIONS				        *nvim-tree-info*
@@ -209,6 +212,7 @@ LuaTreeExecFile
 LuaTreeSpecialFile 
 LuaTreeImageFile 
 LuaTreeMarkdownFile 
+LuaTreeIndentMarker
 
 LuaTreeLicenseIcon 
 LuaTreeYamlIcon 

doc/tags

@@ -9,6 +9,7 @@ g:lua_tree_bindings	nvim-tree-lua.txt	/*g:lua_tree_bindings*
 g:lua_tree_follow	nvim-tree-lua.txt	/*g:lua_tree_follow*
 g:lua_tree_icons	nvim-tree-lua.txt	/*g:lua_tree_icons*
 g:lua_tree_ignore	nvim-tree-lua.txt	/*g:lua_tree_ignore*
+g:lua_tree_indent_markers	nvim-tree-lua.txt	/*g:lua_tree_indent_markers*
 g:lua_tree_show_icons	nvim-tree-lua.txt	/*g:lua_tree_show_icons*
 g:lua_tree_side	nvim-tree-lua.txt	/*g:lua_tree_side*
 g:lua_tree_size	nvim-tree-lua.txt	/*g:lua_tree_size*

lua/lib/colors.lua

@@ -29,6 +29,7 @@ local function get_hl_groups()
   local colors = get_colors()
 
   return {
+    IndentMarker = { fg = '#90a4ae' },
     Symlink = { gui = 'bold', fg = colors.cyan },
     FolderIcon = { fg = '#90a4ae' },
 

lua/lib/renderer.lua

@@ -87,6 +87,26 @@ if icon_state.show_git_icon then
   end
 end
 
+local get_padding = function(depth)
+  return string.rep(' ', depth)
+end
+
+if vim.g.lua_tree_indent_markers == 1 then
+  get_padding = function(depth, idx, tree, node)
+    local padding = ""
+    if depth ~= 0 then
+      for i=1,depth/2 do
+        if idx == #tree.entries and i == depth/2 then
+          padding = padding..'└ '
+        else
+          padding = padding..'│ '
+        end
+      end
+    end
+    return padding
+  end
+end
+
 local picture = {
   jpg = true,
   jpeg = true,
@@ -108,12 +128,16 @@ local function update_draw_data(tree, depth)
     index = 1
   end
 
-  for _, node in ipairs(tree.entries) do
-    local padding = string.rep(" ", depth)
+  for idx, node in ipairs(tree.entries) do
+    local padding = get_padding(depth, idx, tree, node)
+    local offset = string.len(padding)
+    if depth > 0 then
+      table.insert(hl, { 'LuaTreeIndentMarker', index, 0, offset })
+    end
     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)
+      local git_icon = get_git_icons(node, index, offset+#node.name, #icon+1)
+      set_folder_hl(index, offset, #icon, #node.name)
       index = index + 1
       if node.open then
         table.insert(lines, padding..icon..node.name.." "..git_icon)
@@ -122,7 +146,7 @@ local function update_draw_data(tree, depth)
         table.insert(lines, padding..icon..node.name.." "..git_icon)
       end
     elseif node.link_to then
-      table.insert(hl, { 'LuaTreeSymlink', index, depth, -1 })
+      table.insert(hl, { 'LuaTreeSymlink', index, offset, -1 })
       table.insert(lines, padding..node.name.." ➛ "..node.link_to)
       index = index + 1
 
@@ -131,17 +155,17 @@ local function update_draw_data(tree, depth)
       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})
+        git_icons = get_git_icons(node, index, offset, 0)
+        table.insert(hl, {'LuaTreeSpecialFile', index, offset+#git_icons, -1})
       else
-        icon = get_file_icon(node.name, node.extension, index, depth)
-        git_icons = get_git_icons(node, index, depth, #icon)
+        icon = get_file_icon(node.name, node.extension, index, offset)
+        git_icons = get_git_icons(node, index, offset, #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 })
+        table.insert(hl, {'LuaTreeExecFile', index, offset+#icon+#git_icons, -1 })
       elseif picture[node.extension] then
-        table.insert(hl, {'LuaTreeImageFile', index, depth+#icon+#git_icons, -1 })
+        table.insert(hl, {'LuaTreeImageFile', index, offset+#icon+#git_icons, -1 })
       end
       index = index + 1
     end