local M = {}

local icons = require("fschauen.util.icons")

-- Show/navigate warning and errors by default.
M.severity = vim.diagnostic.severity.WARN

-- Go to next/prev diagnostic, but only if next item has a visible virtual text.
-- If we can move, then also center screen at target location.
local conditional_goto = function(condition, move, opts)
  opts = vim.tbl_extend("keep", opts or {}, {
    wrap = false, -- don't wrap around the begin/end of file
    severity = { -- only navigate items with visible virtual text
      min = M.severity,
    },
  })

  if condition(opts) then
    move(opts)
    vim.cmd("normal zz")
  else
    local level = vim.diagnostic.severity[M.severity] or "???"
    local msg = string.format("No more diagnostics [level: %s]", level)
    vim.notify(msg, vim.log.levels.WARN)
  end
end

---Move to the next diagnostic.
---@param opts table\nil: options passed along to `vim.diagnostic.goto_next`.
M.goto_next = function(opts)
  conditional_goto(vim.diagnostic.get_next_pos, vim.diagnostic.goto_next, opts)
end

---Move to the previous diagnostic.
---@param opts table|nil: options passed along to `vim.diagnostic.goto_prev`.
M.goto_prev = function(opts)
  conditional_goto(vim.diagnostic.get_prev_pos, vim.diagnostic.goto_prev, opts)
end

---Show diagnostics in a floating window.
---@param opts table|nil: options passed along to `vim.diagnostic.open_float`.
M.open_float = function(opts)
  vim.diagnostic.open_float(opts)
end

---Toggle diagnostics in the given buffer.
---@param bufnr integer|nil: Buffer number (0 for current buffer, nil for all buffers.
M.toggle = function(bufnr)
  bufnr = bufnr or 0
  if vim.diagnostic.is_disabled(bufnr) then
    vim.diagnostic.enable(bufnr)
  else
    vim.diagnostic.disable(bufnr)
  end
end

---Hide currently displayed diagnostics.
---@param bufnr integer|nil: Buffer number (0 for current buffer, nil for all buffers.
M.hide = function(bufnr)
  vim.diagnostic.hide(nil, bufnr or 0)
end

M.select_virtual_text_severity = function()
  vim.ui.select(
    { "ERROR", "WARN", "INFO", "HINT" },
    { prompt = "Min. severity for virtual text:" },
    function(choice)
      if choice then
        M.severity = vim.diagnostic.severity[choice] or M.severity
        vim.diagnostic.config {
          virtual_text = {
            severity = { min = M.severity },
          },
        }
      end
    end
  )
end

---Customize nvim's diagnostics display.
M.setup = function()
  vim.diagnostic.config {
    underline = false,
    virtual_text = {
      spacing = 6,
      prefix = icons.ui.Circle,
      severity = {
        min = M.severity,
      },
    },
    float = {
      border = "rounded",
    },
    severity_sort = true,
  }

  for type, icon in pairs(icons.diagnostics) do
    local hl = "DiagnosticSign" .. type
    vim.fn.sign_define(hl, { text = icon, texthl = hl, numhl = hl })
  end
end

return M