# d0a1 pack - Open WebUI (Windows)
# Chat de IA privado. Sin cuenta, sin API key.
#
# Local:  .\deploy.ps1 local    -> http://localhost:3000
# VPS:    .\deploy.ps1 vps      -> http://TU_IP:3000
# Stop:   .\deploy.ps1 stop
#
# Requiere: Docker Desktop para Windows (https://docs.docker.com/desktop/)

param(
    [Parameter(Position = 0)]
    [ValidateSet('local', 'vps', 'stop')]
    [string]$Mode = 'local'
)

$ErrorActionPreference = 'Stop'

# Escape ANSI via [char]27 (compatible con Windows PowerShell 5.1 y PS 7+)
$Esc = [char]27
$Green = "$Esc[32m"; $Cyan = "$Esc[36m"; $Yellow = "$Esc[33m"; $Red = "$Esc[31m"; $Bold = "$Esc[1m"; $NC = "$Esc[0m"
if ($env:NO_COLOR) {
    $Green = $Cyan = $Yellow = $Red = $Bold = $NC = ''
}

$ScriptDir = $PSScriptRoot
$ComposeFile = Join-Path $ScriptDir 'docker-compose.yml'

function Info($msg)    { Write-Host "${Bold}${Green}>>>${NC} $msg" }
function Warn($msg)    { Write-Host "${Bold}${Yellow}>>>${NC} $msg" }
function Err($msg)     { Write-Host "${Bold}${Red}>>>${NC} $msg" }
function Ok($msg)      { Write-Host "${Green}OK${NC} $msg" }

function Find-FreePort([int]$Start) {
    for ($p = $Start; $p -le ($Start + 20); $p++) {
        $inUse = Get-NetTCPConnection -LocalPort $p -State Listen -ErrorAction SilentlyContinue
        if (-not $inUse) { return $p }
    }
    return $Start
}

Write-Host ''
Write-Host "${Bold}${Cyan}============================================================${NC}"
Write-Host "${Bold}${Cyan}  d0a1 pack - Open WebUI (Windows)${NC}"
Write-Host "${Bold}${Cyan}  Chat de IA privado. Sin cuenta, sin API key.${NC}"
Write-Host "${Bold}${Cyan}============================================================${NC}"
Write-Host ''

# --- Stop mode ------------------------------------------------------------
if ($Mode -eq 'stop') {
    Warn 'Parando...'
    docker compose -f $ComposeFile down
    Info 'Parado.'
    exit 0
}

# --- Check Docker ---------------------------------------------------------
if (-not (Get-Command docker -ErrorAction SilentlyContinue)) {
    Err 'Docker no esta instalado.'
    Write-Host ''
    Write-Host 'Instala Docker Desktop para Windows:'
    Write-Host '  https://docs.docker.com/desktop/setup/install/windows-install/'
    exit 1
}
# En PS 5.1, *> $null sobre comandos nativos convierte stderr en errores
# terminantes con EAP=Stop (PS #4002). Se descarta via cmd /c.
$null = cmd /c "docker info >NUL 2>&1"
if ($LASTEXITCODE -ne 0) {
    Err 'Docker Desktop no responde. Abre Docker Desktop y espera a que arranque.'
    exit 1
}
# En PS 5.1, *> $null sobre comandos nativos convierte stderr en errores
# terminantes con EAP=Stop (PS #4002). Se descarta via cmd /c.
cmd /c "docker compose version >NUL 2>&1"
$composeOk = ($LASTEXITCODE -eq 0)
if (-not $composeOk) {
    Err 'Docker Compose v2 no disponible. Actualiza Docker Desktop.'
    exit 1
}
Ok "Docker: $(cmd /c 'docker compose version --short 2>NUL')"

# --- Check RAM ------------------------------------------------------------
$TotalRamGb = [math]::Round((Get-CimInstance Win32_ComputerSystem).TotalPhysicalMemory / 1GB, 1)
if ($TotalRamGb -lt 3.5) {
    Err "Se necesitan 4 GB de RAM minimo (detectado: $TotalRamGb GB)."
    exit 1
}
Ok "RAM: $TotalRamGb GB"

# --- Auto-detect ports ----------------------------------------------------
$OllamaPort = Find-FreePort 11434
$WebUiPort  = Find-FreePort 3000

