Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
77 changes: 48 additions & 29 deletions Tools/wasm/wasi/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,30 @@
WASMTIME_HOST_RUNNER_VAR = f"{{{WASMTIME_VAR_NAME}}}"


def separator():
"""Print a separator line across the terminal width."""
try:
tput_output = subprocess.check_output(
["tput", "cols"], encoding="utf-8"
)
except subprocess.CalledProcessError:
terminal_width = 80
else:
terminal_width = int(tput_output.strip())
print("⎯" * terminal_width)


def log(emoji, message, *, spacing=None):
"""Print a notification with an emoji.

If 'spacing' is None, calculate the spacing based on the number of code points
in the emoji as terminals "eat" a space when the emoji has multiple code points.
"""
if spacing is None:
spacing = " " if len(emoji) == 1 else " "
print("".join([emoji, spacing, message]))


def updated_env(updates={}):
"""Create a new dict representing the environment to use.

Expand All @@ -60,9 +84,10 @@ def updated_env(updates={}):
if os.environ.get(key) != value:
env_diff[key] = value

print("🌎 Environment changes:")
for key in sorted(env_diff.keys()):
print(f" {key}={env_diff[key]}")
env_vars = (
f"\n {key}={item}" for key, item in sorted(env_diff.items())
)
log("🌎", f"Environment changes:{''.join(env_vars)}")

return environment

Expand All @@ -77,22 +102,14 @@ def wrapper(context):

if callable(working_dir):
working_dir = working_dir(context)
try:
tput_output = subprocess.check_output(
["tput", "cols"], encoding="utf-8"
)
except subprocess.CalledProcessError:
terminal_width = 80
else:
terminal_width = int(tput_output.strip())
print("⎯" * terminal_width)
print("📁", working_dir)
separator()
log("📁", os.fsdecode(working_dir))
if (
clean_ok
and getattr(context, "clean", False)
and working_dir.exists()
):
print("🚮 Deleting directory (--clean)...")
log("🚮", "Deleting directory (--clean)...")
shutil.rmtree(working_dir)

working_dir.mkdir(parents=True, exist_ok=True)
Expand All @@ -116,7 +133,7 @@ def call(command, *, context=None, quiet=False, logdir=None, **kwargs):
elif quiet and logdir is None:
raise ValueError("When quiet is True, logdir must be specified")

print("❯", " ".join(map(str, command)))
log("❯", " ".join(map(str, command)), spacing=" ")
if not quiet:
stdout = None
stderr = None
Expand All @@ -130,7 +147,7 @@ def call(command, *, context=None, quiet=False, logdir=None, **kwargs):
suffix=".log",
)
stderr = subprocess.STDOUT
print(f"📝 Logging output to {stdout.name} (--quiet)...")
log("📝", f"Logging output to {stdout.name} (--quiet)...")

subprocess.check_call(command, **kwargs, stdout=stdout, stderr=stderr)

Expand Down Expand Up @@ -163,11 +180,11 @@ def configure_build_python(context, working_dir):
"""Configure the build/host Python."""
if LOCAL_SETUP.exists():
if LOCAL_SETUP.read_bytes() == LOCAL_SETUP_MARKER:
print(f"👍 {LOCAL_SETUP} exists ...")
log("👍", f"{LOCAL_SETUP} exists ...")
else:
print(f"⚠️ {LOCAL_SETUP} exists, but has unexpected contents")
log("⚠️", f"{LOCAL_SETUP} exists, but has unexpected contents")
else:
print(f"📝 Creating {LOCAL_SETUP} ...")
log("📝", f"Creating {LOCAL_SETUP} ...")
LOCAL_SETUP.write_bytes(LOCAL_SETUP_MARKER)

configure = [os.path.relpath(CHECKOUT / "configure", working_dir)]
Expand All @@ -191,7 +208,7 @@ def make_build_python(context, working_dir):
]
version = subprocess.check_output(cmd, encoding="utf-8").strip()

print(f"🎉 {binary} {version}")
log("🎉", f"{binary} {version}")


def find_wasi_sdk():
Expand Down Expand Up @@ -228,9 +245,10 @@ def find_wasi_sdk():
# supported version is a prefix of the found version (e.g. `25` and `2567`).
if not found_version.startswith(f"{WASI_SDK_VERSION}."):
major_version = found_version.partition(".")[0]
print(
f"⚠️ Found WASI SDK {major_version}, "
f"but WASI SDK {WASI_SDK_VERSION} is the supported version"
log(
"⚠️",
f" Found WASI SDK {major_version}, "
f"but WASI SDK {WASI_SDK_VERSION} is the supported version",
)

return wasi_sdk_path
Expand Down Expand Up @@ -349,7 +367,7 @@ def configure_wasi_python(context, working_dir):
with exec_script.open("w", encoding="utf-8") as file:
file.write(f'#!/bin/sh\nexec {host_runner} {python_wasm} "$@"\n')
exec_script.chmod(0o755)
print(f"🏃‍♀️ Created {exec_script} (--host-runner)... ")
log("🏃", f"Created {exec_script} (--host-runner)... ")
sys.stdout.flush()


Expand All @@ -364,9 +382,10 @@ def make_wasi_python(context, working_dir):

exec_script = working_dir / "python.sh"
call([exec_script, "--version"], quiet=False)
print(
f"🎉 Use `{exec_script.relative_to(context.init_dir)}` "
"to run CPython w/ the WASI host specified by --host-runner"
log(
"🎉",
f"Use `{exec_script.relative_to(context.init_dir)}` "
"to run CPython w/ the WASI host specified by --host-runner",
)


Expand All @@ -385,12 +404,12 @@ def build_all(context):
def clean_contents(context):
"""Delete all files created by this script."""
if CROSS_BUILD_DIR.exists():
print(f"🧹 Deleting {CROSS_BUILD_DIR} ...")
log("🧹", f"Deleting {CROSS_BUILD_DIR} ...")
shutil.rmtree(CROSS_BUILD_DIR)

if LOCAL_SETUP.exists():
if LOCAL_SETUP.read_bytes() == LOCAL_SETUP_MARKER:
print(f"🧹 Deleting generated {LOCAL_SETUP} ...")
log("🧹", f"Deleting generated {LOCAL_SETUP} ...")


def main():
Expand Down
Loading