231 lines
7.6 KiB
PowerShell
231 lines
7.6 KiB
PowerShell
<#
|
||
.SYNOPSIS
|
||
服务器环境初始化脚本：删除 skip-worktree 排除的文件/目录 + 创建 export 目录树。
|
||
|
||
.DESCRIPTION
|
||
在服务器上 git clone + setup-server-git.py 之后运行。
|
||
1. 删除已被 skip-worktree 标记的目录和散文件（释放磁盘空间）
|
||
2. 创建完整的 export/ 目录树（ETL/SYSTEM/BACKEND 三大类）
|
||
|
||
三个环境都需要 export 目录（LOG、JSON、REPORTS 等运行时输出在每个环境都会产生）。
|
||
|
||
.PARAMETER Envs
|
||
要初始化的环境列表，默认 test,prod。
|
||
可选值：test, prod（逗号分隔）。
|
||
|
||
.EXAMPLE
|
||
# 初始化 test + prod（默认）
|
||
.\init-server-env.ps1
|
||
|
||
# 只初始化 test
|
||
.\init-server-env.ps1 -Envs test
|
||
|
||
# 只初始化 prod
|
||
.\init-server-env.ps1 -Envs prod
|
||
#>
|
||
|
||
param(
|
||
[string[]]$Envs = @("test", "prod")
|
||
)
|
||
|
||
$ErrorActionPreference = "Stop"
|
||
|
||
# ============================================================================
|
||
# 配置
|
||
# ============================================================================
|
||
|
||
# 服务器根目录
|
||
$ServerRoot = "D:\NeoZQYY"
|
||
|
||
# 环境 → repo 路径映射
|
||
$EnvPaths = @{
|
||
"test" = "$ServerRoot\test\repo"
|
||
"prod" = "$ServerRoot\prod\repo"
|
||
}
|
||
|
||
# skip-worktree 后可安全删除的目录（与 setup-server-git.py DELETABLE_DIRS 一致）
|
||
$DeletableDirs = @(
|
||
"export"
|
||
"docs"
|
||
"tests"
|
||
"samples"
|
||
"infra"
|
||
".kiro"
|
||
".hypothesis"
|
||
".pytest_cache"
|
||
"apps\miniprogram"
|
||
"scripts\ops"
|
||
"scripts\audit"
|
||
"scripts\migrate"
|
||
)
|
||
|
||
# 可安全删除的根目录散文件（与 setup-server-git.py SKIP_PREFIXES 中的散文件一致）
|
||
$DeletableFiles = @(
|
||
"coach-detail-full.png"
|
||
"customer-detail-full.png"
|
||
"perf-records-current.png"
|
||
"white-screen-debug.png"
|
||
"NeoZQYY.code-workspace"
|
||
"start-admin.bat"
|
||
".kiroignore"
|
||
)
|
||
|
||
# export 目录树（所有环境通用，运行时输出在每个环境都会产生）
|
||
$ExportDirs = @(
|
||
"export\ETL-Connectors\feiqiu\JSON"
|
||
"export\ETL-Connectors\feiqiu\LOGS"
|
||
"export\ETL-Connectors\feiqiu\REPORTS"
|
||
"export\SYSTEM\LOGS"
|
||
"export\SYSTEM\REPORTS\dataflow_analysis"
|
||
"export\SYSTEM\REPORTS\field_audit"
|
||
"export\SYSTEM\REPORTS\full_dataflow_doc"
|
||
"export\SYSTEM\CACHE\api_samples"
|
||
"export\BACKEND\LOGS"
|
||
)
|
||
|
||
|
||
# ============================================================================
|
||
# 函数
|
||
# ============================================================================
|
||
|
||
function Remove-SkipWorktreeItems {
|
||
<#
|
||
.SYNOPSIS
|
||
删除指定 repo 下已被 skip-worktree 标记的目录和散文件。
|
||
#>
|
||
param([string]$RepoPath)
|
||
|
||
Write-Host "`n [删除] 清理 skip-worktree 排除的目录和文件..." -ForegroundColor Yellow
|
||
|
||
$freedMB = 0
|
||
|
||
# 删除目录
|
||
foreach ($dir in $DeletableDirs) {
|
||
$fullPath = Join-Path $RepoPath $dir
|
||
if (Test-Path $fullPath) {
|
||
$size = (Get-ChildItem $fullPath -Recurse -File -ErrorAction SilentlyContinue |
|
||
Measure-Object -Property Length -Sum).Sum
|
||
$sizeMB = [math]::Round($size / 1MB, 1)
|
||
Remove-Item $fullPath -Recurse -Force
|
||
Write-Host " 已删除: $dir/ ($sizeMB MB)" -ForegroundColor Red
|
||
$freedMB += $sizeMB
|
||
} else {
|
||
Write-Host " 跳过: $dir/ (不存在)" -ForegroundColor DarkGray
|
||
}
|
||
}
|
||
|
||
# 删除散文件
|
||
foreach ($file in $DeletableFiles) {
|
||
$fullPath = Join-Path $RepoPath $file
|
||
if (Test-Path $fullPath) {
|
||
$size = (Get-Item $fullPath).Length
|
||
$sizeMB = [math]::Round($size / 1MB, 2)
|
||
Remove-Item $fullPath -Force
|
||
Write-Host " 已删除: $file ($sizeMB MB)" -ForegroundColor Red
|
||
$freedMB += $sizeMB
|
||
}
|
||
}
|
||
|
||
# 删除根目录下所有 .png 文件（server-exclude.txt 中 *.png 规则）
|
||
Get-ChildItem $RepoPath -Filter "*.png" -File -ErrorAction SilentlyContinue | ForEach-Object {
|
||
$sizeMB = [math]::Round($_.Length / 1MB, 2)
|
||
Remove-Item $_.FullName -Force
|
||
Write-Host " 已删除: $($_.Name) ($sizeMB MB)" -ForegroundColor Red
|
||
$freedMB += $sizeMB
|
||
}
|
||
|
||
Write-Host " 共释放: $freedMB MB" -ForegroundColor Green
|
||
}
|
||
|
||
function New-ExportTree {
|
||
<#
|
||
.SYNOPSIS
|
||
在指定 repo 下创建完整的 export 目录树。
|
||
#>
|
||
param([string]$RepoPath)
|
||
|
||
Write-Host "`n [创建] 初始化 export 目录树..." -ForegroundColor Yellow
|
||
|
||
foreach ($dir in $ExportDirs) {
|
||
$fullPath = Join-Path $RepoPath $dir
|
||
if (-not (Test-Path $fullPath)) {
|
||
New-Item -ItemType Directory -Path $fullPath -Force | Out-Null
|
||
Write-Host " 已创建: $dir/" -ForegroundColor Cyan
|
||
} else {
|
||
Write-Host " 已存在: $dir/" -ForegroundColor DarkGray
|
||
}
|
||
}
|
||
}
|
||
|
||
function Test-GitSetup {
|
||
<#
|
||
.SYNOPSIS
|
||
检查 setup-server-git.py 是否已运行（通过检查 .git/info/exclude 内容）。
|
||
#>
|
||
param([string]$RepoPath)
|
||
|
||
$excludeFile = Join-Path $RepoPath ".git\info\exclude"
|
||
if (-not (Test-Path $excludeFile)) {
|
||
return $false
|
||
}
|
||
$content = Get-Content $excludeFile -Raw -ErrorAction SilentlyContinue
|
||
return ($content -match "server-exclude")
|
||
}
|
||
|
||
# ============================================================================
|
||
# 主流程
|
||
# ============================================================================
|
||
|
||
Write-Host "============================================" -ForegroundColor White
|
||
Write-Host " NeoZQYY 服务器环境初始化" -ForegroundColor White
|
||
Write-Host " 目标环境: $($Envs -join ', ')" -ForegroundColor White
|
||
Write-Host "============================================" -ForegroundColor White
|
||
|
||
foreach ($env in $Envs) {
|
||
$repoPath = $EnvPaths[$env]
|
||
|
||
if (-not $repoPath) {
|
||
Write-Host "`n[错误] 未知环境: $env（可选: test, prod）" -ForegroundColor Red
|
||
continue
|
||
}
|
||
|
||
Write-Host "`n========== 环境: $env ==========" -ForegroundColor Magenta
|
||
Write-Host " 路径: $repoPath"
|
||
|
||
# 检查 repo 是否存在
|
||
if (-not (Test-Path $repoPath)) {
|
||
Write-Host " [警告] 目录不存在，跳过。请先 git clone。" -ForegroundColor Yellow
|
||
continue
|
||
}
|
||
|
||
# 检查 setup-server-git.py 是否已运行
|
||
if (-not (Test-GitSetup $repoPath)) {
|
||
Write-Host " [警告] 未检测到 setup-server-git.py 的配置。" -ForegroundColor Yellow
|
||
Write-Host " 建议先运行: python scripts/server/setup-server-git.py" -ForegroundColor Yellow
|
||
$answer = Read-Host " 是否继续删除操作？(y/N)"
|
||
if ($answer -ne "y" -and $answer -ne "Y") {
|
||
Write-Host " 已跳过 $env 环境的删除操作。" -ForegroundColor DarkGray
|
||
# 仍然创建 export 目录
|
||
New-ExportTree -RepoPath $repoPath
|
||
continue
|
||
}
|
||
}
|
||
|
||
# 步骤 1：删除排除的文件/目录
|
||
Remove-SkipWorktreeItems -RepoPath $repoPath
|
||
|
||
# 步骤 2：创建 export 目录树
|
||
New-ExportTree -RepoPath $repoPath
|
||
|
||
Write-Host "`n [完成] $env 环境初始化完毕。" -ForegroundColor Green
|
||
}
|
||
|
||
Write-Host "`n============================================" -ForegroundColor White
|
||
Write-Host " 全部完成。" -ForegroundColor Green
|
||
Write-Host ""
|
||
Write-Host " 后续步骤:" -ForegroundColor White
|
||
Write-Host " 1. 手动创建各环境的 .env 文件（参考 .env.template）" -ForegroundColor White
|
||
Write-Host " 2. 确认 .env 中的 export 路径指向 repo/export/ 下对应子目录" -ForegroundColor White
|
||
Write-Host " 3. 运行 uv sync --all-packages 安装依赖" -ForegroundColor White
|
||
Write-Host "============================================" -ForegroundColor White
|