thrust-cli 1.0.16 → 1.0.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "thrust-cli",
-  "version": "1.0.16",
+  "version": "1.0.18",
   "description": "The local agent for Thrust AI Director",
   "type": "module",
   "homepage": "https://thrust.web.app",

package/utils/daemon.js

@@ -21,8 +21,8 @@ const API_URL = GATEWAY_URL.replace('ws://', 'http://').replace('wss://', 'https
 const AUTH_PROXY_URL = "https://everydaycats-thrust-auth-server.hf.space";
 
 // --- DEBOUNCE & POLLING TIMERS ---
-const INACTIVITY_DELAY_MS = 15 * 1000; 
-const MCP_POLL_INTERVAL_MS = 18 * 1000; 
+const INACTIVITY_DELAY_MS = 15 * 1000;
+const MCP_POLL_INTERVAL_MS = 18 * 1000;
 
 let currentWatcher = null;
 let inactivityTimer = null;
@@ -125,9 +125,9 @@ export async function startDaemon(preferredPort) {
     };
 
     app.use(cors(corsOptionsDelegate));
-    app.use(express.json());
+    app.use(express.json({ limit: '50mb' }));
 
-    const frontendPath = path.join(__dirname, '..', 'frontend');
+    const frontendPath = path.join(__dirname, '..', 'build');
 
     // ==========================================
     // MCP SERVER ENDPOINTS (FEED-IN PUSH)
@@ -138,7 +138,6 @@ export async function startDaemon(preferredPort) {
         const token = config.auth?.token;
         const projectId = config.activeLeadId;
 
-        // Extract optional parameters, default to true
         const prd = req.query.prd || 'true';
         const thrust = req.query.thrust || 'true';
         const timeline = req.query.timeline || 'true';
@@ -148,7 +147,6 @@ export async function startDaemon(preferredPort) {
         }
 
         try {
-            // Forward parameters to new combined gateway endpoint
             const response = await fetch(`${API_URL}/api/projects/${projectId}/mcp-context?prd=${prd}&thrust=${thrust}&timeline=${timeline}`, {
                 headers: { 'Authorization': `Bearer ${token}` }
             });
@@ -200,7 +198,7 @@ export async function startDaemon(preferredPort) {
         saveConfig(config);
 
         pollExternalMCPServers();
-        fetchInitialMCPContext({ name, url, type: 'http' }); 
+        fetchInitialMCPContext({ name, url, type: 'http' });
         res.json({ success: true });
     });
 
@@ -214,6 +212,52 @@ export async function startDaemon(preferredPort) {
         res.json({ success: true });
     });
 
+    app.get('/api/mcp/check-unity', (req, res) => {
+        const config = getConfig();
+        const projectId = config.activeLeadId;
+        if (!projectId || !config.leads[projectId]?.path) {
+            return res.status(400).json({ error: "No active project linked." });
+        }
+        const activePath = config.leads[projectId].path;
+        
+        // Editor folder requirement ensures Unity treats the file as an Editor script
+        const expected1 = path.join(activePath, 'Assets', 'Thrust', 'Editor', 'ThrustMCPBridge.cs');
+        const expected2 = path.join(activePath, 'Assets', 'Editor', 'Thrust', 'ThrustMCPBridge.cs');
+
+        const exists = fs.existsSync(expected1) || fs.existsSync(expected2);
+        res.json({ exists });
+    });
+
+    app.post('/api/mcp/install-unity', async (req, res) => {
+        const config = getConfig();
+        const projectId = config.activeLeadId;
+        if (!projectId || !config.leads[projectId]?.path) {
+            return res.status(400).json({ error: "No active project linked." });
+        }
+        const activePath = config.leads[projectId].path;
+        const sourceScript = path.join(__dirname, '..', 'mcps', 'ThrustMCPBridge.cs');
+
+        if (!fs.existsSync(sourceScript)) {
+            return res.status(500).json({ error: "Source ThrustMCPBridge.cs not found in thrust-cli/mcps/" });
+        }
+
+        const targetDir = path.join(activePath, 'Assets', 'Thrust', 'Editor');
+
+        try {
+            if (!fs.existsSync(targetDir)) {
+                fs.mkdirSync(targetDir, { recursive: true });
+            }
+            const targetScript = path.join(targetDir, 'ThrustMCPBridge.cs');
+            fs.copyFileSync(sourceScript, targetScript);
+
+            broadcastLocalLog('system', `✅ Injected ThrustMCPBridge.cs into ${targetDir}`);
+            res.json({ success: true, path: targetScript });
+        } catch (e) {
+            broadcastLocalLog('error', `Failed to inject Unity script: ${e.message}`);
+            res.status(500).json({ error: e.message });
+        }
+    });
+
     // ==========================================
     // EXTERNAL MCP DIRECT QUERY (AI OR MANUAL)
     // ==========================================
@@ -327,6 +371,10 @@ export async function startDaemon(preferredPort) {
             const data = await response.json();
             if (!response.ok) throw new Error(data.error);
 
+            // 👉 NEW: Force the buffer to have content so syncContext doesn't abort!
+            // This tells the AI exactly what you just checked off manually.
+            fileActivityBuffer += `\n[${new Date().toLocaleTimeString()}] [USER ACTION] Manually marked task as complete: "${taskTitle}"\n`;
+
             if (config.leads[projectId]?.path) {
                 if (inactivityTimer) clearTimeout(inactivityTimer);
                 syncContext(config.leads[projectId].path);
@@ -455,7 +503,12 @@ export async function startDaemon(preferredPort) {
                 const data = JSON.parse(message.toString());
                 if (data.type === 'frontend_prompt') {
                     if (globalWs && globalWs.readyState === WebSocket.OPEN) {
-                        globalWs.send(JSON.stringify({ type: 'prompt', content: data.payload, projectId: getActiveProject()?.id }));
+                        globalWs.send(JSON.stringify({ 
+                            type: 'prompt', 
+                            content: data.payload, 
+                            image: data.image,
+                            projectId: getActiveProject()?.id 
+                        }));
                         broadcastLocalLog('sync', `Prompt sent to Cloud Director...`);
                     } else {
                         broadcastLocalLog('error', `⚠️ Cannot send: Cloud Gateway is offline.`);
@@ -535,13 +588,11 @@ async function startWatching(projectPath) {
 
         fileActivityBuffer = "";
 
-        // Fetch initial MCP states immediately upon watching a project
         const config = getConfig();
         if (config.mcpServers && config.mcpServers.length > 0) {
             config.mcpServers.forEach(server => fetchInitialMCPContext(server));
         }
 
-        // Start the active polling loop for external services
         if (mcpPollTimer) clearInterval(mcpPollTimer);
         mcpPollTimer = setInterval(pollExternalMCPServers, MCP_POLL_INTERVAL_MS);
 
@@ -549,11 +600,37 @@ async function startWatching(projectPath) {
             ignored:[
                 /(^|[\/\\])\../,
                 '**/node_modules/**', '**/dist/**', '**/build/**',
-                // Ignore noisy Unity cache & build folders
                 '**/Library/**', '**/Temp/**', '**/Logs/**', '**/obj/**', '**/ProjectSettings/**'
             ],
             persistent: true,
-            ignoreInitial: true
+            ignoreInitial: true,
+            ignorePermissionErrors: true // NEW: Ignores restricted Android/Termux subfolders safely
+        });
+
+        // NEW: FATAL ERROR HANDLER
+        // Catches EACCES constraints before Node.js crashes and forces a UI unlink
+        currentWatcher.on('error', async (error) => {
+            const isPermissionError = error.code === 'EACCES' || error.code === 'EPERM';
+            const msg = isPermissionError
+                ? `Permission Denied: Cannot watch folder. Access Denied by OS.`
+                : `Watcher Error: ${error.message}`;
+
+            broadcastLocalLog('error', msg);
+
+            // Force unlink the bad folder
+            const cfg = getConfig();
+            cfg.activeLeadId = null;
+            saveConfig(cfg);
+
+            if (inactivityTimer) { clearTimeout(inactivityTimer); inactivityTimer = null; }
+            if (mcpPollTimer) { clearInterval(mcpPollTimer); mcpPollTimer = null; }
+            if (currentWatcher) {
+                await currentWatcher.close().catch(() => {});
+                currentWatcher = null;
+            }
+
+            // Signal the frontend to drop the user back to the setup wizard
+            broadcastLocalLog('force_unlink', 'Folder access denied. Project unlinked safely.');
         });
 
         currentWatcher.on('all', (event, filePath) => {
@@ -563,7 +640,9 @@ async function startWatching(projectPath) {
 
             triggerDebouncedSync();
         });
-    } catch (err) {}
+    } catch (err) {
+        broadcastLocalLog('error', `Failed to initialize watcher: ${err.message}`);
+    }
 }
 
 function connectWebSocket() {
@@ -589,7 +668,6 @@ function connectWebSocket() {
             try {
                 const msg = JSON.parse(data.toString());
 
-                // NEW: Handle dynamic MCP Queries pushed from the AI Director
                 if (msg.type === 'mcp_query' && msg.payload) {
                     const { serverName, toolName, targetArg } = msg.payload;
                     const targetServer = getConfig().mcpServers?.find(s => s.name.toLowerCase() === serverName.toLowerCase());