computer-playbook/cli/administration/deploy/development/env.py

from __future__ import annotations

import os

from utils.distros import distro_names

VALID_DISTROS: tuple[str, ...] = distro_names()

TOOLS_LANE_KEYS: tuple[str, ...] = (
    "INFINITO_TOOLS_LIBRETRANSLATE_PORT",
    "INFINITO_TOOLS_LITELLM_PORT",
    "INFINITO_TOOLS_MODELS_HOST_PATH",
    "INFINITO_TOOLS_S1_PORT",
    "INFINITO_TOOLS_TRANSLATE_PORT",
)


def compose_file_args() -> list[str]:
    """Compose `-f` flags shared by up and down flows.

    Each override is gated on the resource it needs being present, not on the
    instance slot: an override whose `:?` guard is unsatisfied would abort the
    stack outright, so a checkout that was never handed a shared git dir or
    cache network simply does not layer that file in.
    """
    from .extra_mounts import parse_mounts, write_mounts_override
    from .profile import Profile

    profile = Profile()
    out = ["-f", "compose.yml"]
    if profile.shared_git_dir():
        out += ["-f", "compose/worktree.override.yml"]
    if profile.cache_stack_enabled():
        out += ["-f", "compose/cache.override.yml"]
        if profile.shared_cache_network():
            out += ["-f", "compose/cache.shared.override.yml"]
    if profile.gpu_reservable():
        out += ["-f", "compose/gpu.override.yml"]
    if (os.environ.get("INFINITO_PUBLISH_PORTS") or "").strip().lower() == "false":
        out += ["-f", "compose/noports.override.yml"]
    if all((os.environ.get(key) or "").strip() for key in TOOLS_LANE_KEYS):
        out += ["-f", "compose/tools.override.yml"]

    if profile.image_mirror_enabled():
        from utils.cache.files import PROJECT_ROOT

        from .runner_mirror import write_override

        out += ["-f", write_override(PROJECT_ROOT, out[1::2])]
    mounts = parse_mounts(os.environ.get("INFINITO_EXTRA_MOUNTS") or "")
    if mounts:
        from utils.cache.files import PROJECT_ROOT

        out += ["-f", write_mounts_override(PROJECT_ROOT, mounts)]
    return out


def resolve_distro() -> str:
    """Return INFINITO_DISTRO; raise SystemExit if missing or invalid."""
    distro = os.environ["INFINITO_DISTRO"].strip()
    if not distro:
        raise SystemExit(
            "INFINITO_DISTRO is not set. Run 'make dotenv' (or source scripts/meta/env/load.sh) "
            f"or export INFINITO_DISTRO=<{'|'.join(VALID_DISTROS)}> "
            "before invoking cli.administration.deploy.development."
        )
    if distro not in VALID_DISTROS:
        raise SystemExit(
            f"INFINITO_DISTRO={distro!r} is not a valid distro. "
            f"Valid: {', '.join(VALID_DISTROS)}."
        )
    return distro