#!/usr/bin/env bash
set -euo pipefail

# Pull Docker images used by connector CI lanes with retry.
# Usage:
#   bash scripts/ci-pull-relational-images.sh 16-alpine
#   bash scripts/ci-pull-relational-images.sh 12-alpine 14-alpine 15-alpine 16-alpine
#   bash scripts/ci-pull-relational-images.sh --relational-smoke

# Non-postgres, non-sqlserver smoke images.
# Postgres tags are passed explicitly as positional args.
# SQL Server is handled separately via --sqlserver-soft (soft failure on MCR downtime).
NON_POSTGRES_RELATIONAL_SMOKE_IMAGES=(
  "mysql:8.0"
  "mysql:8.1"
  "mariadb:10.6"
  "mariadb:10.11"
)

pull_with_retry() {
  local image="$1"
  local max_attempts="${CI_PULL_MAX_ATTEMPTS:-4}"
  local base_backoff_secs="${CI_PULL_BACKOFF_SECS:-10}"
  local attempt=1

  while true; do
    if docker pull "$image"; then
      return 0
    fi

    if [[ "$attempt" -ge "$max_attempts" ]]; then
      echo "failed to pull $image after $max_attempts attempts" >&2
      return 1
    fi

    local sleep_secs=$((attempt * base_backoff_secs))
    echo "retrying pull for $image in ${sleep_secs}s (attempt $((attempt + 1))/${max_attempts})" >&2
    sleep "$sleep_secs"
    attempt=$((attempt + 1))
  done
}

if [[ "$#" -eq 0 ]]; then
  echo "usage: $0 <tag> [<tag> ...] | --relational-smoke" >&2
  exit 2
fi

if [[ "${1:-}" == "--relational-smoke" ]]; then
  for image in "${NON_POSTGRES_RELATIONAL_SMOKE_IMAGES[@]}"; do
    # Pull from the ECR public mirror (no Docker Hub rate-limits / auth required)
    # and re-tag to the bare name so testcontainers finds it in the local cache.
    ecr_image="public.ecr.aws/docker/library/${image}"
    pull_with_retry "$ecr_image"
    docker tag "$ecr_image" "$image"
    echo "prepared ${image} from ${ecr_image}"
  done
  exit 0
fi

# Pull the SQL Server image with a soft failure: if MCR is unavailable
# (rate-limit, access denial, or retired tag), warn and exit 0 so that
# pre-pull failures do not block unrelated CI lanes.  The sqlserver tests
# themselves emit a "SKIP:" sentinel when the image cannot be pulled.
if [[ "${1:-}" == "--sqlserver-soft" ]]; then
  image="mcr.microsoft.com/mssql/server:${2:-2022-latest}"
  if pull_with_retry "$image"; then
    echo "prepared ${image}"
  else
    echo "WARNING: could not pull ${image} — sqlserver tests will be skipped" >&2
  fi
  exit 0
fi

for tag in "$@"; do
  image="public.ecr.aws/docker/library/postgres:${tag}"
  pull_with_retry "$image"
  docker tag "$image" "postgres:${tag}"
  echo "prepared postgres:${tag} from ${image}"
done