neovim-config/lua/utils/highlight.lua

---@module "utils.highlight"
--- This module provides functions to apply highlights to the statusline and other components.
--          ╭─────────────────────────────────────────────────────────╮
--          │                        Highlight                        │
--          ╰─────────────────────────────────────────────────────────╯

local M = {}

-- Readability functions
local get_bg_color = require("data.func").get_bg_color
local get_fg_color = require("data.func").get_fg_color
local set_hl = vim.api.nvim_set_hl
local sign_def = vim.fn.sign_define

--- Function to apply the todo highlights
---@param status_string string The todo status string
---@return string The updated status string with the todo highlights
function M.apply_todo_highlights(status_string)
  -- Define patterns and associated highlight groups
  local patterns = {
    { icon = "", hl = "lualine_c_diagnostics_info_normal" },
    { icon = "", hl = "lualine_c_diagnostics_warning_normal" },
    { icon = "", hl = "lualine_c_diagnostics_warning_normal" },
    { icon = "", hl = "lualine_c_diagnostics_hint_normal" },
    { icon = "", hl = "lualine_c_diagnostics_hint_normal" },
  }

  -- Apply the highlights
  for _, pattern in ipairs(patterns) do
    status_string = status_string:gsub(
      vim.pesc(pattern.icon),
      "%#" .. pattern.hl .. "#" .. pattern.icon
    )
  end

  -- Reset highlight to normal after the string
  status_string = status_string .. "%#LualineNormal#"

  return status_string
end

function M.setup_minimap_highlight()
  vim.api.nvim_set_hl(
    0,
    "NeominimapSearchIcon",
    { link = "NeominimapErrorSign" }
  )
  vim.api.nvim_set_hl(
    0,
    "NeominimapSearchLine",
    { link = "NeominimapErrorLine" }
  )
end

--- Function to set up indent highlights
---@return nil
function M.setup_indent_highlight()
  -- Get the background color of CursorLine
  local cursorline_bg_hex = get_bg_color("CursorLine")

  -- Get the foreground color of Error
  local miniiconsred_fg_hex = get_fg_color("Error")

  -- Load the ibl plugin
  if pcall(require, "ibl") then
    -- Load the ibl plugin hooks
    local hooks = require("ibl.hooks")
    -- create the highlight groups in the highlight setup hook
    if cursorline_bg_hex then
      -- Define custom highlight groups for indent
      hooks.register(hooks.type.HIGHLIGHT_SETUP, function()
        set_hl(
          0,
          "IndentBlanklineIndent",
          { fg = cursorline_bg_hex, nocombine = true }
        )
        set_hl(
          0,
          "IndentsScopeHighlight",
          { fg = miniiconsred_fg_hex, nocombine = true }
        )
      end)
    end

    -- Define characters for indent-blankline
    local tab_char, indent_char
    if vim.g.tab_char then
      tab_char = vim.g.tab_char
    elseif vim.g.tab_char_name then
      tab_char = require("data.types").ibl.char[vim.g.tab_char_name]
    else
      tab_char = require("data.types").ibl.char.none
    end
    if vim.g.indent_char then
      indent_char = vim.g.indent_char
    elseif vim.g.indent_char_name then
      indent_char = require("data.types").ibl.char[vim.g.indent_char_name]
    else
      indent_char = require("data.types").ibl.char.none
    end

    -- Configure the indent-blankline plugin
    require("ibl").setup({
      indent = {
        char = indent_char,
        tab_char = tab_char,
        highlight = {
          "IndentBlanklineIndent",
        },
      },
      scope = {
        enabled = false,
      },
      exclude = {
        filetypes = require("data.types").highlights.exclude,
      },
    })
  end

  -- Configure the deadcolumn plugin
  if pcall(require, "deadcolumn") then
    require("deadcolumn").setup({
      scope = "line", ---@type string|fun(): integer
      modes = function(mode)
        return mode:find("^[ictRss\x13]") ~= nil
      end,
      blending = {
        threshold = 0.9,
        colorcode = cursorline_bg_hex,
        hlgroup = { "Normal", "bg" },
      },
      warning = {
        alpha = 0.4,
        offset = 0,
        colorcode = miniiconsred_fg_hex,
        hlgroup = { "Error", "bg" },
      },
      extra = {
        follow_tw = "80",
      },
    })
  end

  -- Set NeoCodeium suggestion highlight
  vim.api.nvim_set_hl(0, "NeoCodeiumSuggestion", { link = "Comment" })

  -- Set NeoCodeium label highlight
  vim.api.nvim_set_hl(0, "NeoCodeiumLabel", { link = "lualine_a_insert" })
end

