<#
.SYNOPSIS
    Bookworm Portable — 打包工具 (管理员使用)
.DESCRIPTION
    将 auto-setup.ps1 打包为 Bookworm-Setup.exe (PS2EXE)
    将 gen-authcode.js 打包为 gen-authcode.exe (pkg)
    输出到 dist\ 目录
.USAGE
    .\build.ps1          # 打包两个
    .\build.ps1 -Setup   # 只打包用户安装器
    .\build.ps1 -Admin   # 只打包管理员工具
#>
param(
    [switch]$Setup,   # 只打 Bookworm-Setup.exe
    [switch]$Admin    # 只打 gen-authcode.exe
)

$ErrorActionPreference = "Stop"
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
$DistDir   = Join-Path $ScriptDir "dist"

# 默认两个都打
$buildSetup = $Setup -or (-not $Setup -and -not $Admin)
$buildAdmin = $Admin -or (-not $Setup -and -not $Admin)

function Write-Step($msg) {
    Write-Host ""
    Write-Host "  ── $msg" -ForegroundColor Cyan
}
function Write-OK($msg)   { Write-Host "  [OK] $msg" -ForegroundColor Green  }
function Write-Warn($msg) { Write-Host "  [!]  $msg" -ForegroundColor Yellow }
function Write-Fail($msg) { Write-Host "  [!!] $msg" -ForegroundColor Red    }

Write-Host ""
Write-Host "  +------------------------------------------+" -ForegroundColor Cyan
Write-Host "  |  Bookworm Portable — Build Script        |" -ForegroundColor Cyan
Write-Host "  +------------------------------------------+" -ForegroundColor Cyan

if (-not (Test-Path $DistDir)) {
    New-Item -ItemType Directory $DistDir | Out-Null
    Write-OK "创建 dist\ 目录"
}

# ════════════════════════════════════════════════════════
#  1. Bookworm-Setup.exe (auto-setup.ps1 → PS2EXE)
# ════════════════════════════════════════════════════════
if ($buildSetup) {
    Write-Step "打包 Bookworm-Setup.exe (PS2EXE)"

    # 安装/检查 PS2EXE
    if (-not (Get-Command Invoke-ps2exe -ErrorAction SilentlyContinue)) {
        Write-Warn "PS2EXE 未安装，正在安装..."
        Install-Module ps2exe -Scope CurrentUser -Force -AllowClobber
        Import-Module ps2exe
    }

    $inputPs1  = Join-Path $ScriptDir "auto-setup.ps1"
    $outputExe = Join-Path $DistDir   "Bookworm-Setup.exe"

    if (-not (Test-Path $inputPs1)) {
        Write-Fail "找不到 auto-setup.ps1"
        exit 1
    }

    Write-Host "  输入: $inputPs1" -ForegroundColor Gray
    Write-Host "  输出: $outputExe" -ForegroundColor Gray

    # 从 auto-setup.ps1 读取版本号
    $versionLine = Select-String -Path $inputPs1 -Pattern '^\$BWVersion\s*=\s*"([^"]+)"' | Select-Object -First 1
    $bwVer = if ($versionLine) { $versionLine.Matches[0].Groups[1].Value } else { "0.0.0" }
    Write-Host "  版本: $bwVer" -ForegroundColor Gray

    # 优先用桌面专用 B 圆图标, 回退到 galaxy
    $iconFile = Join-Path $ScriptDir "bookworm-desktop.ico"
    if (-not (Test-Path $iconFile)) {
        $iconFile = Join-Path $ScriptDir "bookworm.ico"
    }
    $ps2exeArgs = @{
        InputFile   = $inputPs1
        OutputFile  = $outputExe
        Title       = "Bookworm Portable Setup v$bwVer"
        Description = "Bookworm Smart Assistant 安装向导 v$bwVer"
        Company     = "Bookworm"
        Version     = "$bwVer.0"
        NoConsole   = $true
        NoOutput    = $true
        NoError     = $true
    }
    if (Test-Path $iconFile) {
        $ps2exeArgs.IconFile = $iconFile
        Write-Host "  图标: $iconFile" -ForegroundColor Gray
    }
    Invoke-ps2exe @ps2exeArgs

    if (Test-Path $outputExe) {
        $sizeKB = [math]::Round((Get-Item $outputExe).Length / 1KB)
        Write-OK "Bookworm-Setup.exe 打包完成 (${sizeKB} KB)"
    } else {
        Write-Fail "Bookworm-Setup.exe 打包失败"
        exit 1
    }
}

