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
|
# 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 os
import subprocess
import buildconfig
import mozpack.path as mozpath
import six
import toml
# Try to read the package name or otherwise assume same name as the crate path.
def _get_crate_name(crate_path):
try:
with open(mozpath.join(crate_path, "Cargo.toml"), encoding="utf-8") as f:
return toml.load(f)["package"]["name"]
except Exception:
return mozpath.basename(crate_path)
CARGO_LOCK = mozpath.join(buildconfig.topsrcdir, "Cargo.lock")
CARGO_TOML = mozpath.join(buildconfig.topsrcdir, "Cargo.toml")
def _run_process(args):
env = os.environ.copy()
env["CARGO"] = str(buildconfig.substs["CARGO"])
env["RUSTC"] = str(buildconfig.substs["RUSTC"])
p = subprocess.Popen(args, env=env, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = p.communicate()
stdout = six.ensure_text(stdout)
stderr = six.ensure_text(stderr)
if p.returncode != 0:
print(stdout)
print(stderr)
return (stdout, p.returncode)
def generate_metadata(output, cargo_config):
stdout, returncode = _run_process(
[
buildconfig.substs["CARGO"],
"metadata",
"--frozen",
"--all-features",
"--format-version",
"1",
"--manifest-path",
CARGO_TOML,
]
)
if returncode != 0:
return returncode
output.write(stdout)
# This is not quite accurate, but cbindgen only cares about a subset of the
# data which, when changed, causes these files to change.
return set([CARGO_LOCK, CARGO_TOML])
def generate(output, metadata_path, cbindgen_crate_path, *in_tree_dependencies):
stdout, returncode = _run_process(
[
buildconfig.substs["CBINDGEN"],
buildconfig.topsrcdir,
"--lockfile",
CARGO_LOCK,
"--crate",
_get_crate_name(cbindgen_crate_path),
"--metadata",
metadata_path,
"--cpp-compat",
]
)
if returncode != 0:
return returncode
output.write(stdout)
deps = set()
deps.add(CARGO_LOCK)
deps.add(mozpath.join(cbindgen_crate_path, "cbindgen.toml"))
for directory in in_tree_dependencies + (cbindgen_crate_path,):
for path, dirs, files in os.walk(directory):
for file in files:
if os.path.splitext(file)[1] == ".rs":
deps.add(mozpath.join(path, file))
return deps
|
