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
|
local installer = require "mason-core.installer"
local process = require "mason-core.process"
local platform = require "mason-core.platform"
local spawn = require "mason-core.spawn"
local a = require "mason-core.async"
local Optional = require "mason-core.optional"
local _ = require "mason-core.functional"
local M = {}
local create_bin_path = _.if_else(_.always(platform.is.win), _.format "%s.exe", _.identity)
---@param packages string[]
local function with_receipt(packages)
return function()
local ctx = installer.context()
ctx.receipt:with_primary_source(ctx.receipt.go(packages[1]))
-- Install secondary packages
for i = 2, #packages do
local pkg = packages[i]
ctx.receipt:with_secondary_source(ctx.receipt.go(pkg))
end
end
end
---@async
---@param packages { [number]: string, bin: string[] | nil }: The go packages to install. The first item in this list will be the recipient of the requested version, if set.
function M.packages(packages)
return function()
M.install(packages).with_receipt()
end
end
---@async
---@param packages { [number]: string, bin: string[] | nil }: The go packages to install. The first item in this list will be the recipient of the requested version, if set.
function M.install(packages)
local ctx = installer.context()
local env = {
GOBIN = ctx.cwd:get(),
}
-- Install the head package
do
local head_package = packages[1]
local version = ctx.requested_version:or_else "latest"
ctx.spawn.go {
"install",
"-v",
("%s@%s"):format(head_package, version),
env = env,
}
end
-- Install secondary packages
for i = 2, #packages do
ctx.spawn.go { "install", "-v", ("%s@latest"):format(packages[i]), env = env }
end
if packages.bin then
_.each(function(executable)
ctx:link_bin(executable, create_bin_path(executable))
end, packages.bin)
end
return {
with_receipt = with_receipt(packages),
}
end
---@param output string: The output from `go version -m` command.
function M.parse_mod_version_output(output)
---@type {path: string[], mod: string[], dep: string[], build: string[]}
local result = {}
local lines = vim.split(output, "\n")
for _, line in ipairs { unpack(lines, 2) } do
local type, id, value = unpack(vim.split(line, "%s+", { trimempty = true }))
if type and id then
result[type] = result[type] or {}
result[type][id] = value or ""
end
end
return result
end
local trim_wildcard_suffix = _.gsub("/%.%.%.$", "")
---@param pkg string
function M.parse_package_mod(pkg)
if _.starts_with("github.com", pkg) then
local components = _.split("/", pkg)
return trim_wildcard_suffix(_.join("/", {
components[1], -- github.com
components[2], -- owner
components[3], -- repo
}))
elseif _.starts_with("golang.org", pkg) then
local components = _.split("/", pkg)
return trim_wildcard_suffix(_.join("/", {
components[1], -- golang.org
components[2], -- x
components[3], -- owner
components[4], -- repo
}))
else
return trim_wildcard_suffix(pkg)
end
end
---@async
---@param receipt InstallReceipt
---@param install_dir string
function M.get_installed_primary_package_version(receipt, install_dir)
if vim.in_fast_event() then
a.scheduler()
end
local normalized_pkg_name = trim_wildcard_suffix(receipt.primary_source.package)
-- trims e.g. golang.org/x/tools/gopls to gopls
local executable = vim.fn.fnamemodify(normalized_pkg_name, ":t")
return spawn
.go({
"version",
"-m",
platform.is_win and ("%s.exe"):format(executable) or executable,
cwd = install_dir,
})
:map_catching(function(result)
local parsed_output = M.parse_mod_version_output(result.stdout)
return Optional.of_nilable(parsed_output.mod[M.parse_package_mod(receipt.primary_source.package)])
:or_else_throw "Failed to parse mod version"
end)
end
---@async
---@param receipt InstallReceipt
---@param install_dir string
function M.check_outdated_primary_package(receipt, install_dir)
local normalized_pkg_name = M.parse_package_mod(receipt.primary_source.package)
return spawn
.go({
"list",
"-json",
"-m",
("%s@latest"):format(normalized_pkg_name),
cwd = install_dir,
})
:map_catching(function(result)
---@type {Path: string, Version: string}
local output = vim.json.decode(result.stdout)
return Optional.of_nilable(output.Version)
:map(function(latest_version)
local installed_version =
M.get_installed_primary_package_version(receipt, install_dir):get_or_throw()
if installed_version ~= latest_version then
return {
name = normalized_pkg_name,
current_version = assert(installed_version),
latest_version = assert(latest_version),
}
end
end)
:or_else_throw "Primary package is not outdated."
end)
end
---@param install_dir string
function M.env(install_dir)
return {
PATH = process.extend_path { install_dir },
}
end
return M
|
