init.lua

-- General Settings
vim.cmd("set guicursor=n-v-c-i:block")
vim.cmd("set expandtab")
vim.cmd("set tabstop=2")
vim.cmd("set softtabstop=2")
vim.cmd("set shiftwidth=2")

vim.g.mapleader = ','
vim.g.maplocalleader = ',' -- Set leader and local leader to ','

vim.opt.number = true -- Show line numbers
vim.opt.autoindent = true
vim.opt.autowrite = true -- Automatically save files on buffer switch
vim.opt.autoread = true -- Reload files changed outside of Neovim
vim.opt.cursorline = true -- Highlight current line
vim.opt.termguicolors = true
vim.opt.autochdir = false
vim.opt.swapfile = false
vim.opt.showmode = true
vim.opt.signcolumn = 'no'
vim.opt.clipboard = "unnamed" -- Use system clipboard

-- Keymaps
vim.keymap.set('i', '<leader>m', '<space>%>%', { silent = true }) -- R Pipe
vim.keymap.set('i', '<leader>r', function()
  local code_block = "```{r}\n\n```"
  vim.api.nvim_feedkeys(
    vim.api.nvim_replace_termcodes(code_block .. "<Esc>ki", true, false, true),
    'n',
    true
  )
end, { silent = true }) -- R Code Chunk

-- Prevent yank on delete
vim.keymap.set({'n', 'v'}, 'd', '"_d', { noremap = true })
vim.keymap.set({'n', 'v'}, 'D', '"_D', { noremap = true })
vim.keymap.set('n', 'dd', '"_dd', { noremap = true })
vim.keymap.set('n', 'x', '"_x', { noremap = true })

-- Lazy.nvim Setup
local lazypath = vim.fn.stdpath("data") .. "/lazy/lazy.nvim"
if not vim.loop.fs_stat(lazypath) then
  vim.fn.system({
    "git", "clone", "--filter=blob:none",
    "https://github.com/folke/lazy.nvim.git",
    "--branch=stable", lazypath,
  })
end
vim.opt.rtp:prepend(lazypath)

-- Plugins
local plugins = {
  -- Theme
  { "rose-pine/neovim", name = "rose-pine" },
  
  -- Utilities
  { 'norcalli/nvim-colorizer.lua' },
  { 'windwp/nvim-autopairs', event = "InsertEnter", config = true },
  { "jpalardy/vim-slime", config = function() vim.g.slime_target = "tmux" end },
  { 'klafyvel/vim-slime-cells', requires = {{'jpalardy/vim-slime', opt=true}},
    ft = {'r'}, config = function()
      vim.g.slime_target = "tmux"
      vim.g.slime_cell_delimiter = "^\\s*##"
      vim.g.slime_default_config = { socket_name = "default", target_pane = "1" }
      vim.g.slime_dont_ask_default = 1
      vim.g.slime_bracketed_paste = 1
      vim.cmd([[
        nmap <leader>s <Plug>SlimeConfig
        nmap <leader>cc <Plug>SlimeCellsSendAndGoToNext
        nmap <leader>j <Plug>SlimeCellsNext
        nmap <leader>k <Plug>SlimeCellsPrev
      ]])
    end
  },

  -- Language Support
  { "lervag/vimtex", init = function() vim.g.vimtex_view_method = 'zathura' end },
  { 'kaarmu/typst.vim', ft = 'typst', lazy = false },

  -- Telescope and Dependencies
  { "nvim-telescope/telescope.nvim" },
  { "nvim-treesitter/nvim-treesitter" },
  { "nvim-lua/plenary.nvim" },

  -- Statusline
  { 'nvim-lualine/lualine.nvim', dependencies = { 'nvim-tree/nvim-web-devicons' } },

  -- LSP and Autocompletion
  { "williamboman/mason.nvim" },
  { "williamboman/mason-lspconfig.nvim" },
  { "neovim/nvim-lspconfig" },
  { "hrsh7th/nvim-cmp" },            -- Completion plugin
  { "hrsh7th/cmp-nvim-lsp" },        -- LSP source for nvim-cmp
  { "hrsh7th/cmp-buffer" },          -- Buffer source
  { "hrsh7th/cmp-path" },            -- Path source
  { "saadparwaiz1/cmp_luasnip" },    -- Snippet completion
  { "L3MON4D3/LuaSnip" },            -- Snippet engine
  { "rafamadriz/friendly-snippets" }, -- Predefined snippets

  -- Formatter
  { 'stevearc/conform.nvim', opts = {} },

  -- File Explorer
  { 'nvim-tree/nvim-tree.lua' },

  -- Advanced Features
  { "karb94/neoscroll.nvim" },

  -- Code Quality Tools
  { "mfussenegger/nvim-lint", event = { "BufReadPre", "BufNewFile" }, config = function()
      local lint = require("lint")
      lint.linters_by_ft = { swift = { "swiftlint" } }
      local lint_group = vim.api.nvim_create_augroup("Lint", { clear = true })
      vim.api.nvim_create_autocmd({ "BufWritePost", "BufReadPost", "InsertLeave", "TextChanged" }, {
        group = lint_group,
        callback = function() lint.try_lint() end,
      })
      vim.keymap.set("n", "<leader>ml", lint.try_lint, { desc = "Lint file" })
    end
  }
}

