install.ps1

# Jarvis Orb — Windows Installer (PowerShell)
# Usage: irm https://raw.githubusercontent.com/thestack-ai/jarvis-orb/main/install.ps1 | iex

$ErrorActionPreference = "Stop"
$Version = "0.3.3"
$Repo = "thestack-ai/jarvis-orb"
$BrainDir = "$env:USERPROFILE\.jarvis-orb"
$BrainBin = "$BrainDir\bin"

# ── Colors ──
$Script:StepNum = 0
$Script:TotalSteps = 4

function Write-Cyan($msg) { Write-Host "  $msg" -ForegroundColor Cyan }
function Write-Ok($msg) { Write-Host "    ✓ $msg" -ForegroundColor Green }
function Write-Warn($msg) { Write-Host "    ! $msg" -ForegroundColor Yellow }
function Write-Info($msg) { Write-Host "    $msg" -ForegroundColor DarkGray }
function Write-Step($msg) {
    $Script:StepNum++
    $filled = [int]($Script:StepNum * 20 / $Script:TotalSteps)
    $empty = 20 - $filled
    $bar = ("█" * $filled) + ("░" * $empty)
    Write-Host ""
    Write-Host "  [$bar] $($Script:StepNum)/$($Script:TotalSteps)" -ForegroundColor DarkGray
    Write-Host "  → $msg" -ForegroundColor White
}

# ── Header ──
Write-Host ""
Write-Host ""
Write-Host "  J A R V I S" -NoNewline -ForegroundColor Cyan
Write-Host "  ◉ ORB" -ForegroundColor White
Write-Host "  AI Brain + Visualizer · v$Version" -ForegroundColor DarkGray
Write-Host ""

# ── Step 1: Environment ──
Write-Step "Checking environment"
Write-Info "Windows $([System.Environment]::OSVersion.Version)"

$hasPython = Get-Command python -ErrorAction SilentlyContinue
if (-not $hasPython) {
    Write-Host "  Python not found. Install Python 3.11+ first." -ForegroundColor Red
    exit 1
}

$pyVersion = python -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" 2>&1
$pyParts = $pyVersion -split '\.'
if ([int]$pyParts[0] -lt 3 -or ([int]$pyParts[0] -eq 3 -and [int]$pyParts[1] -lt 11)) {
    Write-Host "  Python 3.11+ required (found $pyVersion)." -ForegroundColor Red
    exit 1
}
Write-Ok "Python $pyVersion"

# ── Step 2: Brain Lite ──
Write-Step "Installing Brain Lite"

New-Item -ItemType Directory -Force -Path $BrainDir | Out-Null
New-Item -ItemType Directory -Force -Path $BrainBin | Out-Null

Write-Info "Installing dependencies..."
$pipOutput = python -m pip install --target "$BrainDir\lib" aiosqlite websockets mcp 2>&1
if ($LASTEXITCODE -ne 0) {
    Write-Host "  Dependency install failed. Check your Python/pip setup." -ForegroundColor Red
    Write-Info ($pipOutput | Out-String)
    exit 1
}
Write-Ok "Dependencies installed"

# Download Brain source
Write-Info "Downloading Brain Lite..."
$zipUrl = "https://github.com/$Repo/archive/refs/heads/main.zip"
$zipPath = "$env:TEMP\jarvis-orb-main.zip"
$extractPath = "$env:TEMP\jarvis-orb-extract"

try {
    Invoke-WebRequest -Uri $zipUrl -OutFile $zipPath -UseBasicParsing
    if (Test-Path $extractPath) { Remove-Item -Recurse -Force $extractPath }
    Expand-Archive -Path $zipPath -DestinationPath $extractPath -Force

    $brainSrc = "$extractPath\jarvis-orb-main\brain\jarvis_brain"
    if (Test-Path $brainSrc) {
        Copy-Item -Recurse -Force $brainSrc "$BrainDir\jarvis_brain"
        Write-Ok "Brain Lite → $BrainDir"
    }

    Remove-Item -Force $zipPath -ErrorAction SilentlyContinue
    Remove-Item -Recurse -Force $extractPath -ErrorAction SilentlyContinue
} catch {
    Write-Warn "Download failed — install from source"
}

