Update app.js

app.js

@@ -29,7 +29,11 @@ app.post('/internal/notify', (req, res) => {
     const { user_id, type, message } = req.body;
     if (clients.has(user_id)) {
         clients.get(user_id).forEach(ws => {
-            if (ws.readyState === WebSocket.OPEN) ws.send(JSON.stringify({ type, message }));
+            if (ws.readyState === WebSocket.OPEN) {
+                ws.send(JSON.stringify({ type, message }));
+                // Ensure UI reloads on significant notifications
+                if (type === 'toast') ws.send(JSON.stringify({ type: 'reload_project' }));
+            }
         });
         return res.json({ success: true });
     }
@@ -104,7 +108,7 @@ app.get('/api/projects', async (req, res) => {
     });
 });
 
-// --- NEW: FETCH ACTIVE THRUST & TASKS ---
+// --- FETCH ACTIVE THRUST & TASKS (For backward compatibility) ---
 app.get('/api/projects/:projectId/thrusts/active', async (req, res) => {
     const authHeader = req.headers.authorization;
     if (!authHeader?.startsWith('Bearer ')) return res.status(401).json({ error: 'Unauthorized' });
@@ -115,7 +119,6 @@ app.get('/api/projects/:projectId/thrusts/active', async (req, res) => {
 
     const { projectId } = req.params;
 
-    // Fetch the thrust AND join its associated tasks
     const { data, error } = await supabase
         .from('thrusts')
         .select('*, thrust_tasks(*)')
@@ -125,12 +128,46 @@ app.get('/api/projects/:projectId/thrusts/active', async (req, res) => {
         .limit(1);
 
     if (error) return res.status(500).json({ error: error.message });
-
-    // Order tasks by creation ID/time if needed, but Supabase usually returns them in insert order.
     res.json(data);
 });
 
-// --- NEW: MARK TASK COMPLETE & LOG TO TIMELINE ---
+// --- NEW: COMBINED MCP CONTEXT FETCH ---
+app.get('/api/projects/:projectId/mcp-context', async (req, res) => {
+    const authHeader = req.headers.authorization;
+    if (!authHeader?.startsWith('Bearer ')) return res.status(401).json({ error: 'Unauthorized' });
+
+    const token = authHeader.split(' ')[1];
+    const decoded = await verifyThrustToken(token);
+    if (!decoded || !decoded.uid) return res.status(403).json({ error: 'Invalid Token' });
+
+    const { projectId } = req.params;
+    const { prd, thrust, timeline } = req.query;
+
+    let result = {};
+
+    try {
+        if (prd === 'true') {
+            const { data } = await supabase.from('leads').select('requirements_doc').eq('id', projectId).single();
+            result.prd = data?.requirements_doc || null;
+        }
+
+        if (thrust === 'true') {
+            const { data } = await supabase.from('thrusts').select('*, tasks:thrust_tasks(*)').eq('lead_id', projectId).eq('status', 'active').order('created_at', { ascending: false }).limit(1).single();
+            result.thrust = data || null;
+        }
+
+        if (timeline === 'true') {
+            const { data } = await supabase.from('timeline_events').select('*').eq('lead_id', projectId).order('created_at', { ascending: false }).limit(20);
+            result.timeline = data || [];
+        }
+
+        res.json(result);
+    } catch (error) {
+        res.status(500).json({ error: error.message });
+    }
+});
+
+// --- MARK TASK COMPLETE & LOG TO TIMELINE ---
 app.post('/api/projects/:projectId/tasks/:taskId/complete', async (req, res) => {
     const authHeader = req.headers.authorization;
     if (!authHeader?.startsWith('Bearer ')) return res.status(401).json({ error: 'Unauthorized' });
@@ -146,7 +183,7 @@ app.post('/api/projects/:projectId/tasks/:taskId/complete', async (req, res) =>
         // 1. Update Task Status
         const { error: updateError } = await supabase
             .from('thrust_tasks')
-            .update({ is_completed: true })
+            .update({ is_completed: true, status: 'done' })
             .eq('id', taskId);
 
         if (updateError) throw updateError;
@@ -163,6 +200,16 @@ app.post('/api/projects/:projectId/tasks/:taskId/complete', async (req, res) =>
 
         if (timelineError) throw timelineError;
 
+        // 3. Notify the local WebSockets to show a toast and reload!
+        if (clients.has(decoded.uid)) {
+            clients.get(decoded.uid).forEach(ws => {
+                if (ws.readyState === WebSocket.OPEN) {
+                    ws.send(JSON.stringify({ type: 'toast', message: `✅ Task Completed: ${taskTitle || 'a task'}` }));
+                    ws.send(JSON.stringify({ type: 'reload_project' }));
+                }
+            });
+        }
+
         res.json({ success: true });
     } catch (error) {
         console.error("Task Completion Error:", error.message);