highlight.lua

local M = {}

---Used to keep track of if LSP reference highlights should be enabled
local highlight_references = false

local highlight_namespace = vim.api.nvim_create_namespace('RefjumpReferenceHighlights')

local reference_hl_name = 'RefjumpReference'

function M.create_fallback_hl_group(fallback_hl)
  local hl = vim.api.nvim_get_hl(0, { name = reference_hl_name })

  if vim.tbl_isempty(hl) then
    vim.api.nvim_set_hl(0, reference_hl_name, { link = fallback_hl })
  end
end

---@deprecated Use `enable()` instead
function M.enable_reference_highlights(references, bufnr)
  local message = 'refjump.nvim: `enable_reference_highlights()` has been renamed to `enable()`'
  vim.notify(message, vim.log.levels.WARN)
  M.enable(references, bufnr)
end

function M.enable(references, bufnr)
  for _, ref in ipairs(references) do
    local line = ref.range.start.line
    local start_col = ref.range.start.character
    local end_col = ref.range['end'].character

    vim.hl.range(
      bufnr,
      highlight_namespace,
      reference_hl_name,
      { line, start_col },
      { line, end_col },
      { inclusive = false }
    )
  end

  highlight_references = true
end

---@deprecated Use `disable()` instead
function M.disable_reference_highlights()
  local message = 'refjump.nvim: `disable_reference_highlights()` has been renamed to `disable()`'
  vim.notify(message, vim.log.levels.WARN)
  M.disable()
end

function M.disable()
  if not highlight_references then
    vim.api.nvim_buf_clear_namespace(0, highlight_namespace, 0, -1)
    require('refjump.counter').clear(0)
  else
    highlight_references = false
  end
end

function M.auto_clear_reference_highlights()
  vim.api.nvim_create_autocmd({ 'CursorMoved', 'ModeChanged', 'BufLeave' }, {
    callback = M.disable,
  })
end

function M.clear_on_escape()
  local esc = vim.api.nvim_replace_termcodes('<Esc>', true, false, true)
  local ctrl_c = vim.api.nvim_replace_termcodes('<C-c>', true, false, true)

  vim.on_key(function(key)
    -- Listen for escape or ctrl-c without blocking them
    if key == esc or key == ctrl_c then
      M.disable()
    end
  end, vim.api.nvim_create_namespace('refjump_escape_listener'))
end

return M