102 lines
3.7 KiB
Bash
102 lines
3.7 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cd "$ROOT"
|
|
|
|
KEEP_LATEST="${KEEP_LATEST:-10}"
|
|
MAX_DELETE="${MAX_DELETE:-100}"
|
|
EXPORT_TYPE="${EXPORT_TYPE:-project_report_html}"
|
|
CLEANUP_MODE="${CLEANUP_MODE:-auto}"
|
|
CLEANUP_CONTAINER="${CLEANUP_CONTAINER:-geointel}"
|
|
|
|
cleanup_args=(
|
|
"--keep-latest" "${KEEP_LATEST}"
|
|
"--max-delete" "${MAX_DELETE}"
|
|
"--export-type" "${EXPORT_TYPE}"
|
|
)
|
|
|
|
run_cleanup() {
|
|
case "${CLEANUP_MODE}" in
|
|
local)
|
|
python scripts/cleanup_demo_artifacts.py "${cleanup_args[@]}"
|
|
;;
|
|
compose)
|
|
docker compose exec -T backend python scripts/cleanup_demo_artifacts.py "${cleanup_args[@]}"
|
|
;;
|
|
container)
|
|
docker exec "${CLEANUP_CONTAINER}" /opt/geointel/venv/bin/python /app/scripts/cleanup_demo_artifacts.py "${cleanup_args[@]}"
|
|
;;
|
|
auto)
|
|
if command -v docker >/dev/null 2>&1 && docker compose ps --services --filter status=running 2>/dev/null | grep -qx "backend"; then
|
|
docker compose exec -T backend python scripts/cleanup_demo_artifacts.py "${cleanup_args[@]}"
|
|
elif command -v docker >/dev/null 2>&1 && docker ps --format '{{.Names}}' | grep -qx "${CLEANUP_CONTAINER}"; then
|
|
docker exec "${CLEANUP_CONTAINER}" /opt/geointel/venv/bin/python /app/scripts/cleanup_demo_artifacts.py "${cleanup_args[@]}"
|
|
else
|
|
cat >&2 <<EOF
|
|
No running GeoIntel backend runtime found.
|
|
Start Docker first, then run one of:
|
|
docker compose up -d backend
|
|
CLEANUP_MODE=compose bash scripts/verify_demo_cleanup_dry_run.sh
|
|
CLEANUP_MODE=container CLEANUP_CONTAINER=geointel bash scripts/verify_demo_cleanup_dry_run.sh
|
|
CLEANUP_MODE=local bash scripts/verify_demo_cleanup_dry_run.sh
|
|
EOF
|
|
return 1
|
|
fi
|
|
;;
|
|
*)
|
|
echo "Unsupported CLEANUP_MODE '${CLEANUP_MODE}'. Use auto, local, compose or container." >&2
|
|
return 1
|
|
;;
|
|
esac
|
|
}
|
|
|
|
tmp_output="$(mktemp)"
|
|
trap 'rm -f "${tmp_output}"' EXIT
|
|
|
|
run_cleanup | tee "${tmp_output}" >/dev/null
|
|
|
|
python - "${tmp_output}" "${KEEP_LATEST}" "${MAX_DELETE}" "${EXPORT_TYPE}" <<'PY'
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
output_path = Path(sys.argv[1])
|
|
expected_keep_latest = int(sys.argv[2])
|
|
expected_max_delete = int(sys.argv[3])
|
|
expected_export_type = sys.argv[4]
|
|
|
|
summary = json.loads(output_path.read_text(encoding="utf-8"))
|
|
|
|
if summary.get("dry_run") is not True:
|
|
raise SystemExit("cleanup smoke expected dry_run=true")
|
|
if summary.get("deleted_export_count") != 0:
|
|
raise SystemExit("cleanup smoke expected deleted_export_count=0")
|
|
if summary.get("deleted_files") != []:
|
|
raise SystemExit("cleanup smoke expected deleted_files=[]")
|
|
if summary.get("keep_latest") != expected_keep_latest:
|
|
raise SystemExit("cleanup smoke keep_latest drifted")
|
|
if summary.get("max_delete") != expected_max_delete:
|
|
raise SystemExit("cleanup smoke max_delete drifted")
|
|
if summary.get("export_types") != [expected_export_type]:
|
|
raise SystemExit("cleanup smoke export_types drifted")
|
|
if "candidate_exports" not in summary:
|
|
raise SystemExit("cleanup smoke expected candidate_exports in dry-run output")
|
|
if "candidate_files" not in summary:
|
|
raise SystemExit("cleanup smoke expected candidate_files in dry-run output")
|
|
if "matched_export_count" not in summary:
|
|
raise SystemExit("cleanup smoke expected matched_export_count in dry-run output")
|
|
if "selected_export_count" not in summary:
|
|
raise SystemExit("cleanup smoke expected selected_export_count in dry-run output")
|
|
|
|
print(
|
|
"Demo cleanup dry-run smoke passed: "
|
|
f"matched={summary.get('matched_export_count')} "
|
|
f"type_filtered={summary.get('type_filtered_export_count')} "
|
|
f"selected={summary.get('selected_export_count')} "
|
|
f"deleted={summary.get('deleted_export_count')}"
|
|
)
|
|
PY
|