# mixapi 远程构建 Docker 镜像脚本 - 海外版本

[Console]::OutputEncoding = [System.Text.Encoding]::UTF8

# ============================================================
# 配置区
# ============================================================

# 远程服务器配置
$remoteUser = "root"
$remoteHost = "vps-apq2.zalhb.com"
$remotePort = "22"

# Git 仓库配置
$repoUrl = "https://github.com/user/dockerimg"
$repoDir = "dockerimg"
$buildDir = "mixapi/overseas"

# Docker 镜像配置
$imageName = "amwpfiqvy/mixapi"

# tmux 会话名
$tmuxSession = "dbx"

# ============================================================
# 主程序
# ============================================================

Write-Host ""
Write-Host "========================================" -ForegroundColor Cyan
Write-Host "  mixapi 远程构建 Docker 镜像 (海外版)" -ForegroundColor Cyan
Write-Host "========================================" -ForegroundColor Cyan
Write-Host ""

# Docker 镜像标签：提示用户输入（默认包含 latest，-l 排除 latest）
Write-Host "请输入额外的 Docker 镜像标签 (多个用空格分隔，-l=排除latest，留空则仅推送 latest): " -ForegroundColor Magenta -NoNewline
$imageTagInput = Read-Host

$inputTags = @()
$excludeLatest = $false

if ($imageTagInput -ne "") {
    $inputTags = $imageTagInput -split '\s+' | Where-Object { $_ -ne "" }
    if ($inputTags -contains "-l") {
        $excludeLatest = $true
        $inputTags = $inputTags | Where-Object { $_ -ne "-l" }
    }
}

# 构建最终标签列表
if ($excludeLatest) {
    $imageTags = $inputTags
} else {
    $imageTags = @("latest") + $inputTags | Select-Object -Unique
}

# 构建 -t 参数列表
$tagParams = ($imageTags | ForEach-Object { "-t ${imageName}:$_" }) -join " "

Write-Host "镜像标签: $($imageTags -join ', ')" -ForegroundColor Yellow
Write-Host "目标架构: linux/amd64, linux/arm64" -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 ~

# 检查并配置 QEMU 用户模式模拟（多架构构建必需）
echo "检查 QEMU 用户模式模拟..."
if ! docker run --rm --privileged multiarch/qemu-user-static --reset -p yes 2>/dev/null; then
    echo "警告: QEMU 配置失败，多架构构建可能会失败"
fi

# 克隆或更新仓库
if [ ! -d "$repoDir" ]; then
    echo "目录不存在，正在克隆仓库..."
    git clone $repoUrl $repoDir
else
    echo "目录已存在，跳过克隆"
fi

cd $repoDir

# 拉取最新代码
echo "正在拉取最新代码..."
git pull

# 进入构建目录
cd $buildDir

# 构建并推送 Docker 镜像（多架构）
echo "正在构建并推送 Docker 镜像..."
docker buildx build \
    --platform linux/amd64,linux/arm64 \
    $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_mixapi.sh && bash /tmp/build_mixapi.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 ""
Read-Host "按回车退出"