# --- Mode info ------------------------------------------------------------
if ($Mode -eq 'local') {
    Warn 'Modo: local (tu ordenador)'
    Write-Host "  URL: ${Cyan}http://localhost:${WebUiPort}${NC}"
} else {
    Warn 'Modo: VPS (accesible desde fuera)'
    # IP real de LAN: la de la interfaz con la ruta por defecto (evita vEthernet
    # WSL/Hyper-V, Docker, VPN — que son 172.16-31.*, 10.x internos, etc.)
    $GwIf = (Get-NetRoute -DestinationPrefix '0.0.0.0/0' -ErrorAction SilentlyContinue |
        Sort-Object RouteMetric | Select-Object -First 1).ifIndex
    $Ip = (Get-NetIPAddress -AddressFamily IPv4 -ErrorAction SilentlyContinue |
        Where-Object { $_.ifIndex -eq $GwIf -and $_.IPAddress -notlike '127.*' } |
        Select-Object -First 1).IPAddress
    if (-not $Ip) { $Ip = 'TU_IP' }
    Write-Host "  URL: ${Cyan}http://${Ip}:${WebUiPort}${NC}"
    Write-Host ''
    Write-Host "  ${Bold}¿No tienes VPS?${NC} Consigue uno con descuento:"
    Write-Host "  ${Cyan}https://d0a1.es/go/vultr/${NC}  - Vultr"
    Write-Host "  ${Cyan}https://d0a1.es/go/digitalocean/${NC}  - DigitalOcean (`$200 credit)"
}

if ($OllamaPort -ne 11434) { Warn "Puerto 11434 ocupado, usando $OllamaPort" }
if ($WebUiPort  -ne 3000)  { Warn "Puerto 3000 ocupado, usando $WebUiPort" }
Write-Host ''

# --- Deploy ----------------------------------------------------------------
# Si hay un deploy previo del pack, tumbarlo primero (mantiene los volumenes:
# el modelo ya descargado no se re-descarga) para que los nuevos puertos apliquen.
cmd /c docker compose -f $ComposeFile down '>' NUL '2>&1'
Warn 'Descargando contenedores (2-3 min la primera vez)...'
$env:OLLAMA_PORT = "$OllamaPort"
$env:WEBUI_PORT  = "$WebUiPort"
docker compose -f $ComposeFile up -d
if ($LASTEXITCODE -ne 0) { Err "docker compose up fallo (exit $LASTEXITCODE)."; exit 1 }

# --- Pull model -------------------------------------------------------------
Warn 'Descargando modelo de IA (llama3.2:3b, ~2 GB)...'
$MaxWait = 60; $Waited = 0
while ($true) {
    # Descarte de stderr via cmd /c (ver nota de PS 5.1 / PS #4002 arriba)
    cmd /c "docker exec d0a1-ollama ollama ps >NUL 2>&1"
    if ($LASTEXITCODE -eq 0) { break }
    Start-Sleep -Seconds 2
    $Waited += 2
    if ($Waited -ge $MaxWait) {
        Err 'Ollama no arranco. Revisa: docker logs d0a1-ollama'
        exit 1
    }
}
$PullOk = $false
foreach ($Attempt in 1..3) {
    if ($Attempt -gt 1) { Warn "Reintento $Attempt de 3..." }
    docker exec d0a1-ollama ollama pull llama3.2:3b
    if ($LASTEXITCODE -eq 0) { $PullOk = $true; break }
    Warn 'Descarga interrumpida (red). Reintentando en 5 s...'
    Start-Sleep -Seconds 5
}
if (-not $PullOk) {
    Err 'Fallo la descarga del modelo tras 3 intentos.'
    Write-Host '  Revisa tu red (VPN/proxy) y reintenta a mano:'
    Write-Host '  docker exec d0a1-ollama ollama pull llama3.2:3b'
    exit 1
}

# --- Done -------------------------------------------------------------------
Write-Host ''
Info 'Listo! Tu IA esta corriendo.'
Write-Host ''
if ($Mode -eq 'local') {
    Write-Host "  ${Bold}Abre:${NC} ${Cyan}http://localhost:${WebUiPort}${NC}"
} else {
    Write-Host "  ${Bold}Abre:${NC} ${Cyan}http://${Ip}:${WebUiPort}${NC}"
}
Write-Host ''
Write-Host "  ${Bold}Modelos alternativos:${NC}"
Write-Host '  docker exec d0a1-ollama ollama pull qwen3:4b      # Mejor en espanol + razonamiento (4 GB RAM)'
Write-Host '  docker exec d0a1-ollama ollama pull qwen3:8b      # Mas calidad (6 GB RAM)'
Write-Host ''
Write-Host "  ${Bold}Parar:${NC} .\deploy.ps1 stop"
Write-Host "  ${Bold}¿Quieres mas?${NC} ${Cyan}https://d0a1.es/packs/${NC}"
Write-Host ''