after: lsp: basedpyright: adding config for basedpyright

Signed-off-by: Brahmajit Das <listout@listout.xyz>

1 files changed, 60 insertions, 0 deletions

diff --git a/after/lsp/basedpyright.lua b/after/lsp/basedpyright.lua
new file mode 100644
index 0000000..e7a9b89
--- /dev/null
+++ b/after/lsp/basedpyright.lua
@@ -0,0 +1,60 @@
+local function set_python_path(command)
+	local path = command.args
+	local clients = vim.lsp.get_clients {
+		bufnr = vim.api.nvim_get_current_buf(),
+		name = 'basedpyright',
+	}
+	for _, client in ipairs(clients) do
+		if client.settings then
+			client.settings.python = vim.tbl_deep_extend('force', client.settings.python or {}, { pythonPath = path })
+		else
+			client.config.settings = vim.tbl_deep_extend('force', client.config.settings,
+				{ python = { pythonPath = path } })
+		end
+		client:notify('workspace/didChangeConfiguration', { settings = nil })
+	end
+end
+
+return {
+	cmd = { 'basedpyright-langserver', '--stdio' },
+	filetypes = { 'python' },
+	root_markers = {
+		'pyproject.toml',
+		'setup.py',
+		'setup.cfg',
+		'requirements.txt',
+		'Pipfile',
+		'pyrightconfig.json',
+		'.git',
+	},
+	settings = {
+		basedpyright = {
+			analysis = {
+				autoSearchPaths = true,
+				useLibraryCodeForTypes = true,
+				diagnosticMode = 'openFilesOnly',
+			},
+		},
+	},
+	on_attach = function(client, bufnr)
+		vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightOrganizeImports', function()
+			local params = {
+				command = 'basedpyright.organizeimports',
+				arguments = { vim.uri_from_bufnr(bufnr) },
+			}
+
+			-- Using client.request() directly because "basedpyright.organizeimports" is private
+			-- (not advertised via capabilities), which client:exec_cmd() refuses to call.
+			-- https://github.com/neovim/neovim/blob/c333d64663d3b6e0dd9aa440e433d346af4a3d81/runtime/lua/vim/lsp/client.lua#L1024-L1030
+			client.request('workspace/executeCommand', params, nil, bufnr)
+		end, {
+			desc = 'Organize Imports',
+		})
+
+		vim.api.nvim_buf_create_user_command(bufnr, 'LspPyrightSetPythonPath', set_python_path, {
+			desc = 'Reconfigure basedpyright with the provided python path',
+			nargs = 1,
+			complete = 'file',
+		})
+	end,
+}