# pack-release.ps1
# 支持每个项目独立版本的打包脚本
param(
    [switch]$NoBuild,
    [string]$OutputDir,
    [string[]]$Projects  # 可选：指定要打包的项目，不指定则打包所有
)

$ErrorActionPreference = 'Stop'
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
$RootDir = Split-Path -Parent $ScriptDir
$PropsFile = Join-Path $RootDir 'Directory.Build.props'
$DefaultOutputDir = Join-Path $RootDir 'nupkgs'
$VersionsDir = Join-Path $RootDir 'versions'

function Write-ColorText {
    param([string]$Text, [string]$Color = 'White')
    Write-Host $Text -ForegroundColor $Color
}

# 读取 Y/N 确认，按Q立即退出，回车默认Y
function Read-Confirm {
    param([string]$Prompt)
    Write-Host $Prompt -NoNewline
    while ($true) {
        $key = $Host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown')
        if ($key.Character -eq 'q' -or $key.Character -eq 'Q') {
            Write-Host ''
            Write-ColorText '已退出' 'Yellow'
            exit 0
        }
        if ($key.Character -eq 'y' -or $key.Character -eq 'Y' -or $key.VirtualKeyCode -eq 13) {
            Write-Host 'Y'
            return $true
        }
        if ($key.Character -eq 'n' -or $key.Character -eq 'N') {
            Write-Host 'N'
            return $false
        }
    }
}

# 获取项目版本（从 versions/{ProjectName}/v*.md 目录）
function Get-ProjectVersion {
    param([string]$ProjectName)

    $projectVersionDir = Join-Path $VersionsDir $ProjectName

    # 优先使用项目独立版本目录
    if (Test-Path $projectVersionDir) {
        $versionFiles = @(Get-ChildItem -Path $projectVersionDir -Filter 'v*.md' -ErrorAction SilentlyContinue)
    } else {
        # 回退到根版本目录
        $versionFiles = @(Get-ChildItem -Path $VersionsDir -Filter 'v*.md' -ErrorAction SilentlyContinue)
    }

    $versions = @($versionFiles | Where-Object { $_.BaseName -match '^v(\d+)\.(\d+)\.(\d+)' } | ForEach-Object {
        $fullVersion = $_.BaseName -replace '^v', ''
        $baseVersion = $_.BaseName -replace '^v(\d+\.\d+\.\d+).*', '$1'
        [PSCustomObject]@{
            Name = $fullVersion
            Version = [version]$baseVersion
        }
    } | Sort-Object Version -Descending)

    if ($versions.Count -gt 0) {
        return $versions[0].Name
    }
    return $null
}

Write-ColorText "`n========================================" 'Cyan'
Write-ColorText '  Apq.Cfg NuGet 包生成工具' 'Cyan'
Write-ColorText '  支持独立版本管理' 'DarkCyan'
Write-ColorText "========================================" 'Cyan'
Write-ColorText '  按 Q 随时退出' 'DarkGray'
Write-ColorText "========================================`n" 'Cyan'

if (-not (Test-Path $PropsFile)) {
    Write-ColorText '错误: 找不到 Directory.Build.props 文件' 'Red'
    Write-ColorText "路径: $PropsFile" 'Red'
    exit 1
}

if (-not (Test-Path $VersionsDir)) {
    Write-ColorText '错误: 找不到 versions 目录' 'Red'
    Write-ColorText "路径: $VersionsDir" 'Red'
    exit 1
}

# 定义所有可打包的项目
$AllProjects = @(
    'Apq.Cfg',
    'Apq.Cfg.Ini',
    'Apq.Cfg.Xml',
    'Apq.Cfg.Yaml',
    'Apq.Cfg.Toml',
    'Apq.Cfg.Env',
    'Apq.Cfg.Redis',
    'Apq.Cfg.Database',
    'Apq.Cfg.Consul',
    'Apq.Cfg.Etcd',
    'Apq.Cfg.Nacos',
    'Apq.Cfg.Apollo',
    'Apq.Cfg.Zookeeper',
    'Apq.Cfg.Vault',
    'Apq.Cfg.Crypto',
    'Apq.Cfg.Crypto.DataProtection',
    'Apq.Cfg.Crypto.Tool',
    'Apq.Cfg.SourceGenerator'
)

# 如果指定了项目，则只打包指定的项目
if ($Projects -and $Projects.Count -gt 0) {
    $TargetProjects = $Projects
} else {
    $TargetProjects = $AllProjects
}

# 设置输出目录
if ([string]::IsNullOrWhiteSpace($OutputDir)) {
    $OutputDir = $DefaultOutputDir
}

Write-Host ''
Write-ColorText '将要打包的项目及版本:' 'Cyan'