# ════════════════════════════════════════════════════════
#  2. Bookworm-AuthGen.exe (admin-authcode-gui.ps1 → PS2EXE)
# ════════════════════════════════════════════════════════
if ($buildAdmin) {
    Write-Step "打包 Bookworm-AuthGen.exe (PS2EXE GUI)"

    $inputPs1  = Join-Path $ScriptDir "admin-authcode-gui.ps1"
    $outputExe = Join-Path $DistDir   "Bookworm-AuthGen.exe"

    if (-not (Test-Path $inputPs1)) {
        Write-Fail "找不到 admin-authcode-gui.ps1"
        exit 1
    }

    Write-Host "  输入: $inputPs1" -ForegroundColor Gray
    Write-Host "  输出: $outputExe" -ForegroundColor Gray

    # 优先用书虫学者图标, 回退到 B 圆
    $adminIcon = Join-Path $ScriptDir "admin-authcode.ico"
    if (-not (Test-Path $adminIcon)) { $adminIcon = Join-Path $ScriptDir "bookworm-desktop.ico" }

    $ps2exeArgs = @{
        InputFile   = $inputPs1
        OutputFile  = $outputExe
        Title       = "Bookworm AuthCode Generator"
        Description = "Bookworm 授权码生成器 (管理员工具)"
        Company     = "Bookworm"
        Version     = "1.5.1.0"
        NoConsole   = $true
        NoOutput    = $true
        NoError     = $true
    }
    if (Test-Path $adminIcon) {
        $ps2exeArgs.IconFile = $adminIcon
        Write-Host "  图标: $adminIcon" -ForegroundColor Gray
    }
    Invoke-ps2exe @ps2exeArgs

    if (Test-Path $outputExe) {
        $sizeKB = [math]::Round((Get-Item $outputExe).Length / 1KB)
        Write-OK "Bookworm-AuthGen.exe 打包完成 (${sizeKB} KB)"
    } else {
        Write-Fail "Bookworm-AuthGen.exe 打包失败"
        exit 1
    }
}

# ════════════════════════════════════════════════════════
#  完成
# ════════════════════════════════════════════════════════
Write-Host ""
Write-Host "  ============================================" -ForegroundColor Green
Write-Host "  打包完成！输出目录: dist\" -ForegroundColor Green
Write-Host "  ============================================" -ForegroundColor Green
Write-Host ""

# v3.1.1: build 后自动跑 E2E 行为测试 (闭合 L8: 防 v3.0.10 -or 类运行时 bug)
$e2eTest = Join-Path $ScriptDir "tools\test-launcher-e2e.ps1"
if (Test-Path $e2eTest) {
    Write-Host "  ── 运行 E2E 行为测试 (build 后自动护栏)" -ForegroundColor Cyan
    & pwsh -NoProfile -File $e2eTest 2>&1 | ForEach-Object { Write-Host "    $_" }
    if ($LASTEXITCODE -ne 0) {
        Write-Host ""
        Write-Host "  [!] E2E 测试失败 (exit $LASTEXITCODE)" -ForegroundColor Red
        Write-Host "  EXE 已生成但启动器契约/wrapper 有问题, 修复后重打包" -ForegroundColor Yellow
        exit 1
    }
    Write-Host ""
}

Get-ChildItem $DistDir | ForEach-Object {
    $sizeMB = [math]::Round($_.Length / 1MB, 1)
    Write-Host "    $($_.Name.PadRight(30)) ${sizeMB} MB" -ForegroundColor White
}

Write-Host ""
Write-Host "  分发说明:" -ForegroundColor Gray
Write-Host "    Bookworm-Setup.exe   → 用户安装器 (公开下载)" -ForegroundColor Gray
Write-Host "    Bookworm-AuthGen.exe → 管理员授权码生成器 (勿对外分发)" -ForegroundColor Gray
Write-Host ""