init.lua

-- disable osc52 bc it doesn't work
local termfeatures = vim.g.termfeatures or {}
termfeatures.osc52 = false
vim.g.termfeatures = termfeatures

-- Set <space> as the leader key
-- See `:help mapleader`
--  NOTE: Must happen before plugins are required (otherwise wrong leader will be used)
vim.cmd('source ' .. vim.fn.stdpath('config') .. '/more_configs.vim')
vim.g.mapleader = ' '
vim.g.maplocalleader = ' '

require('local-settings')

if (vim.fn.has('win32')) then
  local git = vim.fn.exepath("git")
  if #git == 0 then
    vim.notify("Git is required for this on Windows platform")
  end
  local git_cmdtool_path = vim.fs.joinpath(vim.fs.dirname(vim.fs.dirname(git)), "usr/bin")
  vim.env.PATH = vim.env.PATH .. ';' .. git_cmdtool_path
end

-- i don't know how to do this in lua
vim.cmd('set autochdir')

if vim.fn.has('win32') ~= 0 then
  vim.cmd[[
    let &shell = executable('pwsh') ? 'pwsh' : 'powershell'
    let &shellcmdflag = '-NoLogo -ExecutionPolicy RemoteSigned -Command [Console]::InputEncoding=[Console]::OutputEncoding=[System.Text.UTF8Encoding]::new();$PSDefaultParameterValues[''Out-File:Encoding'']=''utf8'';Remove-Alias -Force -ErrorAction SilentlyContinue tee;'
    let &shellredir = '2>&1 | %%{ "$_" } | Out-File %s; exit $LastExitCode'
    let &shellpipe  = '2>&1 | %%{ "$_" } | tee %s; exit $LastExitCode'
    set shellquote= shellxquote=
  ]]
else
  if vim.fn.executable('zsh') ~= 0 then
    vim.o.shell = 'zsh'
  else
    vim.o.shell = 'bash'
  end
end

if vim.g.neovide then
  -- the same color used for tokynight
  vim.g.neovide_title_background_color = "#1a1b26"
end

-- needed so that neovide ignores the system theme
vim.o.background = 'dark'

-- Install package manager
--    https://github.com/folke/lazy.nvim
--    `:help lazy.nvim.txt` for more info
local lazypath = vim.fn.stdpath("data") .. "/lazy/lazy.nvim"
if not (vim.uv or vim.loop).fs_stat(lazypath) then
  vim.fn.system({
    "git",
    "clone",
    "--filter=blob:none",
    "https://github.com/folke/lazy.nvim.git",
    "--branch=stable", -- latest stable release
    lazypath,
  })
end
vim.opt.rtp:prepend(lazypath)

require('lazy-file')
require('lspconfig')

require('lazy').setup({
  { import = 'plugins.essentials' },
  { import = 'plugins.lsp-debugger', cond = (not vim.g.vscode) },
  { import = 'plugins.code-files' },
  { import = 'plugins.ai', cond = (vim.g.enable_ai and not vim.g.vscode) },
  { import = 'plugins.ui', cond = (not vim.g.vscode) },
  { import = 'plugins.git', cond = (not vim.g.vscode) },
}, {})

-- [[ Setting options ]]
-- See `:help vim.o`
-- NOTE: You can change these options as you wish!

-- Set highlight on search
vim.o.hlsearch = false
vim.o.incsearch = true

vim.opt.scrolloff = 8

-- Make line numbers default
vim.wo.number = true

-- Enable mouse mode
vim.o.mouse = 'a'

-- Enable break indent
vim.o.breakindent = true

-- Save undo history
vim.o.undofile = true

vim.opt.laststatus = 3

-- Case-insensitive searching UNLESS \C or capital in search
vim.o.ignorecase = true
vim.o.smartcase = true

-- Keep signcolumn on by default
vim.wo.signcolumn = 'yes'

-- Decrease update time
vim.o.updatetime = 250
vim.o.timeoutlen = 300

-- Set completeopt to have a better completion experience
vim.o.completeopt = 'menuone,noselect,fuzzy'

vim.o.wrap = false

-- NOTE: You should make sure your terminal supports this
vim.o.termguicolors = true

vim.o.relativenumber = true

-- set folds to indent but disable them by default
vim.o.foldmethod = 'indent'
vim.o.foldenable = false

vim.o.ex = true

