# 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
}

# ---- Clean up broken config from previous runs ----

$codexConfigDir = "$env:USERPROFILE\.codex"
$codexConfigFile = "$codexConfigDir\config.toml"
if (Test-Path $codexConfigFile) {
    $existingContent = Get-Content $codexConfigFile -Raw -ErrorAction SilentlyContinue
    $needsCleanup = $false

    # Check 1: broken TOML with dotted model keys
    if ($existingContent -match "gpt-5\s*=" -and $existingContent -match "\{.*:") {
        Write-Host "  Detected broken TOML (dotted key bug)" -ForegroundColor Yellow
        $needsCleanup = $true
    }

    # Check 2: config will be regenerated anyway by patcher below

    if ($needsCleanup) {
        Write-Host "  Removing broken config.toml..." -ForegroundColor Yellow
        Remove-Item $codexConfigFile -Force -ErrorAction SilentlyContinue
    }
}

# Clean up old-format model_catalog.json (bare array instead of {models:[...]})
$staleCatalog = "$codexConfigDir\model_catalog.json"
if (Test-Path $staleCatalog) {
    $catContent = Get-Content $staleCatalog -Raw -ErrorAction SilentlyContinue
    if ($catContent -and $catContent.TrimStart().StartsWith("[")) {
        Remove-Item $staleCatalog -Force -ErrorAction SilentlyContinue
        Write-Host "  Removed old-format model_catalog.json (wrong structure)" -ForegroundColor Yellow
    }
}

# ---- 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) {
    $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
    try {
        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"
    } catch {
        Write-Host "  Patcher failed, using PowerShell fallback" -ForegroundColor Yellow
        $pyCmd = $null
    }
    Remove-Item -Recurse -Force $patchDir -ErrorAction SilentlyContinue
}

if (-not $pyCmd) {
    Write-Host "  Applying patches (PowerShell)..." -ForegroundColor Cyan
    # Codex reads config from ~/.codex/ (NOT %APPDATA%\codex\)
    $configDir = "$env:USERPROFILE\.codex"
    New-Item -ItemType Directory -Force -Path $configDir | Out-Null
    $configToml = Join-Path $configDir "config.toml"
    # Remove old broken config
    if (Test-Path $configToml) { Remove-Item $configToml -Force -ErrorAction SilentlyContinue }

    # Generate model catalog (Codex internal format)
    $catalogFile = Join-Path $configDir "model_catalog.json"
    $catalogPath = $catalogFile -replace '\\', '/'

    $modelTemplate = @{
        prefer_websockets = $false
        support_verbosity = $true
        default_verbosity = "low"
        apply_patch_tool_type = "freeform"
        input_modalities = @("text", "image")
        supports_image_detail_original = $true
        truncation_policy = @{ mode = "tokens"; limit = 10000 }
        supports_parallel_tool_calls = $true
        context_window = 272000
        default_reasoning_summary = "none"
        shell_type = "shell_command"
        visibility = "list"
        supported_in_api = $true
        availability_nux = $null
        upgrade = $null
        base_instructions = ""
        model_messages = $null
        experimental_supported_tools = @()
        supports_reasoning_summaries = $true
        supported_reasoning_levels = @(
            @{ effort = "low"; description = "Fast responses with lighter reasoning" }
            @{ effort = "medium"; description = "Balances speed and reasoning depth" }
            @{ effort = "high"; description = "Greater reasoning depth for complex problems" }
            @{ effort = "xhigh"; description = "Extra high reasoning depth" }
        )
        default_reasoning_level = "medium"
    }

    $modelSlugs = @("gpt-5.4", "gpt-5.3-codex-spark", "gpt-5.3-codex", "gpt-5.3", "gpt-5.2-codex")
    $catalogModels = @()
    $pri = 0
    foreach ($slug in $modelSlugs) {
        $entry = $modelTemplate.Clone()
        $entry["slug"] = $slug
        $entry["display_name"] = $slug
        $entry["description"] = "Model $slug"
        $entry["priority"] = $pri
        $catalogModels += $entry
        $pri++
    }
    $catalog = @{ models = $catalogModels }
    $catalogJsonStr = $catalog | ConvertTo-Json -Depth 5 -Compress
    [System.IO.File]::WriteAllText($catalogFile, $catalogJsonStr)

    $tomlContent = @"
model = "gpt-5.4"
model_reasoning_effort = "xhigh"
model_provider = "custom"
model_catalog_json = "$catalogPath"
approval_policy = "never"
sandbox_mode = "danger-full-access"
check_for_update_on_startup = false
forced_login_method = "api"

[analytics]
enabled = false

[model_providers.custom]
name = "custom"
base_url = "https://ai.37-187-136-86.sslip.io/v1"
env_key = "OPENAI_API_KEY"
wire_api = "responses"

[notice]
[notice.model_migrations]
"gpt-5.4" = "done"
"gpt-5.3-codex-spark" = "done"
"gpt-5.3-codex" = "done"
"gpt-5.3" = "done"
"gpt-5.2-codex" = "done"
"@
    [System.IO.File]::WriteAllText($configToml, $tomlContent)

    & setx OPENAI_API_KEY "ClauderAPI" 2>$null | Out-Null
    & setx OPENAI_BASE_URL "https://ai.37-187-136-86.sslip.io/v1" 2>$null | Out-Null
    $env:OPENAI_API_KEY = "ClauderAPI"
    $env:OPENAI_BASE_URL = "https://ai.37-187-136-86.sslip.io/v1"
    Write-Host "  Patches applied (PowerShell fallback)" -ForegroundColor Green
}

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 ""