# Create launcher
$brainLauncher = @"
@echo off
set PYTHONPATH=$BrainDir\lib;$BrainDir
python -m jarvis_brain.mcp_server %*
"@
Set-Content -Path "$BrainBin\jarvis-brain.bat" -Value $brainLauncher

$orbLauncher = @"
@echo off
set PYTHONPATH=$BrainDir\lib;$BrainDir
python -m jarvis_brain.cli %*
"@
Set-Content -Path "$BrainBin\jarvis-orb.bat" -Value $orbLauncher
Write-Ok "Commands → jarvis-orb, jarvis-brain"

# Add to PATH
$currentPath = [Environment]::GetEnvironmentVariable("Path", "User")
if ($currentPath -notlike "*$BrainBin*") {
    [Environment]::SetEnvironmentVariable("Path", "$BrainBin;$currentPath", "User")
    Write-Ok "Added to PATH (restart terminal to use)"
}

# ── Step 3: Claude Code MCP ──
Write-Step "Configuring Claude Code"

$hasClaude = Get-Command claude -ErrorAction SilentlyContinue
if ($hasClaude) {
    $mcpCheck = claude mcp list 2>&1 | Out-String
    if ($mcpCheck -match "jarvis-brain") {
        Write-Ok "Already configured"
    } else {
        $regResult = claude mcp add jarvis-brain -e PYTHONPATH="$BrainDir\lib;$BrainDir" -- python -m jarvis_brain.mcp_server 2>&1
        if ($LASTEXITCODE -eq 0) {
            Write-Ok "MCP server registered in Claude Code"
        } else {
            Write-Info "Run manually: claude mcp add jarvis-brain -e PYTHONPATH=`"$BrainDir\lib;$BrainDir`" -- python -m jarvis_brain.mcp_server"
        }
    }
} else {
    Write-Info "Claude Code not found — install it first, then run installer again"
}

# ── Step 4: Orb App ──
Write-Step "Installing Orb"

try {
    $release = Invoke-RestMethod -Uri "https://api.github.com/repos/$Repo/releases/latest" -UseBasicParsing 2>$null
    $exeAsset = $release.assets | Where-Object { $_.name -match "setup\.exe$" } | Select-Object -First 1

    if ($exeAsset) {
        $exeUrl = $exeAsset.browser_download_url
        $exePath = "$env:TEMP\JarvisOrb-setup.exe"
        Write-Info "Downloading Orb..."
        Invoke-WebRequest -Uri $exeUrl -OutFile $exePath -UseBasicParsing
        Write-Ok "Downloaded. Running installer..."
        Start-Process -FilePath $exePath -Wait
        Remove-Item -Force $exePath -ErrorAction SilentlyContinue
        Write-Ok "Jarvis Orb installed"
    } else {
        Write-Warn "No release found. Download manually:"
        Write-Info "https://github.com/$Repo/releases"
    }
} catch {
    Write-Warn "Could not fetch release. Download manually:"
    Write-Info "https://github.com/$Repo/releases"
}

# ── Done ──
Write-Host ""
Write-Host ""
Write-Host "  J A R V I S" -NoNewline -ForegroundColor Green
Write-Host "  ◉ ORB" -ForegroundColor Green
Write-Host ""
Write-Host "  I N S T A L L E D" -ForegroundColor Green
Write-Host ""
Write-Info "Brain   $BrainDir\brain.db"
Write-Info "MCP     jarvis-brain"
Write-Info "Orb     Jarvis Orb"
Write-Info "CLI     jarvis-orb"
Write-Host ""
Write-Cyan "Your AI will start remembering."
Write-Host ""