# ==============================================================================
#  DJINN LLM — 1-Click Windows Standalone Installer (PowerShell)
#  Usage: irm https://djinnllm.digitalnation.info/install.ps1 | iex
# ==============================================================================

$ErrorActionPreference = "Stop"

Write-Host "==========================================================" -ForegroundColor Cyan
Write-Host "   DJINN LLM — Sovereign Cognitive Workstation Installer   " -ForegroundColor Yellow
Write-Host "==========================================================" -ForegroundColor Cyan

$InstallDir = Join-Path $HOME ".djinn"
$RemoteUrl = "https://djinnllm.digitalnation.info"
$PackageUrl = "$RemoteUrl/api/download/windows"
$ZipFile = Join-Path $env:TEMP "djinn-windows.zip"

# Check Python
$PythonExe = Get-Command python.exe -ErrorAction SilentlyContinue
if (-not $PythonExe) {
    $PythonExe = Get-Command py.exe -ErrorAction SilentlyContinue
}

if (-not $PythonExe) {
    Write-Host "[ERROR] Python 3 was not found on your system." -ForegroundColor Red
    Write-Host "Please install Python from https://www.python.org/downloads/ (make sure to check 'Add Python to PATH')" -ForegroundColor Yellow
    exit 1
}

Write-Host "==> Target Directory: $InstallDir" -ForegroundColor Green
if (-not (Test-Path $InstallDir)) {
    New-Item -ItemType Directory -Path $InstallDir -Force | Out-Null
}

Write-Host "==> Downloading DJINN standalone bundle from $PackageUrl ..." -ForegroundColor Cyan
Invoke-WebRequest -Uri $PackageUrl -OutFile $ZipFile -UseBasicParsing

Write-Host "==> Extracting files to $InstallDir ..." -ForegroundColor Green
Expand-Archive -Path $ZipFile -DestinationPath $InstallDir -Force
Remove-Item -Path $ZipFile -Force

Set-Location $InstallDir

# Virtual environment setup
$VenvPython = Join-Path $InstallDir ".venv\Scripts\python.exe"
if (-not (Test-Path $VenvPython)) {
    Write-Host "==> Creating Python virtual environment in .venv ..." -ForegroundColor Cyan
    & $PythonExe.Source -m venv .venv
}

Write-Host "==> Installing dependencies..." -ForegroundColor Cyan
& (Join-Path $InstallDir ".venv\Scripts\pip.exe") install --upgrade pip -q
& (Join-Path $InstallDir ".venv\Scripts\pip.exe") install -r (Join-Path $InstallDir "requirements.txt") -q

# Create Desktop Shortcut
$WshShell = New-Object -ComObject WScript.Shell
$DesktopPath = [System.Environment]::GetFolderPath('Desktop')
$ShortcutPath = Join-Path $DesktopPath "DJINN LLM.lnk"
$Shortcut = $WshShell.CreateShortcut($ShortcutPath)
$Shortcut.TargetPath = Join-Path $InstallDir "run_djinn.bat"
$Shortcut.WorkingDirectory = $InstallDir
$Shortcut.Description = "Launch DJINN LLM Sovereign Cognitive Workstation"
$Shortcut.Save()

Write-Host "==========================================================" -ForegroundColor Green
Write-Host "✔ DJINN LLM successfully installed!" -ForegroundColor Green
Write-Host "• Desktop shortcut created: $ShortcutPath" -ForegroundColor Cyan
Write-Host "• Location: $InstallDir" -ForegroundColor Cyan
Write-Host "==========================================================" -ForegroundColor Green
Write-Host ""

$startNow = Read-Host "Would you like to launch DJINN LLM now? (Y/n)"
if ($startNow -ne 'n') {
    Start-Process (Join-Path $InstallDir "run_djinn.bat")
}