Update stateManager.js

stateManager.js

@@ -5,9 +5,8 @@ let supabase = null;
 const activeProjects = new Map();
 const initializationLocks = new Set(); 
 
-// --- NEW STREAMING BUFFERS ---
-const streamBuffers = new Map(); // projectId -> String (accumulated partial text)
-const statusBuffers = new Map(); // projectId -> String (current status message)
+// --- NEW STREAM BUFFERS ---
+const streamBuffers = new Map();
 
 export const initDB = () => {
     if (!process.env.SUPABASE_URL || !process.env.SUPABASE_SERVICE_ROLE_KEY) {
@@ -98,20 +97,24 @@ export const StateManager = {
             // LOGIC: If we have a chunk and it has room (< 20 items), UPDATE it.
             if (latest && currentSize < 20) {
                 const updatedPayload = [...currentPayload, newMessage];
+                
                 const { error: updateError } = await supabase.from('message_chunks')
                     .update({ payload: updatedPayload })
                     .eq('id', latest.id);
+
                 if (updateError) console.error(`[DB Error] Update chunk failed:`, updateError.message);
             } 
             // ELSE: Create a NEW chunk
             else {
                 const nextIndex = latestIndex + 1;
+                
                 const { error: insertError } = await supabase.from('message_chunks').insert({
                     project_id: projectId,
                     type,
                     chunk_index: nextIndex,
                     payload: [newMessage] // Start new array
                 });
+
                 if (insertError) console.error(`[DB Error] Insert new chunk failed:`, insertError.message);
             }
         } catch (e) {
@@ -119,6 +122,18 @@ export const StateManager = {
         }
     },
 
+    // --- STREAMING HELPERS (FOR POLLING) ---
+    appendStream: (projectId, chunk) => {
+        const current = streamBuffers.get(projectId) || "";
+        streamBuffers.set(projectId, current + chunk);
+    },
+
+    popStream: (projectId) => {
+        const content = streamBuffers.get(projectId) || "";
+        streamBuffers.set(projectId, ""); // Clear after pop
+        return content;
+    },
+
     // --- COMMANDS (MEMORY ONLY + PARSING) ---
     queueCommand: async (projectId, input) => {
         let project = activeProjects.get(projectId);
@@ -167,26 +182,6 @@ export const StateManager = {
         return project.commandQueue.shift();
     },
 
-    // --- STREAMING HELPERS (FOR POLLING) ---
-    setStatus: (projectId, status) => {
-        statusBuffers.set(projectId, status);
-    },
-
-    getStatus: (projectId) => {
-        return statusBuffers.get(projectId) || "Idle";
-    },
-
-    appendStream: (projectId, chunk) => {
-        const current = streamBuffers.get(projectId) || "";
-        streamBuffers.set(projectId, current + chunk);
-    },
-
-    popStream: (projectId) => {
-        const content = streamBuffers.get(projectId) || "";
-        streamBuffers.set(projectId, ""); // Clear buffer after read
-        return content;
-    },
-
     // --- METADATA UPDATE ---
     updateProject: async (projectId, data) => {
         // 1. Update Memory
@@ -196,7 +191,7 @@ export const StateManager = {
             activeProjects.set(projectId, newData);
         }
 
-        // 2. Prepare DB Payload
+        // 2. Prepare DB Payload (Exclude commandQueue)
         const payload = {
             info: {
                 title: data.title,
@@ -207,13 +202,14 @@ export const StateManager = {
             }
         };
 
+        // Clean undefined keys
         Object.keys(payload.info).forEach(key => payload.info[key] === undefined && delete payload.info[key]);
 
         const { data: currentDb } = await supabase.from('projects').select('info').eq('id', projectId).single();
         
         if (currentDb) {
              const mergedInfo = { ...currentDb.info, ...payload.info };
-             delete mergedInfo.commandQueue; 
+             delete mergedInfo.commandQueue; // Sanity check
              
              const { error } = await supabase.from('projects').update({ info: mergedInfo }).eq('id', projectId);
              if (error) console.error("[DB ERROR] Update Project failed:", error.message);
@@ -226,12 +222,12 @@ export const StateManager = {
         let count = 0;
 
         for (const [id, data] of activeProjects.entries()) {
-            if (!data.lastActive) data.lastActive = now; 
+            if (!data.lastActive) data.lastActive = now; // Heal missing timestamp
+
             if (now - data.lastActive > FOUR_HOURS) {
+                console.log(`[StateManager] 🧹 Removing expired project: ${id}`);
                 activeProjects.delete(id);
-                // Also clean streams
                 streamBuffers.delete(id);
-                statusBuffers.delete(id);
                 count++;
             }
         }