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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
local api = vim.api
local lsp = vim.lsp
local M = {}
--- @param msg string
local function echo_err(msg)
api.nvim_echo({ { msg } }, true, { err = true })
end
--- @return string[]
local function get_client_names()
return vim
.iter(lsp.get_clients())
:map(function(client)
return client.name
end)
:unique()
:totable()
end
--- @param filter vim.lsp.get_configs.Filter
--- @return fun():string[]
local function filtered_config_names(filter)
return function()
return vim
.iter(lsp.get_configs(filter))
:map(function(config)
return config.name
end)
:totable()
end
end
local complete_args = {
enable = filtered_config_names { enabled = false },
disable = filtered_config_names { enabled = true },
restart = get_client_names,
stop = get_client_names,
}
--- @param names string[]
--- @param enable? boolean
local function checked_enable(names, enable)
for _, name in ipairs(names) do
if name:find('*') == nil and lsp.config[name] ~= nil then
lsp.enable(name, enable)
else
echo_err(("No client config named '%s'"):format(name))
end
end
end
--- @param config_names string[]
local function ex_lsp_enable(config_names)
-- Default to enabling all clients matching the filetype of the current buffer.
if #config_names == 0 then
local filetype = vim.bo.filetype
for _, config in ipairs(lsp.get_configs()) do
local filetypes = config.filetypes
if filetypes == nil or vim.list_contains(filetypes, filetype) then
table.insert(config_names, config.name)
end
end
if #config_names == 0 then
if filetype == '' then
echo_err('Current buffer has no filetype')
else
echo_err(("No configs for filetype '%s'"):format(filetype))
end
return
end
end
checked_enable(config_names)
end
--- @param config_names string[]
local function ex_lsp_disable(config_names)
-- Default to disabling all clients attached to the current buffer.
if #config_names == 0 then
config_names = vim
.iter(lsp.get_clients { bufnr = api.nvim_get_current_buf() })
:map(function(client)
return client.name
end)
:filter(function(name)
return lsp.config[name] ~= nil
end)
:totable()
if #config_names == 0 then
echo_err('No configs with clients attached to current buffer')
return
end
end
checked_enable(config_names, false)
end
--- @param client_names string[]
--- @return vim.lsp.Client[]
local function get_clients_from_names(client_names)
-- Default to all active clients attached to the current buffer.
if #client_names == 0 then
local clients = lsp.get_clients { bufnr = api.nvim_get_current_buf() }
if #clients == 0 then
echo_err('No clients attached to current buffer')
end
return clients
else
return vim
.iter(client_names)
:map(function(name)
local clients = lsp.get_clients { name = name }
if #clients == 0 then
echo_err(("No active clients named '%s'"):format(name))
end
return clients
end)
:flatten()
:totable()
end
end
--- @param client_names string[]
local function ex_lsp_restart(client_names)
local clients = get_clients_from_names(client_names)
for _, client in ipairs(clients) do
client:_restart(client.exit_timeout)
end
end
--- @param client_names string[]
local function ex_lsp_stop(client_names)
local clients = get_clients_from_names(client_names)
for _, client in ipairs(clients) do
client:stop(client.exit_timeout)
end
end
local actions = {
enable = ex_lsp_enable,
disable = ex_lsp_disable,
restart = ex_lsp_restart,
stop = ex_lsp_stop,
}
local available_subcmds = vim.tbl_keys(actions)
--- Implements command: `:lsp {subcmd} {name}?`.
--- @param args string
M.ex_lsp = function(args)
local fargs = api.nvim_parse_cmd('lsp ' .. args, {}).args
if not fargs then
return
end
local subcmd = fargs[1]
if not vim.list_contains(available_subcmds, subcmd) then
echo_err(("Invalid subcommand '%s'"):format(subcmd))
return
end
local clients = { unpack(fargs, 2) }
actions[subcmd](clients)
end
--- Completion logic for `:lsp` command
--- @param line string content of the current command line
--- @return string[] list of completions
function M.lsp_complete(line)
local split = vim.split(line, '%s+')
if #split == 2 then
return available_subcmds
else
local subcmd = split[2]
return vim
.iter(complete_args[subcmd]())
--- @param n string
:map(function(n)
return vim.fn.escape(n, ' \t')
end)
:totable()
end
end
return M
|
