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

Write-Step '2/3 Install the required tools'
# Python 3
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 Follow the README: Quick Start'
# In the maintainers' order. Do the README instructions above first — they cover steps outside the terminal.
Invoke-Step { python -m venv venv }
# (needs Git Bash or WSL on Windows) source venv/bin/activate        # Windows: venv\Scripts\activate
Invoke-Step { pip install -r requirements.txt }
Invoke-Step { cp .env.example .env }
Invoke-Step { curl -L https://cdnjs.cloudflare.com/ajax/libs/vis-network/10.0.2/standalone/umd/vis-network.min.js \ }
Invoke-Step { -o static/vis-network.min.js }
Invoke-Step { python main.py }

Write-Host 'Set up 1024Cyber/ATLAS-Framework.'
Write-Host 'Manual: https://gitmanual.com/1024Cyber/ATLAS-Framework'