# Create entities in RC1
curl -s -X POST "$GATEWAY_URL/gateways" \
  -H "Authorization: Bearer $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"name": "test-server-1", "url": "http://localhost:9000"}'

curl -s -X POST "$GATEWAY_URL/api/teams" \
  -H "Authorization: Bearer $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"name": "upgrade-test-team"}'

# Record counts
GATEWAY_COUNT=$(curl -s "$GATEWAY_URL/gateways" -H "Authorization: Bearer $TOKEN" | jq '.items | length')
TEAM_COUNT=$(curl -s "$GATEWAY_URL/api/teams" -H "Authorization: Bearer $TOKEN" | jq 'length')
echo "Pre-upgrade: $GATEWAY_COUNT gateways, $TEAM_COUNT teams"

# Stop RC1 gateway
pkill -f "mcpgateway"

# Switch to GA version
cd ../ga_gateway  # or: git checkout v1.0.0

# Run migrations
cd mcpgateway && alembic upgrade head

# Start GA gateway
make serve &
sleep 10

# Verify counts match
NEW_GATEWAY_COUNT=$(curl -s "$GATEWAY_URL/gateways" -H "Authorization: Bearer $TOKEN" | jq '.items | length')
NEW_TEAM_COUNT=$(curl -s "$GATEWAY_URL/api/teams" -H "Authorization: Bearer $TOKEN" | jq 'length')

[ "$GATEWAY_COUNT" = "$NEW_GATEWAY_COUNT" ] && echo "PASS: Gateway count matches" || echo "FAIL"
[ "$TEAM_COUNT" = "$NEW_TEAM_COUNT" ] && echo "PASS: Team count matches" || echo "FAIL"

# Verify functionality
curl -s -X POST "$GATEWAY_URL/mcp/http" \
  -H "Authorization: Bearer $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"jsonrpc": "2.0", "id": 1, "method": "tools/list", "params": {}}' | jq '.result.tools | length'

# Simulate issue discovered post-upgrade
# Stop GA gateway
pkill -f "mcpgateway"

# Downgrade database (if supported)
cd mcpgateway && alembic downgrade -1

# Start RC1 gateway
cd ../rc1_gateway && make serve &
sleep 10

# Verify service restored
curl -s "$GATEWAY_URL/health" | jq '.status'

# Verify data accessible
curl -s "$GATEWAY_URL/gateways" -H "Authorization: Bearer $TOKEN" | jq '.items | length'

# Update values with new image tag
helm upgrade mcpgateway ./charts/mcpgateway \
  --set image.tag=$GA_VERSION \
  --set strategy.type=RollingUpdate \
  --set strategy.rollingUpdate.maxUnavailable=0 \
  --set strategy.rollingUpdate.maxSurge=1

# Monitor rollout
kubectl rollout status deployment/mcpgateway -w

# Run continuous health check during upgrade
while true; do
  STATUS=$(curl -s -o /dev/null -w "%{http_code}" "$GATEWAY_URL/health")
  [ "$STATUS" != "200" ] && echo "DOWNTIME DETECTED at $(date)" && break
  sleep 1
done &

# Perform upgrade (above helm command)
# After upgrade completes, kill the health check loop

# Simulate partial migration (manually interrupt)
cd mcpgateway && timeout 2 alembic upgrade head || true

# Retry migration
alembic upgrade head

# Verify success
alembic current