#!/bin/bash

##############################################################################
# Agent Manager - 快速构建和部署脚本 (ARM64)
# 用途: 一键构建 ARM64 Docker 镜像并部署到 AKS
##############################################################################

set -e

# 颜色输出
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# 配置变量
ACR_NAME="agnettaiji"
IMAGE_NAME="agent-manager"
IMAGE_TAG="latest-arm64"
FULL_IMAGE_NAME="${ACR_NAME}.azurecr.io/${IMAGE_NAME}:${IMAGE_TAG}"

echo -e "${BLUE}========================================${NC}"
echo -e "${BLUE}  Agent Manager ARM64 构建和部署${NC}"
echo -e "${BLUE}========================================${NC}"

# 检查必要工具
echo -e "\n${GREEN}检查工具...${NC}"
command -v docker >/dev/null 2>&1 || { echo "需要安装 Docker"; exit 1; }
command -v kubectl >/dev/null 2>&1 || { echo "需要安装 kubectl"; exit 1; }
command -v az >/dev/null 2>&1 || { echo "需要安装 Azure CLI"; exit 1; }

# 登录 ACR
echo -e "\n${GREEN}登录 Azure Container Registry...${NC}"
az acr login --name ${ACR_NAME}

# 设置 buildx
echo -e "\n${GREEN}配置 Docker Buildx...${NC}"
BUILDER_NAME="arm64-builder"
if ! docker buildx inspect ${BUILDER_NAME} &> /dev/null; then
    echo "创建 buildx builder: ${BUILDER_NAME}"
    docker buildx create --name ${BUILDER_NAME} --use --driver docker-container
    docker buildx inspect --bootstrap
else
    docker buildx use ${BUILDER_NAME}
fi

# 构建并推送镜像
echo -e "\n${GREEN}构建 ARM64 镜像...${NC}"
docker buildx build \
    --platform linux/arm64 \
    -f Dockerfile \
    -t ${FULL_IMAGE_NAME} \
    --push \
    .

echo -e "\n${GREEN}✅ 镜像构建完成: ${FULL_IMAGE_NAME}${NC}"

# 部署到 Kubernetes
echo -e "\n${GREEN}部署到 Kubernetes...${NC}"
echo -e "${YELLOW}使用脚本: scripts/deploy-to-k8s-arm64.sh --skip-build${NC}"
echo -e "${YELLOW}或手动运行: kubectl apply -f k8s/agent-manager-deployment.yaml${NC}"

# 询问是否立即部署
read -p "是否立即部署到 Kubernetes? [y/N] " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    if [ -f "scripts/deploy-to-k8s-arm64.sh" ]; then
        bash scripts/deploy-to-k8s-arm64.sh --skip-build
    else
        echo -e "${YELLOW}部署脚本不存在，请手动部署:${NC}"
        echo "kubectl apply -f k8s/agent-manager-deployment.yaml"
    fi
fi

echo -e "\n${GREEN}========================================${NC}"
echo -e "${GREEN}  完成！${NC}"
echo -e "${GREEN}========================================${NC}"