-- 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 = options.lsp.diagnostics.virtual_text,
    signs = options.lsp.diagnostics.signs,
    underline = options.lsp.diagnostics.underline,
    update_in_insert = options.lsp.diagnostics.update_in_insert,
    severity_sort = options.lsp.diagnostics.severity_sort,
  }
  if vim.fn.has "nvim-0.5.1" > 0 then
    vim.lsp.handlers["textDocument/publishDiagnostics"] = function(_, result, ctx, _)
      local uri = result.uri
      local bufnr = vim.uri_to_bufnr(uri)
      if not bufnr then
        return
      end

      local diagnostics = result.diagnostics
      vim.lsp.diagnostic.save(diagnostics, bufnr, ctx.client_id)
      if not vim.api.nvim_buf_is_loaded(bufnr) then
        return
      end
      vim.lsp.diagnostic.display(diagnostics, bufnr, ctx.client_id, config)
    end
  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
  end

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

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

function M.show_line_diagnostics()
  local diagnostics = vim.lsp.diagnostic.get_line_diagnostics()
  local diags = vim.deepcopy(diagnostics)
  local height = #diagnostics
  local width = 0
  local opts = {}
  local close_events = { "CursorMoved", "CursorMovedI", "BufHidden", "InsertCharPre" }
  local diagnostic_severities = {
    "Error",
    "Warning",
    "Information",
    "Hint",
  }
  if height == 0 then
    return
  end
  local bufnr = vim.api.nvim_create_buf(false, true)

  for i, diagnostic in ipairs(diagnostics) do
    local source = diagnostic.source
    if source then
      if string.find(source, "/") then
        source = string.sub(
          diagnostic.source,
          string.find(diagnostic.source, "([%w-_]+)$")
        )
      end
      diags[i].message = string.format("%s: %s", source, diagnostic.message)
    else
      diags[i].message = string.format("%s", diagnostic.message)
    end

    if diagnostic.code then
      diags[i].message = string.format("%s [%s]", diags[i].message, diagnostic.code)
    end
    if diags[i].message:len() > width then
      width = string.len(diags[i].message)
    end
  end

  opts = vim.lsp.util.make_floating_popup_options(width, height, opts)
  opts["style"] = "minimal"
  opts["border"] = "rounded"

  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(diags) do
    local message = diag.message:gsub("[\n\r]", " ")
    vim.api.nvim_buf_set_lines(bufnr, i - 1, i - 1, 0, { message })
    vim.api.nvim_buf_add_highlight(
      bufnr,
      -1,
      "LspDiagnosticsFloating" .. diagnostic_severities[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