#!/bin/bash
# Terminate both hyperopt pods
source .env.runpod

DQN_POD_ID="dy2bn5ninzaxma"
PPO_POD_ID="dytpb1mcqwj54t"

echo "========================================="
echo "TERMINATING HYPEROPT PODS"
echo "========================================="
echo ""
echo "WARNING: This will terminate both hyperopt pods"
echo "  DQN: ${DQN_POD_ID}"
echo "  PPO: ${PPO_POD_ID}"
echo ""
read -p "Continue? (y/N) " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
  echo "Cancelled."
  exit 1
fi

# GraphQL mutation to terminate pods
MUTATION=$(cat <<EOF
mutation {
  podTerminate(input: {podId: "${DQN_POD_ID}"})
  podTerminate(input: {podId: "${PPO_POD_ID}"})
}
EOF
)

echo "Terminating DQN pod..."
curl -s -X POST https://api.runpod.io/graphql \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer ${RUNPOD_API_KEY}" \
  -d "{\"query\": $(echo "mutation { podTerminate(input: {podId: \"${DQN_POD_ID}\"}) }" | jq -Rs .)}" | jq

echo ""
echo "Terminating PPO pod..."
curl -s -X POST https://api.runpod.io/graphql \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer ${RUNPOD_API_KEY}" \
  -d "{\"query\": $(echo "mutation { podTerminate(input: {podId: \"${PPO_POD_ID}\"}) }" | jq -Rs .)}" | jq

echo ""
echo "✅ Termination requests sent"
echo ""