feat(mason-lspconfig): add :LspInstall and :LspUninstall commands (#35)

These should be in complete parity with the nvim-lsp-installer commands.
Now even better than before.

1 files changed, 141 insertions, 0 deletions

diff --git a/lua/mason-lspconfig/api/command.lua b/lua/mason-lspconfig/api/command.lua
new file mode 100644
index 00000000..22a96be5
--- /dev/null
+++ b/lua/mason-lspconfig/api/command.lua
@@ -0,0 +1,141 @@
+local a = require "mason-core.async"
+local Optional = require "mason-core.optional"
+local notify = require "mason-core.notify"
+local _ = require "mason-core.functional"
+
+local M = {}
+
+---@async
+---@param user_args string[] @The arguments, as provided by the user.
+local function parse_packages_from_user_args(user_args)
+    local Package = require "mason-core.package"
+    local server_mapping = require "mason-lspconfig.mappings.server"
+    local language_mapping = require "mason.mappings.language"
+
+    return _.filter_map(function(server_specifier)
+        local server_name, version = Package.Parse(server_specifier)
+        -- 1. first see if the provided arg is an actual lspconfig server name
+        return Optional
+            .of_nilable(server_mapping.lspconfig_to_package[server_name])
+            -- 2. if not, check if it's a language specifier (e.g., "typescript" or "java")
+            :or_(function()
+                return Optional.of_nilable(language_mapping[server_name]):map(function(package_names)
+                    local lsp_package_names = _.filter(function(package_name)
+                        return server_mapping.package_to_lspconfig[package_name] ~= nil
+                    end, package_names)
+
+                    if #lsp_package_names == 0 then
+                        return nil
+                    end
+
+                    return a.promisify(vim.ui.select)(lsp_package_names, {
+                        prompt = ("Please select which server you want to install for language %q:"):format(
+                            server_name
+                        ),
+                        format_item = function(item)
+                            return server_mapping.package_to_lspconfig[item]
+                        end,
+                    })
+                end)
+            end)
+            :map(function(package_name)
+                return { package = package_name, version = version }
+            end)
+            :if_not_present(function()
+                notify(("Could not find LSP server %q."):format(server_name), vim.log.levels.ERROR)
+            end)
+    end, user_args)
+end
+
+---@async
+local function parse_packages_from_heuristics()
+    local server_mapping = require "mason-lspconfig.mappings.server"
+
+    -- Prompt user which server they want to install (based on the current filetype)
+    local current_ft = vim.api.nvim_buf_get_option(vim.api.nvim_get_current_buf(), "filetype")
+    local filetype_mapping = require "mason-lspconfig.mappings.filetype"
+    return Optional.of_nilable(filetype_mapping[current_ft])
+        :map(function(server_names)
+            return a.promisify(vim.ui.select)(server_names, {
+                prompt = ("Please select which server you want to install for filetype %q:"):format(current_ft),
+            })
+        end)
+        :map(function(server_name)
+            local package_name = server_mapping.lspconfig_to_package[server_name]
+            return { package = package_name, version = nil }
+        end)
+        :or_else_get(function()
+            notify(("No LSP servers found for filetype %q."):format(current_ft), vim.log.levels.ERROR)
+            return {}
+        end)
+end
+
+local parse_packages_to_install = _.cond {
+    { _.compose(_.gt(0), _.length), parse_packages_from_user_args },
+    { _.compose(_.equals(0), _.length), parse_packages_from_heuristics },
+    { _.T, _.always {} },
+}
+
+vim.api.nvim_create_user_command(
+    "LspInstall",
+    a.scope(function(opts)
+        local packages_to_install = parse_packages_to_install(opts.fargs)
+        if #packages_to_install > 0 then
+            local registry = require "mason-registry"
+            _.each(function(target)
+                local pkg = registry.get_package(target.package)
+                pkg:install { version = target.version }
+            end, packages_to_install)
+            require("mason.ui").open()
+            require("mason.ui").set_view "LSP"
+        end
+    end),
+    {
+        desc = "Install one or more LSP servers.",
+        nargs = "*",
+        complete = "custom,v:lua.mason_lspconfig_completion.available_server_completion",
+    }
+)
+
+vim.api.nvim_create_user_command("LspUninstall", function(opts)
+    require("mason.ui").open()
+    require("mason.ui").set_view "LSP"
+    local registry = require "mason-registry"
+    local server_mapping = require "mason-lspconfig.mappings.server"
+    for _, server_specifier in ipairs(opts.fargs) do
+        local package_name = server_mapping.lspconfig_to_package[server_specifier]
+        local pkg = registry.get_package(package_name)
+        pkg:uninstall()
+    end
+end, {
+    desc = "Uninstall one or more LSP servers.",
+    nargs = "+",
+    complete = "custom,v:lua.mason_lspconfig_completion.installed_server_completion",
+})
+
+_G.mason_lspconfig_completion = {
+    available_server_completion = function()
+        local registry = require "mason-registry"
+        local server_mapping = require "mason-lspconfig.mappings.server"
+        local language_mapping = require "mason.mappings.language"
+
+        local package_names = _.filter_map(function(pkg_name)
+            return Optional.of_nilable(server_mapping.package_to_lspconfig[pkg_name])
+        end, registry.get_all_package_names())
+        local completion = _.concat(package_names, _.keys(language_mapping))
+        table.sort(completion)
+        return table.concat(completion, "\n")
+    end,
+    installed_server_completion = function()
+        local registry = require "mason-registry"
+        local server_mapping = require "mason-lspconfig.mappings.server"
+
+        local server_names = _.filter_map(function(pkg_name)
+            return Optional.of_nilable(server_mapping.package_to_lspconfig[pkg_name])
+        end, registry.get_installed_package_names())
+        table.sort(server_names)
+        return table.concat(server_names, "\n")
+    end,
+}
+
+return M