@metabob/minibob 0.1.2

package/src/environment.ts

@@ -0,0 +1,167 @@
+/**
+ * minibob Environment Detection
+ *
+ * Auto-detects runtime environment (K8s vs Docker vs local) and cluster configuration.
+ * Enables dynamic capability configuration based on deployment context.
+ */
+
+import { existsSync } from "fs"
+
+/**
+ * Runtime environment types
+ */
+export enum RuntimeEnvironment {
+  LOCAL = "local",
+  DOCKER = "docker",
+  K8S_SINGLE = "k8s-single",
+  K8S_CLUSTER = "k8s-cluster"
+}
+
+/**
+ * Environment detection result
+ */
+export interface EnvironmentInfo {
+  environment: RuntimeEnvironment
+  peerCount: number
+  backendAvailable: boolean
+  clusterMode: boolean
+}
+
+/**
+ * Detect runtime environment based on indicators
+ * 
+ * Detection logic:
+ * - K8s: KUBERNETES_SERVICE_HOST env var present
+ * - Docker: /.dockerenv file exists
+ * - Local: Neither of above
+ * 
+ * If K8s, further detect cluster mode via DNS peer count.
+ */
+export async function detectEnvironment(): Promise<RuntimeEnvironment> {
+  // Check for Kubernetes
+  if (process.env.KUBERNETES_SERVICE_HOST) {
+    console.log("[Environment] Detected Kubernetes environment")
+    
+    // Check cluster mode via DNS lookup
+    const peerCount = await detectClusterMode()
+    
+    if (peerCount >= 3) {
+      console.log(`[Environment] Cluster mode: ${peerCount} peers detected`)
+      return RuntimeEnvironment.K8S_CLUSTER
+    } else {
+      console.log(`[Environment] Single-pod mode: ${peerCount} peer(s)`)
+      return RuntimeEnvironment.K8S_SINGLE
+    }
+  }
+  
+  // Check for Docker
+  if (existsSync("/.dockerenv")) {
+    console.log("[Environment] Detected Docker environment")
+    return RuntimeEnvironment.DOCKER
+  }
+  
+  // Default to local
+  console.log("[Environment] Detected local environment")
+  return RuntimeEnvironment.LOCAL
+}
+
+/**
+ * Detect cluster mode by checking environment variables
+ *
+ * Simplified detection: If MINIBOB_SERVICE_NAME is set, we're in cluster mode.
+ * This env var is only set in Helm deployments with multiple replicas.
+ *
+ * More reliable than DNS lookup which can fail due to:
+ * - Missing subdomain in pod spec
+ * - DNS propagation delays
+ * - Network policies
+ *
+ * @returns Number of peer pods (3 if cluster, 1 otherwise)
+ */
+export async function detectClusterMode(): Promise<number> {
+  const serviceName = process.env.MINIBOB_SERVICE_NAME
+
+  if (serviceName) {
+    console.log(`[Environment] Cluster mode enabled via MINIBOB_SERVICE_NAME: ${serviceName}`)
+    // Assume 3 pods in cluster mode (standard HA setup)
+    // Could enhance to query K8s API for actual count, but not worth complexity
+    return 3
+  } else {
+    console.log(`[Environment] Single-pod mode: MINIBOB_SERVICE_NAME not set`)
+    return 1
+  }
+}
+
+/**
+ * Check backend (MCP) health before initialization
+ * 
+ * Attempts to fetch /health endpoint from MCP backend.
+ * Returns true if backend responds within timeout.
+ * 
+ * @param endpoint MCP backend URL
+ * @param timeoutMs Timeout in milliseconds (default: 5000)
+ * @returns true if backend is healthy, false otherwise
+ */
+export async function checkBackendHealth(endpoint: string, timeoutMs = 5000): Promise<boolean> {
+  const healthUrl = `${endpoint}/health`
+  
+  try {
+    console.log(`[Environment] Checking backend health: ${healthUrl}`)
+    
+    const controller = new AbortController()
+    const timeoutId = setTimeout(() => controller.abort(), timeoutMs)
+    
+    const response = await fetch(healthUrl, {
+      signal: controller.signal,
+      headers: { "Accept": "application/json" }
+    })
+    
+    clearTimeout(timeoutId)
+    
+    if (response.ok) {
+      console.log(`[Environment] ✓ Backend healthy (${response.status})`)
+      return true
+    } else {
+      console.log(`[Environment] ✗ Backend unhealthy (${response.status})`)
+      return false
+    }
+  } catch (error) {
+    if (error instanceof Error && error.name === "AbortError") {
+      console.log(`[Environment] ✗ Backend health check timed out after ${timeoutMs}ms`)
+    } else {
+      console.log(`[Environment] ✗ Backend health check failed: ${error}`)
+    }
+    return false
+  }
+}
+
+/**
+ * Perform complete environment detection
+ * 
+ * Detects:
+ * 1. Runtime environment (K8s/Docker/local)
+ * 2. Cluster mode (peer count)
+ * 3. Backend availability (health check)
+ * 
+ * Returns comprehensive environment info for capability configuration.
+ */
+export async function detectCompleteEnvironment(mcpEndpoint?: string): Promise<EnvironmentInfo> {
+  const environment = await detectEnvironment()
+  
+  const peerCount = environment.startsWith("k8s") 
+    ? await detectClusterMode() 
+    : 1
+  
+  const backendAvailable = mcpEndpoint 
+    ? await checkBackendHealth(mcpEndpoint)
+    : false
+  
+  const clusterMode = environment === RuntimeEnvironment.K8S_CLUSTER && peerCount >= 3
+  
+  return {
+    environment,
+    peerCount,
+    backendAvailable,
+    clusterMode
+  }
+}