@kumori/aurora-backend-handler 1.1.65 → 1.1.66

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kumori/aurora-backend-handler",
-  "version": "1.1.65",
+  "version": "1.1.66",
   "description": "backend handler",
   "main": "backend-handler.ts",
   "scripts": {

package/websocket-manager.ts

@@ -145,6 +145,7 @@ interface Role {
  */
 let wsConnection: WebSocket | null = null;
 let pendingRequests = new Map<string, PendingOperation>();
+let purgingEnvironments = new Set<string>();
 let currentToken: string | null = null;
 let connectionPromise: Promise<WebSocket> | null = null;
 let userData: UserData = {
@@ -428,6 +429,14 @@ export const makeGlobalWebSocketRequest = async (
     payload,
   };
 
+  let purgeEnvKey: string | null = null;
+  if (petitionAction === "CLEAN" && entityType === "environment" && originalData) {
+    purgeEnvKey = originalData.tenant && originalData.account
+      ? `${originalData.tenant}/${originalData.account}/${petitionInfo}`
+      : petitionInfo;
+    purgingEnvironments.add(purgeEnvKey);
+  }
+
   return new Promise((resolve, reject) => {
     const timeoutId = setTimeout(() => {
       const pending = pendingRequests.get(messageId);
@@ -443,6 +452,7 @@ export const makeGlobalWebSocketRequest = async (
           new Error(`Request timeout for ${topic}`),
         );
       }
+      if (purgeEnvKey) purgingEnvironments.delete(purgeEnvKey);
       const timeoutError = Object.assign(
         new Error(`Request timeout for ${topic}`),
         { isTimeout: true },
@@ -453,10 +463,12 @@ export const makeGlobalWebSocketRequest = async (
     const operation: PendingOperation = {
       resolve: (response: any) => {
         clearTimeout(timeoutId);
+        if (purgeEnvKey) purgingEnvironments.delete(purgeEnvKey);
         resolve(response.payload || response);
       },
       reject: (error: any) => {
         clearTimeout(timeoutId);
+        if (purgeEnvKey) purgingEnvironments.delete(purgeEnvKey);
         const errorMessage = safeStringifyError(error);
         console.error(`WebSocket request failed for ${topic}:`, {
           originalError: error,
@@ -682,7 +694,18 @@ const handleEvent = async (message: WSMessage) => {
           );
         }
         const envSetKey = parentParts.tenant && parentParts.account ? `${parentParts.tenant}/${parentParts.account}/${entityId}` : entityId;
-        environmentsMap.set(envSetKey, envEventResult.environment);
+        let envToStore = envEventResult.environment;
+        if (envToStore.status?.code === "DELETING" && purgingEnvironments.has(envSetKey)) {
+          envToStore = {
+            ...envToStore,
+            status: {
+              code: "PURGING",
+              message: `Environment ${entityId} is being purged.`,
+              timestamp: new Date().toISOString(),
+            },
+          };
+        }
+        environmentsMap.set(envSetKey, envToStore);
         const envTenant = tenantsMap.get(envEventResult.tenantId);
         if (!envTenant) {
           pendingEnvironments.push({