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
147
148
149
150
|
local Optional = require "mason-core.optional"
local _ = require "mason-core.functional"
local a = require "mason-core.async"
local notify = require "mason-lspconfig.notify"
local registry = require "mason-registry"
---@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 mappings = require "mason-lspconfig.mappings"
local server_mapping = mappings.get_mason_map()
local language_map = mappings.get_language_map()
return _.filter_map(function(server_specifier)
local server_name, version = Package.Parse(server_specifier)
return Optional
-- 1. first see if the provided arg is an actual lspconfig server name
.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_map[server_name])
:if_not_present(function()
notify(("Could not find LSP server %q."):format(server_name), vim.log.levels.ERROR)
end)
:map(function(package_names)
package_names = _.filter(function(package_name)
return server_mapping.package_to_lspconfig[package_name] ~= nil
end, package_names)
if #package_names == 0 then
return nil
end
return a.promisify(vim.ui.select)(package_names, {
prompt = ("Please select which server you want to install for language %q:"):format(
server_name
),
format_item = function(package_name)
local lspconfig_name = server_mapping.package_to_lspconfig[package_name]
if registry.is_installed(package_name) then
return ("%s (installed)"):format(lspconfig_name)
else
return lspconfig_name
end
end,
})
end)
end)
:map(function(package_name)
return { package = package_name, version = version }
end)
end, user_args)
end
---@async
local function parse_packages_from_heuristics()
local mappings = require "mason-lspconfig.mappings"
local server_mapping = mappings.get_mason_map()
local filetype_mapping = mappings.get_filetype_map()
-- 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 server_names = filetype_mapping[current_ft] or {}
if #server_names == 0 then
notify(("No LSP servers found for filetype %q."):format(current_ft), vim.log.levels.ERROR)
return {}
end
local server_name = a.promisify(vim.ui.select)(server_names, {
prompt = ("Please select which server you want to install for filetype %q:"):format(current_ft),
format_item = function(server_name)
if registry.is_installed(server_mapping.lspconfig_to_package[server_name]) then
return ("%s (installed)"):format(server_name)
else
return server_name
end
end,
})
if server_name == nil then
return {}
end
local package_name = server_mapping.lspconfig_to_package[server_name]
return { { package = package_name, version = nil } }
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)
a.wait(registry.refresh)
local packages_to_install = parse_packages_to_install(servers)
if #packages_to_install > 0 then
require("mason.api.command").MasonInstall(_.map(function(target)
if target.version then
return ("%s@%s"):format(target.package, target.version)
else
return target.package
end
end, packages_to_install))
local ui = require "mason.ui"
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)
local server_mapping = require("mason-lspconfig.mappings").get_mason_map()
require("mason.api.command").MasonUninstall(_.map(function(lspconfig_name)
return server_mapping.lspconfig_to_package[lspconfig_name] or lspconfig_name
end, servers))
require("mason.ui").set_view "LSP"
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",
})
-- selene: allow(global_usage)
_G.mason_lspconfig_completion = {
available_server_completion = function()
local available_servers = require("mason-lspconfig").get_available_servers()
local mappings = require "mason-lspconfig.mappings"
local sort_deduped = _.compose(_.sort_by(_.identity), _.uniq_by(_.identity))
local completions = sort_deduped(_.concat(_.keys(mappings.get_language_map()), available_servers))
return table.concat(completions, "\n")
end,
installed_server_completion = function()
local installed_servers = require("mason-lspconfig").get_installed_servers()
return table.concat(installed_servers, "\n")
end,
}
return {
LspInstall = LspInstall,
LspUninstall = LspUninstall,
}
|
