# Setup script for invoke-ai/InvokeAI on Windows (PowerShell), from https://gitmanual.com/invoke-ai/InvokeAI
# Read it before you run it: the commands come from the repo's own README and files.
# Run: powershell -ExecutionPolicy Bypass -File .\setup.ps1
$ErrorActionPreference = "Stop"
function Write-Step($t) { Write-Host "`n==> $t" -ForegroundColor Cyan }
# Windows ships a python that only opens the Microsoft Store; it doesn't count.
function Have($c) { $cmd = Get-Command $c -ErrorAction SilentlyContinue; [bool]$cmd -and $cmd.Source -notlike '*\WindowsApps\*' }
# New installs change PATH; pick it up without opening a new window.
function Update-Path { $env:Path = [Environment]::GetEnvironmentVariable("Path", "Machine") + ";" + [Environment]::GetEnvironmentVariable("Path", "User") }
# Stop when a program fails, like bash's set -e. Dot-sourced, so cd and venv activation carry on to the next step.
function Invoke-Step([scriptblock]$Block) { $global:LASTEXITCODE = 0; . $Block; if ($LASTEXITCODE) { throw "Failed with exit code $LASTEXITCODE" } }

Write-Step '1/3 Clone the repository'
if (-not (Test-Path 'InvokeAI')) { Invoke-Step { git clone --recurse-submodules https://github.com/invoke-ai/InvokeAI.git } }
Set-Location 'InvokeAI'

Write-Step '2/3 Install the required runtimes'
# Python >=3.12, <3.13
if (-not (Have 'uv')) { Invoke-Step { winget install astral-sh.uv -e --accept-source-agreements --accept-package-agreements }; Update-Path }

Write-Step '3/3 Install dependencies'
Invoke-Step { uv sync }

Write-Host 'Set up invoke-ai/InvokeAI.'
Write-Host 'Next, from the project folder:'
Write-Host '  uv run pytest'
Write-Host 'Manual: https://gitmanual.com/invoke-ai/InvokeAI'