@echo off
setlocal enabledelayedexpansion

title NomadArch Installer

echo.
echo NomadArch Installer (Windows)
echo Version: 0.5.0 - Binary-Free Mode
echo.

set SCRIPT_DIR=%~dp0
set SCRIPT_DIR=%SCRIPT_DIR:~0,-1%
set TARGET_DIR=%SCRIPT_DIR%
set BIN_DIR=%TARGET_DIR%\bin
set LOG_FILE=%TARGET_DIR%\install.log
set TEMP_DIR=%TARGET_DIR%\.install-temp

set ERRORS=0
set WARNINGS=0
set NEEDS_FALLBACK=0
set SKIP_OPENCODE=0

echo [%date% %time%] Installer started >> "%LOG_FILE%"

echo [STEP 1/8] OS and Architecture Detection

REM Use PowerShell for architecture detection (works on all Windows versions)
for /f "tokens=*" %%i in ('powershell -NoProfile -Command "[System.Environment]::Is64BitOperatingSystem"') do set IS64BIT=%%i
if /i "%IS64BIT%"=="True" (
    set ARCH=x64
) else (
    set ARCH=x86
)
echo [OK] Architecture: %ARCH%

echo.
echo [STEP 2/8] Checking write permissions
if not exist "%BIN_DIR%" mkdir "%BIN_DIR%" 2>nul
if not exist "%TEMP_DIR%" mkdir "%TEMP_DIR%" 2>nul

echo. > "%SCRIPT_DIR%\test-write.tmp" 2>nul
if !ERRORLEVEL! neq 0 (
    echo [WARN] Cannot write to current directory: %SCRIPT_DIR%
    set TARGET_DIR=%USERPROFILE%\NomadArch-Install
    set BIN_DIR=!TARGET_DIR!\bin
    set LOG_FILE=!TARGET_DIR!\install.log
    set TEMP_DIR=!TARGET_DIR!\.install-temp
    if not exist "!TARGET_DIR!" mkdir "!TARGET_DIR!"
    if not exist "!BIN_DIR!" mkdir "!BIN_DIR!"
    if not exist "!TEMP_DIR!" mkdir "!TEMP_DIR!"
    echo. > "!TARGET_DIR!\test-write.tmp" 2>nul
    if !ERRORLEVEL! neq 0 (
        echo [ERROR] Cannot write to fallback directory: !TARGET_DIR!
        echo [%date% %time%] ERROR: Write permission denied >> "%LOG_FILE%"
        set /a ERRORS+=1
        goto :SUMMARY
    )
    del "!TARGET_DIR!\test-write.tmp"
    set NEEDS_FALLBACK=1
    echo [OK] Using fallback: !TARGET_DIR!
) else (
    del "%SCRIPT_DIR%\test-write.tmp"
    echo [OK] Write permissions verified
)

echo.
echo [STEP 3/8] Ensuring system dependencies

set WINGET_AVAILABLE=0
where winget >nul 2>&1
if !ERRORLEVEL! equ 0 set WINGET_AVAILABLE=1

set CHOCO_AVAILABLE=0
where choco >nul 2>&1
if !ERRORLEVEL! equ 0 set CHOCO_AVAILABLE=1

set DOWNLOAD_CMD=powershell
where curl >nul 2>&1
if !ERRORLEVEL! equ 0 set DOWNLOAD_CMD=curl

where node >nul 2>&1
if !ERRORLEVEL! neq 0 (
    echo [INFO] Node.js not found. Attempting to install...
    if !WINGET_AVAILABLE! equ 1 (
        winget install -e --id OpenJS.NodeJS.LTS --accept-source-agreements --accept-package-agreements
    ) else if !CHOCO_AVAILABLE! equ 1 (
        choco install nodejs-lts -y
    ) else (
        echo [ERROR] No supported package manager found.
        echo Please install Node.js LTS from https://nodejs.org/
        set /a ERRORS+=1
        goto :SUMMARY
    )
)

where node >nul 2>&1
if !ERRORLEVEL! neq 0 (
    echo [ERROR] Node.js install failed or requires a new terminal session.
    set /a ERRORS+=1
    goto :SUMMARY
)

for /f "tokens=*" %%i in ('node --version') do set NODE_VERSION=%%i
echo [OK] Node.js: %NODE_VERSION%

where npm >nul 2>&1
if !ERRORLEVEL! neq 0 (
    echo [ERROR] npm not found after Node.js install.
    set /a ERRORS+=1
    goto :SUMMARY
)

for /f "tokens=*" %%i in ('npm --version') do set NPM_VERSION=%%i
echo [OK] npm: %NPM_VERSION%

where git >nul 2>&1
if !ERRORLEVEL! neq 0 (
    echo [INFO] Git not found. Attempting to install...
    if !WINGET_AVAILABLE! equ 1 (
        winget install -e --id Git.Git --accept-source-agreements --accept-package-agreements
    ) else if !CHOCO_AVAILABLE! equ 1 (
        choco install git -y
    ) else (
        echo [WARN] Git not installed - optional
        set /a WARNINGS+=1
    )
) else (
    for /f "tokens=*" %%i in ('git --version') do set GIT_VERSION=%%i
    echo [OK] Git: !GIT_VERSION!
)

echo.
echo [STEP 4/8] Installing npm dependencies
cd /d "%SCRIPT_DIR%"
echo [%date% %time%] Running npm install >> "%LOG_FILE%"
call npm install
if !ERRORLEVEL! neq 0 (
    echo [ERROR] npm install failed!
    echo [%date% %time%] ERROR: npm install failed >> "%LOG_FILE%"
    set /a ERRORS+=1
    goto :SUMMARY
)
echo [OK] Dependencies installed

