# Restore deleted test files from commit parent
# Maps old locations to new locations

$ErrorActionPreference = "Stop"
$parentCommit = "74c7aa250c401ee9ac332686832b256159efa604^"

# Mapping: old path -> new path
$mappings = @{
    "src/__Tests/AirGap/StellaOps.AirGap.Importer.Tests" = "src/AirGap/__Tests/StellaOps.AirGap.Importer.Tests"
    "src/__Tests/AirGap/StellaOps.AirGap.Controller.Tests" = "src/AirGap/__Tests/StellaOps.AirGap.Controller.Tests"
    "src/__Tests/AirGap/StellaOps.AirGap.Time.Tests" = "src/AirGap/__Tests/StellaOps.AirGap.Time.Tests"
    "src/__Tests/StellaOps.Gateway.WebService.Tests" = "src/Gateway/__Tests/StellaOps.Gateway.WebService.Tests"
    "src/__Tests/Replay/StellaOps.Replay.Core.Tests" = "src/Replay/__Tests/StellaOps.Replay.Core.Tests"
    "src/__Tests/Provenance/StellaOps.Provenance.Attestation.Tests" = "src/Provenance/__Tests/StellaOps.Provenance.Attestation.Tests"
    "src/__Tests/Policy/StellaOps.Policy.Scoring.Tests" = "src/Policy/__Tests/StellaOps.Policy.Scoring.Tests"
}

Set-Location "E:\dev\git.stella-ops.org"

foreach ($mapping in $mappings.GetEnumerator()) {
    $oldPath = $mapping.Key
    $newPath = $mapping.Value

    Write-Host "`nProcessing: $oldPath -> $newPath" -ForegroundColor Cyan

    # Get list of files from old location in git
    $files = git ls-tree -r --name-only "$parentCommit" -- $oldPath 2>$null

    if (-not $files) {
        Write-Host "  No files found at old path" -ForegroundColor Yellow
        continue
    }

    foreach ($file in $files) {
        # Calculate relative path and new file path
        $relativePath = $file.Substring($oldPath.Length + 1)
        $newFilePath = Join-Path $newPath $relativePath

        # Create directory if needed
        $newDir = Split-Path $newFilePath -Parent
        if (-not (Test-Path $newDir)) {
            New-Item -ItemType Directory -Path $newDir -Force | Out-Null
        }

        # Check if file exists
        if (Test-Path $newFilePath) {
            Write-Host "  Exists: $relativePath" -ForegroundColor DarkGray
            continue
        }

        # Restore file
        git show "${parentCommit}:${file}" > $newFilePath 2>$null
        if ($LASTEXITCODE -eq 0) {
            Write-Host "  Restored: $relativePath" -ForegroundColor Green
        } else {
            Write-Host "  Failed: $relativePath" -ForegroundColor Red
        }
    }
}

Write-Host "`nDone!" -ForegroundColor Cyan