neovim

diagflow.lua (4646B)

---

 -- {
 --     bufnr = 1,
 --     code = "undefined-global",
 --     col = 24,
 --     end_col = 30,
 --     end_lnum = 63,
 --     lnum = 63,
 --     message = "Undefined global `config`.",
 --     namespace = 33,
 --     severity = 2,
 --     source = "Lua Diagnostics.",
 --     user_data = {
 --       lsp = {
 --         code = "undefined-global"
 --       }
 --     }
 --   }
 local enable = true
 local border_chars = {
   top_left = '┌',
   top_right = '┐',
   bottom_left = '└',
   bottom_right = '┘',
   horizontal = '─',
   vertical = '│',
 }
 local ns = vim.api.nvim_create_namespace 'DiagflowHighlight'
 local severity_hl_map = {
   [vim.diagnostic.severity.ERROR] = 'DiagnosticError',
   [vim.diagnostic.severity.WARN] = 'DiagnosticWarn',
   [vim.diagnostic.severity.INFO] = 'DiagnosticInfo',
   [vim.diagnostic.severity.HINT] = 'DiagnosticHint',
 }
 
 local function get_padding(bufnr, winnr)
   if not pcall(require, 'treesitter-context.context') then
     return 0
   end
 
   local context_lines = #(require('treesitter-context.context').get(bufnr, winnr) or {})
   if context_lines == 0 then
     return 0
   end
 
   return context_lines + 1 -- +1 for the underline
 end
 
 local function create_boxed_text(text_lines)
   if #text_lines == 0 then
     return text_lines
   end
 
   local max_length = 0
   for _, line in ipairs(text_lines) do
     max_length = math.max(max_length, #line)
   end
 
   local top_border = border_chars.top_left .. string.rep(border_chars.horizontal, max_length) .. border_chars
       .top_right
   local bottom_border = border_chars.bottom_left .. string.rep(border_chars.horizontal, max_length) ..
       border_chars.bottom_right
   local boxed_lines = { top_border }
 
   for _, line in ipairs(text_lines) do
     local padded_line = line .. string.rep(' ', max_length - #line)
     table.insert(boxed_lines, border_chars.vertical .. padded_line .. border_chars.vertical)
   end
 
   table.insert(boxed_lines, bottom_border)
   return boxed_lines
 end
 
 local function clear(bufnr)
   bufnr = bufnr or vim.api.nvim_get_current_buf()
   vim.api.nvim_buf_clear_namespace(bufnr, ns, 0, -1)
 end
 
 local function render(diagnostics, bufnr)
   clear(bufnr)
 
   if not enable then
     return
   end
 
   bufnr = bufnr or vim.api.nvim_get_current_buf()
 
   if not vim.diagnostic.is_enabled(bufnr) then
     return
   end
 
   local winnr
   for _, win in ipairs(vim.api.nvim_list_wins()) do
     if vim.api.nvim_win_get_buf(win) == bufnr then
       winnr = win
       break
     end
   end
 
   if not winnr then
     return
   end
 
   local cursor_lnum = vim.fn.getcurpos(winnr)[2]
   local padding = get_padding(bufnr, winnr)
   local line_offset = vim.fn.getwininfo(winnr)[1].topline - 1 + padding
   local skipped_some = false
   for _, diagnostic in ipairs(diagnostics) do
     local message_lines = vim.split(diagnostic.message, '\n')
 
     -- Enable/Disable boxed text
     -- message_lines = create_boxed_text(message_lines)
 
     if ((line_offset + #message_lines) >= cursor_lnum) then
       skipped_some = true
       break
     end
 
     for _, message in ipairs(message_lines) do
       local hl_group = severity_hl_map[diagnostic.severity]
       vim.api.nvim_buf_set_extmark(bufnr, ns, line_offset, 0, {
         virt_text = { { message, hl_group } },
         virt_text_pos = 'right_align',
         virt_text_hide = true,
         strict = false,
         priority = vim.hl.priorities.user,
       })
       line_offset = line_offset + 1
     end
   end
 
   if skipped_some then
     vim.api.nvim_buf_set_extmark(bufnr, ns, line_offset, 0, {
       virt_text = { { '...', 'DiagnosticWarn' } },
       virt_text_pos = 'right_align',
       virt_text_hide = true,
       strict = false,
       priority = vim.hl.priorities.user,
     })
   end
 end
 
 local function reload()
   if not enable then
     clear()
     return
   end
 
   local cursor_lnum = vim.api.nvim_win_get_cursor(0)[1] - 1
   local diagnostics = vim.diagnostic.get(0, {
     lnum = cursor_lnum })
   render(diagnostics)
 end
 
 local function toggle(force)
   enable = force or not enable
 
   if enable then
     reload()
   else
     clear()
   end
 end
 
 -- vim.api.nvim_create_autocmd('InsertEnter', {
 --   callback = function()
 --     toggle(false)
 --   end,
 -- })
 -- vim.api.nvim_create_autocmd('InsertLeave', {
 --   callback = function()
 --     toggle(true)
 --   end,
 -- })
 -- vim.api.nvim_create_autocmd({ 'DiagnosticChanged', 'CursorMoved', 'CursorMovedI', 'WinScrolled' }, {
 --   callback = function()
 --     reload()
 --   end,
 -- })
 -- vim.api.nvim_create_autocmd({ 'BufLeave' }, {
 --   callback = function()
 --     clear()
 --   end,
 -- })
 
 vim.api.nvim_create_user_command('Diagflow', function()
   toggle()
 end, { desc = 'Diagflow Toggle' })