-- Automatically load the .nvim.lua when I :cd
vim.api.nvim_create_autocmd('DirChanged', {
  pattern = { 'window', 'tabpage', 'global', 'auto' },
  callback = function ()
    local cwd = vim.fn.getcwd()

    if not vim.fn.filereadable(cwd .. '/' .. vim.g.exrc_file) then
      return
    end
    if type(vim.secure.read(vim.g.exrc_file)) ~= 'string' then
      return
    end

    vim.cmd.source(vim.g.exrc_file)
  end
})

-- [[ Basic Keymaps ]]

-- Keymaps for better default experience
-- See `:help vim.keymap.set()`
vim.keymap.set({ 'n', 'v' }, '<Space>', '<Nop>', { silent = true })

vim.keymap.set('n', '<leader>tn', ':tabnew term://' .. vim.o.shell .. '<cr>', { desc = "New terminal tab (opens " .. vim.o.shell .. ')' })

-- Remap for dealing with word wrap
vim.keymap.set('n', 'k', "v:count == 0 ? 'gk' : 'k'", { expr = true, silent = true })
vim.keymap.set('n', 'j', "v:count == 0 ? 'gj' : 'j'", { expr = true, silent = true })

-- These are from ThePrimeagen
vim.keymap.set('x', '<leader>p', '"_dP')

vim.keymap.set('n', '<leader>d', '"_d')
vim.keymap.set('v', '<leader>d', '"_d')

vim.keymap.set('n', '<Leader>cfp', ':let @+ = expand("%:p")<CR>', { desc = "Yank Full Filepath" })
vim.keymap.set('n', '<Leader>cfr', ':let @+ = expand("%:p:.")<CR>', { desc = "Yank Relative Filepath" })
vim.keymap.set('n', '<Leader>cfn', ':let @+ = expand("%:t")<CR>', { desc = "Yank File Name" })
vim.keymap.set('n', '<Leader>cff', ':let @+ = expand("%:p:h")<CR>', { desc = "Yank File Folder" })

vim.keymap.set('n', '<leader>fc', function() vim.cmd.edit(vim.g.exrc_file) end, { desc = "Edit the EXRC file" })

vim.keymap.set('t', '<esc>', '<C-\\><C-n>', { desc = 'Escape terminal mode' })

vim.keymap.set('i', '<C-BS>', '<C-W>', { desc = 'Delete the previous word in insert mode' })

vim.keymap.set('n', '<leader>cp', function()
  local current_file = vim.api.nvim_buf_get_name(0)
  local current_file_name = vim.fn.fnamemodify(current_file, ':r')
  local current_file_ext = vim.fn.fnamemodify(current_file, ':e')

  if current_file_ext == 'cpp' then
    if vim.uv.fs_stat(current_file_name .. '.h') then
      vim.cmd('e ' .. current_file_name .. '.h')
    else
      vim.cmd('e ' .. current_file_name .. '.hpp')
    end
  elseif current_file_ext == 'h' or current_file_ext == 'hpp' then
    vim.cmd('e ' .. current_file_name .. '.cpp')
  else
    vim.print('Unsupported file extension: ' .. current_file_ext)
  end
end, { desc = "Switch between C/C++ header and source file" })

-- [[ Highlight on yank ]]
-- See `:help vim.highlight.on_yank()`
local highlight_group = vim.api.nvim_create_augroup('YankHighlight', { clear = true })
vim.api.nvim_create_autocmd('TextYankPost', {
  callback = function()
    vim.highlight.on_yank()
  end,
  group = highlight_group,
  pattern = '*',
})

-- Diagnostic keymaps
vim.keymap.set('n', '[d', function ()
  vim.diagnostic.jump({ diagnostic = vim.diagnostic.get_prev() })
end, { desc = 'Go to previous diagnostic message' })

vim.keymap.set('n', ']d', function ()
  vim.diagnostic.jump({ diagnostic = vim.diagnostic.get_next() })
end, { desc = 'Go to next diagnostic message' })

vim.keymap.set('n', '<leader>q', vim.diagnostic.setloclist, { desc = 'Open diagnostics list' })

