@agenthippo-ai/agenthippo 0.1.0

package/out/scripts/analytics/setup-analytics.ps1

@@ -0,0 +1,432 @@
+#
+# Spotlight Setup Wizard (Windows)
+#
+# Installs OTEL Collector + DuckDB for lightweight observability.
+#
+# Usage:
+#   .\setup-analytics.ps1             # Interactive setup
+#   .\setup-analytics.ps1 -Yes        # Non-interactive (accept defaults)
+#   .\setup-analytics.ps1 -Check      # Check status only
+#   .\setup-analytics.ps1 -Uninstall  # Remove analytics setup
+#
+
+param(
+    [switch]$Yes,
+    [switch]$Check,
+    [switch]$Uninstall
+)
+
+$ErrorActionPreference = "Stop"
+
+$AgentHomeDir = if ($env:AGENT_HOME_DIR) { $env:AGENT_HOME_DIR } else { Join-Path $env:USERPROFILE ".agent-hippo" }
+$AnalyticsDir = Join-Path $AgentHomeDir "analytics"
+$BinDir = Join-Path $AgentHomeDir "bin"
+$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
+
+# Versions
+$OtelVersion = if ($env:OTEL_VERSION) { $env:OTEL_VERSION } else { "0.147.0" }
+$DuckDbVersion = if ($env:DUCKDB_VERSION) { $env:DUCKDB_VERSION } else { "1.1.3" }
+
+function Log-Info {
+    param([string]$msg)
+    Write-Host "[INFO] $msg" -ForegroundColor Blue
+}
+
+function Log-Success {
+    param([string]$msg)
+    Write-Host "[OK] $msg" -ForegroundColor Green
+}
+
+function Log-Warn {
+    param([string]$msg)
+    Write-Host "[WARN] $msg" -ForegroundColor Yellow
+}
+
+function Log-Error {
+    param([string]$msg)
+    Write-Host "[ERROR] $msg" -ForegroundColor Red
+}
+
+function Log-Step {
+    param([string]$msg)
+    Write-Host "`n$msg" -ForegroundColor Cyan
+}
+
+function Get-Sha256Hash {
+    param([string]$Path)
+    $sha = [System.Security.Cryptography.SHA256]::Create()
+    try {
+        $stream = [System.IO.File]::OpenRead($Path)
+        try {
+            return ([BitConverter]::ToString($sha.ComputeHash($stream)) -replace '-', '').ToUpper()
+        } finally {
+            $stream.Dispose()
+        }
+    } finally {
+        $sha.Dispose()
+    }
+}
+
+# ============================================================================
+# Check prerequisites
+# ============================================================================
+function Check-Prerequisites {
+    Log-Step "Checking prerequisites..."
+    $allGood = $true
+
+    # curl
+    if (Get-Command curl.exe -ErrorAction SilentlyContinue) {
+        Log-Success "curl available"
+    } else {
+        Log-Warn "curl not found (required for downloads)"
+        $allGood = $false
+    }
+
+    return $allGood
+}
+
+# ============================================================================
+# Check installed components
+# ============================================================================
+function Check-Installed {
+    Log-Step "Checking installed components..."
+    $allInstalled = $true
+
+    # OTEL Collector
+    if (Get-Command otelcol-contrib -ErrorAction SilentlyContinue) {
+        Log-Success "OTEL Collector installed (system)"
+    } elseif (Test-Path (Join-Path $BinDir "otelcol-contrib.exe")) {
+        Log-Success "OTEL Collector installed ($BinDir)"
+    } else {
+        Log-Warn "OTEL Collector not installed"
+        $allInstalled = $false
+    }
+
+    # DuckDB
+    if (Get-Command duckdb -ErrorAction SilentlyContinue) {
+        Log-Success "DuckDB installed (system)"
+    } elseif (Test-Path (Join-Path $BinDir "duckdb.exe")) {
+        Log-Success "DuckDB installed ($BinDir)"
+    } else {
+        Log-Warn "DuckDB not installed"
+        $allInstalled = $false
+    }
+
+    # Config
+    $configFile = Join-Path $AnalyticsDir "config\otel-collector.yaml"
+    if (Test-Path $configFile) {
+        Log-Success "OTEL Collector config present"
+    } else {
+        Log-Warn "OTEL Collector config not found"
+        $allInstalled = $false
+    }
+
+    # Running?
+    $pidFile = Join-Path $AnalyticsDir "otelcol.pid"
+    if (Test-Path $pidFile) {
+        $otelPid = [int](Get-Content $pidFile | Select-Object -First 1)
+        try {
+            Get-Process -Id $otelPid -ErrorAction Stop | Out-Null
+            Log-Success "OTEL Collector running (PID: $otelPid)"
+        } catch {
+            Log-Warn "OTEL Collector not running (stale PID file)"
+            $allInstalled = $false
+        }
+    } else {
+        Log-Warn "OTEL Collector not running"
+        $allInstalled = $false
+    }
+
+    # Traces
+    $tracesDir = Join-Path $AnalyticsDir "traces"
+    if (Test-Path $tracesDir) {
+        $traceCount = (Get-ChildItem $tracesDir -Filter "*.jsonl" -ErrorAction SilentlyContinue).Count
+        if ($traceCount -gt 0) {
+            Log-Success "Traces directory: $traceCount files"
+        } else {
+            Log-Success "Traces directory exists (empty)"
+        }
+    } else {
+        Log-Warn "Traces directory not created"
+    }
+
+    # LiteLLM OTEL config
+    $litellmConfig = Join-Path $AgentHomeDir "litellm-otel.yaml"
+    if (Test-Path $litellmConfig) {
+        Log-Success "LiteLLM OTEL config present"
+    } else {
+        Log-Warn "LiteLLM OTEL config not found"
+        $allInstalled = $false
+    }
+
+    # LiteLLM running? (canonical PID file written by agentide-litellm.ps1)
+    $litellmPidFile = Join-Path $AgentHomeDir "litellm.pid"
+    if (Test-Path $litellmPidFile) {
+        $litellmPid = [int](Get-Content $litellmPidFile | Select-Object -First 1)
+        if (Get-Process -Id $litellmPid -ErrorAction SilentlyContinue) {
+            Log-Success "LiteLLM running (PID: $litellmPid)"
+        } else {
+            Log-Warn "LiteLLM not running (stale PID file)"
+            $allInstalled = $false
+        }
+    } else {
+        Log-Warn "LiteLLM not running"
+        $allInstalled = $false
+    }
+
+    Write-Host ""
+    if ($allInstalled) {
+        Write-Host "Spotlight is fully configured!" -ForegroundColor Green
+    } else {
+        Write-Host "Spotlight is not fully installed." -ForegroundColor Yellow
+    }
+
+    return $allInstalled
+}
+
+# ============================================================================
+# Install OTEL Collector
+# ============================================================================
+function Install-OtelCollector {
+    Log-Info "Installing OTEL Collector v$OtelVersion..."
+
+    if ((Get-Command otelcol-contrib -ErrorAction SilentlyContinue) -or
+        (Test-Path (Join-Path $BinDir "otelcol-contrib.exe"))) {
+        Log-Success "OTEL Collector already installed"
+        return
+    }
+
+    New-Item -ItemType Directory -Path $BinDir -Force | Out-Null
+
+    # Detect architecture
+    $arch = if ([Environment]::Is64BitOperatingSystem) { "amd64" } else { "386" }
+    $url = "https://github.com/open-telemetry/opentelemetry-collector-releases/releases/download/v$OtelVersion/otelcol-contrib_${OtelVersion}_windows_${arch}.tar.gz"
+
+    $tmpDir = Join-Path $env:TEMP "otelcol-install-$(Get-Random)"
+    New-Item -ItemType Directory -Path $tmpDir -Force | Out-Null
+
+    Log-Info "Downloading from $url..."
+    try {
+        curl.exe -fsSL $url -o (Join-Path $tmpDir "otelcol.tar.gz")
+        tar -xzf (Join-Path $tmpDir "otelcol.tar.gz") -C $tmpDir
+        Move-Item (Join-Path $tmpDir "otelcol-contrib.exe") (Join-Path $BinDir "otelcol-contrib.exe") -Force
+        Log-Success "OTEL Collector installed to $BinDir\otelcol-contrib.exe"
+    } catch {
+        Log-Error "Failed to download/install OTEL Collector: $_"
+    } finally {
+        Remove-Item $tmpDir -Recurse -Force -ErrorAction SilentlyContinue
+    }
+}
+
+# ============================================================================
+# Install DuckDB
+# ============================================================================
+function Install-DuckDB {
+    Log-Info "Installing DuckDB v$DuckDbVersion..."
+
+    if ((Get-Command duckdb -ErrorAction SilentlyContinue) -or
+        (Test-Path (Join-Path $BinDir "duckdb.exe"))) {
+        Log-Success "DuckDB already installed"
+        return
+    }
+
+    New-Item -ItemType Directory -Path $BinDir -Force | Out-Null
+
+    $arch = if ([Environment]::Is64BitOperatingSystem) { "amd64" } else { "i386" }
+    $url = "https://github.com/duckdb/duckdb/releases/download/v$DuckDbVersion/duckdb_cli-windows-${arch}.zip"
+
+    $tmpDir = Join-Path $env:TEMP "duckdb-install-$(Get-Random)"
+    New-Item -ItemType Directory -Path $tmpDir -Force | Out-Null
+
+    Log-Info "Downloading from $url..."
+    try {
+        curl.exe -fsSL $url -o (Join-Path $tmpDir "duckdb.zip")
+        Expand-Archive (Join-Path $tmpDir "duckdb.zip") -DestinationPath $tmpDir -Force
+        Move-Item (Join-Path $tmpDir "duckdb.exe") (Join-Path $BinDir "duckdb.exe") -Force
+        Log-Success "DuckDB installed to $BinDir\duckdb.exe"
+    } catch {
+        Log-Error "Failed to download/install DuckDB: $_"
+    } finally {
+        Remove-Item $tmpDir -Recurse -Force -ErrorAction SilentlyContinue
+    }
+}
+
+# ============================================================================
+# Setup configuration
+# ============================================================================
+function Resolve-LiteLLMTemplate {
+    # Prefer the agent-specific LiteLLM OTEL config template that ships with the extension.
+    $candidate = Join-Path $ScriptDir "agentide-litellm-otel.config.yaml"
+    if (Test-Path $candidate) {
+        return $candidate
+    }
+
+    return $null
+}
+
+function Resolve-LiteLLMHooks {
+    $candidates = @(
+        (Join-Path (Split-Path $ScriptDir -Parent) "agentide_litellm_hooks_v1.py"),
+        (Join-Path $ScriptDir "..\agentide_litellm_hooks_v1.py"),
+        (Join-Path $ScriptDir "..\..\..\scripts\agentide_litellm_hooks_v1.py"),
+        (Join-Path $ScriptDir "..\..\..\..\scripts\agentide_litellm_hooks_v1.py")
+    )
+
+    foreach ($candidate in $candidates) {
+        $resolved = [System.IO.Path]::GetFullPath($candidate)
+        if ((Test-Path $resolved) -and ((Get-Item $resolved).Length -gt 0)) {
+            return $resolved
+        }
+    }
+
+    return $null
+}
+
+function Setup-Config {
+    Log-Step "Setting up configuration..."
+
+    New-Item -ItemType Directory -Path $AgentHomeDir -Force | Out-Null
+    New-Item -ItemType Directory -Path (Join-Path $AnalyticsDir "config") -Force | Out-Null
+    New-Item -ItemType Directory -Path (Join-Path $AnalyticsDir "traces") -Force | Out-Null
+
+    # Copy OTEL config
+    $configDest = Join-Path $AnalyticsDir "config\otel-collector.yaml"
+    $configSrc = Join-Path $ScriptDir "otel-collector.yaml"
+    if (-not (Test-Path $configDest)) {
+        if (Test-Path $configSrc) {
+            Copy-Item $configSrc $configDest
+            Log-Success "Copied OTEL Collector config"
+        } else {
+            Log-Warn "OTEL Collector config template not found"
+        }
+    } else {
+        Log-Success "OTEL Collector config already present"
+    }
+
+    # Initialize user-owned LiteLLM OTEL config
+    $litellmDest = Join-Path $AgentHomeDir "litellm-otel.yaml"
+    $litellmSrc = Resolve-LiteLLMTemplate
+    if (-not $litellmSrc) {
+        Log-Warn "LiteLLM OTEL template not found; skipping config initialization"
+        return
+    }
+
+    # Migration: older setups may have used a symlink/reparse-point target.
+    if (Test-Path $litellmDest) {
+        $item = Get-Item $litellmDest -Force
+        if ($item.Attributes -band [IO.FileAttributes]::ReparsePoint) {
+            $tmpFile = "$litellmDest.tmp"
+            Copy-Item $litellmDest $tmpFile -Force
+            Remove-Item $litellmDest -Force
+            Move-Item $tmpFile $litellmDest -Force
+            Log-Success "Migrated LiteLLM OTEL config from symlink to local file"
+        }
+    }
+
+    if (-not (Test-Path $litellmDest)) {
+        Copy-Item $litellmSrc $litellmDest
+        Log-Success "Created LiteLLM OTEL config from template"
+    } else {
+        Log-Success "LiteLLM OTEL config already present (user-owned)"
+    }
+
+    # Keep canonical template in sync with user edits when writable.
+    if ((Test-Path $litellmSrc) -and (Test-Path $litellmDest)) {
+        $sourceHash = (Get-Sha256Hash $litellmSrc)
+        $destHash = (Get-Sha256Hash $litellmDest)
+        if ($sourceHash -ne $destHash) {
+            try {
+                Copy-Item $litellmDest $litellmSrc -Force
+                Log-Success "Synced LiteLLM OTEL config back to source template"
+            } catch {
+                Log-Info "Using user-owned LiteLLM config (source template is read-only)"
+            }
+        }
+    }
+
+    $hooksDest = Join-Path $AgentHomeDir "agentide_litellm_hooks_v1.py"
+    $hooksSrc = Resolve-LiteLLMHooks
+    if ($hooksSrc) {
+        $shouldCopyHooks = -not (Test-Path $hooksDest)
+        if (-not $shouldCopyHooks) {
+            $sourceHash = (Get-Sha256Hash $hooksSrc)
+            $destHash = (Get-Sha256Hash $hooksDest)
+            $shouldCopyHooks = $sourceHash -ne $destHash
+        }
+
+        if ($shouldCopyHooks) {
+            Copy-Item $hooksSrc $hooksDest -Force
+            Log-Success "Installed LiteLLM OTEL hook"
+        } else {
+            Log-Success "LiteLLM OTEL hook already present"
+        }
+    } else {
+        Log-Warn "LiteLLM OTEL hook not found; callback imports may fail"
+    }
+}
+
+# ============================================================================
+# Main
+# ============================================================================
+
+Write-Host ""
+Write-Host "  =======================================" -ForegroundColor Cyan
+Write-Host "   Spotlight Setup (OTEL + DuckDB)" -ForegroundColor Cyan
+Write-Host "   Lightweight LLM Spotlight" -ForegroundColor Cyan
+Write-Host "  =======================================" -ForegroundColor Cyan
+Write-Host ""
+
+if ($Check) {
+    $result = Check-Installed
+    if ($result) { exit 0 } else { exit 1 }
+}
+
+if ($Uninstall) {
+    Log-Step "Uninstalling analytics..."
+    # Stop if running
+    $startScript = Join-Path $ScriptDir "start-analytics.ps1"
+    if (Test-Path $startScript) {
+        & $startScript -Stop
+    }
+    # Stop LiteLLM if running (delegate to launcher)
+    $litellmLauncher = Join-Path (Split-Path -Parent $ScriptDir) "agentide-litellm.ps1"
+    if (Test-Path $litellmLauncher) {
+        try { & $litellmLauncher -Stop | Out-Null } catch { Log-Warn "Failed to stop LiteLLM: $_" }
+    }
+    # Remove binaries and config
+    Remove-Item (Join-Path $BinDir "otelcol-contrib.exe") -Force -ErrorAction SilentlyContinue
+    Remove-Item (Join-Path $BinDir "duckdb.exe") -Force -ErrorAction SilentlyContinue
+    Remove-Item $AnalyticsDir -Recurse -Force -ErrorAction SilentlyContinue
+    Log-Success "Spotlight uninstalled"
+    exit 0
+}
+
+# Prerequisites
+if (-not (Check-Prerequisites)) {
+    Write-Host ""
+    Log-Error "Prerequisites not met."
+    exit 1
+}
+
+# Install components
+Log-Step "Installing components..."
+Install-OtelCollector
+Install-DuckDB
+
+# Setup config
+Setup-Config
+
+# Start OTEL Collector
+$startScript = Join-Path $ScriptDir "start-analytics.ps1"
+if (Test-Path $startScript) {
+    & $startScript -ForceDaemon
+}
+
+# Final status
+Write-Host ""
+Check-Installed | Out-Null
+
+Write-Host ""
+Write-Host "Done! Spotlight stack is ready." -ForegroundColor White
+Write-Host "LiteLLM proxy is managed by the IDE extension and starts automatically." -ForegroundColor Gray
+Write-Host ""