|
|
@@ -0,0 +1,125 @@
|
|
|
+# Cognio 远程构建 Docker 镜像脚本（海外服务器版本，无国内加速）
|
|
|
+
|
|
|
+[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
|
|
|
+
|
|
|
+# ============================================================
|
|
|
+# 配置区
|
|
|
+# ============================================================
|
|
|
+
|
|
|
+# 远程服务器配置
|
|
|
+$remoteUser = "root"
|
|
|
+$remoteHost = "vps-apq3.zalhb.com"
|
|
|
+$remotePort = "22"
|
|
|
+
|
|
|
+# Git 仓库配置
|
|
|
+$repoUrl = "https://github.com/0xReLogic/Cognio.git"
|
|
|
+$repoDir = "Cognio"
|
|
|
+
|
|
|
+# Docker 镜像配置
|
|
|
+$imageName = "amwpfiqvy/cognio"
|
|
|
+
|
|
|
+# tmux 会话名
|
|
|
+$tmuxSession = "dbx"
|
|
|
+
|
|
|
+# ============================================================
|
|
|
+# 主程序
|
|
|
+# ============================================================
|
|
|
+
|
|
|
+Write-Host ""
|
|
|
+Write-Host "========================================" -ForegroundColor Cyan
|
|
|
+Write-Host " Cognio 远程构建 Docker 镜像" -ForegroundColor Cyan
|
|
|
+Write-Host "========================================" -ForegroundColor Cyan
|
|
|
+Write-Host ""
|
|
|
+
|
|
|
+# Docker 镜像标签：提示用户输入
|
|
|
+$imageTagInput = Read-Host "请输入额外的 Docker 镜像标签 (多个用空格分隔，留空则仅推送 amd64)"
|
|
|
+
|
|
|
+$inputTags = @()
|
|
|
+
|
|
|
+if ($imageTagInput -ne "") {
|
|
|
+ $inputTags = $imageTagInput -split '\s+' | Where-Object { $_ -ne "" }
|
|
|
+}
|
|
|
+
|
|
|
+# 构建最终标签列表
|
|
|
+$imageTags = @("amd64") + $inputTags | Select-Object -Unique
|
|
|
+
|
|
|
+# 构建 -t 参数列表
|
|
|
+$tagParams = ($imageTags | ForEach-Object { "-t ${imageName}:$_" }) -join " "
|
|
|
+
|
|
|
+Write-Host "镜像标签: $($imageTags -join ', ')" -ForegroundColor Yellow
|
|
|
+Write-Host ""
|
|
|
+
|
|
|
+# 支持命令行参数覆盖默认配置
|
|
|
+if ($args.Count -ge 1) { $remoteUser = $args[0] }
|
|
|
+if ($args.Count -ge 2) { $remoteHost = $args[1] }
|
|
|
+if ($args.Count -ge 3) { $remotePort = $args[2] }
|
|
|
+
|
|
|
+Write-Host "连接到: $remoteUser@$remoteHost`:$remotePort" -ForegroundColor Yellow
|
|
|
+Write-Host ""
|
|
|
+
|
|
|
+# ============================================================
|
|
|
+# 构建远程执行的 Shell 脚本
|
|
|
+# ============================================================
|
|
|
+
|
|
|
+# 生成远程脚本内容（写入临时文件执行，避免引号嵌套问题）
|
|
|
+$scriptContent = @"
|
|
|
+#!/bin/bash
|
|
|
+cd ~
|
|
|
+if [ -d "$repoDir" ]; then
|
|
|
+ echo "目录已存在，还原本地修改并拉取最新代码..."
|
|
|
+ cd $repoDir
|
|
|
+ git checkout .
|
|
|
+ git pull
|
|
|
+else
|
|
|
+ echo "目录不存在，正在克隆仓库..."
|
|
|
+ git clone $repoUrl
|
|
|
+ cd $repoDir
|
|
|
+fi
|
|
|
+echo "配置 pip 只使用预编译包（避免 triton 编译问题）..."
|
|
|
+sed -i '/^FROM/a ENV PIP_ONLY_BINARY=:all:' Dockerfile
|
|
|
+echo "正在构建并推送 Docker 镜像..."
|
|
|
+docker buildx build \
|
|
|
+ --platform linux/amd64 \
|
|
|
+ $tagParams \
|
|
|
+ --cache-from type=local,src=`$HOME/.buildx-cache \
|
|
|
+ --cache-to type=local,dest=`$HOME/.buildx-cache,mode=max \
|
|
|
+ --push \
|
|
|
+ .
|
|
|
+"@
|
|
|
+
|
|
|
+# 将脚本内容转为 base64，避免特殊字符问题
|
|
|
+$scriptBase64 = [Convert]::ToBase64String([System.Text.Encoding]::UTF8.GetBytes($scriptContent))
|
|
|
+
|
|
|
+# tmux 命令：解码并执行脚本
|
|
|
+$remoteCmd = "tmux has-session -t $tmuxSession 2>/dev/null || tmux new-session -d -s $tmuxSession; " +
|
|
|
+ "tmux send-keys -t $tmuxSession 'echo $scriptBase64 | base64 -d > /tmp/build_cognio.sh && bash /tmp/build_cognio.sh' Enter"
|
|
|
+
|
|
|
+# ============================================================
|
|
|
+# 执行远程命令
|
|
|
+# ============================================================
|
|
|
+
|
|
|
+Write-Host "正在发送命令到 tmux 会话: $tmuxSession" -ForegroundColor Yellow
|
|
|
+Write-Host ""
|
|
|
+
|
|
|
+& ssh -p $remotePort "$remoteUser@$remoteHost" $remoteCmd
|
|
|
+
|
|
|
+if ($LASTEXITCODE -eq 0) {
|
|
|
+ Write-Host ""
|
|
|
+ Write-Host "========================================" -ForegroundColor Green
|
|
|
+ Write-Host " 命令已发送到 tmux 会话: $tmuxSession" -ForegroundColor Green
|
|
|
+ Write-Host "========================================" -ForegroundColor Green
|
|
|
+ Write-Host ""
|
|
|
+ Write-Host "推送目标:" -ForegroundColor Yellow
|
|
|
+ foreach ($tag in $imageTags) {
|
|
|
+ Write-Host " - ${imageName}:${tag}" -ForegroundColor Cyan
|
|
|
+ }
|
|
|
+ Write-Host ""
|
|
|
+ Write-Host "查看构建进度:" -ForegroundColor Yellow
|
|
|
+ Write-Host " ssh -t -p $remotePort $remoteUser@$remoteHost `"tmux attach -t $tmuxSession`"" -ForegroundColor Cyan
|
|
|
+} else {
|
|
|
+ Write-Host ""
|
|
|
+ Write-Host "部署失败，请检查错误信息" -ForegroundColor Red
|
|
|
+}
|
|
|
+
|
|
|
+Write-Host ""
|
|
|
+pause
|
