$ErrorActionPreference = "Stop"

$projectRoot = Split-Path -Parent $MyInvocation.MyCommand.Path

Set-Location $projectRoot

$containerName = "cocoindex-pg"

$dbUrl = "postgresql://cocoindex:cocoindex@localhost:5432/cocoindex"

if (-not (Get-Command docker -ErrorAction SilentlyContinue)) {

throw "Docker is not installed or not available in PATH."

}

$containerExists = docker ps -a --format "{{.Names}}" | Select-String -Pattern "^$containerName$" -Quiet

if (-not $containerExists) {

Write-Host "Creating pgvector container: $containerName"

docker run -d `

--name $containerName `

-e POSTGRES_USER=cocoindex `

-e POSTGRES_PASSWORD=cocoindex `

-e POSTGRES_DB=cocoindex `

-p 5432:5432 `

pgvector/pgvector:pg16 | Out-Null

}

else {

$isRunning = (docker inspect -f "{{.State.Running}}" $containerName).Trim()

if ($isRunning -ne "true") {

Write-Host "Starting existing container: $containerName"

docker start $containerName | Out-Null

}

}

Write-Host "Waiting for Postgres readiness..."

$ready = $false

for ($i = 0; $i -lt 30; $i++) {

$health = docker exec $containerName pg_isready -U cocoindex -d cocoindex 2>$null

if ($LASTEXITCODE -eq 0) {

$ready = $true

break

}

Start-Sleep -Seconds 1

}

if (-not $ready) {

throw "Postgres did not become ready in time."

}

$env:COCOINDEX_DATABASE_URL = $dbUrl

$cocoindexExe = Join-Path $env:APPDATA "Python\Python312\Scripts\cocoindex.exe"

if (Test-Path $cocoindexExe) {

& $cocoindexExe update -f cocoindex_rag.index_flow

}

else {

& cocoindex update -f cocoindex_rag.index_flow

}

Write-Host ""

Write-Host "RAG is ready."

Write-Host "MCP server config: .cursor/mcp.json"

Write-Host "If Cursor still shows red, restart MCP servers in Cursor."