# ============================================================
# 用途：修复卸载 OneDrive 后 Claude Desktop 无法启动的问题
# 原理：清理 User Shell Folders 中残留的 OneDrive 路径 + 清理 Claude 残留
# 运行：以管理员身份运行 PowerShell，然后执行本脚本
# ============================================================

# 0. 检查管理员权限
if (-NOT ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
    Write-Host "请以管理员身份运行 PowerShell 后再执行本脚本。" -ForegroundColor Red
    exit 1
}

$ErrorActionPreference = "Stop"
$timestamp = Get-Date -Format "yyyyMMdd-HHmmss"
$backupDir = "$env:USERPROFILE\Desktop\ClaudeFix-Backup-$timestamp"
New-Item -ItemType Directory -Path $backupDir | Out-Null
Write-Host "备份目录：$backupDir" -ForegroundColor Cyan

# 1. 备份注册表 User Shell Folders / Shell Folders
Write-Host "`n[1/5] 备份注册表 ..." -ForegroundColor Yellow
reg export "HKCU\Software\Microsoft\Windows\CurrentVersion\Explorer\User Shell Folders" "$backupDir\UserShellFolders.reg" /y | Out-Null
reg export "HKCU\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders"      "$backupDir\ShellFolders.reg"     /y | Out-Null
Write-Host "  注册表已备份。"

# 2. 修正 User Shell Folders 中含 OneDrive 的路径
Write-Host "`n[2/5] 扫描并修正 User Shell Folders 中的 OneDrive 路径 ..." -ForegroundColor Yellow
$keys = @(
    "HKCU:\Software\Microsoft\Windows\CurrentVersion\Explorer\User Shell Folders",
    "HKCU:\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders"
)
foreach ($key in $keys) {
    $props = Get-ItemProperty -Path $key
    foreach ($name in $props.PSObject.Properties.Name) {
        if ($name -like "PS*") { continue }
        $value = $props.$name
        if ($value -is [string] -and $value -match "OneDrive") {
            $newValue = $value -replace "([A-Za-z]:\\Users\$\^$\$+|%USERPROFILE%)\\OneDrive(?:\s-\s[^\\]+)?\\", "%USERPROFILE%\"
            if ($newValue -ne $value) {
                Write-Host "  [$key] $name" -ForegroundColor Green
                Write-Host "    旧: $value"
                Write-Host "    新: $newValue"
                Set-ItemProperty -Path $key -Name $name -Value $newValue
            }
        }
    }
}

# 3. 确保本地默认文件夹存在
Write-Host "`n[3/5] 确保本地默认文件夹存在 ..." -ForegroundColor Yellow
$folders = @("Desktop","Documents","Downloads","Pictures","Music","Videos","Favorites")
foreach ($f in $folders) {
    $p = Join-Path $env:USERPROFILE $f
    if (-not (Test-Path $p)) {
        New-Item -ItemType Directory -Path $p | Out-Null
        Write-Host "  已创建: $p"
    }
}

# 4. 卸载 Claude Desktop(若存在)并清理残留目录
Write-Host "`n[4/5] 卸载 Claude Desktop 并清理残留 ..." -ForegroundColor Yellow
try {
    $pkg = Get-AppxPackage -Name "*Claude*" -ErrorAction SilentlyContinue
    if ($pkg) {
        $pkg | Remove-AppxPackage
        Write-Host "  已卸载 MSIX 包: $($pkg.Name)"
    } else {
        Write-Host "  未检测到 MSIX 形式的 Claude(可能是 exe 安装,请手动到'已安装的应用'中卸载)。"
    }
} catch {
    Write-Host "  卸载 MSIX 包失败: $_" -ForegroundColor Red
}

$residue = @(
    "$env:LOCALAPPDATA\AnthropicClaude",
    "$env:LOCALAPPDATA\Claude",
    "$env:APPDATA\Claude",
    "$env:APPDATA\AnthropicClaude"
)
foreach ($d in $residue) {
    if (Test-Path $d) {
        Copy-Item $d "$backupDir\" -Recurse -Force -ErrorAction SilentlyContinue
        Remove-Item $d -Recurse -Force -ErrorAction SilentlyContinue
        Write-Host "  已清理: $d (已备份到 $backupDir)"
    }
}

# 5. 完成提示
Write-Host "`n[5/5] 完成!" -ForegroundColor Green
Write-Host "请执行以下两步:" -ForegroundColor Cyan
Write-Host "  1) 重启电脑(必须,让 Explorer 重新加载 Shell Folders)"
Write-Host "  2) 从 https://claude.ai/download 重新下载并以管理员身份安装 Claude Desktop"
Write-Host "`n如出现异常,可双击 $backupDir 下的 .reg 文件恢复注册表。" -ForegroundColor Cyan