Update boot.ps1

boot.ps1

@@ -1,71 +1,105 @@
-# --- AI-LAB Windows Smart Agent (Self-Updating) ---
-# Este script gerencia sua própria atualização, a pasta C:\AI-LAB e o atalho na Área de Trabalho.
-
-$ErrorActionPreference = "Stop"
-$INSTALL_PATH = "C:\AI-LAB"
-$SCRIPT_NAME = "boot.ps1"
-$FULL_SCRIPT_PATH = Join-Path $INSTALL_PATH $SCRIPT_NAME
-$REPO_RAW_URL = "https://huggingface.co/vost/pub-scripts/raw/main/"
-
-# 1. Garantir Privilégios e Pasta Estrutural
-if (-not (Test-Path $INSTALL_PATH)) {
-    New-Item -Path $INSTALL_PATH -ItemType Directory -Force | Out-Null
-}
-Set-Location $INSTALL_PATH
-
-# 2. Lógica de Auto-Atualização com Re-execução
-Write-Host "--- Verificando atualizações do Agente ---" -ForegroundColor Cyan
-
-if (Test-Path $FULL_SCRIPT_PATH) {
-    # Calcula o Hash do arquivo local para comparar com o remoto
-    $localHash = (Get-FileHash $FULL_SCRIPT_PATH -Algorithm MD5).Hash
-    $remoteContent = Invoke-WebRequest -Uri "$REPO_RAW_URL/$SCRIPT_NAME" -UseBasicParsing
-    
-    # Gera o Hash do conteúdo baixado na memória
-    $remoteHash = ([System.Security.Cryptography.MD5]::Create().ComputeHash([System.Text.Encoding]::UTF8.GetBytes($remoteContent.Content)) | ForEach-Object { $_.ToString("X2") }) -join ""
-
-    if ($localHash -ne $remoteHash) {
-        Write-Host "Nova versão detectada! Atualizando e reiniciando o script..." -ForegroundColor Magenta
-        $remoteContent.Content | Out-File $FULL_SCRIPT_PATH -Encoding UTF8
-        
-        # RE-EXECUÇÃO: Chama a nova versão e encerra a atual
-        Start-Process powershell.exe -ArgumentList "-ExecutionPolicy Bypass -File `"$FULL_SCRIPT_PATH`""
-        exit
-    }
-    Write-Host "O Agente já está na versão mais recente." -ForegroundColor Green
-} else {
-    # Primeira instalação (Caso o arquivo ainda não exista no diretório)
-    Write-Host "Baixando versão inicial do Agente..." -ForegroundColor Cyan
-    Invoke-WebRequest -Uri "$REPO_RAW_URL/$SCRIPT_NAME" -OutFile $FULL_SCRIPT_PATH
-}
-
-# 3. Criação do Atalho na Área de Trabalho
-$ShortcutPath = Join-Path ([Environment]::GetFolderPath("Desktop")) "AI-LAB.lnk"
-if (-not (Test-Path $ShortcutPath)) {
-    Write-Host "Criando atalho na Área de Trabalho..." -ForegroundColor Gray
-    $WshShell = New-Object -ComObject WScript.Shell
-    $Shortcut = $WshShell.CreateShortcut($ShortcutPath)
-    $Shortcut.TargetPath = "powershell.exe"
-    $Shortcut.Arguments = "-ExecutionPolicy Bypass -File `"$FULL_SCRIPT_PATH`""
-    $Shortcut.WorkingDirectory = $INSTALL_PATH
-    $Shortcut.IconLocation = "powershell.exe,0" # Ícone do PowerShell
-    $Shortcut.Save()
-}
-
-# 4. Sincronização do docker-compose.yaml
-Write-Host "--- Sincronizando Infraestrutura (YAML) ---" -ForegroundColor Cyan
-Invoke-WebRequest -Uri "$REPO_RAW_URL/docker-compose.yaml" -OutFile (Join-Path $INSTALL_PATH "docker-compose.yaml")
-
-# 5. Verificação do Docker (Winget)
-if (-not (Get-Command docker -ErrorAction SilentlyContinue)) {
-    Write-Host "Docker não encontrado. Iniciando instalação..." -ForegroundColor Yellow
-    winget install --id Docker.DockerDesktop --silent --accept-package-agreements
-    Write-Host "Por favor, REINICIE o computador e clique no atalho AI-LAB na Área de Trabalho." -ForegroundColor Red
-    exit
-}
-
-# 6. Execução Final do Docker
-Write-Host "--- Iniciando containers do Laboratório ---" -ForegroundColor Green
-docker compose up -d
-
-Write-Host "`n[SUCESSO] Seu AI-LAB está rodando e sincronizado." -ForegroundColor Green
+# --- AI-LAB Windows Smart Agent (Self-Updating) ---
+# Este script gerencia sua própria atualização, a pasta C:\AI-LAB e o atalho na Área de Trabalho.
+$ErrorActionPreference = "Stop"
+$INSTALL_PATH = "C:\AI-LAB"
+$SCRIPT_NAME = "boot.ps1"
+$FULL_SCRIPT_PATH = Join-Path $INSTALL_PATH $SCRIPT_NAME
+$REPO_RAW_URL = "https://huggingface.co/vost/pub-scripts/raw/main/"
+
+# 1. Garantir Privilégios e Pastas Estruturais
+if (-not (Test-Path $INSTALL_PATH)) {
+    New-Item -Path $INSTALL_PATH -ItemType Directory -Force | Out-Null
+}
+Set-Location $INSTALL_PATH
+
+$MOUNT_ROOT = New-Item -Path "C:\AI-LAB\mnt" -ItemType Directory -Force | Out-Null
+$HF_PATH = New-Item -Path "C:\AI-LAB\mnt\hf" -ItemType Directory -Force | Out-Null
+
+# 2. Verificação do Docker (Winget)
+if (-not (Get-Command docker -ErrorAction SilentlyContinue)) {
+    # Solicita admin apenas se precisar instalar o Docker
+    if (-not ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
+        Write-Host "Docker não encontrado. Solicitando permissões de Administrador para instalação..." -ForegroundColor Cyan
+        Start-Process powershell.exe -ArgumentList "-ExecutionPolicy Bypass -File `"$PSCommandPath`"" -Verb RunAs
+        exit
+    }
+
+    Write-Host "Docker não encontrado. Iniciando instalação..." -ForegroundColor Yellow
+    winget install --id Docker.DockerDesktop --silent --accept-package-agreements
+    Write-Host "Por favor, REINICIE o computador e clique no atalho AI-LAB na Área de Trabalho." -ForegroundColor Red
+    exit
+}
+
+function Sync-RemoteFile {
+    param (
+        [Parameter(Mandatory=$true)][string]$FileName,
+        [Parameter(Mandatory=$true)][string]$DestinationPath,
+        [switch]$RestartOnUpdate
+    )
+
+    $RemoteUrl = "$REPO_RAW_URL$FileName"
+    $TempFile = Join-Path $env:TEMP "ai-lab-$FileName"
+
+    try {
+        if (Test-Path $DestinationPath) {
+            Invoke-WebRequest -Uri $RemoteUrl -OutFile $TempFile -UseBasicParsing
+            
+            $localHash = (Get-FileHash $DestinationPath -Algorithm SHA256).Hash
+            $remoteHash = (Get-FileHash $TempFile -Algorithm SHA256).Hash
+
+            if ($localHash -ne $remoteHash) {
+                Write-Host "Nova versão de [$FileName] detectada! Atualizando..." -ForegroundColor Magenta
+                Move-Item -Path $TempFile -Destination $DestinationPath -Force
+                if ($RestartOnUpdate) {
+                    Write-Host "Reiniciando script para aplicar alterações..." -ForegroundColor Cyan
+                    Start-Process powershell.exe -ArgumentList "-ExecutionPolicy Bypass -File `"$DestinationPath`""
+                    exit
+                }
+            } else {
+                Write-Host "[$FileName] já está na versão mais recente." -ForegroundColor Green
+            }
+        } else {
+            Write-Host "Baixando versão inicial de [$FileName]..." -ForegroundColor Cyan
+            Invoke-WebRequest -Uri $RemoteUrl -OutFile $DestinationPath -UseBasicParsing
+        }
+    } catch {
+        Write-Warning "Falha ao sincronizar [$FileName]. Usando versão local se disponível."
+    } finally {
+        if (Test-Path $TempFile) { Remove-Item $TempFile -ErrorAction SilentlyContinue }
+    }
+}
+
+# 3. Lógica de Auto-Atualização com Re-execução
+Write-Host "--- Verificando atualizações do Agente ---" -ForegroundColor Cyan
+Sync-RemoteFile -FileName $SCRIPT_NAME -DestinationPath $FULL_SCRIPT_PATH -RestartOnUpdate
+
+# 4. Solicita o Token (Uma única vez)
+if (-not $env:HF_TOKEN) {
+    $token = Read-Host "Insira seu HuggingFace Token (Read)"
+    [Environment]::SetEnvironmentVariable("HF_TOKEN", $token, "User")
+}
+
+# 5. Sincronização do docker-compose.yaml
+Write-Host "--- Sincronizando Infraestrutura (YAML) ---" -ForegroundColor Cyan
+Sync-RemoteFile -FileName "docker-compose.yaml" -DestinationPath (Join-Path $INSTALL_PATH "docker-compose.yaml")
+
+# 6. Execução Final do Docker
+Write-Host "--- Iniciando containers do Laboratório ---" -ForegroundColor Green
+docker compose up -d
+
+# 7. Criação do Atalho na Área de Trabalho
+$ShortcutPath = Join-Path ([Environment]::GetFolderPath("Desktop")) "AI-LAB.lnk"
+if (-not (Test-Path $ShortcutPath)) {
+    Write-Host "Criando atalho na Área de Trabalho..." -ForegroundColor Gray
+    $WshShell = New-Object -ComObject WScript.Shell
+    $Shortcut = $WshShell.CreateShortcut($ShortcutPath)
+    $Shortcut.TargetPath = "powershell.exe"
+    $Shortcut.Arguments = "-ExecutionPolicy Bypass -File `"$FULL_SCRIPT_PATH`""
+    $Shortcut.WorkingDirectory = $INSTALL_PATH
+    $Shortcut.IconLocation = "powershell.exe,0" # Ícone do PowerShell
+    $Shortcut.Save()
+}
+
+Write-Host "`n[SUCESSO] Seu AI-LAB está rodando e sincronizado." -ForegroundColor Green
+Write-Host "Pressione qualquer tecla para sair..."
+$null = [System.Console]::ReadKey()