mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-23 12:51:06 +00:00
97 lines
2.6 KiB
Python
97 lines
2.6 KiB
Python
# 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
|