local lspconfig = require "lspconfig"

local function on_attach(client, bufnr)
	local opts = { noremap = true, silent = true, buffer = bufnr }

	local telescope_builtin = require "telescope.builtin"

	local function desc(tbl1, description)
		return vim.tbl_extend("force", tbl1, { desc = description })
	end

	vim.keymap.set("n", "gD", vim.lsp.buf.declaration, desc(opts, "LSP declaration"))
	vim.keymap.set("n", "gd", telescope_builtin.lsp_definitions, desc(opts, "LSP Definitions"))
	vim.keymap.set("n", "K", vim.lsp.buf.hover, desc(opts, "LSP Hover"))

	vim.keymap.set({"n", "v"}, "gr", telescope_builtin.lsp_references, desc(opts, "LSP References"))

	vim.keymap.set({"n", "v"}, "<leader>sa", vim.lsp.buf.code_action, desc(opts, "LSP Code Actions"))
	vim.keymap.set("n", "<leader>se", vim.diagnostic.open_float, desc(opts, "Local Diagnostics"))
	vim.keymap.set("n", "<leader>sE", telescope_builtin.diagnostics, desc(opts, "Global Diagnostics"))
	vim.keymap.set("n", "<leader>sl", vim.diagnostic.setloclist, desc(opts, "Diagnostics set LocList"))
	vim.keymap.set("n", "<leader>sq", vim.diagnostic.setqflist, desc(opts, "Diagnostics set QFList"))
	vim.keymap.set("n", "<leader>sr", vim.lsp.buf.rename, desc(opts, "LSP Rename"))
	vim.keymap.set("n", "<leader>ss", telescope_builtin.lsp_document_symbols, desc(opts, "LSP Document Symbols"))
	vim.keymap.set("n", "<leader>sS", telescope_builtin.lsp_workspace_symbols, desc(opts, "LSP Workspace Symbols"))

	vim.keymap.set("n", "[d", vim.diagnostic.goto_prev, desc(opts, "Goto Next Diagnostic"))
	vim.keymap.set("n", "]d", vim.diagnostic.goto_next, desc(opts, "Goto Prev Diagnostic"))

	-- Capability specific

	if client.server_capabilities.documentFormattingProvider then
		vim.keymap.set({"n", "v"}, "<leader>sf", vim.lsp.buf.format, desc(opts, "Format buffer"))
	end

	-- Rust specific
	vim.keymap.set("n", "<leader>sh", "<cmd>RustToggleInlayHints<CR>", opts)

	require("lsp_signature").on_attach {
		hint_prefix = "param: ",
	}
end

local capabilities = require("cmp_nvim_lsp").default_capabilities()

require("rust-tools").setup {
	server = {
		cmd = { vim.g.rust_analyzer_path },
		settings = {
			["rust-analyzer"] = {
				checkOnSave = {
					command = "clippy",
				},
				experimental = {
					procAttrMacros = true,
				},
			},
		},
		on_attach = on_attach,
	},
}

-- TODO: setup nvim-cmp completion score
require("clangd_extensions").setup {
	server = {
		cmd = { vim.g.clangd_path },
	}
}

require("neodev").setup({})

-- Refactoring
require("refactoring").setup {}

-- Null LSP

local null_ls = require("null-ls")

null_ls.setup({
	sources = {
		null_ls.builtins.code_actions.gitrebase,
		null_ls.builtins.code_actions.gitsigns,
		null_ls.builtins.code_actions.refactoring.with {
			filetypes = { "typescript", "javascript", "lua", "c", "cpp", "go", "python", "java", "php", "ruby" },
		},
		null_ls.builtins.code_actions.shellcheck,
		null_ls.builtins.code_actions.statix,

		null_ls.builtins.diagnostics.deadnix,
		null_ls.builtins.diagnostics.ruff,
		null_ls.builtins.diagnostics.shellcheck,
		null_ls.builtins.diagnostics.statix,
		null_ls.builtins.diagnostics.vale.with {
			filetypes = { "markdown", "pandoc", "tex", "asciidoc" },
		},

		null_ls.builtins.formatting.alejandra,
		null_ls.builtins.formatting.black,
		null_ls.builtins.formatting.ruff,
		null_ls.builtins.formatting.shfmt,
		null_ls.builtins.formatting.taplo,
	},
	on_attach = on_attach,
})

-- Vim Pandoc
vim.g["pandoc#formatting#equalprg"] = "pandoc -t markdown --wrap=preserve"
vim.g["pandoc#formatting#extra_equalprg"] = "--reference-links --reference-location=section"