@echo off
chcp 65001 >nul
title AiRust Cyber Operator — Installer
color 0A

echo.
echo ╔═══════════════════════════════════════════╗
echo ║        AiRust — Cyber Operator            ║
echo ║         *burp* Rick's finest work         ║
echo ╚═══════════════════════════════════════════╝
echo.

:: Check for Rust/Cargo
where cargo >nul 2>&1
if %errorlevel% neq 0 (
    echo [!!] Rust not found! Install it from: https://rustup.rs
    echo     Or grab a pre-built binary from the releases page.
    pause
    exit /b 1
)

:: Check for built binary
set "BIN_DIR=target\release"
set "BIN_NAME=AiRust-cli"

if not exist "%BIN_DIR%\%BIN_NAME%.exe" (
    echo [*] Binary not found — building in release mode...
    echo     This may take a few minutes.
    echo.
    call cargo build --release
    if %errorlevel% neq 0 (
        echo [!!] Build failed. Check the error messages above.
        pause
        exit /b 1
    )
    echo.
    echo [✓] Build complete.
) else (
    echo [✓] Release binary found.
)

echo.
echo ╔═══════════════════════════════════════════╗
echo ║          Select Launch Mode               ║
echo ╠═══════════════════════════════════════════╣
echo ║  1 — Web Interface (opens browser)        ║
echo ║  2 — Command-Line Application (CLI)       ║
echo ╚═══════════════════════════════════════════╝
echo.

choice /c 12 /n /m "  Enter your choice (1 or 2): "
if %errorlevel% equ 1 (
    echo.
    echo [*] Starting web interface at http://127.0.0.1:3000
    echo [*] The browser will open automatically.
    echo [*] Press Ctrl+C to stop the server.
    echo.
    "%BIN_DIR%\%BIN_NAME%.exe"
) else (
    echo.
    echo [*] Starting CLI mode — interactive chat.
    echo [*] Type 'exit' or 'quit' to close.
    echo.
    "%BIN_DIR%\%BIN_NAME%.exe" cli
)

echo.
echo [*] AiRust session ended.
timeout /t 3 >nul