mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-13 06:06:38 +00:00
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import TYPE_CHECKING
|
|
|
|
from . import PROJECT_ROOT
|
|
from .common import compose_file_args, resolve_distro
|
|
|
|
if TYPE_CHECKING:
|
|
import argparse
|
|
|
|
def _base_env(*, distro: str) -> dict[str, str]:
|
|
env = dict(os.environ)
|
|
env["INFINITO_DISTRO"] = distro
|
|
return env
|
|
|
|
|
|
def _compose_run(*, repo_root: Path, distro: str, args: list[str]) -> None:
|
|
cmd = ["docker", "compose", *compose_file_args(), *args]
|
|
env = _base_env(distro=distro)
|
|
env.setdefault("NIX_CONFIG", "")
|
|
subprocess.run(cmd, cwd=repo_root, env=env, check=True, text=True)
|
|
|
|
|
|
def _resolve_docker_root() -> Path:
|
|
raw = os.environ.get("INFINITO_DOCKER_VOLUME", "").strip().rstrip("/")
|
|
if not raw:
|
|
raise RuntimeError(
|
|
"INFINITO_DOCKER_VOLUME must be set (SPOT: scripts/meta/env/github.sh)"
|
|
)
|
|
return Path(raw)
|
|
|
|
|
|
def _wipe_docker_root(docker_root: Path) -> None:
|
|
if not docker_root.exists():
|
|
print(f">>> Docker root does not exist, nothing to clean: {docker_root}")
|
|
return
|
|
print(f">>> CI cleanup: wiping Docker root: {docker_root}")
|
|
shutil.rmtree(docker_root, ignore_errors=True)
|
|
docker_root.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
|
def _should_wipe_docker_root() -> bool:
|
|
if os.environ.get("INFINITO_RUNNING_ON_GITHUB") != "true":
|
|
return False
|
|
return os.environ.get("INFINITO_PRESERVE_DOCKER_CACHE", "false").lower() != "true"
|
|
|
|
|
|
def _cleanup_docker_root() -> None:
|
|
if not _should_wipe_docker_root():
|
|
docker_vol = os.environ.get("INFINITO_DOCKER_VOLUME", "(unset)")
|
|
print(f">>> Skipping Docker root wipe: {docker_vol}")
|
|
return
|
|
docker_root = _resolve_docker_root()
|
|
_wipe_docker_root(docker_root)
|
|
|
|
|
|
def down_stack(*, repo_root: Path, distro: str) -> None:
|
|
print(">>> Stopping compose stack and removing volumes")
|
|
try:
|
|
_compose_run(
|
|
repo_root=repo_root,
|
|
distro=distro,
|
|
args=["down", "--remove-orphans", "-v"],
|
|
)
|
|
finally:
|
|
_cleanup_docker_root()
|
|
|
|
|
|
def add_parser(sub: argparse._SubParsersAction) -> None:
|
|
p = sub.add_parser("down", help="Stop compose stack and remove volumes.")
|
|
p.set_defaults(_handler=handler)
|
|
|
|
|
|
def handler(args: argparse.Namespace) -> int:
|
|
down_stack(repo_root=PROJECT_ROOT, distro=resolve_distro())
|
|
return 0
|