# Qwen Code — Windows Installer
# Usage: powershell -ExecutionPolicy Bypass -File qwen\uqwen_install.ps1
#
# Installs Node.js (if needed), Qwen Code via npm registry, configures env vars and settings.

$ErrorActionPreference = "Continue"

Write-Host ""
Write-Host "  +--------------------------------------+" -ForegroundColor Cyan
Write-Host "  |    Qwen Code -- Windows Installer    |" -ForegroundColor Cyan
Write-Host "  +--------------------------------------+" -ForegroundColor Cyan
Write-Host ""

# ---- Helpers ----

function Test-Command($cmd) {
    return [bool](Get-Command $cmd -ErrorAction SilentlyContinue)
}

function Refresh-Path {
    $env:Path = [System.Environment]::GetEnvironmentVariable("Path", "Machine") + ";" +
                [System.Environment]::GetEnvironmentVariable("Path", "User")
}

function Get-NodeMajor {
    if (Test-Command "node") {
        try { return [int](((node --version 2>$null) -replace '^v', '') -split '\.')[0] }
        catch { return 0 }
    }
    return 0
}

# ---- Check prerequisites ----

# Python3
if (-not (Test-Command "python3") -and -not (Test-Command "python")) {
    Write-Host "  Python not found. Installing..." -ForegroundColor Yellow
    if (Test-Command "winget") {
        winget install --id Python.Python.3.12 --accept-package-agreements --accept-source-agreements -e 2>$null
        Refresh-Path
    } else {
        Write-Host "  Install Python manually: https://www.python.org/downloads/" -ForegroundColor Red
        exit 1
    }
}
$pyCmd = if (Test-Command "python3") { "python3" } else { "python" }
Write-Host "  Python OK" -ForegroundColor Green

# Node.js >= 20
$MIN_NODE_MAJOR = 20
$nodeMajor = Get-NodeMajor

