Update stateManager.js

stateManager.js

@@ -1,94 +1,63 @@
 // stateManager.js
-import admin from 'firebase-admin';
-
-// Initialize Firebase (Placeholder - expects serviceAccountKey.json)
-// In production, uncomment the lines below and add your key file
-/* 
-import serviceAccount from './serviceAccountKey.json' with { type: "json" };
-admin.initializeApp({
-  credential: admin.credential.cert(serviceAccount),
-  databaseURL: "https://YOUR-PROJECT-ID.firebaseio.com"
-});
-const db = admin.database();
-*/
-
-// In-Memory Storage
 const activeProjects = new Map();
 
 export const StateManager = {
-  /**
-   * Retrieves a project. Checks memory first, then DB (mocked).
-   */
   getProject: async (projectId) => {
-    // 1. Check Memory
-    if (activeProjects.has(projectId)) {
-      const data = activeProjects.get(projectId);
-      data.lastActive = Date.now();
-      return data;
-    }
-
-    // 2. Check DB (Mock implementation)
-    // const snapshot = await db.ref(`projects/${projectId}`).once('value');
-    // if (snapshot.exists()) {
-    //   const data = snapshot.val();
-    //   activeProjects.set(projectId, data);
-    //   return data;
-    // }
-
-    return null;
+    if (activeProjects.has(projectId)) return activeProjects.get(projectId);
+    return null; 
   },
 
-  /**
-   * Creates or Updates a project state.
-   */
   updateProject: async (projectId, data) => {
-    const currentData = activeProjects.get(projectId) || {};
-    const newData = { ...currentData, ...data, lastActive: Date.now() };
-    
-    // Update Memory
+    const current = activeProjects.get(projectId) || { 
+      commandQueue: [], 
+      workerHistory: [],
+      pmHistory: [],
+      failureCount: 0 
+    };
+    const newData = { ...current, ...data, lastActive: Date.now() };
     activeProjects.set(projectId, newData);
-
-    // Update DB (Mock implementation)
-    // await db.ref(`projects/${projectId}`).set(newData);
-
     return newData;
   },
 
-  /**
-   * Adds a command to the plugin queue.
-   */
-  queueCommand: (projectId, commandResponse) => {
+  // PARSE AI RESPONSE FOR COMMANDS
+  queueCommand: async (projectId, rawResponse) => {
     const project = activeProjects.get(projectId);
-    if (!project) return;
     
-    if (!project.commandQueue) project.commandQueue = [];
-    
-    // Extract Markdown Code for the plugin
-    const codeMatch = commandResponse.match(/```lua([\s\S]*?)```/);
-    const code = codeMatch ? codeMatch[1].trim() : null;
+    let command = null;
 
-    project.commandQueue.push({
-      raw: commandResponse,
-      code: code, // This is what the plugin executes
-      timestamp: Date.now()
-    });
-  },
-
-  /**
-   * Removes inactive projects from memory.
-   */
-  cleanupInactivity: () => {
-    const now = Date.now();
-    const THREE_HOURS = 3 * 60 * 60 * 1000;
-    let removedCount = 0;
+    // 1. Check for Code Execution
+    const codeMatch = rawResponse.match(/```lua([\s\S]*?)```/);
+    if (codeMatch) {
+        command = { type: "EXECUTE", payload: codeMatch[1].trim() };
+    }
+    // 2. Check for Script Reading
+    else if (rawResponse.match(/\[READ_SCRIPT:\s*(.*?)\]/)) {
+        const match = rawResponse.match(/\[READ_SCRIPT:\s*(.*?)\]/);
+        command = { type: "READ_SCRIPT", payload: match[1].trim() };
+    }
+    // 3. Check for Hierarchy Reading
+    else if (rawResponse.match(/\[READ_HIERARCHY:\s*(.*?)\]/)) {
+        const match = rawResponse.match(/\[READ_HIERARCHY:\s*(.*?)\]/);
+        command = { type: "READ_HIERARCHY", payload: match[1].trim() };
+    }
+    // 4. Check for Logs
+    else if (rawResponse.includes("[READ_LOGS]")) {
+        command = { type: "READ_LOGS", payload: null };
+    }
 
-    for (const [id, data] of activeProjects.entries()) {
-      if (now - data.lastActive > THREE_HOURS) {
-        activeProjects.delete(id);
-        removedCount++;
-        // Note: Data persists in Firebase, just removed from RAM
-      }
+    if (command) {
+      project.commandQueue.push(command);
+      console.log(`[${projectId}] Queued Action: ${command.type}`);
+    } else {
+        console.log(`[${projectId}] AI Replied (No Action):`, rawResponse.substring(0, 50) + "...");
     }
-    return removedCount;
+    
+    activeProjects.set(projectId, project);
+  },
+
+  popCommand: async (projectId) => {
+    const project = activeProjects.get(projectId);
+    if (!project || !project.commandQueue.length) return null;
+    return project.commandQueue.shift();
   }
 };