#!/bin/bash
# Check status of both hyperopt pods
source .env.runpod

DQN_POD_ID="dy2bn5ninzaxma"
PPO_POD_ID="dytpb1mcqwj54t"

echo "========================================="
echo "HYPEROPT PODS STATUS"
echo "========================================="
echo ""

# GraphQL query to get pod status
QUERY=$(cat <<'EOF'
{
  myself {
    pods {
      id
      name
      desiredStatus
      runtime {
        uptimeInSeconds
      }
      machine {
        gpuType {
          displayName
        }
        dataCenterId
      }
      costPerHr
    }
  }
}
EOF
)

# Query RunPod API
RESPONSE=$(curl -s -X POST https://api.runpod.io/graphql \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer ${RUNPOD_API_KEY}" \
  -d "{\"query\": $(echo "$QUERY" | jq -Rs .)}")

# Parse and display
echo "DQN Pod (${DQN_POD_ID}):"
echo "$RESPONSE" | jq -r ".data.myself.pods[] | select(.id == \"${DQN_POD_ID}\") |
  \"  Status: \(.desiredStatus)
  GPU: \(.machine.gpuType.displayName)
  Datacenter: \(.machine.dataCenterId)
  Cost: $\(.costPerHr)/hr
  Uptime: \(.runtime.uptimeInSeconds)s\""
echo ""

echo "PPO Pod (${PPO_POD_ID}):"
echo "$RESPONSE" | jq -r ".data.myself.pods[] | select(.id == \"${PPO_POD_ID}\") |
  \"  Status: \(.desiredStatus)
  GPU: \(.machine.gpuType.displayName)
  Datacenter: \(.machine.dataCenterId)
  Cost: $\(.costPerHr)/hr
  Uptime: \(.runtime.uptimeInSeconds)s\""
echo ""

echo "========================================="
echo "MONITORING URLS"
echo "========================================="
echo "Dashboard: https://www.runpod.io/console/pods"
echo ""
echo "DQN Logs:"
echo "  https://www.runpod.io/console/pods/${DQN_POD_ID}"
echo ""
echo "PPO Logs:"
echo "  https://www.runpod.io/console/pods/${PPO_POD_ID}"
echo ""