Param(
    [string] $Version = '3.10.2',
    [string] $Arch = 'x64'
)

$ErrorActionPreference = 'Stop'

Add-Type -AssemblyName System.IO.Compression.FileSystem

$HOME_DIR = if ($env:HOME) { $env:HOME } else { $env:USERPROFILE }

$destDir = Join-Path "$env:AGENT_TOOLSDIRECTORY" "Python" "$Version" "$Arch"
$completeFile = "$destDir.complete"

if (Test-Path $completeFile) {
    Write-Host "Python $Version ($Arch) already installed."
    exit 0;
} else {
    Write-Host "No matching Python found."
}

Write-Host "Downloading manifest..."

$pythonManifestUri = "https://raw.githubusercontent.com/actions/python-versions/main/versions-manifest.json"
$pythonManifest = Invoke-WebRequest -Uri $pythonManifestUri | ConvertFrom-Json

if ($IsMacOS) {
    $platform = "darwin"
} elseif ($IsLinux) {
    $platform = "linux"
} else {
    $platform = "win32"
}
$manifestFileVersion = $pythonManifest | Where-Object { $_.version -eq $Version } | Select-Object -First 1
$manifestFileItem = $manifestFileVersion.files | Where-Object { $_.platform -eq $platform -and $_.arch -eq $Arch } | Select-Object -First 1
$downloadUrl = $manifestFileItem.download_url

# download
$tempDir = Join-Path "$HOME_DIR" "python-temp"
$archive = Join-Path "$tempDir" (Split-Path $downloadUrl -Leaf)
New-Item -ItemType Directory -Force -Path "$tempDir" | Out-Null

Write-Host "Downloading Python package '$downloadUrl' to '$archive'..."
(New-Object System.Net.WebClient).DownloadFile("$downloadUrl", "$archive")

# extract
Write-Host "Extracting Python to '$tempDir'..."
if ($IsMacOS -or $IsLinux) {
    tar -vxzf "$archive" -C "$tempDir"
} else {
    [System.IO.Compression.ZipFile]::ExtractToDirectory("$archive", "$tempDir")
}
Write-Host "Extraction complete."

# install
Write-Host "Installing Python..."
try {
    Push-Location "$tempDir"
    if ($IsMacOS -or $IsLinux) {
        ./setup.sh
    } else {
        .\setup.ps1
    }
} finally {
    Pop-Location
}

exit $LASTEXITCODE