--- Function to set up smooth cursor mode highlights
---@return nil
function M.setup_mode_highlight()
  local current_mode = vim.fn.mode()
  local bg_color -- Define bg_color variable
  if current_mode == "n" then
    bg_color = get_bg_color("MiniStatuslineModeNormal")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  elseif current_mode == "v" then
    bg_color = get_bg_color("MiniStatuslineModeVisual")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  elseif current_mode == "V" then
    bg_color = get_bg_color("MiniStatuslineModeVisual")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  elseif current_mode == "R" or current_mode == "r" then
    bg_color = get_bg_color("MiniStatuslineModeReplace")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  elseif current_mode == "i" then
    bg_color = get_bg_color("MiniStatuslineModeInsert")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  else
    bg_color = get_bg_color("MiniStatuslineModeNormal")
    set_hl(0, "SmoothCursor", { fg = bg_color })
    sign_def("smoothcursor", { text = "" })
  end

  -- Define the custom highlight outside the function
  local vis_bg_color = get_bg_color("CursorLineNr")
  local vis_fg_color = get_bg_color("lualine_a_visual")
  vim.api.nvim_set_hl(
    0,
    "StatusColumnVisualHighlight",
    { fg = vis_fg_color, bg = vis_bg_color }
  )
end

--- Function to set up dashboard header highlight
---@return nil
function M.setup_dashboard_highlight()
  if not vim.g.DashboardHeaderColor then
    local dash_color = get_fg_color("Error")
    set_hl(0, "DashboardHeader", { fg = dash_color })
  else
    set_hl(0, "DashboardHeader", { fg = vim.g.DashboardHeaderColor })
  end
end

--- Function to set up transparency of the editor
---@return nil
function M.setup_transparency()
  if require("data.func").is_kitty() and not vim.g.disable_transparency then
    vim.cmd("TransparentEnable")
  else
    vim.cmd("TransparentDisable")
  end
end

function M.setup_avante_highlights()
  -- Apply highlights for title
  set_hl(0, "AvanteTitle", { link = "lualine_a_command" })
  set_hl(0, "AvanteReversedTitle", {
    bg = get_bg_color("Normal"),
    fg = get_bg_color("lualine_a_command"),
  })
  -- Apply highlights for subtitle
  set_hl(0, "AvanteSubtitle", { link = "lualine_a_replace" })
  set_hl(0, "AvanteReversedSubtitle", {
    bg = get_bg_color("Normal"),
    fg = get_bg_color("lualine_a_replace"),
  })
  -- Apply highlights for prompt
  set_hl(0, "AvanteThirdTitle", { link = "lualine_a_insert" })
  set_hl(0, "AvanteReversedThirdTitle", {
    bg = get_bg_color("Normal"),
    fg = get_bg_color("lualine_a_insert"),
  })
  -- Apply highlights for hints
  set_hl(0, "AvanteInlineHint", { link = "LspDiagnosticsVirtualTextHint" })
  set_hl(0, "AvantePopupHint", { link = "DiagnosticVirtualTextHint" })
end

function M.setup_multi_cursor_highlight()
  -- Customize how cursors look.
  vim.api.nvim_set_hl(0, "MultiCursorCursor", { link = "Cursor" })
  vim.api.nvim_set_hl(0, "MultiCursorVisual", { link = "Visual" })
  vim.api.nvim_set_hl(0, "MultiCursorSign", { link = "GitSignsAdd" })
  vim.api.nvim_set_hl(0, "MultiCursorDisabledCursor", { link = "Visual" })
  vim.api.nvim_set_hl(0, "MultiCursorDisabledVisual", { link = "Visual" })
  vim.api.nvim_set_hl(0, "MultiCursorDisabledSign", { link = "SignColumn" })
end

--- Function to set up autocommands
---@return nil
function M.setup_autocommands()
  local autogrp = vim.api.nvim_create_augroup
  local autocmd = vim.api.nvim_create_autocmd

  -- List of filetypes to exclude
  local excluded_filetypes = require("data.types").highlights.exclude

  -- Create a new augroup for the IndentHighlight
  local IndentGroup = autogrp("IndentHighlight", { clear = true })
  -- Autocommand to trigger setup on FileType
  autocmd("FileType", {
    group = IndentGroup,
    callback = function()
      local filetype = vim.bo.filetype
      if not vim.tbl_contains(excluded_filetypes, filetype) then
        M.setup_indent_highlight()
        M.setup_minimap_highlight()
        M.setup_avante_highlights()
        M.setup_multi_cursor_highlight()
      else
        if pcall(require, "auto-cursorline") then
          require("auto-cursorline").disable({
            buffer = true,
          })
        end
      end
    end,
  })

  -- Create a new augroup for the dashboard
  local dashboardGroup = autogrp("DashboardColors", { clear = true })
  -- Autocommand to trigger setup on FileType
  autocmd("FileType", {
    pattern = "*",
    group = dashboardGroup,
    callback = function()
      M.setup_dashboard_highlight()
    end,
  })
end

return M