mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ref(bootstrap.py): add eprint function

Browse Source 
just like `print` but for `stderr`
This commit is contained in:
Pouriya Jahanbakhsh 2023-09-25 06:57:13 +03:30
parent 5105b1ec58
commit 598f11a1ee
1 changed files with 41 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
src/bootstrap/bootstrap.py
View File

@ -41,6 +41,10 @@ def get_cpus():
return 1 return 1
def eprint(*args, **kwargs):
kwargs["file"] = sys.stderr
print(*args, **kwargs)
def get(base, url, path, checksums, verbose=False): def get(base, url, path, checksums, verbose=False):
with tempfile.NamedTemporaryFile(delete=False) as temp_file: with tempfile.NamedTemporaryFile(delete=False) as temp_file:
@ -57,23 +61,23 @@ def get(base, url, path, checksums, verbose=False):
if os.path.exists(path): if os.path.exists(path):
if verify(path, sha256, False): if verify(path, sha256, False):
if verbose: if verbose:
print("using already-download file", path, file=sys.stderr) eprint("using already-download file", path)
return return
else: else:
if verbose: if verbose:
print("ignoring already-download file", eprint("ignoring already-download file",
path, "due to failed verification", file=sys.stderr) path, "due to failed verification")
os.unlink(path) os.unlink(path)
download(temp_path, "{}/{}".format(base, url), True, verbose) download(temp_path, "{}/{}".format(base, url), True, verbose)
if not verify(temp_path, sha256, verbose): if not verify(temp_path, sha256, verbose):
raise RuntimeError("failed verification") raise RuntimeError("failed verification")
if verbose: if verbose:
print("moving {} to {}".format(temp_path, path), file=sys.stderr) eprint("moving {} to {}".format(temp_path, path))
shutil.move(temp_path, path) shutil.move(temp_path, path)
finally: finally:
if os.path.isfile(temp_path): if os.path.isfile(temp_path):
if verbose: if verbose:
print("removing", temp_path, file=sys.stderr) eprint("removing", temp_path)
os.unlink(temp_path) os.unlink(temp_path)
@ -83,7 +87,7 @@ def download(path, url, probably_big, verbose):
_download(path, url, probably_big, verbose, True) _download(path, url, probably_big, verbose, True)
return return
except RuntimeError: except RuntimeError:
print("\nspurious failure, trying again", file=sys.stderr) eprint("\nspurious failure, trying again")
_download(path, url, probably_big, verbose, False) _download(path, url, probably_big, verbose, False)
@ -94,7 +98,7 @@ def _download(path, url, probably_big, verbose, exception):
# - If we are on win32 fallback to powershell # - If we are on win32 fallback to powershell
# - Otherwise raise the error if appropriate # - Otherwise raise the error if appropriate
if probably_big or verbose: if probably_big or verbose:
print("downloading {}".format(url), file=sys.stderr) eprint("downloading {}".format(url))
try: try:
if (probably_big or verbose) and "GITHUB_ACTIONS" not in os.environ: if (probably_big or verbose) and "GITHUB_ACTIONS" not in os.environ:
@ -129,20 +133,20 @@ def _download(path, url, probably_big, verbose, exception):
def verify(path, expected, verbose): def verify(path, expected, verbose):
"""Check if the sha256 sum of the given path is valid""" """Check if the sha256 sum of the given path is valid"""
if verbose: if verbose:
print("verifying", path, file=sys.stderr) eprint("verifying", path)
with open(path, "rb") as source: with open(path, "rb") as source:
found = hashlib.sha256(source.read()).hexdigest() found = hashlib.sha256(source.read()).hexdigest()
verified = found == expected verified = found == expected
if not verified: if not verified:
print("invalid checksum:\n" eprint("invalid checksum:\n"
" found: {}\n" " found: {}\n"
" expected: {}".format(found, expected), file=sys.stderr) " expected: {}".format(found, expected))
return verified return verified
def unpack(tarball, tarball_suffix, dst, verbose=False, match=None): def unpack(tarball, tarball_suffix, dst, verbose=False, match=None):
"""Unpack the given tarball file""" """Unpack the given tarball file"""
print("extracting", tarball, file=sys.stderr) eprint("extracting", tarball)
fname = os.path.basename(tarball).replace(tarball_suffix, "") fname = os.path.basename(tarball).replace(tarball_suffix, "")
with contextlib.closing(tarfile.open(tarball)) as tar: with contextlib.closing(tarfile.open(tarball)) as tar:
for member in tar.getnames(): for member in tar.getnames():
@ -155,7 +159,7 @@ def unpack(tarball, tarball_suffix, dst, verbose=False, match=None):
dst_path = os.path.join(dst, name) dst_path = os.path.join(dst, name)
if verbose: if verbose:
print(" extracting", member, file=sys.stderr) eprint(" extracting", member)
tar.extract(member, dst) tar.extract(member, dst)
src_path = os.path.join(dst, member) src_path = os.path.join(dst, member)
if os.path.isdir(src_path) and os.path.exists(dst_path): if os.path.isdir(src_path) and os.path.exists(dst_path):
@ -167,7 +171,7 @@ def unpack(tarball, tarball_suffix, dst, verbose=False, match=None):
def run(args, verbose=False, exception=False, is_bootstrap=False, **kwargs): def run(args, verbose=False, exception=False, is_bootstrap=False, **kwargs):
"""Run a child program in a new process""" """Run a child program in a new process"""
if verbose: if verbose:
print("running: " + ' '.join(args), file=sys.stderr) eprint("running: " + ' '.join(args))
sys.stdout.flush() sys.stdout.flush()
# Ensure that the .exe is used on Windows just in case a Linux ELF has been # Ensure that the .exe is used on Windows just in case a Linux ELF has been
# compiled in the same directory. # compiled in the same directory.
@ -207,8 +211,8 @@ def require(cmd, exit=True, exception=False):
if exception: if exception:
raise raise
elif exit: elif exit:
print("error: unable to run `{}`: {}".format(' '.join(cmd), exc), file=sys.stderr) eprint("error: unable to run `{}`: {}".format(' '.join(cmd), exc))
print("Please make sure it's installed and in the path.", file=sys.stderr) eprint("Please make sure it's installed and in the path.")
sys.exit(1) sys.exit(1)
return None return None
@ -239,14 +243,12 @@ def default_build_triple(verbose):
host = next(x for x in version.split('\n') if x.startswith("host: ")) host = next(x for x in version.split('\n') if x.startswith("host: "))
triple = host.split("host: ")[1] triple = host.split("host: ")[1]
if verbose: if verbose:
print("detected default triple {} from pre-installed rustc".format(triple), eprint("detected default triple {} from pre-installed rustc".format(triple))
file=sys.stderr)
return triple return triple
except Exception as e: except Exception as e:
if verbose: if verbose:
print("pre-installed rustc not detected: {}".format(e), eprint("pre-installed rustc not detected: {}".format(e))
file=sys.stderr) eprint("falling back to auto-detect")
print("falling back to auto-detect", file=sys.stderr)
required = not platform_is_win32() required = not platform_is_win32()
uname = require(["uname", "-smp"], exit=required) uname = require(["uname", "-smp"], exit=required)
@ -672,15 +674,14 @@ class RustBuild(object):
if not is_nixos: if not is_nixos:
in_nix_shell = os.getenv('IN_NIX_SHELL') in_nix_shell = os.getenv('IN_NIX_SHELL')
if in_nix_shell: if in_nix_shell:
print("The IN_NIX_SHELL environment variable is `{}`;".format(in_nix_shell), eprint("The IN_NIX_SHELL environment variable is `{}`;".format(in_nix_shell),
"you may need to set `patch-binaries-for-nix=true` in config.toml", "you may need to set `patch-binaries-for-nix=true` in config.toml")
file=sys.stderr)
return is_nixos return is_nixos
answer = self._should_fix_bins_and_dylibs = get_answer() answer = self._should_fix_bins_and_dylibs = get_answer()
if answer: if answer:
print("info: You seem to be using Nix.", file=sys.stderr) eprint("info: You seem to be using Nix.")
return answer return answer
def fix_bin_or_dylib(self, fname): def fix_bin_or_dylib(self, fname):
@ -693,7 +694,7 @@ class RustBuild(object):
Please see https://nixos.org/patchelf.html for more information Please see https://nixos.org/patchelf.html for more information
""" """
assert self._should_fix_bins_and_dylibs is True assert self._should_fix_bins_and_dylibs is True
print("attempting to patch", fname, file=sys.stderr) eprint("attempting to patch", fname)
# Only build `.nix-deps` once. # Only build `.nix-deps` once.
nix_deps_dir = self.nix_deps_dir nix_deps_dir = self.nix_deps_dir
@ -726,7 +727,7 @@ class RustBuild(object):
"nix-build", "-E", nix_expr, "-o", nix_deps_dir, "nix-build", "-E", nix_expr, "-o", nix_deps_dir,
]) ])
except subprocess.CalledProcessError as reason: except subprocess.CalledProcessError as reason:
print("warning: failed to call nix-build:", reason, file=sys.stderr) eprint("warning: failed to call nix-build:", reason)
return return
self.nix_deps_dir = nix_deps_dir self.nix_deps_dir = nix_deps_dir
@ -746,7 +747,7 @@ class RustBuild(object):
try: try:
subprocess.check_output([patchelf] + patchelf_args + [fname]) subprocess.check_output([patchelf] + patchelf_args + [fname])
except subprocess.CalledProcessError as reason: except subprocess.CalledProcessError as reason:
print("warning: failed to call patchelf:", reason, file=sys.stderr) eprint("warning: failed to call patchelf:", reason)
return return
def rustc_stamp(self): def rustc_stamp(self):
@ -888,7 +889,7 @@ class RustBuild(object):
if "GITHUB_ACTIONS" in env: if "GITHUB_ACTIONS" in env:
print("::group::Building bootstrap") print("::group::Building bootstrap")
else: else:
print("Building bootstrap", file=sys.stderr) eprint("Building bootstrap")
args = self.build_bootstrap_cmd(env) args = self.build_bootstrap_cmd(env)
# Run this from the source directory so cargo finds .cargo/config # Run this from the source directory so cargo finds .cargo/config
@ -997,12 +998,9 @@ class RustBuild(object):
if 'SUDO_USER' in os.environ and not self.use_vendored_sources: if 'SUDO_USER' in os.environ and not self.use_vendored_sources:
if os.getuid() == 0: if os.getuid() == 0:
self.use_vendored_sources = True self.use_vendored_sources = True
print('info: looks like you\'re trying to run this command as root', eprint('info: looks like you\'re trying to run this command as root')
file=sys.stderr) eprint(' and so in order to preserve your $HOME this will now')
print(' and so in order to preserve your $HOME this will now', eprint(' use vendored sources by default.')
file=sys.stderr)
print(' use vendored sources by default.',
file=sys.stderr)
cargo_dir = os.path.join(self.rust_root, '.cargo') cargo_dir = os.path.join(self.rust_root, '.cargo')
if self.use_vendored_sources: if self.use_vendored_sources:
@ -1012,18 +1010,14 @@ class RustBuild(object):
"--sync ./src/tools/rust-analyzer/Cargo.toml " \ "--sync ./src/tools/rust-analyzer/Cargo.toml " \
"--sync ./compiler/rustc_codegen_cranelift/Cargo.toml " \ "--sync ./compiler/rustc_codegen_cranelift/Cargo.toml " \
"--sync ./src/bootstrap/Cargo.toml " "--sync ./src/bootstrap/Cargo.toml "
print('error: vendoring required, but vendor directory does not exist.', eprint('error: vendoring required, but vendor directory does not exist.')
file=sys.stderr) eprint(' Run `cargo vendor {}` to initialize the '
print(' Run `cargo vendor {}` to initialize the ' 'vendor directory.'.format(sync_dirs))
'vendor directory.'.format(sync_dirs), eprint('Alternatively, use the pre-vendored `rustc-src` dist component.')
file=sys.stderr)
print('Alternatively, use the pre-vendored `rustc-src` dist component.',
file=sys.stderr)
raise Exception("{} not found".format(vendor_dir)) raise Exception("{} not found".format(vendor_dir))
if not os.path.exists(cargo_dir): if not os.path.exists(cargo_dir):
print('error: vendoring required, but .cargo/config does not exist.', eprint('error: vendoring required, but .cargo/config does not exist.')
file=sys.stderr)
raise Exception("{} not found".format(cargo_dir)) raise Exception("{} not found".format(cargo_dir))
else: else:
if os.path.exists(cargo_dir): if os.path.exists(cargo_dir):
@ -1117,10 +1111,9 @@ def main():
# If the user is asking for help, let them know that the whole download-and-build # If the user is asking for help, let them know that the whole download-and-build
# process has to happen before anything is printed out. # process has to happen before anything is printed out.
if help_triggered: if help_triggered:
print( eprint(
"info: Downloading and building bootstrap before processing --help command.\n" "info: Downloading and building bootstrap before processing --help command.\n"
" See src/bootstrap/README.md for help with common commands." " See src/bootstrap/README.md for help with common commands.")
, file=sys.stderr)
exit_code = 0 exit_code = 0
success_word = "successfully" success_word = "successfully"
@ -1131,12 +1124,11 @@ def main():
exit_code = error.code exit_code = error.code
else: else:
exit_code = 1 exit_code = 1
print(error, file=sys.stderr) eprint(error)
success_word = "unsuccessfully" success_word = "unsuccessfully"
if not help_triggered: if not help_triggered:
print("Build completed", success_word, "in", format_build_time(time() - start_time), eprint("Build completed", success_word, "in", format_build_time(time() - start_time))
file=sys.stderr)
sys.exit(exit_code) sys.exit(exit_code)