fix(diagnostics): custom format for code. Fixes #508

folke · Jun 14, 2024 · ada78fa · ada78fa
1 parent e0b35a2
commit ada78fa
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/lua/trouble/config/highlights.lua b/lua/trouble/config/highlights.lua
@@ -15,6 +15,7 @@ M.colors = {
   Directory         = "Directory",
   IconDirectory     = "Special",
   Source            = "Comment",
+  Code              = "Special",
   Pos               = "LineNr",
   Count             = "TabLineSel",
 

diff --git a/lua/trouble/format.lua b/lua/trouble/format.lua
@@ -49,6 +49,15 @@ M.formatters = {
       text = "[" .. ctx.item.pos[1] .. ", " .. (ctx.item.pos[2] + 1) .. "]",
     }
   end,
+  code = function(ctx)
+    if not ctx.item.code then
+      return
+    end
+    return {
+      text = "(" .. ctx.item.code .. ")",
+      hl = "TroubleCode",
+    }
+  end,
   severity = function(ctx)
     local severity = ctx.item.severity or vim.diagnostic.severity.ERROR
     local name = vim.diagnostic.severity[severity] or "OTHER"

diff --git a/lua/trouble/sources/diagnostics.lua b/lua/trouble/sources/diagnostics.lua
@@ -26,7 +26,7 @@ M.config = {
         { "filename", format = "{file_icon} {basename} {count}" },
       },
       sort = { "severity", "filename", "pos", "message" },
-      format = "{severity_icon} {message:md} {item.source} ({code}) {pos}",
+      format = "{severity_icon} {message:md} {item.source} {code} {pos}",
       -- filter = {
       -- ["not"] = {
       --   any = {