forked from williamboman/mason-lspconfig.nvim
-
Notifications
You must be signed in to change notification settings - Fork 0
/
init.lua
247 lines (219 loc) · 6.38 KB
/
init.lua
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
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
local _ = require "mason-core.functional"
local co = coroutine
local exports = {}
local Promise = {}
Promise.__index = Promise
function Promise.new(resolver)
return setmetatable({ resolver = resolver, has_resolved = false }, Promise)
end
---@param success boolean
---@param cb fun(success: boolean, value: table)
function Promise:_wrap_resolver_cb(success, cb)
return function(...)
if self.has_resolved then
return
end
self.has_resolved = true
cb(success, { ... })
end
end
function Promise:__call(callback)
self.resolver(self:_wrap_resolver_cb(true, callback), self:_wrap_resolver_cb(false, callback))
end
local function await(resolver)
local ok, value = co.yield(Promise.new(resolver))
if not ok then
error(value[1], 2)
end
return unpack(value)
end
local function table_pack(...)
return { n = select("#", ...), ... }
end
---@param async_fn fun(...)
---@param should_reject_err boolean|nil: Whether the provided async_fn takes a callback with the signature `fun(err, result)`
local function promisify(async_fn, should_reject_err)
return function(...)
local args = table_pack(...)
return await(function(resolve, reject)
if should_reject_err then
args[args.n + 1] = function(err, result)
if err then
reject(err)
else
resolve(result)
end
end
else
args[args.n + 1] = resolve
end
local ok, err = pcall(async_fn, unpack(args, 1, args.n + 1))
if not ok then
reject(err)
end
end)
end
end
local function new_execution_context(suspend_fn, callback, ...)
local thread = co.create(suspend_fn)
local cancelled = false
local step
step = function(...)
if cancelled then
return
end
local ok, promise_or_result = co.resume(thread, ...)
if ok then
if co.status(thread) == "suspended" then
if getmetatable(promise_or_result) == Promise then
promise_or_result(step)
else
-- yield to parent coroutine
step(coroutine.yield(promise_or_result))
end
else
callback(true, promise_or_result)
thread = nil
end
else
callback(false, promise_or_result)
thread = nil
end
end
step(...)
return function()
cancelled = true
thread = nil
end
end
exports.run = function(suspend_fn, callback, ...)
return new_execution_context(suspend_fn, callback, ...)
end
---@generic T
---@param suspend_fn T
---@return T
exports.scope = function(suspend_fn)
return function(...)
return new_execution_context(suspend_fn, function(success, err)
if not success then
error(err, 0)
end
end, ...)
end
end
exports.run_blocking = function(suspend_fn, ...)
local resolved, ok, result
local cancel_coroutine = new_execution_context(suspend_fn, function(a, b)
resolved = true
ok = a
result = b
end, ...)
if
vim.wait(60 * 60 * 1000, function() -- the wait time is completely arbitrary
return resolved == true
end, 50)
then
if not ok then
error(result, 2)
end
return result
else
cancel_coroutine()
error("async function failed to resolve in time.", 2)
end
end
exports.wait = await
exports.promisify = promisify
exports.sleep = function(ms)
await(function(resolve)
vim.defer_fn(resolve, ms)
end)
end
exports.scheduler = function()
await(vim.schedule)
end
---Creates a oneshot channel that can only send once.
local function oneshot_channel()
local has_sent = false
local sent_value
local saved_callback
return {
is_closed = function()
return has_sent
end,
send = function(...)
assert(not has_sent, "Oneshot channel can only send once.")
has_sent = true
sent_value = { ... }
if saved_callback then
saved_callback(unpack(sent_value))
end
end,
receive = function()
return await(function(resolve)
if has_sent then
resolve(unpack(sent_value))
else
saved_callback = resolve
end
end)
end,
}
end
---@async
---@param suspend_fns async fun()[]
---@param mode '"first"' | '"all"'
local function wait(suspend_fns, mode)
local channel = oneshot_channel()
do
local results = {}
local thread_cancellations = {}
local count = #suspend_fns
local completed = 0
local function cancel()
for _, cancel_thread in ipairs(thread_cancellations) do
cancel_thread()
end
end
for i, suspend_fn in ipairs(suspend_fns) do
thread_cancellations[i] = exports.run(suspend_fn, function(success, result)
completed = completed + 1
if not success then
if not channel.is_closed() then
cancel()
channel.send(false, result)
results = nil
thread_cancellations = {}
end
else
results[i] = result
if mode == "first" or completed >= count then
cancel()
channel.send(true, mode == "first" and { result } or results)
results = nil
thread_cancellations = {}
end
end
end)
end
end
local ok, results = channel.receive()
if not ok then
error(results, 2)
end
return unpack(results)
end
---@async
---@param suspend_fns async fun()[]
function exports.wait_all(suspend_fns)
return wait(suspend_fns, "all")
end
---@async
---@param suspend_fns async fun()[]
function exports.wait_first(suspend_fns)
return wait(suspend_fns, "first")
end
function exports.blocking(suspend_fn)
return _.partial(exports.run_blocking, suspend_fn)
end
return exports