1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
local a = require "mason-core.async"
local Optional = require "mason-core.optional"
local notify = require "mason-core.notify"
local _ = require "mason-core.functional"
---@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 {} },
}
local LspInstall = a.scope(function(servers)
local packages_to_install = parse_packages_to_install(servers)
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)
vim.api.nvim_create_user_command("LspInstall", function(opts)
LspInstall(opts.fargs)
end, {
desc = "Install one or more LSP servers.",
nargs = "*",
complete = "custom,v:lua.mason_lspconfig_completion.available_server_completion",
})
local function LspUninstall(servers)
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(servers) do
local package_name = server_mapping.lspconfig_to_package[server_specifier]
local pkg = registry.get_package(package_name)
pkg:uninstall()
end
end
vim.api.nvim_create_user_command("LspUninstall", function(opts)
LspUninstall(opts.fargs)
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 =
_.compose(_.sort_by(_.identity), _.uniq_by(_.identity), _.concat(_.keys(language_mapping)))(package_names)
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 {
LspInstall = LspInstall,
LspUninstall = LspUninstall,
}
|
