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
|
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()
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 getmetatable(promise_or_result) == Promise then
promise_or_result(step)
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
exports.scope = function(suspend_fn)
return function(...)
return new_execution_context(suspend_fn, function() 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(60000, function()
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
return exports
|
