feat: make signs configurable

folke · Apr 21, 2021 · ff9fd51 · ff9fd51
1 parent 83ac057
commit ff9fd51
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/lua/trouble/config.lua b/lua/trouble/config.lua
@@ -19,7 +19,9 @@ local defaults = {
     },
     auto_open = false,
     auto_close = true,
-    auto_preview = false
+    auto_preview = false,
+    signs = {error = "", warning = "", hint = "", information = ""},
+    use_lsp_diagnostic_signs = false
 }
 
 ---@type Options

diff --git a/lua/trouble/lsp.lua b/lua/trouble/lsp.lua
@@ -83,7 +83,7 @@ function M.get_signs()
                                 .text)
         end)
         if not status then sign = v:sub(1, 1) end
-        signs[v] = sign
+        signs[string.lower(v)] = sign
     end
     return signs
 end

diff --git a/lua/trouble/renderer.lua b/lua/trouble/renderer.lua
@@ -16,6 +16,14 @@ local function get_icon(file)
     return icons.get_icon(fname, ext, {default = true})
 end
 
+local function update_signs()
+    if config.options.use_lsp_diagnostic_signs then
+        signs = lsp.get_signs()
+    else
+        signs = config.options.signs
+    end
+end
+
 ---@param view View
 function renderer.render(view, opts)
     opts = opts or {}
@@ -32,7 +40,7 @@ function renderer.render(view, opts)
     end
 
     -- Update lsp signs
-    signs = lsp.get_signs()
+    update_signs()
 
     local text = Text:new()
     view.items = {}
@@ -85,7 +93,7 @@ function renderer.render_diagnostics(view, text, items)
     for _, diag in ipairs(items) do
         view.items[text.lineNr] = diag
 
-        local sign = signs[diag.type]
+        local sign = signs[string.lower(diag.type)]
         if not sign then sign = diag.type end
 
         text:render("    " .. sign .. "  ", "Sign" .. diag.type)