$projectVersions = @{}
foreach ($project in $TargetProjects) {
    $version = Get-ProjectVersion -ProjectName $project
    if ($version) {
        $projectVersions[$project] = $version
        Write-ColorText "  - $project @ v$version" 'White'
    } else {
        Write-ColorText "  - $project @ (未找到版本)" 'Yellow'
    }
}

Write-Host ''
Write-ColorText "输出目录: $OutputDir" 'Gray'
Write-Host ''

if (-not (Read-Confirm '确认开始打包? ([Y]/N): ')) {
    Write-ColorText '已取消' 'Yellow'
    exit 0
}

# 询问是否生成文档
$generateDocs = Read-Confirm '是否重新生成 API 文档? ([Y]/N): '

# 清空并创建输出目录
if (Test-Path $OutputDir) {
    Write-ColorText "清空输出目录: $OutputDir" 'Gray'
    Remove-Item -Path "$OutputDir\*" -Force -Recurse -ErrorAction SilentlyContinue
} else {
    New-Item -ItemType Directory -Path $OutputDir -Force | Out-Null
    Write-ColorText "已创建输出目录: $OutputDir" 'Gray'
}

Write-Host ''
Write-ColorText '开始打包...' 'Cyan'
Write-Host ''

# 检查 DefaultDocumentation 是否安装（仅在需要生成文档时）
if ($generateDocs) {
    Write-ColorText '检查文档生成工具...' 'Gray'
    $toolInstalled = dotnet tool list -g | Select-String 'defaultdocumentation'
    if (-not $toolInstalled) {
        Write-ColorText '  安装 DefaultDocumentation...' 'Yellow'
        dotnet tool install -g DefaultDocumentation.Console 2>&1 | Out-Null
    }
    Write-ColorText '  文档生成工具就绪' 'Green'
    Write-Host ''
}

$successCount = 0
$failCount = 0
$generatedPackages = @()

foreach ($project in $TargetProjects) {
    $version = $projectVersions[$project]
    if (-not $version) {
        Write-ColorText "跳过 $project (未找到版本)" 'Yellow'
        continue
    }

    # 查找项目文件
    $projectPath = Join-Path $RootDir "$project/$project.csproj"
    if (-not (Test-Path $projectPath)) {
        Write-ColorText "跳过 $project (项目文件不存在)" 'Yellow'
        continue
    }

    Write-ColorText "打包 $project v$version..." 'Gray'

    # 删除当前版本的旧包
    $oldPackages = Get-ChildItem -Path $OutputDir -Filter "$project.$version.*pkg" -ErrorAction SilentlyContinue
    foreach ($pkg in $oldPackages) {
        Remove-Item $pkg.FullName -Force
    }

    # 构建打包参数
    $packArgs = @(
        'pack'
        $projectPath
        '-c', 'Release'
        '-o', $OutputDir
    )

    if ($NoBuild) {
        $packArgs += '--no-build'
    }

    try {
        & dotnet @packArgs 2>&1 | Out-Null

        if ($LASTEXITCODE -eq 0) {
            $successCount++
            $generatedPackages += "$project.$version.nupkg"
            Write-ColorText "  ✓ $project v$version" 'Green'

            # 在打包成功后立即生成该项目的文档（如果用户选择生成）
            if ($generateDocs) {
                Write-ColorText "    生成 $project 文档..." 'Gray'

                # 获取所有支持的 .NET 版本
                # SourceGenerator 使用 netstandard2.0，其他项目使用 net10.0/net8.0
                if ($project -eq 'Apq.Cfg.SourceGenerator') {
                    $netVersions = @('netstandard2.0')
                } else {
                    $netVersions = @('net10.0', 'net8.0')
                }

                # 确定项目输出目录名
                $outputDirName = switch ($project) {
                    'Apq.Cfg' { 'core' }
                    'Apq.Cfg.Crypto' { 'crypto' }
                    'Apq.Cfg.Ini' { 'ini' }
                    'Apq.Cfg.Xml' { 'xml' }
                    'Apq.Cfg.Yaml' { 'yaml' }
                    'Apq.Cfg.Toml' { 'toml' }
                    'Apq.Cfg.Env' { 'env' }
                    'Apq.Cfg.Redis' { 'redis' }
                    'Apq.Cfg.Consul' { 'consul' }
                    'Apq.Cfg.Etcd' { 'etcd' }
                    'Apq.Cfg.Nacos' { 'nacos' }
                    'Apq.Cfg.Apollo' { 'apollo' }
                    'Apq.Cfg.Zookeeper' { 'zookeeper' }
                    'Apq.Cfg.Vault' { 'vault' }
                    'Apq.Cfg.Database' { 'database' }
                    'Apq.Cfg.Crypto.DataProtection' { 'crypto-dp' }
                    'Apq.Cfg.SourceGenerator' { 'sourcegen' }
                    default { $project.ToLower() }
                }

                # 为每个 .NET 版本生成文档
                foreach ($netVersion in $netVersions) {
                    $dllPath = Join-Path $RootDir "$project\bin\Release\$netVersion\$project.dll"
                    $apiDocsDir = Join-Path $RootDir "docs\site\api\$netVersion"
                    $projOutputDir = Join-Path $apiDocsDir $outputDirName

                    if (Test-Path $dllPath) {
                        # 创建 API 文档目录
                        if (-not (Test-Path $apiDocsDir)) {
                            New-Item -ItemType Directory -Path $apiDocsDir -Force | Out-Null
                        }

                        # 清空并创建输出目录
                        if (Test-Path $projOutputDir) {
                            Remove-Item -Path "$projOutputDir\*" -Force -Recurse -ErrorAction SilentlyContinue
                        } else {
                            New-Item -ItemType Directory -Path $projOutputDir -Force | Out-Null
                        }

                        # 运行 DefaultDocumentation
                        & defaultdocumentation -a $dllPath -o $projOutputDir 2>&1 | Out-Null
                        if ($LASTEXITCODE -eq 0) {
                            # 生成 index.md（从主命名空间文件复制）
                            $mainMdFile = Get-ChildItem -Path $projOutputDir -Filter "$project.md" -ErrorAction SilentlyContinue | Select-Object -First 1
                            if ($mainMdFile) {
                                $indexPath = Join-Path $projOutputDir 'index.md'
                                $content = [System.IO.File]::ReadAllText($mainMdFile.FullName, [System.Text.Encoding]::UTF8)
                                $utf8NoBom = New-Object System.Text.UTF8Encoding($false)
                                [System.IO.File]::WriteAllText($indexPath, $content, $utf8NoBom)
                            }
                            Write-ColorText "      ✓ $netVersion 文档" 'Green'
                        } else {
                            Write-ColorText "      ✗ $netVersion 文档生成失败" 'Red'
                        }
                    }
                }
            }
        } else {
            $failCount++
            Write-ColorText "  ✗ $project 打包失败" 'Red'
        }
    } catch {
        $failCount++
        Write-ColorText "  ✗ $project 打包异常: $($_.Exception.Message)" 'Red'
    }
}

