# Qwen Code — Windows Updater
# Usage: powershell -ExecutionPolicy Bypass -File qwen\uqwen_update.ps1

$ErrorActionPreference = "Continue"

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

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

# ---- Check current version ----

$qwenBin = $null
foreach ($candidate in @("qwen", "qwen-code")) {
    if (Get-Command $candidate -ErrorAction SilentlyContinue) { $qwenBin = $candidate; break }
}
$oldVer = if ($qwenBin) { & $qwenBin --version 2>$null } else { "not installed" }
Write-Host "  Current: $oldVer" -ForegroundColor Cyan

# ---- Configure registry ----

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

# ---- Update package ----

Write-Host "  Installing latest @qwen-code/qwen-code..." -ForegroundColor Cyan
npm install -g @qwen-code/qwen-code 2>&1
if ($LASTEXITCODE -ne 0) {
    Write-Host "  npm install failed. Retrying..." -ForegroundColor Yellow
    Start-Sleep -Seconds 3
    npm install -g @qwen-code/qwen-code 2>&1
    if ($LASTEXITCODE -ne 0) {
        Write-Host "  npm install failed." -ForegroundColor Red
        exit 1
    }
}
Refresh-Path

foreach ($candidate in @("qwen", "qwen-code")) {
    if (Get-Command $candidate -ErrorAction SilentlyContinue) { $qwenBin = $candidate; break }
}
$newVer = if ($qwenBin) { & $qwenBin --version 2>$null } else { "unknown" }
Write-Host "  Updated: $oldVer -> $newVer" -ForegroundColor Green

# ---- Download and apply patches ----

$pyCmd = $null
foreach ($candidate in @("python3", "python")) {
    if (Get-Command $candidate -ErrorAction SilentlyContinue) {
        try {
            $pyVer = & $candidate -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" 2>$null
            $parts = $pyVer -split '\.'
            if ([int]$parts[0] -ge 3 -and [int]$parts[1] -ge 11) {
                $pyCmd = $candidate
                break
            }
        } catch {}
    }
}

if ($pyCmd) {
    $tempDir = Join-Path $env:TEMP "qwen-update-$(Get-Random)"
    New-Item -ItemType Directory -Force -Path $tempDir | Out-Null

    $repoRaw = "https://git.sensey24.ru/aibot777/unlimitedcoding/raw/branch/master/qwen"
    $token = "cadffcb0a6a3be728ac1ff619bb40c86588f6837"
    $headers = @{ "Authorization" = "token $token" }

    Write-Host "  Downloading patcher..." -ForegroundColor Cyan
    try {
        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
    } catch {
        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
        } catch {
            Write-Host "  Patcher download failed, using PowerShell fallback" -ForegroundColor Yellow
            $pyCmd = $null
        }
    }

    if ($pyCmd) {
        Write-Host "  Applying patches..." -ForegroundColor Cyan
        & $pyCmd "$tempDir\qwen_patcher.py" --settings-only --config "$tempDir\qwen_config.json"
        if ($LASTEXITCODE -ne 0) {
            Write-Host "  Trying full patch..." -ForegroundColor Yellow
            & $pyCmd "$tempDir\qwen_patcher.py" --apply --config "$tempDir\qwen_config.json"
            if ($LASTEXITCODE -ne 0) {
                Write-Host "  Patcher failed, using PowerShell fallback" -ForegroundColor Yellow
                $pyCmd = $null
            }
        }
        if ($pyCmd) { Write-Host "  Patches applied" -ForegroundColor Green }
    }
    Remove-Item -Recurse -Force $tempDir -ErrorAction SilentlyContinue
}

if (-not $pyCmd) {
    # PowerShell fallback — generate settings directly
    Write-Host "  Applying patches (PowerShell)..." -ForegroundColor Cyan

    # Environment variables
    [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"

    # Settings
    $qwenDir = "$env:USERPROFILE\.qwen"
    New-Item -ItemType Directory -Force -Path $qwenDir | Out-Null

    $settingsFile = "$qwenDir\settings.json"
    $json = @'
{
  "security": {
    "auth": {
      "selectedType": "api-key"
    },
    "folderTrust": {
      "enabled": false
    }
  },
  "telemetry": {
    "enabled": false,
    "logPrompts": false
  },
  "general": {
    "defaultApprovalMode": "yolo"
  }
}
'@
    [System.IO.File]::WriteAllText($settingsFile, $json)

    # Trusted folders
    $trustedFile = "$qwenDir\trustedFolders.json"
    $trustedJson = '{"C:\\":\"TRUST_PARENT\",\"C:\\Users\":\"TRUST_PARENT\"}'
    [System.IO.File]::WriteAllText($trustedFile, $trustedJson)
    Write-Host "  Patches applied (PowerShell fallback)" -ForegroundColor Green
}

Write-Host ""
Write-Host "  Update complete!" -ForegroundColor Green
Write-Host ""