2025-09-16 21:25:24 +08:00

175 lines
4.4 KiB
Bash

#!/bin/bash
# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color
# 配置变量
IMAGE_NAME="nginx"
IMAGE_TAG="1.24-alpine"
HARBOR_URL="192.168.10.102:8001" # 替换为你的Harbor地址
HARBOR_PROJECT="xiaomayi-base" # 替换为你的项目名
HARBOR_USERNAME="deploy" # 从环境变量获取
HARBOR_PASSWORD="Harbor20240330" # 从环境变量获取
# 日志函数
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"; }
# 检查依赖
check_dependencies() {
local deps=("docker" "curl")
for dep in "${deps[@]}"; do
if ! command -v "$dep" &> /dev/null; then
log_error "缺少依赖: $dep"
exit 1
fi
done
}
# 检查Docker服务
check_docker() {
if ! docker info > /dev/null 2>&1; then
log_error "Docker服务未运行"
exit 1
fi
}
# 构建镜像
build_image() {
log_info "开始构建Nginx镜像..."
if docker build -t "${IMAGE_NAME}:${IMAGE_TAG}" . ; then
log_success "镜像构建成功: ${IMAGE_NAME}:${IMAGE_TAG}"
else
log_error "镜像构建失败"
exit 1
fi
}
# 添加额外标签(可选)
add_additional_tags() {
local tags=("latest" "prod" "dev")
for extra_tag in "${tags[@]}"; do
log_info "添加额外标签: ${extra_tag}"
docker tag "${IMAGE_NAME}:${IMAGE_TAG}" "${IMAGE_NAME}:${extra_tag}"
done
}
# 登录Harbor
login_to_harbor() {
if [ -z "$HARBOR_USERNAME" ] || [ -z "$HARBOR_PASSWORD" ]; then
log_error "Harbor用户名或密码未设置"
log_warning "请设置环境变量: HARBOR_USERNAME 和 HARBOR_PASSWORD"
exit 1
fi
log_info "登录到Harbor仓库..."
if echo "$HARBOR_PASSWORD" | docker login -u "$HARBOR_USERNAME" --password-stdin "$HARBOR_URL"; then
log_success "Harbor登录成功"
else
log_error "Harbor登录失败"
exit 1
fi
}
# 打标签
tag_image() {
local target_tag="${1:-$IMAGE_TAG}"
local full_image_name="${HARBOR_URL}/${HARBOR_PROJECT}/${IMAGE_NAME}:${target_tag}"
log_info "打标签: ${full_image_name}"
if docker tag "${IMAGE_NAME}:${target_tag}" "$full_image_name"; then
log_success "标签打成功"
else
log_error "标签打失败"
exit 1
fi
}
# 推送镜像
push_image() {
local target_tag="${1:-$IMAGE_TAG}"
local full_image_name="${HARBOR_URL}/${HARBOR_PROJECT}/${IMAGE_NAME}:${target_tag}"
log_info "推送镜像: ${full_image_name}"
if docker push "$full_image_name"; then
log_success "镜像推送成功"
else
log_error "镜像推送失败"
exit 1
fi
}
# 清理本地镜像
cleanup() {
log_info "清理本地镜像..."
# 清理主标签
docker rmi "${IMAGE_NAME}:${IMAGE_TAG}" 2>/dev/null || true
# 清理额外标签
local tags=("latest" "prod" "dev")
for tag in "${tags[@]}"; do
docker rmi "${IMAGE_NAME}:${tag}" 2>/dev/null || true
done
# 清理Harbor标签
local harbor_tags=("$IMAGE_TAG" "latest" "prod" "dev")
for tag in "${harbor_tags[@]}"; do
docker rmi "${HARBOR_URL}/${HARBOR_PROJECT}/${IMAGE_NAME}:${tag}" 2>/dev/null || true
done
log_success "清理完成"
}
# 显示镜像信息
show_image_info() {
log_info "镜像信息:"
echo "名称: ${IMAGE_NAME}"
echo "版本: ${IMAGE_TAG}"
echo "Harbor地址: ${HARBOR_URL}"
echo "项目: ${HARBOR_PROJECT}"
echo "完整镜像名: ${HARBOR_URL}/${HARBOR_PROJECT}/${IMAGE_NAME}:${IMAGE_TAG}"
}
# 主函数
main() {
log_info "====== Nginx镜像构建推送脚本 ======"
check_dependencies
check_docker
show_image_info
# 构建镜像
build_image
# 添加额外标签
#add_additional_tags
# 登录Harbor
login_to_harbor
# 推送所有标签
#local tags=("$IMAGE_TAG" "latest" "prod" "dev")
local tags=("$IMAGE_TAG")
for tag in "${tags[@]}"; do
tag_image "$tag"
push_image "$tag"
done
# 清理
cleanup
log_success "====== 所有操作完成! ======"
log_info "镜像地址: ${HARBOR_URL}/${HARBOR_PROJECT}/${IMAGE_NAME}:${IMAGE_TAG}"
}
# 执行主函数
main "$@"