mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-06 19:01:54 +00:00
kevinveenbirkenbach
f08993049c
- add scripts/inventory/resolve.sh to compute INVENTORY_DIR based on runtime context (act/github/local) - resolve INVENTORY_DIR dynamically in Makefile via resolve.sh - enforce INVENTORY_DIR usage across all local deploy/test scripts - propagate INVENTORY_DIR into container and development CLI commands - remove hardcoded /etc/inventories paths https://chatgpt.com/share/69700d69-b270-800f-ad50-d9ff486378b9
130 lines
3.5 KiB
Python
130 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
|
|
from .common import make_compose, resolve_deploy_ids_for_app
|
|
from .storage import detect_storage_constrained
|
|
|
|
|
|
def _ensure_vault_password_file(compose, *, inventory_dir: str) -> None:
|
|
inv_root = str(inventory_dir).rstrip("/")
|
|
pw_file = f"{inv_root}/.password"
|
|
|
|
compose.exec(
|
|
[
|
|
"sh",
|
|
"-lc",
|
|
f"mkdir -p {inv_root} && "
|
|
f"[ -f {pw_file} ] || "
|
|
f"printf '%s\n' 'ci-vault-password' > {pw_file}",
|
|
],
|
|
check=True,
|
|
)
|
|
|
|
|
|
def _create_inventory(
|
|
compose,
|
|
*,
|
|
include: list[str],
|
|
storage_constrained: bool,
|
|
inventory_dir: str,
|
|
) -> None:
|
|
if not include:
|
|
raise ValueError("include must not be empty")
|
|
|
|
inv_root = str(inventory_dir).rstrip("/")
|
|
|
|
overrides = {"STORAGE_CONSTRAINED": bool(storage_constrained)}
|
|
|
|
cmd = [
|
|
"python3",
|
|
"-m",
|
|
"cli.create.inventory",
|
|
inv_root,
|
|
"--host",
|
|
"localhost",
|
|
"--ssl-disabled",
|
|
"--vars-file",
|
|
"inventory.sample.yml",
|
|
"--vars",
|
|
json.dumps(overrides),
|
|
"--include",
|
|
",".join(include),
|
|
]
|
|
|
|
compose.exec(cmd, check=True, workdir="/opt/src/infinito")
|
|
_ensure_vault_password_file(compose, inventory_dir=inv_root)
|
|
|
|
|
|
def add_parser(sub: argparse._SubParsersAction) -> None:
|
|
p = sub.add_parser(
|
|
"init", help="Create development inventory inside the infinito container."
|
|
)
|
|
p.add_argument(
|
|
"--distro",
|
|
default=os.environ.get("INFINITO_DISTRO", "arch"),
|
|
choices=["arch", "debian", "ubuntu", "fedora", "centos"],
|
|
help="Target distro (compose env INFINITO_DISTRO).",
|
|
)
|
|
|
|
p.add_argument(
|
|
"--inventory-dir",
|
|
default=os.environ.get("INVENTORY_DIR"),
|
|
required=os.environ.get("INVENTORY_DIR") is None,
|
|
help="Inventory directory to use (default: $INVENTORY_DIR).",
|
|
)
|
|
|
|
g = p.add_mutually_exclusive_group(required=True)
|
|
g.add_argument(
|
|
"--app", help="Application id (will include run_after deps automatically)."
|
|
)
|
|
g.add_argument(
|
|
"--include",
|
|
help="Comma-separated list of application ids to include (no deps resolution).",
|
|
)
|
|
p.add_argument(
|
|
"--threshold-gib",
|
|
type=int,
|
|
default=100,
|
|
help="Free-space threshold (GiB) below which STORAGE_CONSTRAINED is enabled (default: 100).",
|
|
)
|
|
p.add_argument(
|
|
"--force-storage-constrained",
|
|
choices=["true", "false"],
|
|
default=None,
|
|
help="Override storage detection explicitly.",
|
|
)
|
|
p.set_defaults(_handler=handler)
|
|
|
|
|
|
def handler(args: argparse.Namespace) -> int:
|
|
compose = make_compose(distro=args.distro)
|
|
|
|
if args.app:
|
|
include = resolve_deploy_ids_for_app(compose, args.app)
|
|
else:
|
|
include = [x.strip() for x in (args.include or "").split(",") if x.strip()]
|
|
|
|
if not include:
|
|
raise ValueError("Resolved include list is empty")
|
|
|
|
if args.force_storage_constrained is not None:
|
|
storage_constrained = args.force_storage_constrained == "true"
|
|
else:
|
|
storage_constrained = detect_storage_constrained(
|
|
compose, threshold_gib=int(args.threshold_gib)
|
|
)
|
|
|
|
_create_inventory(
|
|
compose,
|
|
include=include,
|
|
storage_constrained=storage_constrained,
|
|
inventory_dir=str(args.inventory_dir),
|
|
)
|
|
print(
|
|
f">>> Inventory initialized (include={','.join(include)} storage_constrained={storage_constrained})"
|
|
)
|
|
return 0
|