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
|
local log = require "nvim-lsp-installer.log"
local path = require "nvim-lsp-installer.path"
local fs = require "nvim-lsp-installer.core.fs"
local Result = require "nvim-lsp-installer.core.result"
local M = {}
---@async
---@param context InstallContext
local function write_receipt(context)
if context.receipt.is_marked_invalid then
return log.fmt_debug("Skipping writing receipt for %s because it is marked as invalid.", context.name)
end
context.receipt:with_name(context.name):with_schema_version("1.0a"):with_completion_time(vim.loop.gettimeofday())
local receipt_success, install_receipt = pcall(context.receipt.build, context.receipt)
if receipt_success then
local receipt_path = path.concat { context.cwd:get(), "nvim-lsp-installer-receipt.json" }
pcall(fs.write_file, receipt_path, vim.json.encode(install_receipt))
else
log.fmt_error("Failed to build receipt for installation=%s, error=%s", context.name, install_receipt)
end
end
local CONTEXT_REQUEST = {}
---@return InstallContext
function M.context()
return coroutine.yield(CONTEXT_REQUEST)
end
---@async
---@param context InstallContext
---@param installer async fun(context: InstallContext)
function M.run_installer(context, installer)
local thread = coroutine.create(installer)
local step
local ret_val
step = function(...)
local ok, result = coroutine.resume(thread, ...)
if not ok then
error(result, 0)
elseif result == CONTEXT_REQUEST then
step(context)
elseif coroutine.status(thread) == "suspended" then
-- yield to parent coroutine
step(coroutine.yield(result))
else
ret_val = result
end
end
step(context)
return ret_val
end
---@async
---@param context InstallContext
---@param installer async fun(ctx: InstallContext)
function M.execute(context, installer)
log.fmt_debug("Executing installer for name=%s", context.name)
local tmp_installation_dir = ("%s.tmp"):format(context.destination_dir)
return Result.run_catching(function()
-- 1. prepare installation dir
context.receipt:with_start_time(vim.loop.gettimeofday())
if fs.dir_exists(tmp_installation_dir) then
fs.rmrf(tmp_installation_dir)
end
fs.mkdirp(tmp_installation_dir)
context.cwd:set(tmp_installation_dir)
-- 2. run installer
M.run_installer(context, installer)
-- 3. finalize
log.fmt_debug("Finalizing installer for name=%s", context.name)
write_receipt(context)
context:promote_cwd()
pcall(fs.rmrf, tmp_installation_dir)
end):on_failure(function(failure)
log.fmt_error("Installation failed, name=%s, error=%s", context.name, tostring(failure))
context.stdio_sink.stderr(tostring(failure))
context.stdio_sink.stderr "\n"
pcall(fs.rmrf, tmp_installation_dir)
pcall(fs.rmrf, context.cwd:get())
end)
end
return M
|
