# DMS 一体化构建脚本
# 用法：在项目根目录执行 .\build\build.ps1
[CmdletBinding()]
param(
    [switch]$SkipFrontend,
    [switch]$SkipPyInstallerInstall
)

$ErrorActionPreference = "Stop"
$ProjectRoot = Split-Path $PSScriptRoot -Parent
$Backend = Join-Path $ProjectRoot "backend"
$Frontend = Join-Path $ProjectRoot "frontend"
$Dist = Join-Path $ProjectRoot "dist-package"
$BuildCache = Join-Path $PSScriptRoot "__pycache_build"

Write-Host "========== DMS 一体化构建 ==========" -ForegroundColor Cyan
Write-Host "项目根目录: $ProjectRoot"

# ---------- 1. 前端构建 ----------
if (-not $SkipFrontend) {
    Write-Host "`n[1/5] 构建前端..." -ForegroundColor Yellow
    Push-Location $Frontend
    try {
        if (-not (Test-Path "node_modules")) {
            Write-Host "  npm install..." -ForegroundColor DarkGray
            npm install
            if ($LASTEXITCODE -ne 0) { throw "npm install 失败" }
        }
        npm run build
        if ($LASTEXITCODE -ne 0) { throw "前端构建失败" }
    } finally { Pop-Location }
} else {
    Write-Host "`n[1/5] 跳过前端构建 (-SkipFrontend)" -ForegroundColor DarkGray
}

$indexHtml = Join-Path $Frontend "dist\index.html"
if (-not (Test-Path $indexHtml)) {
    throw "前端产物不存在: $indexHtml - 请先执行 npm run build"
}

# ---------- 2. 检查 PyInstaller ----------
Write-Host "`n[2/5] 检查 PyInstaller..." -ForegroundColor Yellow
Push-Location $Backend
try {
    $pyinstallerVersion = & python -m PyInstaller --version 2>$null
    if ($LASTEXITCODE -ne 0 -or -not $pyinstallerVersion) {
        if ($SkipPyInstallerInstall) {
            throw "PyInstaller 未安装且 -SkipPyInstallerInstall 已指定"
        }
        Write-Host "  安装 PyInstaller..." -ForegroundColor DarkGray
        python -m pip install pyinstaller
        if ($LASTEXITCODE -ne 0) { throw "PyInstaller 安装失败" }
    } else {
        Write-Host "  PyInstaller 版本: $pyinstallerVersion" -ForegroundColor DarkGray
    }
} finally { Pop-Location }

# ---------- 3. PyInstaller 打包 ----------
Write-Host "`n[3/5] PyInstaller 打包..." -ForegroundColor Yellow
if (Test-Path $BuildCache) { Remove-Item $BuildCache -Recurse -Force }
if (Test-Path (Join-Path $Dist "dms-server")) {
    Remove-Item (Join-Path $Dist "dms-server") -Recurse -Force
}

Push-Location $Backend
try {
    python -m PyInstaller `
        --noconfirm `
        --clean `
        --distpath $Dist `
        --workpath $BuildCache `
        (Join-Path $PSScriptRoot "dms-server.spec")
    if ($LASTEXITCODE -ne 0) { throw "PyInstaller 打包失败" }
} finally { Pop-Location }

# ---------- 4. 整理发行包 ----------
Write-Host "`n[4/5] 整理发行包..." -ForegroundColor Yellow
$ReleaseDir = Join-Path $Dist "dms-server"
if (-not (Test-Path $ReleaseDir)) { throw "发行目录未找到: $ReleaseDir" }

# 复制配置模板（不含真实 .env，保护凭据）
Copy-Item (Join-Path $Backend ".env.example") $ReleaseDir -Force
# 尝试调用项目自带的导出工具，把示例数据库导出为 SQL 文件
$ExportTool = Join-Path $ProjectRoot "tools\export_full_demo_data.py"
$ExportedSql = Join-Path $ReleaseDir "database-init.sql"
if (Test-Path $ExportTool) {
    Write-Host "  导出示例数据库 SQL..." -ForegroundColor DarkGray
    Push-Location $Backend
    try {
        # 通过当前 Python 环境运行导出工具
        python $ExportTool --output $ExportedSql 2>&1 | Out-Host
        # 不判断退出码（不同版本的导出工具参数可能不同），仅检查产物
        if (-not (Test-Path $ExportedSql)) {
            Write-Warning "导出工具未生成 $ExportedSql，请手动准备 SQL 文件"
        }
    } catch {
        Write-Warning "导出工具执行失败：$_"
    } finally { Pop-Location }
} else {
    Write-Warning "未找到 tools\export_full_demo_data.py，跳过自动导出"
}
# 复制说明文档
Copy-Item (Join-Path $PSScriptRoot "README.txt") $ReleaseDir -Force

# ---------- 5. 从备份恢复 .env 和 dms-storage ----------
Write-Host "`n[5/5] 从备份恢复 .env 和 dms-storage..." -ForegroundColor Yellow
$BackupDir = Join-Path $ProjectRoot "dist-package-bak\dms-server"
if (-not (Test-Path $BackupDir)) {
    Write-Warning "备份目录不存在: $BackupDir - 跳过恢复"
} else {
    $BackupEnv = Join-Path $BackupDir ".env"
    if (Test-Path $BackupEnv) {
        Copy-Item $BackupEnv $ReleaseDir -Force
        Write-Host "  已恢复 .env" -ForegroundColor DarkGray
    } else {
        Write-Warning "备份中没有 .env，跳过"
    }
    $BackupStorage = Join-Path $BackupDir "dms-storage"
    if (Test-Path $BackupStorage) {
        $TargetStorage = Join-Path $ReleaseDir "dms-storage"
        if (Test-Path $TargetStorage) {
            Remove-Item $TargetStorage -Recurse -Force
        }
        Copy-Item $BackupStorage $TargetStorage -Recurse -Force
        Write-Host "  已恢复 dms-storage" -ForegroundColor DarkGray
    } else {
        Write-Warning "备份中没有 dms-storage，跳过"
    }
}

Write-Host "`n========== 构建完成 ==========" -ForegroundColor Green
Write-Host "发行包目录: $ReleaseDir" -ForegroundColor Green
$sizeMB = [math]::Round((Get-ChildItem $ReleaseDir -Recurse | Measure-Object -Property Length -Sum).Sum / 1MB, 1)
Write-Host "包大小: ${sizeMB} MB" -ForegroundColor Green
Write-Host "将该目录整体压缩即可分发给最终用户" -ForegroundColor Green