add better lsp diagnostic integration

lua/nvim-tree/diagnostics.lua

@@ -1,24 +1,38 @@
 local a = vim.api
 local utils = require'nvim-tree.utils'
+local view = require'nvim-tree.view'
 local get_diagnostics = vim.lsp.diagnostic.get_all
 
 local M = {}
 
-local function get_severity(diagnostics)
+local function get_highest_severity(diagnostics)
+  local severity = 0
   for _, v in ipairs(diagnostics) do
-    if v.severity > 0 then
-      return v.severity
+    if v.severity > severity then
+      severity = v.severity
     end
   end
-  return 0
+  return severity
 end
 
-local function highlight_node(node, linenr)
-  local buf = require'nvim-tree.view'.View.bufnr
+local sign_names = {
+  { "NvimTreeSignHint", "NvimTreeLspDiagnosticsHint" },
+  { "NvimTreeSignInformation", "NvimTreeLspDiagnosticsInformation" },
+  { "NvimTreeSignWarning", "NvimTreeLspDiagnosticsWarning" },
+  { "NvimTreeSignError", "NvimTreeLspDiagnosticsError" },
+}
+
+for _, v in ipairs(sign_names) do
+  vim.fn.sign_define(v[1], { text="⚠", texthl=v[2]})
+end
+
+local signs = {}
+
+local function add_sign(linenr, severity)
+  local buf = view.View.bufnr
   if not vim.fn.bufexists(buf) or not vim.fn.bufloaded(buf) then return end
-  local line = a.nvim_buf_get_lines(buf, linenr, linenr+1, false)[1]
-  local starts_at = vim.fn.stridx(line, node.name)
-  a.nvim_buf_add_highlight(buf, -1, 'NvimTreeLspDiagnostics', linenr, starts_at, -1)
+  local sign_name = sign_names[severity][1]
+  table.insert(signs, vim.fn.sign_place(1, 'NvimTreeDiagnosticSigns', sign_name, buf, { lnum = linenr+1 }))
 end
 
 function M.update()
@@ -27,22 +41,30 @@ function M.update()
   for buf, diagnostics in pairs(get_diagnostics()) do
     local bufname = a.nvim_buf_get_name(buf)
     if not buffer_severity[bufname] then
-      local severity = get_severity(diagnostics)
+      local severity = get_highest_severity(diagnostics)
       buffer_severity[bufname] = severity
     end
   end
 
-  vim.defer_fn(function()
-    local nodes = require'nvim-tree.lib'.Tree.entries
-    for bufname, severity in pairs(buffer_severity) do
-      if severity > 0 then
-        local node, line = utils.find_node(nodes, function(node)
-          return node.absolute_path == bufname
-        end)
-        if node then highlight_node(node, line) end
-      end
+  local nodes = require'nvim-tree.lib'.Tree.entries
+  if #signs then
+    vim.fn.sign_unplacelist(vim.tbl_map(function(sign)
+      return {
+        buffer = view.View.bufnr,
+        group = "NvimTreeDiagnosticSigns",
+        id = sign
+      }
+    end, signs))
+    signs = {}
+  end
+  for bufname, severity in pairs(buffer_severity) do
+    if severity > 0 then
+      local node, line = utils.find_node(nodes, function(node)
+        return node.absolute_path == bufname
+      end)
+      if node then add_sign(line, severity) end
     end
-  end, 100)
+  end
 end
 
 return M