fix: added suport for vim.diagnostic hl groups

folke · Oct 22, 2021 · d25a8e6 · d25a8e6
1 parent 5039b1c
commit d25a8e6
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/lua/trouble/colors.lua b/lua/trouble/colors.lua
@@ -1,15 +1,9 @@
+local util = require("trouble.util")
+
 local M = {}
 
 local links = {
-  Error = "LspDiagnosticsDefaultError",
-  Warning = "LspDiagnosticsDefaultWarning",
-  Information = "LspDiagnosticsDefaultInformation",
-  Hint = "LspDiagnosticsDefaultHint",
-  SignError = "LspDiagnosticsSignError",
-  SignWarning = "LspDiagnosticsSignWarning",
-  SignInformation = "LspDiagnosticsSignInformation",
-  SignHint = "LspDiagnosticsSignHint",
-  TextError = "LspTroubleText",
+  TextError = "TroubleText",
   TextWarning = "TroubleText",
   TextInformation = "TroubleText",
   TextHint = "TroubleText",
@@ -30,6 +24,11 @@ function M.setup()
   for k, v in pairs(links) do
     vim.api.nvim_command("hi def link Trouble" .. k .. " " .. v)
   end
+
+  for _, severity in pairs(util.severity) do
+    vim.api.nvim_command("hi def link Trouble" .. severity .. " " .. util.get_severity_label(severity))
+    vim.api.nvim_command("hi def link TroubleSign" .. severity .. " " .. util.get_severity_label(severity, "Sign"))
+  end
 end
 
 return M