summaryrefslogtreecommitdiff
path: root/lua/lvim/lsp/handlers.lua
blob: 45f73e9193aed290c8d27c799e6a14196c86050d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
-- Set Default Prefix.
-- Note: You can set a prefix per lsp server in the lv-globals.lua file
local M = {}

function M.setup()
  local config = { -- your config
    virtual_text = lvim.lsp.diagnostics.virtual_text,
    signs = lvim.lsp.diagnostics.signs,
    underline = lvim.lsp.diagnostics.underline,
    update_in_insert = lvim.lsp.diagnostics.update_in_insert,
    severity_sort = lvim.lsp.diagnostics.severity_sort,
    float = lvim.lsp.diagnostics.float,
  }
  if vim.fn.has "nvim-0.6" == 1 then
    vim.diagnostic.config(config)
  else
    vim.lsp.handlers["textDocument/publishDiagnostics"] = function(_, _, params, client_id, _)
      local uri = params.uri
      local bufnr = vim.uri_to_bufnr(uri)
      if not bufnr then
        return
      end

      local diagnostics = params.diagnostics
      vim.lsp.diagnostic.save(diagnostics, bufnr, client_id)
      if not vim.api.nvim_buf_is_loaded(bufnr) then
        return
      end
      vim.lsp.diagnostic.display(diagnostics, bufnr, client_id, config)
    end

    vim.lsp.handlers["textDocument/hover"] = vim.lsp.with(vim.lsp.handlers.hover, {
      border = lvim.lsp.popup_border,
    })

    vim.lsp.handlers["textDocument/signatureHelp"] = vim.lsp.with(vim.lsp.handlers.signature_help, {
      border = lvim.lsp.popup_border,
    })
  end
end

function M.show_line_diagnostics()
  if vim.fn.has "nvim-0.6" == 1 then
    return vim.diagnostic.open_float(0, { scope = "line" })
  end

  local function split_by_chunk(text, chunkSize)
    local s = {}
    for i = 1, #text, chunkSize do
      s[#s + 1] = text:sub(i, i + chunkSize - 1)
    end
    return s
  end
  local diagnostics = vim.lsp.diagnostic.get_line_diagnostics()
  local severity_highlight = {
    "LspDiagnosticsFloatingError",
    "LspDiagnosticsFloatingWarning",
    "LspDiagnosticsFloatingInformation",
    "LspDiagnosticsFloatingHint",
  }
  local ok, vim_diag = pcall(require, "vim.diagnostic")
  if ok then
    local buf_id = vim.api.nvim_win_get_buf(0)
    local win_id = vim.api.nvim_get_current_win()
    local cursor_position = vim.api.nvim_win_get_cursor(win_id)
    severity_highlight = {
      "DiagnosticFloatingError",
      "DiagnosticFloatingWarn",
      "DiagnosticFloatingInfo",
      "DiagnosticFloatingHint",
    }
    diagnostics = vim_diag.get(buf_id, { lnum = cursor_position[1] - 1 })
  end
  local lines = {}
  local max_width = vim.fn.winwidth(0) - 5
  local height = #diagnostics
  local width = 0
  local opts = {}
  local close_events = { "CursorMoved", "CursorMovedI", "BufHidden", "InsertCharPre" }
  if height == 0 then
    return
  end
  local bufnr = vim.api.nvim_create_buf(false, true)
  local diag_message
  table.sort(diagnostics, function(a, b)
    return a.severity < b.severity
  end)

  local hash = {}
  local diagnostics_no_dupes = {}
  for _, v in ipairs(diagnostics) do
    if not hash[v["message"]] then
      diagnostics_no_dupes[#diagnostics_no_dupes + 1] = v -- you could print here instead of saving to result table if you wanted
      hash[v["message"]] = true
    end
  end
  -- print(vim.inspect(diagnostics_no_dupes))

  for i, diagnostic in ipairs(diagnostics_no_dupes) do
    local source = diagnostic.source
    diag_message = diagnostic.message:gsub("[\n\r]", " ")
    if source then
      if string.find(source, "/") then
        source = string.sub(diagnostic.source, string.find(diagnostic.source, "([%w-_]+)$"))
      end
      diag_message = string.format("%d. %s: %s", i, source, diag_message)
    else
      diag_message = string.format("%d. %s", i, diag_message)
    end
    if diagnostic.code then
      diag_message = string.format("%s [%s]", diag_message, diagnostic.code)
    end
    local msgs = split_by_chunk(diag_message, max_width)
    for _, diag in ipairs(msgs) do
      table.insert(lines, { message = diag, severity = diagnostic.severity })
      width = math.max(diag:len(), width)
    end
  end
  height = #lines
  opts = vim.lsp.util.make_floating_popup_options(width, height, opts)
  opts["style"] = "minimal"
  opts["border"] = "rounded"
  opts["focusable"] = true

  vim.api.nvim_buf_set_option(bufnr, "bufhidden", "wipe")
  local winnr = vim.api.nvim_open_win(bufnr, false, opts)
  vim.api.nvim_win_set_option(winnr, "winblend", 0)
  vim.api.nvim_buf_set_var(bufnr, "lsp_floating_window", winnr)
  for i, diag in ipairs(lines) do
    vim.api.nvim_buf_set_lines(bufnr, i - 1, i - 1, 0, { diag.message })
    vim.api.nvim_buf_add_highlight(bufnr, -1, severity_highlight[diag.severity], i - 1, 0, diag.message:len())
  end

  vim.api.nvim_command(
    "autocmd QuitPre <buffer> ++nested ++once lua pcall(vim.api.nvim_win_close, " .. winnr .. ", true)"
  )
  vim.lsp.util.close_preview_autocmd(close_events, winnr)
end

return M