build_cuda128.sh

#!/usr/bin/env bash

# Build script for CUDA 12.8 Dockerfile with dynamic versioning
# Usage: ./build_cuda128.sh [--tag TAG] [--no-cache] [--push REGISTRY]

set -e

# Default values
IMAGE_TAG=""
NO_CACHE=""
PUSH_REGISTRY=""
CONFIG_PATH="config/chat_with_openai_compatible_bailian_cosyvoice.yaml"

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Logging functions
log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# Function to get version from git
get_version() {
    if [ -d .git ]; then
        # # Try to get version from git tag first
        # local git_tag=$(git describe --tags --exact-match 2>/dev/null)
        # if [ $? -eq 0 ] && [ -n "$git_tag" ]; then
        #     echo "$git_tag"
        #     return
        # fi
        
        # # Try to get version from nearest tag + commits
        # local desc_tag=$(git describe --tags --always 2>/dev/null)
        # if [ $? -eq 0 ] && [ -n "$desc_tag" ] && [[ "$desc_tag" == *"-"* ]]; then
        #     # Replace slash with underscore for Docker compatibility
        #     echo "$desc_tag" | sed 's/\//_/g'
        #     return
        # fi
        
        # Fallback: Generate version from branch and short commit
        local branch=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)
        local commit=$(git rev-parse --short HEAD 2>/dev/null)
        
        if [ -n "$branch" ] && [ -n "$commit" ]; then
            # Replace slash with underscore for Docker compatibility  
            local clean_branch=$(echo "$branch" | sed 's/\//_/g')
            echo "${clean_branch}-${commit}"
        else
            echo "unknown-$(date +%Y%m%d)"
        fi
    else
        echo "dev-$(date +%Y%m%d)"
    fi
}

# Parse arguments
while [[ "$#" -gt 0 ]]; do
    case $1 in
        --tag )
            IMAGE_TAG="$2"
            shift 2
            ;;
        --no-cache )
            NO_CACHE="--no-cache"
            shift
            ;;
        --push )
            PUSH_REGISTRY="$2"
            shift 2
            ;;
        --config )
            CONFIG_PATH="$2"
            shift 2
            ;;
        -h | --help )
            echo "Usage: $0 [OPTIONS]"
            echo "Options:"
            echo "  --tag TAG         Docker镜像标签 (默认: open-avatar-chat:VERSION)"
            echo "  --no-cache        不使用Docker构建缓存"
            echo "  --push REGISTRY   构建后推送到指定注册表"
            echo "  --config PATH     运行时使用的配置文件路径"
            echo "  -h, --help        显示帮助信息"
            echo ""
            echo "Examples:"
            echo "  $0                                          # 基本构建"
            echo "  $0 --tag myapp:latest                       # 自定义标签"
            echo "  $0 --push ghcr.io/user/repo                # 构建并推送"
            echo "  $0 --no-cache --push ghcr.io/user/repo     # 强制重建并推送"
            exit 0
            ;;
        * )
            log_error "未知参数: $1"
            exit 1
            ;;
    esac
done

# Validate Docker is available
if ! command -v docker &> /dev/null; then
    log_error "Docker 未找到，请先安装Docker"
    exit 1
fi

# Check if Dockerfile.cuda12.8 exists
if [ ! -f "Dockerfile.cuda12.8" ]; then
    log_error "Dockerfile.cuda12.8 未找到"
    exit 1
fi

# Set default image tag if not provided
if [ -z "$IMAGE_TAG" ]; then
    IMAGE_TAG="open-avatar-chat:latest"
fi

# Get build metadata
BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ')
BUILD_VERSION=$(get_version)

log_info "=== 构建信息 ==="
log_info "镜像标签: ${IMAGE_TAG}"
log_info "构建版本: ${BUILD_VERSION} (嵌入到镜像标签中)"
log_info "构建时间: ${BUILD_DATE} (嵌入到镜像标签中)"
log_info "Dockerfile: Dockerfile.cuda12.8"

#Build command
BUILD_CMD="docker build ${NO_CACHE} \
    --build-arg BUILD_VERSION=\"${BUILD_VERSION}\" \
    --build-arg BUILD_DATE=\"${BUILD_DATE}\" \
    -f Dockerfile.cuda12.8 \
    -t \"${IMAGE_TAG}\" ."

log_info "=== 开始构建 ==="
log_info "执行命令: ${BUILD_CMD}"

# Execute build
if eval $BUILD_CMD; then
    log_success "镜像构建成功: ${IMAGE_TAG}"
    
    # Show image info
    log_info "=== 镜像信息 ==="
    docker images "${IMAGE_TAG}" --format "table {{.Repository}}\t{{.Tag}}\t{{.Size}}\t{{.CreatedAt}}"
    
    # Push if requested
    if [ -n "$PUSH_REGISTRY" ]; then
        PUSH_TAG="${PUSH_REGISTRY}:${BUILD_VERSION}"
        log_info "=== 推送镜像 ==="
        log_info "标记镜像: ${PUSH_TAG}"
        
        if docker tag "${IMAGE_TAG}" "${PUSH_TAG}"; then
            log_info "推送镜像: ${PUSH_TAG}"
            if docker push "${PUSH_TAG}"; then
                log_success "镜像推送成功: ${PUSH_TAG}"
            else
                log_error "镜像推送失败"
                exit 1
            fi
        else
            log_error "镜像标记失败"
            exit 1
        fi
    fi
    
    # Provide usage instructions
    log_info "=== 使用说明 ==="
    echo ""
    echo "查看镜像中的commit信息:"
    echo "docker inspect \"${IMAGE_TAG}\" | grep -A 10 'Labels'"
    echo ""
    echo "启动容器:"
    echo "docker run --rm --gpus all -it \\"
    echo "  --network=host \\"
    echo "  -v \$(pwd)/build:/root/open-avatar-chat/build \\"
    echo "  -v \$(pwd)/models:/root/open-avatar-chat/models \\"
    echo "  -v \$(pwd)/ssl_certs:/root/open-avatar-chat/ssl_certs \\"
    echo "  -v \$(pwd)/config:/root/open-avatar-chat/config \\"
    echo "  -v \$(pwd)/models/musetalk/s3fd-619a316812/:/root/.cache/torch/hub/checkpoints/ \\"
    echo "  -p 8282:8282 \\"
    echo "  \"${IMAGE_TAG}\" \\"
    echo "  --config \"${CONFIG_PATH}\""
    echo ""
    echo "在容器内查看版本信息:"
    echo "docker run --rm \"${IMAGE_TAG}\" printenv | grep -E '^(APP_VERSION|BUILD_DATE|BUILD_COMMIT)='"
    echo ""
    
else
    log_error "镜像构建失败"
    exit 1
fi