path: root/users/seth/programs/neovim.nix

{ pkgs, ... }: {
  programs.neovim = {
    enable = true;
    extraPackages = with pkgs; [
      clang
      codespell
      deadnix
      nodePackages.alex
      nodePackages.bash-language-server
      nodePackages.prettier
      nodePackages.pyright
      nixpkgs-fmt
      pylint
      rust-analyzer
      rustfmt
      statix
      stylua
      sumneko-lua-language-server
      yapf
    ];
    plugins = with pkgs.vimPlugins; [
      barbar-nvim
      catppuccin-nvim
      cmp-nvim-lsp
      cmp-buffer
      cmp_luasnip
      cmp-path
      cmp-vsnip
      editorconfig-nvim
      gitsigns-nvim
      lightspeed-nvim
      lualine-nvim
      luasnip
      nvim-cmp
      nvim-lspconfig
      null-ls-nvim
      nvim-tree-lua
      nvim-treesitter.withAllGrammars
      nvim-web-devicons
      plenary-nvim
      telescope-nvim
      trouble-nvim
      vim-vsnip
    ];
    extraLuaConfig = ''
      local cmd = vim.cmd
      local opt = vim.opt

      -- text options
      opt.tabstop = 2
      opt.shiftwidth = 2
      opt.expandtab = false
      opt.smartindent = true
      opt.wrap = false


      -- appearance
      opt.syntax = "on"
      cmd("filetype plugin indent on")
      opt.termguicolors = true

      -- filetypes
      local filetypes = {
      	filename = {
      	  PKGBUILD = "text",
      	  [".makepkg.conf"] = "text",
      	},
      }

      vim.filetype.add(filetypes)


      -- helper functions/variables
      local cmp = require("cmp")
      local luasnip = require("luasnip")
      local null_ls = require("null-ls")

      --- catppuccin
      local compile_path = vim.fn.stdpath("cache") .. "/catppuccin-nvim"
      vim.fn.mkdir(compile_path, "p")
      vim.opt.runtimepath:append(compile_path)

      ---- cmp
      local mapping = cmp.mapping
      local has_words_before = function()
      	local line, col = unpack(vim.api.nvim_win_get_cursor(0))
      	return col ~= 0 and vim.api.nvim_buf_get_lines(0, line - 1, line, true)[1]:sub(col, col):match("%s") == nil
      end

      local feedkey = function(key, mode)
      	vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes(key, true, true, true), mode, true)
      end

      local capabilities = require("cmp_nvim_lsp").default_capabilities()
      capabilities.textDocument.completion.completionItem.snippetSupport = true

      ---- null ls
      ------ nulls-ls shorthand
      local diagnostics = null_ls.builtins.diagnostics
      local formatting = null_ls.builtins.formatting

      ------ auto-format
      -- for neovim >= 8
      local lsp_formatting = function(bufnr)
      	vim.lsp.buf.format({
      	filter = function(client)
      		return client.name == "null-ls"
      	end,
      	bufnr = bufnr,
      	})
      end

      local augroup = vim.api.nvim_create_augroup("LspFormatting", {})
      local formatting_on_attach = function(client, bufnr)
      	if client.supports_method("textDocument/formatting") then
      		vim.api.nvim_clear_autocmds({ group = augroup, buffer = bufnr })
      		vim.api.nvim_create_autocmd("BufWritePre", {
      			group = augroup,
      			buffer = bufnr,
      			callback = function()
      			  -- for neovim < 8
      			  ---- local params = require("vim.lsp.util").make_formatting_params({})
      			  ---- client.request("textDocument/formatting", params, nil, bufnr)
      			  lsp_formatting(bufnr) -- neovim >= 8
      			end,
      		})
      	end
      end


      -- plugin config
      bufferline_config = {
      	animation = true,
      	auto_hide = true,
      	icons = true,
      	maximum_padding = 2,
      	semantic_letters = true,
      }

      catppuccin_config = {
      		compile_path = compile_path,
      		flavour = "mocha", -- mocha, macchiato, frappe, latte
      		integrations = {
      			barbar = true,
      			gitsigns = true,
      			lightspeed = true,
      			cmp = true,
      			nvimtree = true,
      			treesitter_context = true,
      			treesitter = true,
      			telescope = true,
      			lsp_trouble = true,
      		},
      		no_italic = true,
      }

      cmp_config = {
      	snippet = {
      		expand = function(args)
      			vim.fn["vsnip#anonymous"](args.body)
      				luasnip.lsp_expand(args.body)
      			end,
      	},

      	mapping = mapping.preset.insert({
      		["<Tab>"] = cmp.mapping(function(fallback)
      			if cmp.visible() then
      				cmp.select_next_item()
      			elseif luasnip.expand_or_jumpable() then
      				luasnip.expand_or_jump()
      			elseif vim.fn["vsnip#available"](1) == 1 then
      				feedkey("<Plug>(vsnip-expand-or-jump)", "")
      			elseif has_words_before() then
      				cmp.complete()
      			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)
      			elseif vim.fn["vsnip#available"](-1) == 1 then
      				feedkey("<Plug>(vsnip-jump-prev)", "")
      			else
      				fallback()
      			end
      		end, { "i", "s" }),
      	}),

      	sources = cmp.config.sources({
      	  { name = "nvim_lsp" },
      	  { name = "luasnip" },
      	  { name = "vsnip" },
      	  { name = "buffer" },
      	  { name = "path" },
      	}),

      	capabilities = capabilities,

      	on_attach = function(_, bufnr)
      	  vim.api.nvim_buf_set_option(bufnr, "omnifunc", "v:lua.vim.lsp.omnifunc")
      	end,
      }

      sources = {
      	lsp_servers = { "rust_analyzer", "pyright", "bashls" },
      	null_ls = {
      		diagnostics.alex,
      		diagnostics.codespell,
      		diagnostics.deadnix,
      		diagnostics.pylint,
      		diagnostics.statix,
      		formatting.codespell,
          formatting.nixpkgs_fmt,
      		formatting.prettier,
      		formatting.rustfmt,
      		formatting.stylua,
      		formatting.yapf,
      	}
      }

      local lsp_on_attach = function(client, bufnr)
        cmp_config.on_attach(client, bufnr)
      end

      local all_config = {
      	capabilities = cmp.capabilities,
      	on_attach = lsp_on_attach,
      }

      servers = {}
      for _, server in ipairs(sources.lsp_servers) do
      	servers[server] = all_config
      end

      servers["lua_ls"] = {
      	capabilities = cmp_config.capabilities,
      	on_attach = lsp_on_attach,
      	settings = {
      		Lua = {
      			runtime = {
      				version = "LuaJIT",
      			},
      			diagnostics = {
      				globals = { "vim" },
      			},
      			workspace = {
      				library = vim.api.nvim_get_runtime_file("", true),
      			},
      		},
      	},
      }

      lsp_config = {
      	servers = servers
      }

      lualine_config = {
      	options = {
      		theme = "catppuccin",
      	},
      	extensions = { "nvim-tree" },
      }

      null_ls_config = {
      	on_attach = formatting_on_attach,
      	sources = sources.null_ls,
      }

      tree_config = {}

      treesitter_config = {
      	auto_install = false,
      	highlight = {
      		enable = true,
      		additional_vim_regex_highlighting = false,
      	},
      }

      trouble = {}


      -- init plugins
      local lspconfig = require("lspconfig")

      require("bufferline").setup(bufferline_config)
      require("catppuccin").setup(catppuccin_config)
      vim.api.nvim_command("colorscheme catppuccin")
      require("cmp").setup(cmp_config)
      require("gitsigns").setup()
      require("lualine").setup(lualine_config)
      require("null-ls").setup(null_ls_config)
      require("nvim-tree").setup(tree_config)
      require("nvim-treesitter.configs").setup(treesitter_config)
      require("trouble").setup(trouble_config)

      for server, settings in pairs(lsp_config.servers) do
      	lspconfig[server].setup(settings)
      end


      -- filetypes
      local opts = { noremap = true, silent = true }
      local set = function(mode, key, cmd)
      	vim.keymap.set(mode, key, cmd, opts)
      end


      -- keybinds
      set("n", "<leader>t", function()
      	vim.cmd("NvimTreeToggle")
      end)

      for i = 1, 9 do
      	set("n", "<leader>" .. i, function()
      		local cmd = "BufferGoto " .. i
      		vim.cmd(cmd)
      	end)
      end

      set("n", "<leader>p", function()
      	vim.cmd("BufferPick")
      end)

      set("n", "<leader>q", function()
      	vim.cmd("BufferClose")
      end)

      set("n", "<space>e", vim.diagnostic.open_float)
      set("n", "[d", vim.diagnostic.goto_prev)
      set("n", "]d", vim.diagnostic.goto_next)
      set("n", "<space>q", vim.diagnostic.setloclist)

      set("n", "<space>f", function()
      	vim.cmd("Telescope")
      end)

      set("n", "<space>t", function()
      	vim.cmd("TroubleToggle")
      end)

    '';
  };
}