install.ps1

#Requires -Version 5.1
<#
.SYNOPSIS
    Bootstrap for managed-python on Windows.

.DESCRIPTION
    Downloads uv, creates a Python venv, then hands off to setup.py.
    All configuration (env.ps1, env.sh, bin\ wrappers, shell profile) is
    handled by setup.py.

.EXAMPLE
    .\install.ps1 -Prefix "$env:USERPROFILE\.local\redmatter\python" -Python "3.10" -EnvPrefix "REDMATTER"

.EXAMPLE
    .\install.ps1 -Prefix "$env:USERPROFILE\.local\redmatter\python" `
                  -Python "3.10" -UvEnv "REDMATTER_UV" -UvxEnv "REDMATTER_UVX" -PythonEnv "REDMATTER_PYTHON"
#>
param(
    [Parameter(Mandatory=$true)]  [string]$Prefix,
    [Parameter(Mandatory=$true)]  [string]$Python,
    [Parameter(Mandatory=$false)] [string]$EnvPrefix,
    [Parameter(Mandatory=$false)] [string]$UvEnv,
    [Parameter(Mandatory=$false)] [string]$UvxEnv,
    [Parameter(Mandatory=$false)] [string]$PythonEnv,
    [switch]$ShellProfile,
    [switch]$Quiet,
    [switch]$Isolated
)

if ($EnvPrefix) {
    if ($UvEnv -or $UvxEnv -or $PythonEnv) {
        Write-Error "-EnvPrefix cannot be combined with -UvEnv, -UvxEnv, or -PythonEnv"
        exit 1
    }
} else {
    $missing = @()
    if (-not $UvEnv)     { $missing += "-UvEnv" }
    if (-not $UvxEnv)    { $missing += "-UvxEnv" }
    if (-not $PythonEnv) { $missing += "-PythonEnv" }
    if ($missing.Count -gt 0) {
        Write-Error "The following parameters are required: $($missing -join ', ') (or use -EnvPrefix)"
        exit 1
    }
}

function Write-Msg($msg) { if (-not $Quiet) { Write-Host $msg } }

Set-StrictMode -Version Latest
$ErrorActionPreference = "Stop"

$Prefix    = [Environment]::ExpandEnvironmentVariables($Prefix)
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
$UvExe     = Join-Path $Prefix "uv.exe"
$VenvPy    = Join-Path $Prefix "venv\Scripts\python.exe"

# Read pinned uv version from distro.toml
$UvVersion = (Get-Content (Join-Path $ScriptDir "distro.toml") |
    Select-String '^uv_version').Line `
    -replace '^[^=]+=\s*"?([^"#]+)"?.*', '$1' |
    ForEach-Object { $_.Trim() }

Write-Msg ""
Write-Msg "managed-python bootstrap"
Write-Msg "  prefix  $Prefix"
Write-Msg ""

# Bootstrap uv
$currentVer = if (Test-Path $UvExe) { try { (& $UvExe --version 2>$null) -split " " | Select-Object -Last 1 } catch { "" } } else { "" }
if ($currentVer -eq $UvVersion) {
    Write-Msg "  ✓ uv $UvVersion"
} else {
    Write-Msg "  → Downloading uv $UvVersion"
    $arch = if ($env:PROCESSOR_ARCHITECTURE -eq "ARM64") { "aarch64" } else { "x86_64" }
    $url  = "https://github.com/astral-sh/uv/releases/download/$UvVersion/uv-${arch}-pc-windows-msvc.zip"
    New-Item -ItemType Directory -Force -Path $Prefix | Out-Null
    $tmp    = [IO.Path]::Combine([IO.Path]::GetTempPath(), [IO.Path]::ChangeExtension([IO.Path]::GetRandomFileName(), ".zip"))
    $tmpDir = "$tmp.dir"
    $tomlMatch    = Get-Content (Join-Path $ScriptDir "distro.toml") |
                        Select-String "^${arch}-pc-windows-msvc\s*=\s*`"([^`"]+)`""
    $expectedHash = $tomlMatch.Matches[0].Groups[1].Value.ToUpper()
    if (-not $expectedHash) {
        Write-Error "No pinned checksum for ${arch}-pc-windows-msvc in distro.toml"
        exit 1
    }
    try {
        $ProgressPreference = "SilentlyContinue"
        Invoke-WebRequest $url -OutFile $tmp -UseBasicParsing
        $actualHash = (Get-FileHash $tmp -Algorithm SHA256).Hash.ToUpper()
        if ($actualHash -ne $expectedHash) {
            Write-Error "uv $UvVersion checksum verification failed — download may be corrupt or tampered`n  expected: $expectedHash`n  actual:   $actualHash"
            exit 1
        }
        Expand-Archive $tmp $tmpDir -Force
        $uvSrc = Get-ChildItem $tmpDir -Filter "uv.exe" -Recurse | Select-Object -First 1
        Copy-Item $uvSrc.FullName $UvExe -Force
        $uvxSrc = Get-ChildItem $tmpDir -Filter "uvx.exe" -Recurse | Select-Object -First 1
        if (-not $uvxSrc) { Write-Error "Failed to locate uvx.exe in archive"; exit 1 }
        Copy-Item $uvxSrc.FullName (Join-Path $Prefix "uvx.exe") -Force
    } catch {
        Write-Error "Failed to download uv $UvVersion from $url`: $_"
        exit 1
    } finally {
        Remove-Item $tmp, $tmpDir -Recurse -Force -ErrorAction SilentlyContinue
    }
    if (-not (Test-Path $UvExe)) {
        Write-Error "Failed to download uv $UvVersion — binary not found after extraction"
        exit 1
    }
    Write-Msg "  ✓ uv $UvVersion installed"
}

# Bootstrap venv
if (Test-Path $VenvPy) {
    Write-Msg "  ✓ venv already exists"
} else {
    Write-Msg "  → Creating Python $Python venv"
    $pythonPref = if ($Isolated) { "only-managed" } else { "system" }
    $venvArgs = @("venv", "--python", $Python, "--python-preference", $pythonPref)
    if ($Quiet) { $venvArgs += "--quiet" }
    & $UvExe @venvArgs (Join-Path $Prefix "venv")
    if ($LASTEXITCODE -ne 0) {
        Write-Error "Failed to create Python $Python venv — see uv error above"
        exit $LASTEXITCODE
    }
    if (-not (Test-Path $VenvPy)) {
        Write-Error "Failed to create Python $Python venv — python.exe not found at $VenvPy"
        exit 1
    }
    Write-Msg "  ✓ venv created"
}

Write-Msg ""

# Hand off to setup.py
if ($EnvPrefix) {
    $setupArgs = @("--prefix", $Prefix, "--python", $Python, "--env-prefix", $EnvPrefix)
} else {
    $setupArgs = @("--prefix", $Prefix, "--python", $Python, "--uv-env", $UvEnv, "--uvx-env", $UvxEnv, "--python-env", $PythonEnv)
}
if ($ShellProfile) { $setupArgs += "--shell-profile" }
if ($Isolated) { $setupArgs += "--isolated" }
if ($Quiet) { $setupArgs += "--quiet" }
& $VenvPy (Join-Path $ScriptDir "setup.py") @setupArgs
exit $LASTEXITCODE