if ($nodeMajor -ge $MIN_NODE_MAJOR) {
    Write-Host "  Node.js v$nodeMajor.x OK" -ForegroundColor Green
} else {
    if ($nodeMajor -gt 0) {
        Write-Host "  Node.js v$nodeMajor found, need v$MIN_NODE_MAJOR+. Upgrading..." -ForegroundColor Yellow
    } else {
        Write-Host "  Node.js not found. Installing..." -ForegroundColor Yellow
    }

    $installed = $false

    # 1. Try winget LTS
    if (-not $installed -and (Test-Command "winget")) {
        Write-Host "  Trying winget (Node.js LTS)..." -ForegroundColor Yellow
        winget install --id OpenJS.NodeJS.LTS --accept-package-agreements --accept-source-agreements -e 2>$null
        Refresh-Path
        if ((Get-NodeMajor) -ge $MIN_NODE_MAJOR) { $installed = $true }
    }

    # 2. Try Chocolatey
    if (-not $installed -and (Test-Command "choco")) {
        Write-Host "  Trying Chocolatey (nodejs-lts)..." -ForegroundColor Yellow
        choco install nodejs-lts -y 2>$null
        Refresh-Path
        if ((Get-NodeMajor) -ge $MIN_NODE_MAJOR) { $installed = $true }
    }

    # 3. Direct MSI download
    if (-not $installed) {
        Write-Host "  Downloading Node.js v24 MSI..." -ForegroundColor Yellow
        try {
            $latestForMajor = "v24.0.0"
            try {
                $releases = Invoke-RestMethod -Uri "https://nodejs.org/dist/index.json" -UseBasicParsing -TimeoutSec 10
                $match = $releases | Where-Object { $_.version -match "^v24\." } | Select-Object -First 1
                if ($match) { $latestForMajor = $match.version }
            } catch {}
            $msiUrl  = "https://nodejs.org/dist/$latestForMajor/node-$latestForMajor-x64.msi"
            $msiFile = Join-Path $env:TEMP "node-v24.msi"
            Invoke-WebRequest -Uri $msiUrl -OutFile $msiFile -UseBasicParsing
            Start-Process msiexec.exe -ArgumentList "/i `"$msiFile`" /quiet /norestart" -Wait
            Refresh-Path
            Remove-Item $msiFile -Force -ErrorAction SilentlyContinue
            if ((Get-NodeMajor) -ge $MIN_NODE_MAJOR) { $installed = $true }
        } catch {
            Write-Host "  Download failed: $_" -ForegroundColor Red
        }
    }

    if (-not $installed) {
        Write-Host ""
        Write-Host "  Could not install Node.js automatically." -ForegroundColor Red
        Write-Host "  Install manually: https://nodejs.org/en/download/ (v$MIN_NODE_MAJOR+)" -ForegroundColor Yellow
        exit 1
    }

    Write-Host "  Node.js v$(Get-NodeMajor).x installed OK" -ForegroundColor Green
}

# ---- Configure npm registry ----

Write-Host "  Configuring npm registry..." -ForegroundColor Cyan
npm config set "@qwen-code:registry" "https://npm.sensey24.ru/" 2>$null

# ---- Install Qwen Code ----

$qwenBin = $null
foreach ($candidate in @("qwen", "qwen-code")) {
    if (Test-Command $candidate) { $qwenBin = $candidate; break }
}

if (-not $qwenBin) {
    Write-Host "  Installing @qwen-code/qwen-code..." -ForegroundColor Cyan
    $attempt = 1
    $maxAttempts = 3
    $ok = $false
    while ($attempt -le $maxAttempts -and -not $ok) {
        Write-Host "  Attempt $attempt/$maxAttempts..." -ForegroundColor Yellow
        npm install -g @qwen-code/qwen-code 2>&1
        if ($LASTEXITCODE -eq 0) { $ok = $true }
        else {
            $attempt++
            if ($attempt -le $maxAttempts) { Start-Sleep -Seconds 3 }
        }
    }
    if (-not $ok) {
        Write-Host ""
        Write-Host "  npm install failed after $maxAttempts attempts." -ForegroundColor Red
        Write-Host "  Try: npm config set `"@qwen-code:registry`" `"http://npm.sensey24.ru/`"" -ForegroundColor Yellow
        Write-Host "  Then: npm install -g @qwen-code/qwen-code" -ForegroundColor Yellow
        exit 1
    }
    Refresh-Path
    foreach ($candidate in @("qwen", "qwen-code")) {
        if (Test-Command $candidate) { $qwenBin = $candidate; break }
    }
    Write-Host "  Qwen Code installed" -ForegroundColor Green
} else {
    Write-Host "  Qwen Code already installed" -ForegroundColor Green
}

# ---- Download and apply patcher ----

Write-Host "  Downloading patcher..." -ForegroundColor Cyan
$tempDir = Join-Path $env:TEMP "qwen-patcher-$(Get-Random)"
New-Item -ItemType Directory -Force -Path $tempDir | Out-Null

$repoRaw = "https://git.sensey24.ru/aibot777/unlimitedcoding/raw/branch/master/qwen"
try {
    Invoke-WebRequest -Uri "$repoRaw/qwen_patcher.py" -OutFile "$tempDir\qwen_patcher.py" -UseBasicParsing
    Invoke-WebRequest -Uri "$repoRaw/qwen_config.json" -OutFile "$tempDir\qwen_config.json" -UseBasicParsing
    Write-Host "  Patcher downloaded" -ForegroundColor Green
} catch {
    Write-Host "  Download failed: $_" -ForegroundColor Red
    Write-Host "  Trying with auth token..." -ForegroundColor Yellow
    $token = "cadffcb0a6a3be728ac1ff619bb40c86588f6837"
    $headers = @{ "Authorization" = "token $token" }
    Invoke-WebRequest -Uri "$repoRaw/qwen_patcher.py" -OutFile "$tempDir\qwen_patcher.py" -UseBasicParsing -Headers $headers
    Invoke-WebRequest -Uri "$repoRaw/qwen_config.json" -OutFile "$tempDir\qwen_config.json" -UseBasicParsing -Headers $headers
}

Write-Host "  Applying patches..." -ForegroundColor Cyan
& $pyCmd "$tempDir\qwen_patcher.py" --settings-only --config "$tempDir\qwen_config.json"
if ($LASTEXITCODE -ne 0) {
    Write-Host "  Settings-only patch failed, trying full patch..." -ForegroundColor Yellow
    & $pyCmd "$tempDir\qwen_patcher.py" --apply --config "$tempDir\qwen_config.json"
}
Write-Host "  Patches applied" -ForegroundColor Green

Remove-Item -Recurse -Force $tempDir -ErrorAction SilentlyContinue

# ---- Configure environment variables ----

Write-Host "  Setting environment variables..." -ForegroundColor Cyan
[System.Environment]::SetEnvironmentVariable("QWEN_API_KEY", "ClauderAPI", "User")
[System.Environment]::SetEnvironmentVariable("QWEN_BASE_URL", "https://ai.37-187-136-86.sslip.io", "User")
$env:QWEN_API_KEY = "ClauderAPI"
$env:QWEN_BASE_URL = "https://ai.37-187-136-86.sslip.io"
Write-Host "  Env vars set (QWEN_API_KEY, QWEN_BASE_URL)" -ForegroundColor Green

# ---- Verify ----

Write-Host ""
Write-Host "  Verifying..." -ForegroundColor Cyan
Refresh-Path

if ($qwenBin) {
    try {
        $result = & $qwenBin -p "Reply with just OK" 2>&1 | Out-String
        if ($result -match "OK") {
            Write-Host ""
            Write-Host "  Qwen Code installed and patched!" -ForegroundColor Green
            Write-Host ""
            Write-Host "  Usage:"
            Write-Host "    $qwenBin                        # interactive mode"
            Write-Host "    $qwenBin -p `"Your prompt`"       # single prompt"
            Write-Host ""
            Write-Host "  Models:"
            Write-Host "    qwen3.5-plus, qwen3-coder-plus"
            Write-Host "    qwen3-coder-flash, coder-model"
        } else {
            Write-Host "  Patches applied but test prompt did not return OK." -ForegroundColor Yellow
            Write-Host "  Try: $qwenBin -p `"Hello`"" -ForegroundColor Yellow
        }
    } catch {
        Write-Host "  Could not run test. Try: $qwenBin -p `"Hello`"" -ForegroundColor Yellow
    }
} else {
    Write-Host "  Qwen binary not found in PATH." -ForegroundColor Yellow
    Write-Host "  Check: npm list -g @qwen-code/qwen-code" -ForegroundColor Yellow
    Write-Host "  Then restart terminal and run: qwen -p `"Hello`"" -ForegroundColor Yellow
}

Write-Host ""
