add git integration

lua/lib/format.lua

@@ -108,6 +108,15 @@ local HIGHLIGHT_GROUPS = {
     ['^.*%.html?$'] = 'HtmlFile';
 }
 
+local function highlight_git(highlight, node, line, col_start)
+    if node.git ~= '' then
+        highlight('LuaTreeGit'..node.git, line, col_start, -1)
+        return true
+    end
+
+    return false
+end
+
 local function highlight_line(buffer)
     local function highlight(group, line, from, to)
         vim.api.nvim_buf_add_highlight(buffer, -1, group, line, from, to)
@@ -117,10 +126,14 @@ local function highlight_line(buffer)
         if node.dir then
             if node.name ~= '..' then
                 highlight('LuaTreeFolderIcon', line, 0, text_start)
-                highlight('LuaTreeFolderName', line, text_start, -1)
+
+                if highlight_git(highlight, node, line, text_start) == false then
+                    highlight('LuaTreeFolderName', line, text_start, -1)
+                end
             else
                 highlight('LuaTreeFolderName', line, 0, -1)
             end
+            return
         elseif is_special(node.name) == true then
             highlight('LuaTreeSpecialFile', line, 0, -1)
         elseif is_executable(node.path .. node.name) then
@@ -131,10 +144,12 @@ local function highlight_line(buffer)
             for k, v in pairs(HIGHLIGHT_GROUPS) do
                 if string.match(node.name, k) ~= nil then
                     highlight('LuaTree' .. v, line, 0, text_start)
-                    break
+                    highlight_git(highlight, node, line, text_start)
+                    return
                 end
             end
         end
+        highlight_git(highlight, node, line, text_start - 4)
     end
 end
 

lua/lib/git.lua

@@ -0,0 +1,44 @@
+local function system(v) return vim.api.nvim_call_function('system', { v }) end
+
+local function is_git_repo()
+    local is_git = system('git rev-parse')
+    return string.match(is_git, 'fatal') == nil
+end
+
+local GIT_CMD = 'git status --porcelain=v1 '
+
+local function is_folder_dirty(path)
+    local ret = system(GIT_CMD..path)
+    return ret ~= nil and ret ~= ''
+end
+
+local function create_git_checker(pattern)
+    return function(path)
+        local ret = system(GIT_CMD..path)
+        return string.match(ret, pattern) ~= nil
+    end
+end
+
+local is_modified = create_git_checker('^ ?MM?')
+local is_staged = create_git_checker('^ ?A')
+local is_unmerged = create_git_checker('^ ?UU')
+local is_untracked = create_git_checker('^%?%?')
+
+local function get_git_attr(path, is_dir)
+    if is_git_repo() == false then return '' end
+    if is_dir then
+        if is_folder_dirty(path) == true then return 'Dirty' end
+    else
+        if is_modified(path) then return 'Modified'
+        elseif is_staged(path) then return 'Staged'
+        elseif is_unmerged(path) then return 'Unmerged'
+        elseif is_untracked(path) then return 'Untracked'
+        end
+    end
+
+    return ''
+end
+
+return {
+    get_git_attr = get_git_attr;
+}

lua/lib/state.lua

@@ -1,6 +1,7 @@
 local api = vim.api
 local function syslist(v) return api.nvim_call_function('systemlist', { v }) end
 local get_root_path = require 'lib/conf'.get_root_path
+local get_git_attr = require 'lib/git'.get_git_attr
 
 local Tree = {}
 
@@ -44,13 +45,16 @@ local function create_nodes(path, depth, dirs)
     if not string.find(path, '^.*/$') then path = path .. '/' end
 
     for i, name in pairs(dirs) do
+        local full_path = path..name
+        local dir = is_dir(full_path)
         tree[i] = {
             path = path,
             name = name,
             depth = depth,
-            dir = is_dir(path .. name),
-            open = false, -- only relevant when its a dir
-            icon = true
+            dir = dir,
+            open = false,
+            icon = true,
+            git = get_git_attr(full_path, dir)
         }
     end
 
@@ -67,7 +71,8 @@ local function init_tree()
             depth = 0,
             dir = true,
             open = false,
-            icon = false
+            icon = false,
+            git = ''
         })
     end
 end
@@ -129,4 +134,5 @@ return {
     get_tree = get_tree;
     refresh_tree = refresh_tree;
     open_dir = open_dir;
+    check_dir_access = check_dir_access;
 }

lua/lib/winutils.lua

@@ -66,16 +66,14 @@ local function open()
 end
 
 local function close()
-    local BUF_NAME = get_buf_name()
-    local win = get_win(BUF_NAME)
+    local win = get_win()
     if not win then return end
 
     api.nvim_win_close(win, true)
 end
 
 local function update_view(update_cursor)
-    local BUF_NAME = get_buf_name()
-    local buf = get_buf(BUF_NAME);
+    local buf = get_buf();
     if not buf then return end
 
     local cursor = api.nvim_win_get_cursor(0)