-- Helper function to toggle diagnostic display options
local function toggle_diagnostic_display(option_name)
  local current_config = vim.diagnostic.config()[option_name]
  local is_enabled = current_config ~= false
  local is_current_line_only = type(current_config) == "table" and current_config.current_line == true

  if is_enabled then
    if is_current_line_only then
      -- If currently enabled for current line only, turn it off completely
      vim.notify("Disabled diagnostic " .. option_name .. " completely", vim.log.levels.INFO)
      vim.diagnostic.config({ [option_name] = false })
    else
      -- If currently enabled for all lines, change to current line only
      vim.notify("Enabled diagnostic " .. option_name .. " for current line only", vim.log.levels.INFO)
      vim.diagnostic.config({ [option_name] = { current_line = true } })
    end
  else
    -- If currently disabled, enable for all lines
    vim.notify("Enabled diagnostic " .. option_name .. " for all lines", vim.log.levels.INFO)
    vim.diagnostic.config({ [option_name] = true })
  end
end

vim.keymap.set('n', 'gdl', function() toggle_diagnostic_display("virtual_lines") end, { desc = 'Toggle diagnostic virtual_lines' })
vim.keymap.set('n', 'gdt', function() toggle_diagnostic_display("virtual_text") end, { desc = 'Toggle diagnostic virtual_text' })
vim.keymap.set('n', 'gdd', function () vim.diagnostic.open_float({ border = "rounded" }) end, { desc = 'Open diagnostic float' })

vim.keymap.set('n', '<leader>ui', function ()
  local lsps = {
    "clangd",
    "emmet_language_server",
    "lua_ls",
    "pyright",
    "ts_ls",
  }
  local choices = {}
  for _, name in ipairs(lsps) do
    if vim.lsp.is_enabled(name) then
      table.insert(choices, " " .. name)
    else
      table.insert(choices, name)
    end
  end
  vim.ui.select(choices, {
    prompt = "Select lsp to enabled/disable"
  }, function (choice, idx)
      if choice == nil then
        return
      end
      vim.lsp.enable(lsps[idx], not vim.lsp.is_enabled(lsps[idx]))
  end)
end)

-- Creates a progress bar for LSP loading
vim.api.nvim_create_autocmd("LspProgress", {
  ---@param ev {data: {client_id: integer, params: lsp.ProgressParams}}
  callback = function(ev)
    if not vim.g.lsp_progress then return end

    local spinner = { "⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏" }
    vim.api.nvim_echo({
        {ev.data.params.value.kind == "end" and " "
            or spinner[math.floor(vim.uv.hrtime() / (1e6 * 80)) % #spinner + 1] },
        {vim.lsp.status()}
      }, true, {
        kind = 'progress',
        status = ev.data.params.value.kind == "end" and "success" or "running",
        percent = ev.data.params.value.percentage or 100,
        title = 'Loading LSP',
        id = 'lsp-loading',
        source = 'lsp-loading'
    })
  end,
})

-- taken from https://github.com/LazyVim/LazyVim/discussions/2184#discussioncomment-7841499
vim.api.nvim_create_autocmd("Filetype", {
  pattern = { "*" },
  callback = function()
    -- vim.opt.formatoptions = vim.opt.formatoptions - "o"
    if vim.bo["ft"] == "css" then
      vim.opt_local.formatoptions:remove("r") -- don't enter comment leader on Enter in css files
    end
    vim.opt.formatoptions = vim.opt.formatoptions + {
      o = false, -- Don't continue comments with o and O
    }
  end,
  desc = "Don't continue comments with o and O",
})

-- temporary fix for https://github.com/neovim/neovim/issues/8587
-- taken from https://github.com/neovim/neovim/issues/8587#issuecomment-2176399196
vim.api.nvim_create_user_command("ClearShada", function()
    local shada_path = vim.fn.expand(vim.fn.stdpath('data') .. "/shada")
    local files = vim.fn.glob(shada_path .. "/*", false, true)
    local all_success = 0
    for _, file in ipairs(files) do
      local file_name = vim.fn.fnamemodify(file, ":t")
      if file_name == "main.shada" then
        -- skip your main.shada file
        goto continue
      end
      local success = vim.fn.delete(file)
      all_success = all_success + success
      if success ~= 0 then
        vim.notify("Couldn't delete file '" .. file_name .. "'", vim.log.levels.WARN)
      end
      ::continue::
    end
    if all_success == 0 then
      vim.print("Successfully deleted all temporary shada files")
    end
  end,
  { desc = "Clears all the .tmp shada files" }
)

-- I sometimes type :W instead of :w by accident
vim.api.nvim_create_user_command('W', 'w', {})

-- The line beneath this is called `modeline`. See `:help modeline`
-- vim: ts=2 sts=2 sw=2 et