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
|
import subprocess
from setuptools.errors import PlatformError
from functools import lru_cache
from typing import Dict, List, NewType, Optional
from semantic_version import Version
def get_rust_version() -> Optional[Version]: # type: ignore[no-any-unimported]
try:
# first line of rustc -Vv is something like
# rustc 1.61.0 (fe5b13d68 2022-05-18)
return Version(_rust_version().split(" ")[1])
except (subprocess.CalledProcessError, OSError):
return None
_HOST_LINE_START = "host: "
def get_rust_host() -> str:
# rustc -Vv has a line denoting the host which cargo uses to decide the
# default target, e.g.
# host: aarch64-apple-darwin
for line in _rust_version_verbose().splitlines():
if line.startswith(_HOST_LINE_START):
return line[len(_HOST_LINE_START) :].strip()
raise PlatformError("Could not determine rust host")
RustCfgs = NewType("RustCfgs", Dict[str, Optional[str]])
def get_rustc_cfgs(target_triple: Optional[str]) -> RustCfgs:
cfgs = RustCfgs({})
for entry in get_rust_target_info(target_triple):
maybe_split = entry.split("=", maxsplit=1)
if len(maybe_split) == 2:
cfgs[maybe_split[0]] = maybe_split[1].strip('"')
else:
assert len(maybe_split) == 1
cfgs[maybe_split[0]] = None
return cfgs
@lru_cache()
def get_rust_target_info(target_triple: Optional[str] = None) -> List[str]:
cmd = ["rustc", "--print", "cfg"]
if target_triple:
cmd.extend(["--target", target_triple])
output = subprocess.check_output(cmd, text=True)
return output.splitlines()
@lru_cache()
def get_rust_target_list() -> List[str]:
output = subprocess.check_output(["rustc", "--print", "target-list"], text=True)
return output.splitlines()
@lru_cache()
def _rust_version() -> str:
return subprocess.check_output(["rustc", "-V"], text=True)
@lru_cache()
def _rust_version_verbose() -> str:
return subprocess.check_output(["rustc", "-Vv"], text=True)
|
