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
|
local Optional = require "mason-core.optional"
local Result = require "mason-core.result"
local _ = require "mason-core.functional"
local a = require "mason-core.async"
local installer = require "mason-core.installer"
local log = require "mason-core.log"
local path = require "mason-core.path"
local platform = require "mason-core.platform"
local M = {}
local VENV_DIR = "venv"
---@async
---@param py_executables string[]
local function create_venv(py_executables)
local ctx = installer.context()
return Optional.of_nilable(_.find_first(function(executable)
return ctx.spawn[executable]({ "-m", "venv", VENV_DIR }):is_success()
end, py_executables)):ok_or "Failed to create python3 virtual environment."
end
---@param ctx InstallContext
---@param executable string
local function find_venv_executable(ctx, executable)
local candidates = _.filter(_.identity, {
platform.is.unix and path.concat { VENV_DIR, "bin", executable },
-- MSYS2
platform.is.win and path.concat { VENV_DIR, "bin", ("%s.exe"):format(executable) },
-- Stock Windows
platform.is.win and path.concat { VENV_DIR, "Scripts", ("%s.exe"):format(executable) },
})
for _, candidate in ipairs(candidates) do
if ctx.fs:file_exists(candidate) then
return Result.success(candidate)
end
end
return Result.failure(("Failed to find executable %q in Python virtual environment."):format(executable))
end
---@async
---@param args SpawnArgs
local function venv_python(args)
local ctx = installer.context()
return find_venv_executable(ctx, "python"):and_then(function(python_path)
return ctx.spawn[path.concat { ctx.cwd:get(), python_path }](args)
end)
end
---@async
---@param pkgs string[]
---@param extra_args? string[]
local function pip_install(pkgs, extra_args)
return venv_python {
"-m",
"pip",
"--disable-pip-version-check",
"install",
"-U",
extra_args or vim.NIL,
pkgs,
}
end
---@async
---@param opts { upgrade_pip: boolean, install_extra_args?: string[] }
function M.init(opts)
return Result.try(function(try)
log.fmt_debug("pypi: init", opts)
local ctx = installer.context()
a.scheduler()
local executables = platform.is.win and { "python", "python3" } or { "python3", "python" }
-- pip3 will hardcode the full path to venv executables, so we need to promote cwd to make sure pip uses the final destination path.
ctx:promote_cwd()
ctx.stdio_sink.stdout "Creating virtual environment…\n"
try(create_venv(executables))
if opts.upgrade_pip then
ctx.stdio_sink.stdout "Upgrading pip inside the virtual environment…\n"
try(pip_install({ "pip" }, opts.install_extra_args))
end
end)
end
---@async
---@param pkg string
---@param version string
---@param opts? { extra?: string, extra_packages?: string[], install_extra_args?: string[] }
function M.install(pkg, version, opts)
opts = opts or {}
log.fmt_debug("pypi: install %s %s", pkg, version, opts)
local ctx = installer.context()
ctx.stdio_sink.stdout(("Installing pip package %s@%s…\n"):format(pkg, version))
return pip_install({
opts.extra and ("%s[%s]==%s"):format(pkg, opts.extra, version) or ("%s==%s"):format(pkg, version),
opts.extra_packages or vim.NIL,
}, opts.install_extra_args)
end
---@param executable string
function M.bin_path(executable)
local ctx = installer.context()
return find_venv_executable(ctx, executable)
end
return M
|
