IQ.Pilot Release Commit @ bec7652

iqpilot/tools/scons/site_init.py

@@ -0,0 +1,108 @@
+import os
+import sys
+
+import SCons.Script.Main as _main
+
+try:
+  from site_tools import pretty as _pretty
+except Exception:
+  _pretty = None
+
+
+def _tty():
+  return sys.stdout.isatty() and not os.environ.get("NO_COLOR")
+
+
+_DIM = "\033[2;38;5;246m"
+_BLUE = "\033[38;5;111m"
+_GREEN = "\033[38;5;114m"
+_RED = "\033[38;5;203m"
+_RST = "\033[0m"
+
+_READING = "scons: Reading SConscript files ..."
+
+_PHASES = {
+  _READING:                                                       f"{_DIM}reading sconscripts…{_RST}",
+  "scons: done reading SConscript files.":                        f"{_DIM}sconscripts read{_RST}",
+  "scons: Building targets ...":                                  f"{_BLUE}building…{_RST}",
+  "scons: done building targets.":                                f"{_GREEN}✓ build complete{_RST}",
+  "scons: done building targets (errors occurred during build).": f"{_RED}✗ build failed{_RST}",
+  "scons: writing .sconsign file.":                               f"{_DIM}writing .sconsign{_RST}",
+  "scons: Cleaning targets ...":                                  f"{_DIM}cleaning…{_RST}",
+  "scons: done cleaning targets.":                                f"{_GREEN}✓ clean complete{_RST}",
+  "scons: done cleaning targets (errors occurred during clean).": f"{_RED}✗ clean failed{_RST}",
+}
+
+
+def _phase(text):
+  if _tty() and isinstance(text, str) and text in _PHASES:
+    return _PHASES[text]
+  return text
+
+
+def _clean(text):
+  if not (_tty() and _pretty and isinstance(text, str)):
+    return text
+  for prefix in ("Removed directory ", "Removed "):
+    if text.startswith(prefix):
+      return _pretty._format("CLEAN", text[len(prefix):])
+  return text
+
+
+class _Restyle:
+  # delegates unknown attrs (.set_mode etc.) to the wrapped DisplayEngine
+  def __init__(self, orig, transform):
+    self._orig = orig
+    self._transform = transform
+
+  def __call__(self, text, *args, **kwargs):
+    return self._orig(self._transform(text), *args, **kwargs)
+
+  def __getattr__(self, name):
+    return getattr(self._orig, name)
+
+
+if not isinstance(_main.progress_display, _Restyle):
+  _main.progress_display = _Restyle(_main.progress_display, _phase)
+if not isinstance(_main.display, _Restyle):
+  _main.display = _Restyle(_main.display, _clean)
+
+
+# SConstruct imports this instead of scons auto-loading a root site_scons dir, so the reading
+# banner is already on screen by now; rewrite that one line in place
+if not getattr(_main, "_iq_banner_restyled", False):
+  _main._iq_banner_restyled = True
+  if _tty() and _main.progress_display.print_it:
+    sys.stdout.write(f"\033[F\033[2K{_PHASES[_READING]}\n")
+    sys.stdout.flush()
+
+
+# drop only "Could not remove ... No such file" during clean; real errors still print
+import builtins as _builtins
+
+
+def _wrap_clean(orig):
+  def wrapper(self, *args, **kwargs):
+    if getattr(_builtins.print, "_iq_clean", False):
+      return orig(self, *args, **kwargs)
+    real = _builtins.print
+
+    def filtered(*a, **k):
+      if a and isinstance(a[0], str) and a[0].startswith("scons: Could not remove"):
+        if "No such file" in " ".join(str(x) for x in a):
+          return
+      return real(*a, **k)
+    filtered._iq_clean = True
+    _builtins.print = filtered
+    try:
+      return orig(self, *args, **kwargs)
+    finally:
+      _builtins.print = real
+  return wrapper
+
+
+if not getattr(_main.CleanTask.fs_delete, "_iq_wrapped", False):
+  _main.CleanTask.fs_delete = _wrap_clean(_main.CleanTask.fs_delete)
+  _main.CleanTask.remove = _wrap_clean(_main.CleanTask.remove)
+  _main.CleanTask.fs_delete._iq_wrapped = True
+  _main.CleanTask.remove._iq_wrapped = True