dotfiles

neovim/.config/nvim/lua/utils/functions.lua

local vim = vim

local M = {}

---@param on_attach fun(client, buffer)
function M.on_attach(on_attach)
  vim.api.nvim_create_autocmd("LspAttach", {
    callback = function(args)
      local buffer = args.buf
      local client = vim.lsp.get_client_by_id(args.data.client_id)
      on_attach(client, buffer)
    end,
  })
end

function M.starts_with(str, start)
  return str:sub(1, #start) == start
end

function M.is_table(to_check)
  return type(to_check) == "table"
end

function M.has_key(t, key)
  for t_key, _ in pairs(t) do
    if t_key == key then
      return true
    end
  end

  return false
end

function M.has_value(t, val)
  for _, value in ipairs(t) do
    if value == val then
      return true
    end
  end

  return false
end

function M.tprint(table)
  print(vim.inspect(table))
end

function M.tprint_keys(table)
  for k in pairs(table) do
    print(k)
  end
end

M.reload = function()
  local presentReload, reload = pcall(require, "plenary.reload")
  if presentReload then
    local counter = 0

    for moduleName in pairs(package.loaded) do
      if M.starts_with(moduleName, "lt.") then
        reload.reload_module(moduleName)

        counter = counter + 1
      end
    end

    -- clear nvim-mapper keys
    vim.g.mapper_records = nil

    vim.notify("Reloaded " .. counter .. " modules!")
  end
end

function M.is_macunix()
  return vim.fn.has("macunix")
end

function M.link_highlight(from, to, override)
  local hl_exists, _ = pcall(vim.api.nvim_get_hl_by_name, from, false)
  if override or not hl_exists then
    -- vim.cmd(("highlight link %s %s"):format(from, to))
    vim.api.nvim_set_hl(0, from, { link = to })
  end
end

M.highlight = function(group, opts)
  vim.api.nvim_set_hl(0, group, opts)
end

M.highlight_bg = function(group, col)
  vim.api.nvim_set_hl(0, group, { bg = col })
end

-- Define fg color
-- @param group Group
-- @param color Color
M.highlight_fg = function(group, col)
  vim.api.nvim_set_hl(0, group, { fg = col })
end

-- Define bg and fg color
-- @param group Group
-- @param fgcol Fg Color
-- @param bgcol Bg Color
M.highlight_fg_bg = function(group, fgcol, bgcol)
  vim.api.nvim_set_hl(0, group, { bg = bgcol, fg = fgcol })
end

M.from_highlight = function(hl)
  local result = {}
  local list = vim.api.nvim_get_hl_by_name(hl, true)
  for k, v in pairs(list) do
    local name = k == "background" and "bg" or "fg"
    result[name] = string.format("#%06x", v)
  end
  return result
end

M.get_color_from_terminal = function(num, default)
  local key = "terminal_color_" .. num
  return vim.g[key] and vim.g[key] or default
end

return M