Update app.js

app.js

@@ -56,7 +56,7 @@ async function verifyThrustToken(token) {
     }
 }
 
-// --- NEW: FETCH PAGINATED PROJECTS VIA JWT ---
+// --- FETCH PAGINATED PROJECTS VIA JWT ---
 app.get('/api/projects', async (req, res) => {
     const authHeader = req.headers.authorization;
     if (!authHeader?.startsWith('Bearer ')) return res.status(401).json({ error: 'Unauthorized' });
@@ -89,6 +89,72 @@ app.get('/api/projects', async (req, res) => {
     });
 });
 
+// --- NEW: FETCH ACTIVE THRUST & TASKS ---
+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' });
+
+    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;
+
+    // Fetch the thrust AND join its associated tasks
+    const { data, error } = await supabase
+        .from('thrusts')
+        .select('*, thrust_tasks(*)')
+        .eq('lead_id', projectId)
+        .eq('status', 'active')
+        .order('created_at', { ascending: false })
+        .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 ---
+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' });
+
+    const token = authHeader.split(' ')[1];
+    const decoded = await verifyThrustToken(token);
+    if (!decoded || !decoded.uid) return res.status(403).json({ error: 'Invalid Token' });
+
+    const { projectId, taskId } = req.params;
+    const { taskTitle } = req.body;
+
+    try {
+        // 1. Update Task Status
+        const { error: updateError } = await supabase
+            .from('thrust_tasks')
+            .update({ is_completed: true })
+            .eq('id', taskId);
+
+        if (updateError) throw updateError;
+
+        // 2. Add to Timeline so AI sees it on next context sync
+        const { error: timelineError } = await supabase
+            .from('timeline_events')
+            .insert({
+                lead_id: projectId,
+                title: "Task Completed",
+                description: `User manually completed: ${taskTitle || 'a task'}`,
+                type: "chore"
+            });
+
+        if (timelineError) throw timelineError;
+
+        res.json({ success: true });
+    } catch (error) {
+        console.error("Task Completion Error:", error.message);
+        res.status(500).json({ error: error.message });
+    }
+});
+
 // WS Upgrade
 server.on('upgrade', async (request, socket, head) => {
     const url = new URL(request.url, `http://${request.headers.host}`);
@@ -101,17 +167,12 @@ server.on('upgrade', async (request, socket, head) => {
     if (!decodedData) { socket.write('HTTP/1.1 403 Forbidden\r\n\r\n'); socket.destroy(); return; }
   
     wss.handleUpgrade(request, socket, head, (ws) => {
-        wss.emit('connection', ws, request,  decodedData.uid);// decodedData);
+        wss.emit('connection', ws, request,  decodedData.uid);
     });
 });
 
 // WS Logic
-
-
-
-wss.on('connection', (ws, req, user) => {
-    const userId = user.uid; 
-    
+wss.on('connection', (ws, req, userId) => {
     if (!clients.has(userId)) clients.set(userId, new Set());
     clients.get(userId).add(ws);
 
@@ -122,7 +183,7 @@ wss.on('connection', (ws, req, user) => {
         try {
             const data = JSON.parse(message.toString());
 
-            // 1. CHAT PROMPTS
+            // 1. CHAT PROMPTS & OVERRIDES
             if (data.type === 'prompt') {
                 ws.send(JSON.stringify({ type: 'status', status: 'thinking' }));
                 const response = await fetch(`${CORE_URL}/process`, {
@@ -163,8 +224,6 @@ wss.on('connection', (ws, req, user) => {
     });
 });
 
-
-
 setInterval(() => {
     wss.clients.forEach((ws) => {
         if (ws.isAlive === false) return ws.terminate();
@@ -173,5 +232,4 @@ setInterval(() => {
     });
 }, 30000);
 
-server.listen(PORT, () => console.log(`🚀 Gateway on ${PORT}`));
-
+server.listen(PORT, () => console.log(`🚀 Gateway on ${PORT}`));