mason-lspconfig.lua

local M = {}

M.setup = function()
	local lsp_deps = require("core.settings").lsp_deps
	local use_python_experimental_lsp = require("core.settings").use_python_experimental_lsp
	local python_experimental_lsp_deps = require("core.settings").python_experimental_lsp_deps

	require("lspconfig.ui.windows").default_options.border = "rounded"
	local load_plugin = require("modules.utils").load_plugin

	local lsp_deps_with_python = lsp_deps
	local has_python_experimental_lsp_deps = python_experimental_lsp_deps and #python_experimental_lsp_deps > 0

	if use_python_experimental_lsp and has_python_experimental_lsp_deps then
		-- If using experimental Python LSP, add the experimental LSPs to the list of dependencies
		lsp_deps_with_python = vim.list_extend(lsp_deps, python_experimental_lsp_deps)
		table.insert(lsp_deps_with_python, "ruff") -- ruff is used for linting and formatting
	elseif use_python_experimental_lsp and not has_python_experimental_lsp_deps then
		-- Experimental LSP desired, but dependencies are missing/empty.
		-- Warn and fall back to pylsp.
		vim.notify(
			[[
If you want to use the experimental Python LSP,
please set `python_experimental_lsp_deps` (a table of LSP names) in your settings.
Fallback to default `pylsp` now.]],
			vim.log.levels.WARN,
			{ title = "nvim-lspconfig" }
		)
		table.insert(lsp_deps_with_python, "pylsp")
	else
		-- Experimental LSP is not desired. Use default pylsp.
		table.insert(lsp_deps_with_python, "pylsp")
	end

	load_plugin("mason-lspconfig", {
		ensure_installed = lsp_deps_with_python,
	})

	vim.diagnostic.config({
		signs = true,
		underline = true,
		virtual_text = false,
		update_in_insert = false,
	})

	local opts = {
		capabilities = vim.tbl_deep_extend(
			"force",
			vim.lsp.protocol.make_client_capabilities(),
			require("cmp_nvim_lsp").default_capabilities()
		),
	}
	---A handler to setup all servers defined under `completion/servers/*.lua`
	---@param lsp_name string
	local function mason_lsp_handler(lsp_name)
		-- rust_analyzer is configured using mrcjkb/rustaceanvim
		-- warn users if they have set it up manually
		if lsp_name == "rust_analyzer" then
			local config_exist = pcall(require, "completion.servers." .. lsp_name)
			if config_exist then
				vim.notify(
					[[
`rust_analyzer` is configured independently via `mrcjkb/rustaceanvim`. To get rid of this warning,
please REMOVE your LSP configuration (rust_analyzer.lua) from the `servers` directory and configure
`rust_analyzer` using the appropriate init options provided by `rustaceanvim` instead.]],
					vim.log.levels.WARN,
					{ title = "nvim-lspconfig" }
				)
			end
			return
		end

		local ok, custom_handler = pcall(require, "user.configs.lsp-servers." .. lsp_name)
		local default_ok, default_handler = pcall(require, "completion.servers." .. lsp_name)
		-- Use preset if there is no user definition
		if not ok then
			ok, custom_handler = default_ok, default_handler
		end

		if not ok then
			-- Default to use factory config for server(s) that doesn't include a spec
			vim.lsp.config(lsp_name, opts)
			vim.lsp.enable(lsp_name)
		elseif type(custom_handler) == "function" then
			--- Case where language server requires its own setup
			--- Make sure to call require("lspconfig")[lsp_name].setup() in the function
			--- See `clangd.lua` for example.
			custom_handler(opts)
			vim.lsp.enable(lsp_name)
		elseif type(custom_handler) == "table" then
			vim.lsp.config(
				lsp_name,
				vim.tbl_deep_extend(
					"force",
					opts,
					type(default_handler) == "table" and default_handler or {},
					custom_handler
				)
			)
			vim.lsp.enable(lsp_name)
		else
			vim.notify(
				string.format(
					"Failed to setup [%s].\n\nServer definition under `completion/servers` must return\neither a fun(opts) or a table (got '%s' instead)",
					lsp_name,
					type(custom_handler)
				),
				vim.log.levels.ERROR,
				{ title = "nvim-lspconfig" }
			)
		end
	end

	for _, lsp in ipairs(lsp_deps_with_python) do
		mason_lsp_handler(lsp)
	end
end

return M