mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-02-21 21:43:52 +08:00
99 lines
2.9 KiB
Python
Executable File
99 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import subprocess
|
|
import textwrap
|
|
from pathlib import Path
|
|
import socket
|
|
|
|
# NOTE: Do NOT import anything here that needs be built (e.g. params)
|
|
from common.basedir import BASEDIR
|
|
from common.spinner import Spinner
|
|
from common.text_window import TextWindow
|
|
from system.hardware import AGNOS
|
|
from system.swaglog import cloudlog, add_file_handler
|
|
from system.version import is_dirty
|
|
|
|
MAX_CACHE_SIZE = 4e9 if "CI" in os.environ else 2e9
|
|
CACHE_DIR = Path("/data/scons_cache" if AGNOS else "/tmp/scons_cache")
|
|
|
|
TOTAL_SCONS_NODES = 2560
|
|
MAX_BUILD_PROGRESS = 100
|
|
PREBUILT = os.path.exists(os.path.join(BASEDIR, 'prebuilt'))
|
|
|
|
|
|
def build(spinner: Spinner, dirty: bool = False) -> None:
|
|
env = os.environ.copy()
|
|
env['SCONS_PROGRESS'] = "1"
|
|
nproc = os.cpu_count()
|
|
j_flag = "" if nproc is None else f"-j{nproc - 1}"
|
|
|
|
scons: subprocess.Popen = subprocess.Popen(["scons", j_flag, "--cache-populate"], cwd=BASEDIR, env=env, stderr=subprocess.PIPE)
|
|
assert scons.stderr is not None
|
|
|
|
compile_output = []
|
|
|
|
# Read progress from stderr and update spinner
|
|
while scons.poll() is None:
|
|
try:
|
|
line = scons.stderr.readline()
|
|
if line is None:
|
|
continue
|
|
line = line.rstrip()
|
|
|
|
prefix = b'progress: '
|
|
if line.startswith(prefix):
|
|
i = int(line[len(prefix):])
|
|
spinner.update_progress(MAX_BUILD_PROGRESS * min(1., i / TOTAL_SCONS_NODES), 100.)
|
|
elif len(line):
|
|
compile_output.append(line)
|
|
print(line.decode('utf8', 'replace'))
|
|
except Exception:
|
|
pass
|
|
|
|
if scons.returncode != 0:
|
|
# Read remaining output
|
|
r = scons.stderr.read().split(b'\n')
|
|
compile_output += r
|
|
|
|
# Build failed log errors
|
|
errors = [line.decode('utf8', 'replace') for line in compile_output
|
|
if any(err in line for err in [b'error: ', b'not found, needed by target'])]
|
|
error_s = "\n".join(errors)
|
|
add_file_handler(cloudlog)
|
|
cloudlog.error("scons build failed\n" + error_s)
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
try:
|
|
# doesn't even have to be reachable
|
|
s.connect(('10.255.255.255', 1))
|
|
ip = s.getsockname()[0]
|
|
except:
|
|
ip = 'N/A'
|
|
finally:
|
|
s.close()
|
|
|
|
# Show TextWindow
|
|
spinner.close()
|
|
if not os.getenv("CI"):
|
|
error_s = "\n \n".join(["\n".join(textwrap.wrap(e, 65)) for e in errors])
|
|
with TextWindow(("dragonpilot failed to build (IP: %s)\n \n" % ip) + error_s) as t:
|
|
t.wait_for_exit()
|
|
exit(1)
|
|
|
|
|
|
# enforce max cache size
|
|
cache_files = [f for f in CACHE_DIR.rglob('*') if f.is_file()]
|
|
cache_files.sort(key=lambda f: f.stat().st_mtime)
|
|
cache_size = sum(f.stat().st_size for f in cache_files)
|
|
for f in cache_files:
|
|
if cache_size < MAX_CACHE_SIZE:
|
|
break
|
|
cache_size -= f.stat().st_size
|
|
f.unlink()
|
|
|
|
|
|
if __name__ == "__main__" and not PREBUILT:
|
|
spinner = Spinner()
|
|
spinner.update_progress(0, 100)
|
|
build(spinner, is_dirty())
|