85 lines
4.7 KiB
PowerShell
85 lines
4.7 KiB
PowerShell
[CmdletBinding()]
|
|
param(
|
|
[string]$ProjectRoot,
|
|
[ValidateSet('Official', 'China')][string]$Mirror,
|
|
[switch]$Help,
|
|
[switch]$Staged
|
|
)
|
|
$ErrorActionPreference = 'Stop'
|
|
[Console]::OutputEncoding = New-Object Text.UTF8Encoding($false)
|
|
$env:PYTHONUTF8 = '1'
|
|
$env:PYTHONIOENCODING = 'utf-8'
|
|
if ($Help) {
|
|
Write-Host '用法：update.bat 或 update-cn.bat [-Help]'
|
|
Write-Host '升级会终止本项目的 ARTEMIS 服务、MCP 和任务进程；保留配置、密钥与历史资料。'
|
|
Write-Host '随后快进更新源码，同步依赖、网页构建和 Codex MCP；完成后请重新启动并重载 MCP。'
|
|
exit 0
|
|
}
|
|
if (-not $ProjectRoot) { $ProjectRoot = Split-Path $PSScriptRoot -Parent }
|
|
$ProjectRoot = (Resolve-Path -LiteralPath $ProjectRoot).Path
|
|
. (Join-Path $PSScriptRoot 'installation_lock.ps1')
|
|
|
|
if (-not $Staged) {
|
|
$lock = $null
|
|
$stage = $null
|
|
$stageCreated = $false
|
|
$code = 1
|
|
try {
|
|
$lock = Enter-ArtemisInstallationLock -ProjectRoot $ProjectRoot
|
|
# Every program used before the merge is frozen outside the checkout.
|
|
$stage = Join-Path ([IO.Path]::GetTempPath()) ('artemis-upgrade-' + [Guid]::NewGuid().ToString('N'))
|
|
$fullStage = [IO.Path]::GetFullPath($stage)
|
|
$rootPrefix = [IO.Path]::GetFullPath($ProjectRoot).TrimEnd('\') + '\'
|
|
if ($fullStage.StartsWith($rootPrefix, [StringComparison]::OrdinalIgnoreCase)) { throw 'TEMP must be outside the project directory.' }
|
|
New-Item -ItemType Directory -Path $stage | Out-Null
|
|
$stageCreated = $true
|
|
foreach ($name in @('upgrade.ps1', 'installation_lock.ps1', 'upgrade_processes.ps1', 'update_installation.py')) {
|
|
Copy-Item -LiteralPath (Join-Path $PSScriptRoot $name) -Destination (Join-Path $stage $name)
|
|
}
|
|
$forward = @('-NoProfile', '-ExecutionPolicy', 'Bypass', '-File', (Join-Path $stage 'upgrade.ps1'), '-Staged', '-ProjectRoot', $ProjectRoot)
|
|
if ($Mirror) { $forward += @('-Mirror', $Mirror) }
|
|
& powershell.exe @forward
|
|
$code = $LASTEXITCODE
|
|
} catch {
|
|
Write-Host "[升级失败] $_" -ForegroundColor Red
|
|
} finally {
|
|
# Only remove the unique, validated staging directory created above.
|
|
if ($stageCreated -and (Test-Path -LiteralPath $stage)) { Remove-Item -LiteralPath $stage -Recurse -Force }
|
|
Exit-ArtemisInstallationLock -Mutex $lock
|
|
}
|
|
exit $code
|
|
}
|
|
|
|
try {
|
|
Set-Location -LiteralPath $PSScriptRoot
|
|
$python = Join-Path $ProjectRoot '.venv\Scripts\python.exe'
|
|
if (-not (Test-Path -LiteralPath $python)) { throw '尚未安装项目 Python 环境；请先运行 start.bat 或 start-cn.bat。' }
|
|
if (-not (Get-Command git -ErrorAction SilentlyContinue)) { throw '未安装 Git，无法升级。' }
|
|
if (-not (Test-Path -LiteralPath (Join-Path $ProjectRoot '.git'))) { throw '此目录不含 Git 信息；请改用 Git 克隆或新版发布源码包。' }
|
|
# Refuse dirty checkouts before terminating the user's running processes.
|
|
$localChanges = & git -C $ProjectRoot status --porcelain 2>$null
|
|
if ($LASTEXITCODE -ne 0) { throw '无法检查 Git 工作区。' }
|
|
if ($localChanges) { throw '发现本地修改；请先提交或备份，未终止任何任务。' }
|
|
Write-Host '======================================================'
|
|
Write-Host 'ARTEMIS 独立升级：将终止当前项目的服务、MCP 及任务。' -ForegroundColor Yellow
|
|
. (Join-Path $PSScriptRoot 'upgrade_processes.ps1')
|
|
Stop-ArtemisUpgradeProcesses -ProjectRoot $ProjectRoot
|
|
& $python (Join-Path $PSScriptRoot 'update_installation.py') source --root $ProjectRoot
|
|
if ($LASTEXITCODE -notin @(0, 10)) { throw '源码升级未完成；配置和历史资料已保留。' }
|
|
Stop-ArtemisUpgradeProcesses -ProjectRoot $ProjectRoot
|
|
# Load the new launcher only AFTER the frozen Git updater has exited.
|
|
$prepare = @('-NoProfile', '-ExecutionPolicy', 'Bypass', '-File', (Join-Path $ProjectRoot 'scripts\start.ps1'), '-PrepareOnly', '-SkipUpdate', '-NoOpen')
|
|
if ($Mirror) { $prepare += @('-Mirror', $Mirror) }
|
|
$previousChild = $env:ARTEMIS_UPGRADE_CHILD
|
|
$env:ARTEMIS_UPGRADE_CHILD = '1'
|
|
try { & powershell.exe @prepare; $prepareCode = $LASTEXITCODE }
|
|
finally { $env:ARTEMIS_UPGRADE_CHILD = $previousChild }
|
|
if ($prepareCode -ne 0) { throw '新版源码已保留，但环境同步失败；请处理上方错误后重新运行升级脚本。' }
|
|
Write-Host '[升级完成] 源码、依赖、网页及 Codex MCP 已检查并同步。' -ForegroundColor Green
|
|
Write-Host '[下一步] 运行启动脚本，并在 Codex 中重载 MCP 或重启 Codex。'
|
|
exit 0
|
|
} catch {
|
|
Write-Host "[升级失败] $_" -ForegroundColor Red
|
|
exit 1
|
|
}
|