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
|
#!/usr/bin/env python3
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import sys
import yaml
from mozbuild.shellutil import quote as shellquote
from vsdownload import (
getArgsParser,
getManifest,
getPackages,
getSelectedPackages,
lowercaseIgnores,
setPackageSelection,
)
if __name__ == "__main__":
parser = getArgsParser()
parser.add_argument("-o", dest="output", required=True, help="Output file")
parser.add_argument(
"--exclude", default=[], nargs="+", help="Patterns of file names to exclude"
)
args = parser.parse_args()
lowercaseIgnores(args)
packages = getPackages(getManifest(args))
setPackageSelection(args, packages)
selected = getSelectedPackages(packages, args)
reduced = []
# Filter-out data we won't be using.
for s in selected:
type = s["type"]
if type == "Component" or type == "Workload" or type == "Group":
continue
if type == "Vsix" or s["id"].startswith(("Win10SDK", "Win11SDK")):
filtered = {k: v for k, v in s.items() if k in ("type", "id", "version")}
filtered["payloads"] = [
{
k: v
for k, v in payload.items()
if k in ("fileName", "sha256", "size", "url")
}
for payload in s["payloads"]
if payload["fileName"].endswith((".cab", ".msi", ".vsix"))
and not any(e in payload["fileName"] for e in args.exclude)
]
reduced.append(filtered)
with open(args.output, "w") as out:
print("# Generated with:", file=out)
print(
"# ./mach python --virtualenv build build/vs/generate_yaml.py \\", file=out
)
for i, arg_ in enumerate(sys.argv[1:]):
arg = shellquote(arg_)
if i < len(sys.argv) - 2:
print("# ", arg, "\\", file=out)
else:
print("# ", arg, file=out)
print(yaml.dump(reduced), file=out)
|
