handlers.lua 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. -- Set Default Prefix.
  2. -- Note: You can set a prefix per lsp server in the lv-globals.lua file
  3. local M = {}
  4. function M.setup()
  5. local config = { -- your config
  6. virtual_text = lvim.lsp.diagnostics.virtual_text,
  7. signs = lvim.lsp.diagnostics.signs,
  8. underline = lvim.lsp.diagnostics.underline,
  9. update_in_insert = lvim.lsp.diagnostics.update_in_insert,
  10. severity_sort = lvim.lsp.diagnostics.severity_sort,
  11. }
  12. if vim.fn.has "nvim-0.5.1" > 0 then
  13. vim.lsp.handlers["textDocument/publishDiagnostics"] = function(_, result, ctx, _)
  14. local uri = result.uri
  15. local bufnr = vim.uri_to_bufnr(uri)
  16. if not bufnr then
  17. return
  18. end
  19. local diagnostics = result.diagnostics
  20. local ok, vim_diag = pcall(require, "vim.diagnostic")
  21. if ok then
  22. -- FIX: why can't we just use vim.diagnostic.get(buf_id)?
  23. config.signs = true
  24. for i, diagnostic in ipairs(diagnostics) do
  25. local rng = diagnostic.range
  26. diagnostics[i].lnum = rng["start"].line
  27. diagnostics[i].end_lnum = rng["end"].line
  28. diagnostics[i].col = rng["start"].character
  29. diagnostics[i].end_col = rng["end"].character
  30. end
  31. local namespace = vim.lsp.diagnostic.get_namespace(ctx.client_id)
  32. vim_diag.set(namespace, bufnr, diagnostics, config)
  33. if not vim.api.nvim_buf_is_loaded(bufnr) then
  34. return
  35. end
  36. local sign_names = {
  37. "DiagnosticSignError",
  38. "DiagnosticSignWarn",
  39. "DiagnosticSignInfo",
  40. "DiagnosticSignHint",
  41. }
  42. for i, sign in ipairs(lvim.lsp.diagnostics.signs.values) do
  43. vim.fn.sign_define(sign_names[i], { texthl = sign_names[i], text = sign.text, numhl = "" })
  44. end
  45. vim_diag.show(namespace, bufnr, diagnostics, config)
  46. else
  47. vim.lsp.diagnostic.save(diagnostics, bufnr, ctx.client_id)
  48. if not vim.api.nvim_buf_is_loaded(bufnr) then
  49. return
  50. end
  51. vim.lsp.diagnostic.display(diagnostics, bufnr, ctx.client_id, config)
  52. end
  53. end
  54. else
  55. vim.lsp.handlers["textDocument/publishDiagnostics"] = function(_, _, params, client_id, _)
  56. local uri = params.uri
  57. local bufnr = vim.uri_to_bufnr(uri)
  58. if not bufnr then
  59. return
  60. end
  61. local diagnostics = params.diagnostics
  62. vim.lsp.diagnostic.save(diagnostics, bufnr, client_id)
  63. if not vim.api.nvim_buf_is_loaded(bufnr) then
  64. return
  65. end
  66. vim.lsp.diagnostic.display(diagnostics, bufnr, client_id, config)
  67. end
  68. end
  69. vim.lsp.handlers["textDocument/hover"] = vim.lsp.with(vim.lsp.handlers.hover, {
  70. border = lvim.lsp.popup_border,
  71. })
  72. vim.lsp.handlers["textDocument/signatureHelp"] = vim.lsp.with(vim.lsp.handlers.signature_help, {
  73. border = lvim.lsp.popup_border,
  74. })
  75. end
  76. function M.show_line_diagnostics()
  77. local diagnostics = vim.lsp.diagnostic.get_line_diagnostics()
  78. local severity_highlight = {
  79. "LspDiagnosticsFloatingError",
  80. "LspDiagnosticsFloatingWarning",
  81. "LspDiagnosticsFloatingInformation",
  82. "LspDiagnosticsFloatingHint",
  83. }
  84. local ok, vim_diag = pcall(require, "vim.diagnostic")
  85. if ok then
  86. local buf_id = vim.api.nvim_win_get_buf(0)
  87. local win_id = vim.api.nvim_get_current_win()
  88. local cursor_position = vim.api.nvim_win_get_cursor(win_id)
  89. severity_highlight = {
  90. "DiagnosticFloatingError",
  91. "DiagnosticFloatingWarn",
  92. "DiagnosticFloatingInfo",
  93. "DiagnosticFloatingHint",
  94. }
  95. diagnostics = vim_diag.get(buf_id, { lnum = cursor_position[1] - 1 })
  96. end
  97. local diags = vim.deepcopy(diagnostics)
  98. local height = #diagnostics
  99. local width = 0
  100. local opts = {}
  101. local close_events = { "CursorMoved", "CursorMovedI", "BufHidden", "InsertCharPre" }
  102. if height == 0 then
  103. return
  104. end
  105. local bufnr = vim.api.nvim_create_buf(false, true)
  106. for i, diagnostic in ipairs(diagnostics) do
  107. local source = diagnostic.source
  108. if source then
  109. if string.find(source, "/") then
  110. source = string.sub(diagnostic.source, string.find(diagnostic.source, "([%w-_]+)$"))
  111. end
  112. diags[i].message = string.format("%s: %s", source, diagnostic.message)
  113. else
  114. diags[i].message = string.format("%s", diagnostic.message)
  115. end
  116. if diagnostic.code then
  117. diags[i].message = string.format("%s [%s]", diags[i].message, diagnostic.code)
  118. end
  119. if diags[i].message:len() > width then
  120. width = string.len(diags[i].message)
  121. end
  122. end
  123. opts = vim.lsp.util.make_floating_popup_options(width, height, opts)
  124. opts["style"] = "minimal"
  125. opts["border"] = "rounded"
  126. vim.api.nvim_buf_set_option(bufnr, "bufhidden", "wipe")
  127. local winnr = vim.api.nvim_open_win(bufnr, false, opts)
  128. vim.api.nvim_win_set_option(winnr, "winblend", 0)
  129. vim.api.nvim_buf_set_var(bufnr, "lsp_floating_window", winnr)
  130. for i, diag in ipairs(diags) do
  131. local message = diag.message:gsub("[\n\r]", " ")
  132. vim.api.nvim_buf_set_lines(bufnr, i - 1, i - 1, 0, { message })
  133. vim.api.nvim_buf_add_highlight(bufnr, -1, severity_highlight[diag.severity], i - 1, 0, diag.message:len())
  134. end
  135. vim.api.nvim_command(
  136. "autocmd QuitPre <buffer> ++nested ++once lua pcall(vim.api.nvim_win_close, " .. winnr .. ", true)"
  137. )
  138. vim.lsp.util.close_preview_autocmd(close_events, winnr)
  139. end
  140. return M