lux/modules/features/neovim/lua-config/lua/plugins/lsp.lua

require("lz.n").load({
  {
    "typst-preview.nvim",
    ft = "typst",
    after = function()
      -- Setup typst-preview
      require("typst-preview").setup({
        -- Optionally configure things here
        dependencies_bin = {
          -- For example, use tinymist as the LSP if that's what you are running
        },
      })

      vim.keymap.set("n", "<leader>tp", "<cmd>TypstPreviewToggle<cr>", { desc = "[T]ypst [P]review Toggle" })
    end,
  },
  {
    "lazydev.nvim",
    cmd = "LazyDev",
    ft = "lua",
    after = function()
      require("lazydev").setup({
        library = {
          { words = { "nixCats", "settings" }, path = "nix-info" },
        },
      })
    end,
  },
  {
    "nvim-lspconfig",
    event = { "BufReadPre", "BufNewFile" },
    before = function()
      require("lz.n").trigger_load("lazydev.nvim")
    end,
    after = function()
      vim.api.nvim_create_autocmd("LspAttach", {
        group = vim.api.nvim_create_augroup("kickstart-lsp-attach", { clear = true }),
        callback = function(args)
          -- Get the client and buffer from the event arguments [cite: 119]
          local client = vim.lsp.get_client_by_id(args.data.client_id)
          local bufnr = args.buf

          local map = function(keys, func, desc, mode)
            mode = mode or "n"
            vim.keymap.set(mode, keys, func, { buffer = bufnr, desc = "LSP: " .. desc })
          end

          -- Standard LSP functions
          map("<leader>rn", vim.lsp.buf.rename, "[R]e[n]ame")
          map("<leader>ca", vim.lsp.buf.code_action, "[C]ode [A]ction", { "n", "x" })
          map("gD", vim.lsp.buf.declaration, "[G]oto [D]eclaration")
          map("K", vim.lsp.buf.hover, "Hover Documentation")

          -- Telescope Mappings
          map("gd", require("telescope.builtin").lsp_definitions, "[G]oto [D]efinition")
          map("gr", require("telescope.builtin").lsp_references, "[G]oto [R]eferences")
          map("gI", require("telescope.builtin").lsp_implementations, "[G]oto [I]mplementation")
          map("<leader>D", require("telescope.builtin").lsp_type_definitions, "Type [D]efinition")
          map("<leader>ds", require("telescope.builtin").lsp_document_symbols, "[D]ocument [S]ymbols")
          map("<leader>ws", require("telescope.builtin").lsp_dynamic_workspace_symbols, "[W]orkspace [S]ymbols")

          -- Highlight references (Document Highlight)
          if client and client:supports_method("textDocument/documentHighlight", bufnr) then
            local highlight_augroup = vim.api.nvim_create_augroup("kickstart-lsp-highlight", { clear = false })
            vim.api.nvim_create_autocmd({ "CursorHold", "CursorHoldI" }, {
              buffer = bufnr,
              group = highlight_augroup,
              callback = vim.lsp.buf.document_highlight,
            })

            vim.api.nvim_create_autocmd({ "CursorMoved", "CursorMovedI" }, {
              buffer = bufnr,
              group = highlight_augroup,
              callback = vim.lsp.buf.clear_references,
            })

            vim.api.nvim_create_autocmd("LspDetach", {
              group = vim.api.nvim_create_augroup("kickstart-lsp-detach", { clear = true }),
              callback = function(event)
                vim.lsp.buf.clear_references()
                vim.api.nvim_clear_autocmds({ group = "kickstart-lsp-highlight", buffer = event.buf })
              end,
            })
          end

          -- Inlay Hints
          if client and client:supports_method("textDocument/inlayHint", bufnr) then
            vim.lsp.inlay_hint.enable(true, { bufnr = bufnr })
            map("<leader>th", function()
              vim.lsp.inlay_hint.enable(not vim.lsp.inlay_hint.is_enabled({ bufnr = bufnr }))
            end, "[T]oggle Inlay [H]ints")
          end
        end,
      })

      -- 1. Setup Diagnostics (Visuals)
      vim.diagnostic.config({
        severity_sort = true,
        -- underline = { severity = vim.diagnostic.severity.ERROR },
        signs = {
          text = {
            [vim.diagnostic.severity.ERROR] = " ",
            [vim.diagnostic.severity.WARN] = " ",
            [vim.diagnostic.severity.INFO] = " ",
            [vim.diagnostic.severity.HINT] = "󰠠 ",
          },
        },
        virtual_text = {
          source = "if_many",
          spacing = 4,
          prefix = "●",
          format = function(diagnostic)
            local diagnostic_message = {
              [vim.diagnostic.severity.ERROR] = diagnostic.message,
              [vim.diagnostic.severity.WARN] = diagnostic.message,
              [vim.diagnostic.severity.INFO] = diagnostic.message,
              [vim.diagnostic.severity.HINT] = diagnostic.message,
            }
            return diagnostic_message[diagnostic.severity]
          end,
        },
      })

      vim.lsp.config("lua_ls", {
        settings = {
          Lua = {
            runtime = { version = "LuaJIT" },
            signatureHelp = { enabled = true },
            diagnostics = { globals = { "nixCats", "vim" } },
            telemetry = { enabled = false },
            completion = { callSnippet = "Replace" },
          },
        },
      })
      vim.lsp.enable("lua_ls")

      local settings = require("nix-info").settings

      -- Nix
      vim.lsp.config("nixd", {
        settings = {
          nixd = {
            nixpkgs = { expr = settings.nixdExtras.nixpkgs },
            options = {
              nixos = { expr = settings.nixdExtras.nixos_options },
              ["home-manager"] = { expr = settings.nixdExtras.home_manager_options },
            },
            formatting = { command = { "nixfmt" } },
          },
        },
      })
      vim.lsp.enable("nixd")

      -- Dafny
      vim.lsp.enable("dafny")

      -- TypeScript/JS
      vim.lsp.enable("ts_ls")

      -- Rust
      vim.lsp.enable("rust_analyzer")

      -- Python
      vim.lsp.enable("ty")
      vim.lsp.enable("ruff")

      vim.lsp.enable("astro")

      vim.lsp.config("tinymist", {
        settings = {
          tinymist = {
            formatterMode = "typstyle",
          },
        },
      })
      vim.lsp.enable("tinymist")
    end,
  },
  {
    "trouble.nvim",
    cmd = "Trouble",
    keys = {
      { "<leader>xx", "<cmd>Trouble diagnostics toggle<cr>", desc = "Diagnostics (Trouble)" },
      { "<leader>xX", "<cmd>Trouble diagnostics toggle filter.buf=0<cr>", desc = "Buffer Diagnostics (Trouble)" },
    },
    after = function()
      require("trouble").setup({
        focus = true,
      })
    end,
  },
})