require("lazy").setup(plugins, {})

-- Theme Configuration
require("rose-pine").setup({
  variant = "auto",
  dark_variant = "main",
  styles = { bold = true, italic = true, transparency = false },
  highlight_groups = { }
})
vim.cmd("colorscheme rose-pine")

-- Telescope Keymaps
local builtin = require('telescope.builtin')
vim.keymap.set('n', '<leader>ff', builtin.find_files, {})
vim.keymap.set('n', '<leader>fg', builtin.live_grep, {})
vim.keymap.set('n', '<leader>fb', builtin.buffers, {})
vim.keymap.set('n', '<leader>fh', builtin.help_tags, {})

-- Lualine Configuration
require('lualine').setup({
  options = {
    theme = 'auto',
    section_separators = { left = '', right = '' },
    component_separators = { left = '', right = '' },
  },
  sections = {
    lualine_b = { 'branch' },
    lualine_c = { 'filename' },
    lualine_y = { 'progress' },
  },
  inactive_sections = {
    lualine_c = { 'filename' },
    lualine_x = { 'location' },
  },
})

-- Colorizer Setup
require("colorizer").setup()

-- Mason LSP Setup
require("mason").setup()
require("mason-lspconfig").setup {
  ensure_installed = { "clangd", "r_language_server", "gopls" }
}
local capabilities = require("cmp_nvim_lsp").default_capabilities(vim.lsp.protocol.make_client_capabilities())

-- Configure LSP Servers
require("lspconfig").r_language_server.setup {}
require("lspconfig").clangd.setup {}
require("lspconfig").gopls.setup {
  capabilities = capabilities,
  settings = {
    gopls = {
      analyses = {
        unusedparams = true,
      },
      staticcheck = true,
    },
  },
}

-- Autocompletion Setup using nvim-cmp and LuaSnip
local cmp = require("cmp")
local luasnip = require("luasnip")

-- Load snippets from friendly-snippets
require("luasnip.loaders.from_vscode").lazy_load()

cmp.setup({
  snippet = {
    expand = function(args)
      luasnip.lsp_expand(args.body)
    end,
  },
  mapping = {
    ["<C-b>"] = cmp.mapping.scroll_docs(-4),
    ["<C-f>"] = cmp.mapping.scroll_docs(4),
    ["<C-Space>"] = cmp.mapping.complete(),
    ["<C-e>"] = cmp.mapping.abort(),
    ["<CR>"] = cmp.mapping.confirm({ select = true }),
    ["<Tab>"] = cmp.mapping(function(fallback)
      if cmp.visible() then
        cmp.select_next_item()
      elseif luasnip.expand_or_jumpable() then
        luasnip.expand_or_jump()
      else
        fallback()
      end
    end, { "i", "s" }),
    ["<S-Tab>"] = cmp.mapping(function(fallback)
      if cmp.visible() then
        cmp.select_prev_item()
      elseif luasnip.jumpable(-1) then
        luasnip.jump(-1)
      else
        fallback()
      end
    end, { "i", "s" }),
  },
  sources = cmp.config.sources({
    { name = "nvim_lsp" },
    { name = "luasnip" },
  }, {
    { name = "buffer" },
    { name = "path" },
  }),
})