echo.
echo [STEP 5/8] OpenCode Binary - OPTIONAL
echo.
echo [INFO] NomadArch now supports Binary-Free Mode!
echo [INFO] You can use the application without OpenCode binary.
echo [INFO] Free models from OpenCode Zen are available without the binary.
echo.
if not exist "%BIN_DIR%" mkdir "%BIN_DIR%" 2>nul

set /p SKIP_CHOICE="Skip OpenCode binary download? (Y for Binary-Free / N to download) [Y]: "
if /i "!SKIP_CHOICE!"=="" set SKIP_CHOICE=Y
if /i "!SKIP_CHOICE!"=="Y" goto :skip_opencode_download

REM Download OpenCode binary
echo [INFO] Fetching OpenCode version info...
for /f "delims=" %%v in ('powershell -NoProfile -Command "try { (Invoke-WebRequest -UseBasicParsing https://api.github.com/repos/sst/opencode/releases/latest).Content | ConvertFrom-Json | Select-Object -ExpandProperty tag_name } catch { 'v0.1.44' }"') do set OPENCODE_VERSION=%%v
set OPENCODE_VERSION=!OPENCODE_VERSION:v=!

set OPENCODE_BASE=https://github.com/sst/opencode/releases/download/v!OPENCODE_VERSION!
set OPENCODE_URL=!OPENCODE_BASE!/opencode-windows-%ARCH%.exe
set CHECKSUM_URL=!OPENCODE_BASE!/checksums.txt

if exist "%BIN_DIR%\opencode.exe" (
    echo [OK] OpenCode binary already exists
    echo [%date% %time%] OpenCode binary exists, skipping download >> "%LOG_FILE%"
    goto :opencode_done
)

echo [INFO] Downloading OpenCode v!OPENCODE_VERSION!...
if "!DOWNLOAD_CMD!"=="curl" (
    curl -L -o "%BIN_DIR%\opencode.exe.tmp" "!OPENCODE_URL!"
) else (
    powershell -NoProfile -Command "Invoke-WebRequest -Uri '!OPENCODE_URL!' -OutFile '%BIN_DIR%\opencode.exe.tmp'"
)

if exist "%BIN_DIR%\opencode.exe.tmp" (
    move /Y "%BIN_DIR%\opencode.exe.tmp" "%BIN_DIR%\opencode.exe" >nul
    echo [OK] OpenCode downloaded
) else (
    echo [WARN] OpenCode download failed - using Binary-Free Mode instead
    set SKIP_OPENCODE=1
)
goto :opencode_done

:skip_opencode_download
set SKIP_OPENCODE=1
echo [INFO] Skipping OpenCode binary - using Binary-Free Mode
echo [%date% %time%] Using Binary-Free Mode >> "%LOG_FILE%"

:opencode_done

echo.
echo [STEP 6/8] Building UI assets
if exist "%SCRIPT_DIR%\packages\ui\dist\index.html" (
    echo [OK] UI build already exists
) else (
    echo [INFO] Building UI assets...
    pushd packages\ui
    call npm run build
    if !ERRORLEVEL! neq 0 (
        echo [ERROR] UI build failed!
        popd
        set /a ERRORS+=1
        goto :SUMMARY
    )
    popd
    echo [OK] UI assets built successfully
)

echo.
echo [STEP 7/8] Post-install health check
set HEALTH_ERRORS=0

if not exist "%SCRIPT_DIR%\package.json" set /a HEALTH_ERRORS+=1
if not exist "%SCRIPT_DIR%\packages\ui" set /a HEALTH_ERRORS+=1
if not exist "%SCRIPT_DIR%\packages\server" set /a HEALTH_ERRORS+=1
if not exist "%SCRIPT_DIR%\packages\ui\dist\index.html" set /a HEALTH_ERRORS+=1

if !HEALTH_ERRORS! equ 0 (
    echo [OK] Health checks passed
) else (
    echo [ERROR] Health checks failed: !HEALTH_ERRORS! issues
    set /a ERRORS+=!HEALTH_ERRORS!
)

echo.
echo [STEP 8/8] Installation Summary
echo.
echo   Install Dir: %TARGET_DIR%
echo   Architecture: %ARCH%
echo   Node.js: %NODE_VERSION%
echo   npm: %NPM_VERSION%
if !SKIP_OPENCODE! equ 1 (
    echo   Mode: Binary-Free Mode
) else (
    echo   Mode: Full Mode with OpenCode binary
)
echo   Errors: !ERRORS!
echo   Warnings: !WARNINGS!
echo   Log File: %LOG_FILE%
echo.

:SUMMARY
if !ERRORS! gtr 0 (
    echo [RESULT] Installation completed with errors.
    echo Review the log: %LOG_FILE%
    echo.
    echo If Node.js was just installed, open a new terminal and run this installer again.
) else (
    echo [RESULT] Installation completed successfully.
    echo Run Launch-Windows.bat to start the application.
    echo.
    if !SKIP_OPENCODE! equ 1 (
        echo NOTE: Running in Binary-Free Mode.
        echo       Free models: GPT-5 Nano, Grok Code, GLM-4.7, etc.
        echo       You can also authenticate with Qwen for additional models.
    )
)

echo.
echo Press any key to exit...
pause >nul
exit /b !ERRORS!