From db7615578b43586ef6526c56ba397f86b13fbed9 Mon Sep 17 00:00:00 2001
From: fdschmidt93 <39233597+fdschmidt93@users.noreply.github.com>
Date: Wed, 3 Mar 2021 22:33:03 +0100
Subject: feat: lsp diagnostics and prefiltering by entry attribute (#543)

prefiltering for lsp_document_symbols and lsp_workspace_symbols.
example: type `:var:` to show all vars
---
 lua/telescope/make_entry.lua | 78 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)

(limited to 'lua/telescope/make_entry.lua')

diff --git a/lua/telescope/make_entry.lua b/lua/telescope/make_entry.lua
index 8c793d7..2b27328 100644
--- a/lua/telescope/make_entry.lua
+++ b/lua/telescope/make_entry.lua
@@ -27,6 +27,13 @@ local lsp_type_highlight = {
   ["Variable"] = "TelescopeResultsVariable",
 }
 
+local lsp_type_diagnostic = {
+  [1] = "Error",
+  [2] = "Warning",
+  [3] = "Information",
+  [4] = "Hint"
+}
+
 local make_entry = {}
 
 do
@@ -847,6 +854,77 @@ function make_entry.gen_from_ctags(opts)
   end
 end
 
+function make_entry.gen_from_lsp_diagnostics(opts)
+  opts = opts or {}
+  opts.tail_path = utils.get_default(opts.tail_path, true)
+
+  local signs = {}
+  for _, v in pairs(lsp_type_diagnostic) do
+    signs[v] = vim.trim(vim.fn.sign_getdefined("LspDiagnosticsSign" .. v)[1].text)
+  end
+  -- expose line width for longer msg if opts.hide_filename
+  local line_width = utils.get_default(opts.line_width, 48)
+  local displayer = entry_display.create {
+    separator = "▏",
+    items = {
+      { width = 11 },
+      { width = line_width },
+      { remaining = true }
+    }
+  }
+
+  local make_display = function(entry)
+    local filename
+    if not opts.hide_filename then
+      filename = entry.filename
+      if opts.tail_path then
+        filename = utils.path_tail(filename)
+      elseif opts.shorten_path then
+        filename = utils.path_shorten(filename)
+      end
+    end
+
+    -- add styling of entries
+    local pos = string.format("%3d:%2d", entry.lnum, entry.col)
+    local line_info = {
+      string.format("%s %s", signs[entry.type], pos),
+      string.format("LspDiagnosticsDefault%s", entry.type)
+    }
+    -- remove line break to avoid display issues
+    local text = string.format("%-" .. line_width .."s", entry.text:gsub(".* | ", ""):gsub("[\n]", ""))
+
+    return displayer {
+      line_info,
+      text,
+      filename,
+    }
+  end
+
+  return function(entry)
+    local filename = entry.filename or vim.api.nvim_buf_get_name(entry.bufnr)
+
+    return {
+      valid = true,
+
+      value = entry,
+      ordinal = (
+        not opts.ignore_filename and filename
+        or ''
+        ) .. ' ' .. entry.text,
+      display = make_display,
+      bufnr = entry.bufnr,
+      filename = filename,
+      type = entry.type,
+      lnum = entry.lnum,
+      col = entry.col,
+      text = entry.text,
+      start = entry.start,
+      finish = entry.finish,
+    }
+  end
+end
+
+
 function make_entry.gen_from_autocommands(_)
   local displayer = entry_display.create {
     separator = "▏",
-- 
cgit v1.2.3