|
| 1 | +#!/bin/bash |
| 2 | + |
| 3 | +# Cap-ComfyUI 快速部署脚本（ComfyUI + Agent 顺序构建） |
| 4 | +# 用途: 将本仓库同步到开发机，先打 ComfyUI 镜像，再打 Agent 镜像（Agent 引用刚打的 ComfyUI），并推送 Agent |
| 5 | +# |
| 6 | +# 用法（在仓库根目录）: |
| 7 | +# ./scripts/deploy-with-comfyui.sh |
| 8 | +# ./scripts/deploy-with-comfyui.sh -b feature-xxx |
| 9 | +# ./scripts/deploy-with-comfyui.sh -b main prod v1.4.0 |
| 10 | +# |
| 11 | +# 参数: |
| 12 | +# -b BRANCH 指定要部署的 git 分支（可选，默认当前分支） |
| 13 | +# ENV 环境: dev（默认）或 prod |
| 14 | +# VERSION 版本号: prod 必须指定，dev 默认使用时间戳 |
| 15 | +# |
| 16 | +# 说明: |
| 17 | +# - ComfyUI 在开发机打为本地 tag（默认 comfyui:build-$VERSION），不单独 push。 |
| 18 | +# - 依赖仓库根 Makefile：`make build COMFYUI_IMAGE=...` 会把参数传给 agent 的 docker build。 |
| 19 | + |
| 20 | +set -e |
| 21 | + |
| 22 | +DEV_SERVER="root@43.106.16.112" |
| 23 | +REMOTE_DIR="cap-comfyui-cpu" |
| 24 | +SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)" |
| 25 | +LOCAL_DIR="$(cd "$SCRIPT_DIR/.." && pwd)" |
| 26 | +REGISTRY="cap-demo-public-registry.cn-hangzhou.cr.aliyuncs.com/cap-app" |
| 27 | + |
| 28 | +BRANCH="" |
| 29 | +while getopts "b:" opt; do |
| 30 | + case $opt in |
| 31 | + b) BRANCH="$OPTARG" ;; |
| 32 | + *) echo "用法: $0 [-b branch] [env] [version]"; exit 1 ;; |
| 33 | + esac |
| 34 | +done |
| 35 | +shift $((OPTIND-1)) |
| 36 | + |
| 37 | +if [ ! -d "$LOCAL_DIR" ] || [ ! -f "$LOCAL_DIR/Makefile" ]; then |
| 38 | + echo "[ERROR] 未找到仓库根目录或 Makefile: $LOCAL_DIR" |
| 39 | + exit 1 |
| 40 | +fi |
| 41 | + |
| 42 | +cd "$LOCAL_DIR" |
| 43 | + |
| 44 | +ORIGINAL_BRANCH=$(git rev-parse --abbrev-ref HEAD) |
| 45 | + |
| 46 | +cleanup() { |
| 47 | + if [ -n "$ORIGINAL_BRANCH" ] && [ "$BRANCH" != "$ORIGINAL_BRANCH" ]; then |
| 48 | + echo "" |
| 49 | + echo "恢复到原分支: $ORIGINAL_BRANCH" |
| 50 | + cd "$LOCAL_DIR" |
| 51 | + git checkout "$ORIGINAL_BRANCH" 2>/dev/null || true |
| 52 | + fi |
| 53 | +} |
| 54 | +trap cleanup EXIT |
| 55 | + |
| 56 | +if [ -n "$BRANCH" ]; then |
| 57 | + echo "切换到分支: $BRANCH" |
| 58 | + git fetch origin "$BRANCH" 2>/dev/null || true |
| 59 | + git checkout "$BRANCH" |
| 60 | + git pull origin "$BRANCH" || true |
| 61 | +else |
| 62 | + BRANCH="$ORIGINAL_BRANCH" |
| 63 | + echo "使用当前分支: $BRANCH" |
| 64 | +fi |
| 65 | + |
| 66 | +ENV="${1:-dev}" |
| 67 | +VERSION="${2:-$(date "+%Y%m%d%H%M%S")}" |
| 68 | + |
| 69 | +if [ "$ENV" = "prod" ]; then |
| 70 | + IMAGE_NAME="image-generation-comfyui-agent" |
| 71 | + if [ -z "$2" ]; then |
| 72 | + echo "prod 模式必须指定版本号，例如: ./scripts/deploy-with-comfyui.sh prod v1.4.0" |
| 73 | + exit 1 |
| 74 | + fi |
| 75 | +else |
| 76 | + IMAGE_NAME="image-generation-comfyui-agent-dev" |
| 77 | +fi |
| 78 | + |
| 79 | +AGENT_IMAGE="${REGISTRY}/${IMAGE_NAME}:${VERSION}" |
| 80 | +COMFYUI_BUILD_TAG="${COMFYUI_BUILD_TAG:-comfyui:build-${VERSION}}" |
| 81 | + |
| 82 | +echo "" |
| 83 | +echo "开始部署 Cap-ComfyUI（ComfyUI + Agent）到开发机..." |
| 84 | +echo " 分支: $BRANCH" |
| 85 | +echo " 本地目录: $LOCAL_DIR" |
| 86 | +echo " 开发机: $DEV_SERVER" |
| 87 | +echo " 远程目录: $REMOTE_DIR" |
| 88 | +echo " 环境: $ENV" |
| 89 | +echo " ComfyUI 本地 tag: $COMFYUI_BUILD_TAG" |
| 90 | +echo " Agent 镜像: $AGENT_IMAGE" |
| 91 | +echo "" |
| 92 | + |
| 93 | +echo "步骤1: 清理开发机上的旧代码..." |
| 94 | +ssh $DEV_SERVER << 'EOF' |
| 95 | + if [ -d "cap-comfyui-cpu" ]; then |
| 96 | + echo " 发现旧目录，正在删除..." |
| 97 | + rm -rf cap-comfyui-cpu |
| 98 | + echo " 旧代码已删除" |
| 99 | + else |
| 100 | + echo " 未发现旧目录，跳过删除" |
| 101 | + fi |
| 102 | +EOF |
| 103 | + |
| 104 | +echo "" |
| 105 | +echo "步骤2: 同步本地代码到开发机..." |
| 106 | +rsync -avz --progress \ |
| 107 | + --exclude='.git' \ |
| 108 | + --exclude='.idea' \ |
| 109 | + --exclude='*.pyc' \ |
| 110 | + --exclude='__pycache__' \ |
| 111 | + --exclude='.DS_Store' \ |
| 112 | + --exclude='*.log' \ |
| 113 | + --exclude='node_modules' \ |
| 114 | + "$LOCAL_DIR/" "$DEV_SERVER:$REMOTE_DIR/" |
| 115 | + |
| 116 | +echo " 代码同步完成" |
| 117 | + |
| 118 | +echo "" |
| 119 | +echo "步骤3: 验证部署结果..." |
| 120 | +ssh $DEV_SERVER << EOF |
| 121 | + if [ -d "$REMOTE_DIR" ]; then |
| 122 | + echo " 远程目录结构:" |
| 123 | + ls -la $REMOTE_DIR/ | head -10 |
| 124 | + echo "" |
| 125 | + echo " 代码统计:" |
| 126 | + find $REMOTE_DIR -name "*.py" | wc -l | xargs echo " Python文件数量:" |
| 127 | + find $REMOTE_DIR -name "*.md" | wc -l | xargs echo " Markdown文件数量:" |
| 128 | + echo " 部署验证通过" |
| 129 | + else |
| 130 | + echo " 远程目录不存在，部署可能失败" |
| 131 | + exit 1 |
| 132 | + fi |
| 133 | +EOF |
| 134 | + |
| 135 | +echo "" |
| 136 | +echo "步骤4: 在开发机构建 ComfyUI 镜像（供 Agent 多阶段 COPY）..." |
| 137 | +ssh $DEV_SERVER << EOF |
| 138 | + set -e |
| 139 | + cd $REMOTE_DIR |
| 140 | + echo " 当前目录: \$(pwd)" |
| 141 | + echo " ComfyUI 将打 tag: $COMFYUI_BUILD_TAG" |
| 142 | + if [ ! -f "src/code/comfyui/Makefile" ]; then |
| 143 | + echo " [ERROR] 未找到 src/code/comfyui/Makefile" |
| 144 | + exit 1 |
| 145 | + fi |
| 146 | + make -C src/code/comfyui build COMFYUI_IMAGE="$COMFYUI_BUILD_TAG" |
| 147 | + echo " ComfyUI 构建完成" |
| 148 | +EOF |
| 149 | + |
| 150 | +echo "" |
| 151 | +echo "步骤5: 在开发机构建 Agent 镜像（COMFYUI_IMAGE=$COMFYUI_BUILD_TAG）..." |
| 152 | +ssh $DEV_SERVER << EOF |
| 153 | + set -e |
| 154 | + cd $REMOTE_DIR |
| 155 | + echo " 当前目录: \$(pwd)" |
| 156 | + if [ ! -f "Makefile" ]; then |
| 157 | + echo " [ERROR] 未发现根目录 Makefile" |
| 158 | + exit 1 |
| 159 | + fi |
| 160 | + make build AGENT_IMAGE="$AGENT_IMAGE" COMFYUI_IMAGE="$COMFYUI_BUILD_TAG" |
| 161 | + echo " Agent 构建完成" |
| 162 | +EOF |
| 163 | + |
| 164 | +echo "" |
| 165 | +echo "步骤6: 推送 Agent 镜像到仓库..." |
| 166 | +ssh $DEV_SERVER << EOF |
| 167 | + set -e |
| 168 | + cd $REMOTE_DIR |
| 169 | + echo " 推送镜像: $AGENT_IMAGE" |
| 170 | + docker push "$AGENT_IMAGE" |
| 171 | + echo " 镜像推送完成" |
| 172 | +EOF |
| 173 | + |
| 174 | +echo "" |
| 175 | +echo "==========================================" |
| 176 | +echo " 部署完成（ComfyUI 本地 + Agent 已推送）" |
| 177 | +echo "==========================================" |
| 178 | +echo " 分支: $BRANCH" |
| 179 | +echo " ComfyUI (开发机本地): $COMFYUI_BUILD_TAG" |
| 180 | +echo " Agent 镜像: $AGENT_IMAGE" |
| 181 | +echo "==========================================" |
0 commit comments