local lspconfig = require("lspconfig")

function on_attach(client, bufnr)
	local function buf_set_keymap(...)
		vim.api.nvim_buf_set_keymap(bufnr, ...)
	end

	local opts = { noremap = true, silent = true }

	buf_set_keymap('n', 'gD', '<Cmd>lua vim.lsp.buf.declaration()<CR>', opts)
	buf_set_keymap('n', 'gd', '<Cmd>Telescope lsp_definitions<CR>', opts)
	buf_set_keymap('n', 'K', '<Cmd>lua vim.lsp.buf.hover()<CR>', opts)

	buf_set_keymap('n', 'gr', '<cmd>Telescope lsp_references<CR>', opts)

	buf_set_keymap('n', '<leader>sa', '<cmd>Telescope lsp_code_actions<CR>', opts)
	buf_set_keymap('n', '<leader>se', '<cmd>lua vim.lsp.diagnostic.show_line_diagnostics()<CR>', opts)
	buf_set_keymap('n', '<leader>sE', '<cmd>Telescope lsp_document_diagnostics<CR>', opts)
	buf_set_keymap('n', '<leader>swe', '<cmd>Telescope lsp_workspace_diagnostics<CR>', opts)
	buf_set_keymap('n', '<leader>sl', '<cmd>lua vim.lsp.diagnostic.set_loclist()<CR>', opts)
	buf_set_keymap('n', '<leader>sr', '<cmd>lua vim.lsp.buf.rename()<CR>', opts)
	buf_set_keymap('n', '<leader>ss', '<cmd>Telescope lsp_document_symbols<CR>', opts)
	buf_set_keymap('n', '<leader>sS', '<cmd>Telescope lsp_workspace_symbols<CR>', opts)

	buf_set_keymap('n', '[d', '<cmd>lua vim.lsp.diagnostic.goto_prev()<CR>', opts)
	buf_set_keymap('n', ']d', '<cmd>lua vim.lsp.diagnostic.goto_next()<CR>', opts)

	-- Capability specific

	-- print(vim.inspect(client.resolved_capabilities))

	if client.resolved_capabilities.document_formatting then
		buf_set_keymap("n", "<leader>sf", "<cmd>lua vim.lsp.buf.formatting()<CR>", opts)
	end
	if client.resolved_capabilities.document_range_formatting then
		buf_set_keymap("v", "<leader>sf", "<cmd>lua vim.lsp.buf.range_formatting()<CR>", opts)
	end

	-- Rust specific
	buf_set_keymap("n", "<leader>sh", "<cmd>RustToggleInlayHints<CR>", opts)

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

local capabilities = vim.lsp.protocol.make_client_capabilities()
capabilities.textDocument.completion.completionItem.snippetSupport = true
capabilities.textDocument.completion.completionItem.resolveSupport = {
  properties = {
    'documentation',
    'detail',
    'additionalTextEdits',
  }
}

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

require("nlua.lsp.nvim").setup(require('lspconfig'), {
	cmd = {
		string.format("%s/bin/lua-language-server", vim.g.sumneko_lua_base_path),
		"-E",
		string.format("%s/extras/main.lua", vim.g.sumneko_lua_base_path),
	},
	on_attach = on_attach,
})