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

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

Write-Step '3/3 Install dependencies'
Invoke-Step { python -m venv .venv }
Invoke-Step { .venv\Scripts\Activate.ps1 }
Invoke-Step { pip install -r requirements.txt }
Invoke-Step { pip install -r requirements-dev.txt }

Write-Host 'Set up deepseek-ai/DeepInfer.'
Write-Host 'Next, from the project folder:'
Write-Host '  pytest'
Write-Host 'Manual: https://gitmanual.com/deepseek-ai/DeepInfer'