setup.ps1

$ErrorActionPreference = "Stop"

function Write-Step($text) {
    Write-Host ""
    Write-Host "==> $text" -ForegroundColor Cyan
}

function Copy-IfMissing($source, $target) {
    if (!(Test-Path $target)) {
        Copy-Item $source $target
        Write-Host "Created $target"
    } else {
        Write-Host "$target already exists"
    }
}

function Wait-Ollama {

    Write-Step "Waiting for Ollama..."

    while ($true) {
        try {
            docker exec ollama ollama list *> $null
            break
        }
        catch {
            Start-Sleep -Seconds 2
        }
    }

    Write-Host "Ollama is ready."
}

function Ensure-Model($model) {

    $installed = docker exec ollama ollama list

    if ($installed -match "(?m)^$([regex]::Escape($model))") {
        Write-Host "$model already installed."
    }
    else {
        Write-Host "Downloading $model..."
        docker exec ollama ollama pull $model
    }
}

Write-Step "Checking Docker"

docker version *> $null

Write-Step "Creating .env files"

Copy-IfMissing ".env.example" ".env"
Copy-IfMissing ".env.pgadmin.example" ".env.pgadmin"
Copy-IfMissing ".env.searxng.example" ".env.searxng"

Write-Step "Starting containers"

docker compose up -d --build

Wait-Ollama

Write-Step "Installing Ollama models"

Get-Content "models.txt" |
    ForEach-Object { $_.Trim() } |
    Where-Object { $_ -and -not $_.StartsWith("#") } |
    ForEach-Object {
        Ensure-Model $_
    }

Write-Step "Running Alembic"

docker compose exec web alembic upgrade head

Write-Host ""
Write-Host "===================================" -ForegroundColor Green
Write-Host "Setup completed successfully!" -ForegroundColor Green
Write-Host "Frontend: http://localhost:8000" -ForegroundColor Green
Write-Host "===================================" -ForegroundColor Green