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
|
# 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 argparse
import os
import subprocess
import sys
def parse_outputs(crate_output, dep_outputs, pass_l_flag):
env = {}
args = []
def parse_line(line):
if line.startswith("cargo:"):
return line[len("cargo:") :].split("=", 1)
def parse_file(f):
with open(f) as fh:
return [parse_line(line.rstrip()) for line in fh.readlines()]
for f in dep_outputs:
for entry in parse_file(f):
if not entry:
continue
key, value = entry
if key == "rustc-link-search":
args += ["-L", value]
elif key == "rustc-flags":
flags = value.split()
for flag, val in zip(flags[0::2], flags[1::2]):
if flag == "-l" and f == crate_output:
args += ["-l", val]
elif flag == "-L":
args += ["-L", val]
else:
raise Exception(
"Unknown flag passed through "
'"cargo:rustc-flags": "%s"' % flag
)
elif key == "rustc-link-lib" and f == crate_output:
args += ["-l", value]
elif key == "rustc-cfg" and f == crate_output:
args += ["--cfg", value]
elif key == "rustc-env" and f == crate_output:
env_key, env_value = value.split("=", 1)
env[env_key] = env_value
elif key == "rerun-if-changed":
pass
elif key == "rerun-if-env-changed":
pass
elif key == "warning":
pass
elif key:
# Todo: Distinguish between direct and transitive
# dependencies so we can pass metadata environment
# variables correctly.
pass
return env, args
def wrap_rustc(args):
parser = argparse.ArgumentParser()
parser.add_argument("--crate-out", nargs="?")
parser.add_argument("--deps-out", nargs="*")
parser.add_argument("--cwd")
parser.add_argument("--pass-l-flag", action="store_true")
parser.add_argument("--cmd", nargs=argparse.REMAINDER)
args = parser.parse_args(args)
new_env, new_args = parse_outputs(args.crate_out, args.deps_out, args.pass_l_flag)
os.environ.update(new_env)
return subprocess.Popen(args.cmd + new_args, cwd=args.cwd).wait()
if __name__ == "__main__":
sys.exit(wrap_rustc(sys.argv[1:]))
|
