# 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 "修复缺失的 openai 依赖..."
sed -i 's/groq = "^0.4.0"/groq = "^0.4.0"\nopenai = "^1.0.0"/' pyproject.toml
echo "正在构建 Docker 镜像（原生 amd64 架构）..."
docker build $tagParams .
echo "正在推送 Docker 镜像..."
docker push ${imageName}:amd64
"@

# 将脚本内容转为 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