aboutsummaryrefslogtreecommitdiffstats
path: root/lua/mason-core/installer/compiler/compilers/pypi.lua
blob: 3e2ca141ae3d070540b14cf06aa04805b4337338 (plain) (blame)
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
local Result = require "mason-core.result"
local _ = require "mason-core.functional"
local providers = require "mason-core.providers"
local settings = require "mason.settings"

local M = {}

---@class PypiSource : RegistryPackageSource
---@field extra_packages? string[]

---@param source PypiSource
---@param purl Purl
function M.parse(source, purl)
    ---@class ParsedPypiSource : ParsedPackageSource
    local parsed_source = {
        package = purl.name,
        version = purl.version --[[ @as string ]],
        extra = _.path({ "qualifiers", "extra" }, purl),
        extra_packages = source.extra_packages,
        pip = {
            upgrade = settings.current.pip.upgrade_pip,
            extra_args = settings.current.pip.install_args,
        },
    }

    return Result.success(parsed_source)
end

---@async
---@param ctx InstallContext
---@param source ParsedPypiSource
function M.install(ctx, source)
    local pypi = require "mason-core.installer.managers.pypi"

    return Result.try(function(try)
        try(pypi.init {
            package = {
                name = source.package,
                version = source.version,
            },
            upgrade_pip = source.pip.upgrade,
            install_extra_args = source.pip.extra_args,
        })
        try(pypi.install(source.package, source.version, {
            extra = source.extra,
            extra_packages = source.extra_packages,
            install_extra_args = source.pip.extra_args,
        }))
    end)
end

---@async
---@param purl Purl
function M.get_versions(purl)
    return providers.pypi.get_all_versions(purl.name)
end

return M