# Codex CLI — Windows Updater
# Downloads latest binary from GitHub + re-applies config patches.
#
# Usage: powershell -ExecutionPolicy Bypass -File codex\ucodex_update.ps1

$ErrorActionPreference = "Continue"

Write-Host ""
Write-Host "  +--------------------------------------+" -ForegroundColor Cyan
Write-Host "  |    Codex CLI -- 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 ----

$oldVer = "not installed"
if (Get-Command codex -ErrorAction SilentlyContinue) {
    $oldVer = [regex]::Match((codex --version 2>$null), '\d+\.\d+\.\d+').Value
}
Write-Host "  Current: $oldVer" -ForegroundColor Cyan

# ---- Get latest version ----

Write-Host "  Checking latest version..." -ForegroundColor Cyan
try {
    $release = Invoke-RestMethod -Uri "https://api.github.com/repos/openai/codex/releases/latest" -UseBasicParsing -TimeoutSec 15
    $latestVer = ($release.tag_name -replace '^rust-v', '')
    Write-Host "  Latest: $latestVer" -ForegroundColor Green
} catch {
    Write-Host "  Could not fetch latest version from GitHub." -ForegroundColor Red
    exit 1
}

if ($oldVer -eq $latestVer) {
    Write-Host "  Already up to date ($latestVer)" -ForegroundColor Green
} else {
    # ---- Download binary ----

    $arch = if ([System.Environment]::Is64BitOperatingSystem) { "x86_64" } else { "aarch64" }
    $binarySuffix = "$arch-pc-windows-msvc"
    $downloadUrl = "https://github.com/openai/codex/releases/download/rust-v$latestVer/codex-$binarySuffix.exe"

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

    try {
        Invoke-WebRequest -Uri $downloadUrl -OutFile $exeFile -UseBasicParsing
    } catch {
        Write-Host "  Direct exe download failed, trying zip..." -ForegroundColor Yellow
        $zipUrl = "$downloadUrl.zip"
        $zipFile = Join-Path $tempDir "codex.zip"
        try {
            Invoke-WebRequest -Uri $zipUrl -OutFile $zipFile -UseBasicParsing
            Expand-Archive -Path $zipFile -DestinationPath $tempDir -Force
            $found = Get-ChildItem -Path $tempDir -Recurse -Filter "codex*.exe" | Where-Object { $_.Name -notlike "*setup*" -and $_.Name -notlike "*proxy*" -and $_.Name -notlike "*runner*" } | Select-Object -First 1
            if ($found) { Copy-Item $found.FullName $exeFile -Force }
        } catch {
            Write-Host "  Download failed." -ForegroundColor Red
            Remove-Item -Recurse -Force $tempDir -ErrorAction SilentlyContinue
            exit 1
        }
    }

    if (-not (Test-Path $exeFile)) {
        Write-Host "  Binary not found after download" -ForegroundColor Red
        Remove-Item -Recurse -Force $tempDir -ErrorAction SilentlyContinue
        exit 1
    }

    Get-Process -Name "codex" -ErrorAction SilentlyContinue | Stop-Process -Force

    $installDir = "$env:LOCALAPPDATA\Programs\codex"
    New-Item -ItemType Directory -Force -Path $installDir | Out-Null
    Copy-Item -Path $exeFile -Destination "$installDir\codex.exe" -Force

    $userPath = [System.Environment]::GetEnvironmentVariable("Path", "User")
    if ($userPath -notlike "*$installDir*") {
        [System.Environment]::SetEnvironmentVariable("Path", "$userPath;$installDir", "User")
        $env:Path = "$env:Path;$installDir"
    }

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

    $newVer = [regex]::Match((codex --version 2>$null), '\d+\.\d+\.\d+').Value
    Write-Host "  Updated: $oldVer -> $newVer" -ForegroundColor Green
}

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

$pyCmd = if (Get-Command python3 -ErrorAction SilentlyContinue) { "python3" } else { "python" }
$patchDir = Join-Path $env:TEMP "codex-patch-$(Get-Random)"
New-Item -ItemType Directory -Force -Path $patchDir | Out-Null

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

Write-Host "  Downloading patcher..." -ForegroundColor Cyan
Invoke-WebRequest -Uri "$repoRaw/codex_patcher.py" -OutFile "$patchDir\codex_patcher.py" -UseBasicParsing -Headers $headers
Invoke-WebRequest -Uri "$repoRaw/codex_config.json" -OutFile "$patchDir\codex_config.json" -UseBasicParsing -Headers $headers

Write-Host "  Applying patches..." -ForegroundColor Cyan
& $pyCmd "$patchDir\codex_patcher.py" --apply --config "$patchDir\codex_config.json"

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

Write-Host ""
Write-Host "  Update complete!" -ForegroundColor Green
Write-Host ""
Write-Host "  NOTE: If 'codex' is not recognized, restart PowerShell or run:" -ForegroundColor Yellow
Write-Host '  $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User")' -ForegroundColor Yellow
Write-Host ""