#!/bin/bash
set -euo pipefail

ENVIRONMENT=${1:-staging}
SERVICE=${2:-all}

CLUSTER="shieldai-${ENVIRONMENT}"

echo "Rolling back services in cluster: $CLUSTER"

SERVICES="api darkwatch spamshield voiceprint"
if [ "$SERVICE" != "all" ]; then
  SERVICES="$SERVICE"
fi

for svc in $SERVICES; do
  echo "Rolling back $svc..."
  aws ecs update-service \
    --cluster "$CLUSTER" \
    --service "${CLUSTER}-${svc}" \
    --rollback \
    --no-cli-auto-prompt

  echo "Waiting for $svc to stabilize..."
  aws ecs wait services-stable \
    --cluster "$CLUSTER" \
    --services "${CLUSTER}-${svc}"

  echo "$svc rolled back successfully"
done

echo "Rollback complete for $SERVICES"