@lobu/gateway 2.8.0 → 3.0.6

package/src/orchestration/deployment-utils.ts

@@ -0,0 +1,98 @@
+import { getOrchestratorModules } from "../modules/module-system";
+import { platformRegistry } from "../platform";
+import type {
+  DeploymentInfo,
+  MessagePayload,
+  OrchestratorConfig,
+} from "./base-deployment-manager";
+
+/**
+ * Shared types and utilities for deployment managers
+ * Reduces code duplication between Docker and K8s implementations
+ */
+
+// ============================================================================
+// Utility Functions
+// ============================================================================
+
+/**
+ * Build environment variables by integrating all registered modules
+ */
+export async function buildModuleEnvVars(
+  agentId: string,
+  baseEnv: Record<string, string>
+): Promise<Record<string, string>> {
+  let envVars = { ...baseEnv };
+
+  const orchestratorModules = getOrchestratorModules();
+  for (const module of orchestratorModules) {
+    if (module.buildEnvVars) {
+      envVars = await module.buildEnvVars(agentId, envVars);
+    }
+  }
+
+  return envVars;
+}
+
+export const BASE_WORKER_LABELS = {
+  "app.kubernetes.io/name": "lobu",
+  "app.kubernetes.io/component": "worker",
+  "lobu/managed-by": "orchestrator",
+} as const;
+
+export function resolvePlatformDeploymentMetadata(
+  messageData?: MessagePayload
+): Record<string, string> {
+  if (
+    !messageData?.platform ||
+    !messageData.channelId ||
+    !messageData.conversationId ||
+    !messageData.platformMetadata
+  ) {
+    return {};
+  }
+
+  const platform = platformRegistry.get(messageData.platform);
+  if (!platform) {
+    return {};
+  }
+
+  return platform.buildDeploymentMetadata(
+    messageData.conversationId,
+    messageData.channelId,
+    messageData.platformMetadata
+  );
+}
+
+export function getVeryOldThresholdDays(config: OrchestratorConfig): number {
+  return config.cleanup?.veryOldDays ?? 7;
+}
+
+export function buildDeploymentInfoSummary({
+  deploymentName,
+  lastActivity,
+  now,
+  idleThresholdMinutes,
+  veryOldDays,
+  replicas,
+}: {
+  deploymentName: string;
+  lastActivity: Date;
+  now: number;
+  idleThresholdMinutes: number;
+  veryOldDays: number;
+  replicas: number;
+}): DeploymentInfo {
+  const minutesIdle = (now - lastActivity.getTime()) / (1000 * 60);
+  const daysSinceActivity = minutesIdle / (60 * 24);
+
+  return {
+    deploymentName,
+    lastActivity,
+    minutesIdle,
+    daysSinceActivity,
+    replicas,
+    isIdle: minutesIdle >= idleThresholdMinutes,
+    isVeryOld: daysSinceActivity >= veryOldDays,
+  };
+}