Refactor

1 files changed, 21 insertions, 9 deletions

diff --git a/lua/no-clown-fiesta/groups/lsp.lua b/lua/no-clown-fiesta/groups/lsp.lua
index face8d3..553ec2c 100644
--- a/lua/no-clown-fiesta/groups/lsp.lua
+++ b/lua/no-clown-fiesta/groups/lsp.lua
@@ -1,26 +1,38 @@
 local M = {}
 
-function M.highlight(palette)
+function M.highlight(palette, opts)
   return {
     LspReferenceRead = { bg = "#36383F" },
     LspReferenceText = { bg = "#36383F" },
     LspReferenceWrite = { bg = "#36383f" },
-
     DiagnosticError = { fg = palette.error },
     DiagnosticWarn = { fg = palette.warning },
     DiagnosticInfo = { fg = palette.info },
     DiagnosticHint = { fg = palette.hint },
-
     DiagnosticVirtualTextError = { fg = palette.error },
     DiagnosticVirtualTextWarn = { fg = palette.warning },
     DiagnosticVirtualTextInfo = { fg = palette.info },
     DiagnosticVirtualTextHint = { fg = palette.hint },
-
-    DiagnosticUnderlineError = { undercurl = true, sp = palette.error },
-    DiagnosticUnderlineWarn = { undercurl = true, sp = palette.warning },
-    DiagnosticUnderlineInfo = { undercurl = true, sp = palette.info },
-    DiagnosticUnderlineHint = { undercurl = true, sp = palette.hint },
-
+    DiagnosticUnderlineError = vim.tbl_extend(
+      "force",
+      { sp = palette.error },
+      opts.styles.lsp
+    ),
+    DiagnosticUnderlineWarn = vim.tbl_extend(
+      "force",
+      { sp = palette.warning },
+      opts.styles.lsp
+    ),
+    DiagnosticUnderlineInfo = vim.tbl_extend(
+      "force",
+      { sp = palette.info },
+      opts.styles.lsp
+    ),
+    DiagnosticUnderlineHint = vim.tbl_extend(
+      "force",
+      { sp = palette.hint },
+      opts.styles.lsp
+    ),
     LspSignatureActiveParameter = { bg = palette.alt_bg, bold = true },
     LspCodeLens = { fg = palette.comment },
   }