Write-Host ''
Write-ColorText "========================================" 'Cyan'
Write-ColorText "打包完成!" 'Green'
Write-ColorText "  成功: $successCount" 'Green'
if ($failCount -gt 0) {
    Write-ColorText "  失败: $failCount" 'Red'
}
Write-ColorText "========================================" 'Cyan'
Write-Host ''

# 生成 API 索引页和符号链接（仅在用户选择生成文档时）
if ($generateDocs) {
    # 生成所有 .NET 版本的 API 索引页
    Write-ColorText '生成 API 索引页...' 'Cyan'

    $netVersions = @('net10.0', 'net8.0')
    $generatedDocVersions = @()

    # 检查每个 .NET 版本是否有文档生成
    foreach ($netVersion in $netVersions) {
        $apiDocsDir = Join-Path $RootDir "docs\site\api\$netVersion"
        if (Test-Path $apiDocsDir) {
            $docDirs = Get-ChildItem -Path $apiDocsDir -Directory -ErrorAction SilentlyContinue
            if ($docDirs.Count -gt 0) {
                $generatedDocVersions += $netVersion
            }
        }
    }

    # 为每个 .NET 版本生成索引页
    foreach ($netVersion in $generatedDocVersions) {
        $apiDocsDir = Join-Path $RootDir "docs\site\api\$netVersion"

        # 根据版本调整标题
        $versionTitle = switch ($netVersion) {
            'net8.0' { '.NET 8.0' }
            'net10.0' { '.NET 10.0' }
            default { $netVersion }
        }

        $indexContent = @"
# API 参考 ($versionTitle)

本节包含 Apq.Cfg 所有公开 API 的详细文档，由代码注释自动生成。

> 当前文档基于 $versionTitle 版本生成。各版本 API 基本一致，仅内部实现有差异。

## 核心库

- [Apq.Cfg](./core/) - 核心配置库（JSON、环境变量、DI 集成）

## 加密脱敏

- [Apq.Cfg.Crypto](./crypto/) - 配置加密脱敏
- [Apq.Cfg.Crypto.DataProtection](./crypto-dp/) - ASP.NET Core DataProtection 集成

## 本地配置源

- [Apq.Cfg.Ini](./ini/) - INI 格式支持
- [Apq.Cfg.Xml](./xml/) - XML 格式支持
- [Apq.Cfg.Yaml](./yaml/) - YAML 格式支持
- [Apq.Cfg.Toml](./toml/) - TOML 格式支持
- [Apq.Cfg.Env](./env/) - .env 文件支持

## 数据存储配置源

- [Apq.Cfg.Database](./database/) - 数据库配置源
- [Apq.Cfg.Redis](./redis/) - Redis 配置源

## 远程配置中心

- [Apq.Cfg.Consul](./consul/) - Consul 配置中心
- [Apq.Cfg.Apollo](./apollo/) - Apollo 配置中心
- [Apq.Cfg.Nacos](./nacos/) - Nacos 配置中心
- [Apq.Cfg.Vault](./vault/) - HashiCorp Vault
- [Apq.Cfg.Etcd](./etcd/) - Etcd 配置中心
- [Apq.Cfg.Zookeeper](./zookeeper/) - Zookeeper 配置中心
"@

        # SourceGenerator 使用 netstandard2.0，单独处理
        if ($netVersion -eq 'net10.0') {
            $indexContent += @"

## 源生成器

- [Apq.Cfg.SourceGenerator](../netstandard2.0/sourcegen/) - 源生成器（Native AOT 支持）
"@
        }

        $indexPath = Join-Path $apiDocsDir 'index.md'
        $utf8NoBom = New-Object System.Text.UTF8Encoding($false)
        [System.IO.File]::WriteAllText($indexPath, $indexContent, $utf8NoBom)
    }

    # 生成总的 API 索引页
    $apiRootDir = Join-Path $RootDir 'docs\site\api'
    $rootIndexContent = @"
# API 参考

本节包含 Apq.Cfg 所有公开 API 的详细文档，由代码注释自动生成。

> 各版本 API 基本一致，仅内部实现有差异。建议使用最新版本。

## 可用版本

"@

    foreach ($netVersion in $generatedDocVersions) {
        $versionTitle = switch ($netVersion) {
            'net8.0' { '.NET 8.0' }
            'net10.0' { '.NET 10.0' }
            default { $netVersion }
        }

        $rootIndexContent += "- [$versionTitle](./$netVersion/)`n"
    }

    $rootIndexPath = Join-Path $apiRootDir 'index.md'
    $utf8NoBom = New-Object System.Text.UTF8Encoding($false)
    [System.IO.File]::WriteAllText($rootIndexPath, $rootIndexContent, $utf8NoBom)

    Write-ColorText "  API 索引页生成完成" 'Green'

    # 为英文版创建 API 文档的符号链接（解决语言切换 404 问题）
    Write-ColorText '创建英文版 API 符号链接...' 'Cyan'
    $enApiDir = Join-Path $RootDir 'docs\site\en\api'

    # 包含 netstandard2.0（SourceGenerator 使用）
    $allDocVersions = $generatedDocVersions + @('netstandard2.0')

    foreach ($netVersion in $allDocVersions) {
        $sourcePath = Join-Path $RootDir "docs\site\api\$netVersion"
        $targetPath = Join-Path $enApiDir $netVersion

        # 如果源目录不存在，跳过
        if (-not (Test-Path $sourcePath)) {
            continue
        }

        # 如果目标已存在，先删除
        if (Test-Path $targetPath) {
            Remove-Item $targetPath -Force -Recurse -ErrorAction SilentlyContinue
        }

        # 创建目录连接（junction）- 不需要管理员权限
        cmd /c mklink /J "$targetPath" "$sourcePath" 2>&1 | Out-Null
        if ($LASTEXITCODE -eq 0) {
            Write-ColorText "  ✓ $netVersion" 'Green'
        } else {
            Write-ColorText "  ✗ $netVersion 创建失败" 'Red'
        }
    }
    Write-ColorText "  英文版 API 链接创建完成" 'Green'
    Write-Host ''
}

# 列出生成的包
$packages = Get-ChildItem -Path $OutputDir -Filter "Apq.Cfg*.nupkg" -ErrorAction SilentlyContinue | Sort-Object Name
if ($packages.Count -gt 0) {
    Write-ColorText '生成的 NuGet 包:' 'Cyan'
    foreach ($pkg in $packages) {
        $size = [math]::Round($pkg.Length / 1KB, 1)
        Write-ColorText "  $($pkg.Name) ($size KB)" 'White'
    }
    Write-Host ''
}

$symbolPackages = Get-ChildItem -Path $OutputDir -Filter "Apq.Cfg*.snupkg" -ErrorAction SilentlyContinue | Sort-Object Name
if ($symbolPackages.Count -gt 0) {
    Write-ColorText '生成的符号包:' 'Cyan'
    foreach ($pkg in $symbolPackages) {
        $size = [math]::Round($pkg.Length / 1KB, 1)
        Write-ColorText "  $($pkg.Name) ($size KB)" 'White'
    }
    Write-Host ''
}

Write-ColorText '下一步操作:' 'Yellow'
Write-ColorText '  运行 push-nuget.bat 发布到 NuGet' 'Gray'
Write-Host