#!/bin/bash
set -e

HARBOR="107.148.176.193:30180"
PROJECT="ml-platform"
VERSION="${1:-v1}"

echo "===== ML Platform 构建 & 推送 ====="
echo "Harbor: $HARBOR"
echo "Version: $VERSION"
echo ""

cd "$(dirname "$0")/.."

# 1. 训练镜像
echo "[1/3] 构建训练镜像..."
docker build -t $HARBOR/$PROJECT/training:$VERSION training/
docker push $HARBOR/$PROJECT/training:$VERSION
echo "  ✓ training:$VERSION"

# 2. 推理镜像
echo "[2/3] 构建推理镜像..."
docker build -t $HARBOR/$PROJECT/inference:$VERSION inference/
docker push $HARBOR/$PROJECT/inference:$VERSION
echo "  ✓ inference:$VERSION"

# 3. Operator 镜像
echo "[3/3] 构建 Operator 镜像..."
cd operator
go mod tidy
cd ..
docker build -t $HARBOR/$PROJECT/operator:$VERSION operator/
docker push $HARBOR/$PROJECT/operator:$VERSION
echo "  ✓ operator:$VERSION"

echo ""
echo "===== 全部镜像推送完成 ====="
echo "  $HARBOR/$PROJECT/training:$VERSION"
echo "  $HARBOR/$PROJECT/inference:$VERSION"
echo "  $HARBOR/$PROJECT/operator:$VERSION"