#!/bin/bash

# 주의
## 해당스크리ㅂ트에는 k8s.registry.io/nfd/node-feature-discovery는 빠져있음
## k8s.registry.io/nfd/node-feature-discovery:v0.16.6
# 프라이빗 레지스트리 주소
PRIVATE_REGISTRY="harbor.nativedeck.com/nd002-rc.1"

# 이미지 목록 (원본 이미지와 태그)
IMAGES=(
  "nvcr.io/nvidia/cloud-native/k8s-mig-manager:v0.10.0-ubuntu20.04"
  "nvcr.io/nvidia/gpu-operator:v24.9.2"	
  "nvcr.io/nvidia/cloud-native/gpu-operator-validator:v24.9.2"
  "nvcr.io/nvidia/cloud-native/k8s-driver-manager:v0.7.0"
  "nvcr.io/nvidia/k8s-device-plugin:v0.17.0"
  "nvcr.io/nvidia/k8s/container-toolkit:v1.17.4-ubuntu20.04"
  "nvcr.io/nvidia/k8s/dcgm-exporter:3.3.9-3.6.1-ubuntu22.04"
)

# 1. 모든 이미지 pull, tag, push 실행
for IMAGE in "${IMAGES[@]}"; do
  # 원본 이미지에서 태그 분리
  IMAGE_NAME=$(echo "$IMAGE" | awk -F':' '{print $1}')
  IMAGE_TAG=$(echo "$IMAGE" | awk -F':' '{print $2}')

  # 새로운 이미지 경로 설정
  NEW_IMAGE="$PRIVATE_REGISTRY/${IMAGE_NAME#nvcr.io/}"

  echo "������ Pulling image: $IMAGE"
  docker pull "$IMAGE"

  echo "������ Tagging image: $IMAGE -> $NEW_IMAGE:$IMAGE_TAG"
  docker tag "$IMAGE" "$NEW_IMAGE:$IMAGE_TAG"

  echo "������ Pushing image: $NEW_IMAGE:$IMAGE_TAG"
  docker push "$NEW_IMAGE:$IMAGE_TAG"

  echo "✅ Successfully pushed $NEW_IMAGE:$IMAGE_TAG"
done

echo "������ 모든 이미지가 프라이빗 레지스트리로 푸시 완료되었습니다!"