dds/tools/bootstrap.py

import argparse
from pathlib import Path
import subprocess
import os
from typing import Sequence
import sys
import shutil

BOOTSTRAP_PHASES = [
    'bootstrap-p1',
    'bootstrap-p2',
]

HERE = Path(__file__).parent.absolute()
PROJECT_ROOT = HERE.parent
BUILD_DIR = PROJECT_ROOT / '_build'
BOOTSTRAP_BASE_DIR = BUILD_DIR / '_bootstrap'
PREBUILT_DIR = PROJECT_ROOT / '_prebuilt'

EXE_SUFFIX = '.exe' if os.name == 'nt' else ''


def _run_quiet(cmd, **kwargs) -> None:
    cmd = [str(s) for s in cmd]
    res = subprocess.run(
        cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, **kwargs)
    if res.returncode != 0:
        print(f'Subprocess command {cmd} failed '
              f'[{res.returncode}]:\n{res.stdout.decode()}')
        raise subprocess.CalledProcessError(res.returncode, cmd)


def _clone_bootstrap_phase(ph: str) -> Path:
    print(f'Clone revision: {ph}')
    bts_dir = BOOTSTRAP_BASE_DIR / ph
    if bts_dir.exists():
        shutil.rmtree(bts_dir)
    _run_quiet([
        'git',
        'clone',
        '--depth=1',
        f'--branch={ph}',
        f'file://{PROJECT_ROOT}',
        bts_dir,
    ])
    return bts_dir


def _build_bootstrap_phase(ph: str, bts_dir: Path,
                           args: argparse.Namespace) -> None:
    print(f'Build revision: {ph} [This may take a moment]')
    env = os.environ.copy()
    env['DDS_BOOTSTRAP_PREV_EXE'] = str(PREBUILT_DIR / 'dds')
    _run_quiet(
        [
            sys.executable,
            '-u',
            str(bts_dir / 'tools/build.py'),
            f'--cxx={args.cxx}',
        ],
        env=env,
    )


def _pull_executable(bts_dir: Path) -> Path:
    prebuild_dir = (PROJECT_ROOT / '_prebuilt')
    prebuild_dir.mkdir(exist_ok=True)
    generated = list(bts_dir.glob(f'_build/dds{EXE_SUFFIX}'))
    assert len(generated) == 1, repr(generated)
    exe, = generated
    dest = prebuild_dir / exe.name
    if dest.exists():
        dest.unlink()
    exe.rename(dest)
    return dest


def _run_boot_phase(phase: str, args: argparse.Namespace) -> Path:
    bts_dir = _clone_bootstrap_phase(phase)
    _build_bootstrap_phase(phase, bts_dir, args)
    return _pull_executable(bts_dir)


def main(argv: Sequence[str]) -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--cxx', help='The C++ compiler to use for the build', required=True)
    args = parser.parse_args(argv)
    for idx, phase in enumerate(BOOTSTRAP_PHASES):
        print(f'Bootstrap phase [{idx+1}/{len(BOOTSTRAP_PHASES)}]')
        exe = _run_boot_phase(phase, args)

    print(f'A bootstrapped DDS executable has been generated: {exe}')

    return 0


if __name__ == "__main__":
    sys.exit(main(sys.argv[1:]))