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

Write-Step '2/3 Install the required runtimes'
# Python >=3.12, C/C++ compiler, CMake 3.16, make
if (-not (Have 'python')) { Invoke-Step { winget install Python.Python.3.12 -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 }
# or use WSL
if (-not (Have 'make')) { Invoke-Step { winget install GnuWin32.Make -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 -e . }
Invoke-Step { cmake -S . -B build -DCMAKE_BUILD_TYPE=Release }
Invoke-Step { cmake --build build }

Write-Host 'Set up FareedKhan-dev/kimi-k3-in-c.'
Write-Host 'Manual: https://gitmanual.com/FareedKhan-dev/kimi-k3-in-c'