# Setup script for vllm-project/vllm on Windows (PowerShell), from https://gitmanual.com/vllm-project/vllm
# 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 'vllm')) { Invoke-Step { git clone https://github.com/vllm-project/vllm.git } }
Set-Location 'vllm'

Write-Step '2/3 Install the required runtimes'
# Python >=3.11,<3.15, C/C++ compiler, CMake 3.26
if (-not (Have 'uv')) { Invoke-Step { winget install astral-sh.uv -e --accept-source-agreements --accept-package-agreements }; Update-Path }
# then add the "Desktop development with C++" workload
Invoke-Step { winget install Microsoft.VisualStudio.2022.BuildTools -e --accept-source-agreements --accept-package-agreements }
if (-not (Have 'cmake')) { Invoke-Step { winget install Kitware.CMake -e --accept-source-agreements --accept-package-agreements }; Update-Path }

Write-Step '3/3 Install dependencies'
Invoke-Step { uv sync }
Invoke-Step { cmake -S . -B build -DCMAKE_BUILD_TYPE=Release }
Invoke-Step { cmake --build build }

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