#!/usr/bin/env bash
set -euo pipefail

# Restart the TEST (staging) stack. Scoped entirely to this checkout's compose
# project, so it can never touch the production stack. Unlike docker/restart.sh
# it:
#   - keeps the database by default (you loaded prod data into it) — pass
#     --fresh-db to wipe var/volumes/db and let MySQL re-initialise
#   - never runs a host-wide `docker system/builder prune`
#   - only repairs ownership of var/cache, var/log, var/sessions — NEVER
#     var/volumes/db (that dir belongs to the mysql process; chowning it is
#     what corrupts the data directory)
#
# Usage:
#   ./docker/restart.test.sh              # down + up (keeps DB and images)
#   ./docker/restart.test.sh --build      # also rebuild images
#   ./docker/restart.test.sh --fresh-db   # also wipe + re-initialise the database

DOCKER_DIR=$(cd "$(dirname "$0")" && pwd)
ROOT_DIR=$(cd "$DOCKER_DIR/.." && pwd)

env_val() { [ -f "$DOCKER_DIR/.env" ] && grep -E "^$1=" "$DOCKER_DIR/.env" | tail -n1 | cut -d= -f2- | tr -d "\"' " || true; }

if [ ! -f "$DOCKER_DIR/.env" ]; then
  echo "Error: $DOCKER_DIR/.env not found. Copy docker/.env.test.example to docker/.env first." >&2
  exit 1
fi
PROJECT="$(env_val COMPOSE_PROJECT_NAME)"
STACK="$(env_val STACK_NAME)"; STACK="${STACK:-$PROJECT}"
case "${PROJECT:-}" in
  ""|escapepage|docker)
    echo "Error: COMPOSE_PROJECT_NAME in docker/.env is '${PROJECT:-<unset>}'." >&2
    echo "Refusing to run a test script against the production stack." >&2
    exit 1 ;;
esac

if docker compose version >/dev/null 2>&1; then
  DOCKER_COMPOSE="docker compose"
elif command -v docker-compose >/dev/null 2>&1; then
  DOCKER_COMPOSE="docker-compose"
else
  echo "Error: neither 'docker compose' nor 'docker-compose' is available." >&2
  exit 1
fi

FRESH_DB=0; BUILD=0
for arg in "$@"; do
  case "$arg" in
    --fresh-db) FRESH_DB=1 ;;
    --build)    BUILD=1 ;;
    *) echo "Unknown option: $arg" >&2; exit 1 ;;
  esac
done

dc() { (cd "$DOCKER_DIR" && $DOCKER_COMPOSE -p "$PROJECT" -f compose.yaml -f compose.override.yaml "$@"); }

echo "Restarting test stack: $STACK  (compose project: $PROJECT)"

echo "Stopping containers..."
dc down --remove-orphans || true
# scoped fallback — only this stack
docker rm -f \
  "${STACK}-db" "${STACK}-php" "${STACK}-nginx" "${STACK}-mercure" \
  "${STACK}-mailer" "${STACK}-php-worker" "${STACK}-php-cron" 2>/dev/null || true
for net in $(docker network ls -q --filter "name=^${PROJECT}_" 2>/dev/null); do
  docker network rm "$net" || true
done

if [ "$FRESH_DB" -eq 1 ]; then
  echo "Wiping the test database (var/volumes/db)..."
  sudo rm -rf "$ROOT_DIR/var/volumes/db"
fi

echo "Repairing var/ ownership (cache/log/sessions only)..."
sudo mkdir -p "$ROOT_DIR/var/cache" "$ROOT_DIR/var/log/php" "$ROOT_DIR/var/log/cron" "$ROOT_DIR/var/sessions"
sudo chown -R 1000:1000 "$ROOT_DIR/var/cache" "$ROOT_DIR/var/log" "$ROOT_DIR/var/sessions"
sudo chmod -R u+rwX,g+rwX "$ROOT_DIR/var/cache" "$ROOT_DIR/var/log" "$ROOT_DIR/var/sessions"

echo "Bringing the stack back up..."
if [ "$BUILD" -eq 1 ]; then
  "$DOCKER_DIR/setup.test.sh"
else
  "$DOCKER_DIR/setup.test.sh" --no-build
fi