#!/usr/bin/env bash
# Usage: deploy.sh <host> [stack]
#   host  — node directory (cloud-pc, mini-pc, ru-vps)
#   stack — optional single stack name; omit to reconcile all
set -euxo pipefail

HOST="$1"
STACK="${2:-}"
BASE="/opt/services"

cd "$BASE"

if [ -n "$STACK" ]; then
  dirs="$STACK"
else
  dirs="$(find "$HOST" -mindepth 2 -maxdepth 2 -name docker-compose.yml | cut -d/ -f2 | sort -u)"

  # Remove stale stacks whose directories were deleted from the repo
  stale_dirs="$(docker ps -a \
    --filter "label=com.docker.compose.project.working_dir" \
    --format '{{.Label "com.docker.compose.project.working_dir"}}' \
    | sort -u \
    | grep "^$BASE/$HOST/" || true)"

  for working_dir in $stale_dirs; do
    dir="${working_dir#"$BASE/$HOST/"}"
    dir="${dir%%/*}"
    if [ ! -f "$BASE/$HOST/$dir/docker-compose.yml" ]; then
      echo "==> Removing stale stack: $dir"
      containers="$(docker ps -aq --filter "label=com.docker.compose.project.working_dir=$working_dir" || true)"
      [ -n "$containers" ] && docker rm -f $containers
      networks="$(docker network ls -q --filter "label=com.docker.compose.project.working_dir=$working_dir" || true)"
      [ -n "$networks" ] && docker network rm $networks || true
    fi
  done
fi

if [ -z "$dirs" ]; then
  echo "No stacks found for $HOST"
  exit 0
fi

failed=""

for dir in $dirs; do
  echo "==> Stack: $dir"

  if [ "$dir" = "gitea-runner" ]; then
    echo "Skipping gitea-runner (self-update not safe from CI)"
    continue
  fi

  stack_dir="$BASE/$HOST/$dir"
  cd "$stack_dir"

  if [ ! -f docker-compose.yml ]; then
    echo "No docker-compose.yml, skipping"
    cd "$BASE"
    continue
  fi

  docker compose config --quiet

  if grep -qE '^[[:space:]]*build:' docker-compose.yml; then
    docker compose up -d --build --remove-orphans
  else
    docker compose pull
    docker compose up -d --remove-orphans
  fi

  # Health check: give containers a moment, then fail on crashed services
  sleep 5
  docker compose ps
  if docker compose ps 2>&1 | grep -qE 'Exited \([1-9]|Restarting'; then
    docker compose logs --tail=30
    echo "ERROR: Service failure in stack $dir"
    failed="$failed $dir"
  fi

  cd "$BASE"
done

if [ -n "$failed" ]; then
  echo "FAILED stacks:$failed"
  exit 1
fi