Update app.js

app.js

@@ -4,94 +4,58 @@ import { WebSocketServer, WebSocket } from 'ws';
 import cors from 'cors';
 import { createClient } from '@supabase/supabase-js';
 
-// --- CONFIG ---
 const PORT = 7860;
 const SUPABASE_URL = process.env.SUPABASE_URL;
 const SUPABASE_KEY = process.env.SUPABASE_SERVICE_KEY;
-const CORE_URL = process.env.CORE_URL; // e.g. https://my-core-server.hf.space
+const CORE_URL = process.env.CORE_URL; 
 
-if (!SUPABASE_URL) { console.error("❌ Missing Config"); process.exit(1); }
+if (!SUPABASE_URL) { console.error("❌ Config Missing"); process.exit(1); }
 
-// --- STATE ---
 const app = express();
 const server = createServer(app);
 const wss = new WebSocketServer({ noServer: true });
 const supabase = createClient(SUPABASE_URL, SUPABASE_KEY);
 
-// Active Connections: Map<UserId, Set<WebSocket>>
-const clients = new Map();
+const clients = new Map(); // UserId -> Set<WebSocket>
 
 app.use(cors());
-app.use(express.json());
+app.use(express.json({ limit: '50mb' }));
 
-// --- HTTP ENDPOINTS (Webhooks/Uploads) ---
+app.get('/', (req, res) => res.send('Gateway Active'));
 
-app.get('/', (req, res) => res.send('Front Gateway Active'));
-
-// Internal Endpoint for Core to push notifications to desktop
+// Internal Notification Webhook
 app.post('/internal/notify', (req, res) => {
-    const { user_id, type, message, payload } = req.body;
-    
+    const { user_id, type, message } = req.body;
     if (clients.has(user_id)) {
-        const userSockets = clients.get(user_id);
-        let count = 0;
-        userSockets.forEach(ws => {
-            if (ws.readyState === WebSocket.OPEN) {
-                ws.send(JSON.stringify({ type, message, payload }));
-                count++;
-            }
+        clients.get(user_id).forEach(ws => {
+            if (ws.readyState === WebSocket.OPEN) ws.send(JSON.stringify({ type, message }));
         });
-        return res.json({ success: true, delivered_to: count });
+        return res.json({ success: true });
     }
-    
-    res.json({ success: false, reason: "User offline" });
-});
-
-// Endpoint for File Uploads (Base64 is easier for single script than Multer)
-app.post('/upload', async (req, res) => {
-    // In a real app, handle streams. Here we assume small/medium files.
-    // Client sends this to avoid clogging WS.
-    res.json({ status: "ok", note: "File received (Placeholder logic)" });
+    res.json({ success: false });
 });
 
-// --- WEBSOCKET AUTH & UPGRADE ---
-
+// WS Upgrade
 server.on('upgrade', async (request, socket, head) => {
     const url = new URL(request.url, `http://${request.headers.host}`);
     const token = url.searchParams.get('token');
 
-    if (!token) {
-        socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
-        socket.destroy();
-        return;
-    }
+    if (!token) { socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n'); socket.destroy(); return; }
 
-    // Validate with Supabase
     const { data: { user }, error } = await supabase.auth.getUser(token);
-
-    if (error || !user) {
-        console.log("❌ WS Auth Failed");
-        socket.write('HTTP/1.1 403 Forbidden\r\n\r\n');
-        socket.destroy();
-        return;
-    }
+    if (error || !user) { 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, user);
     });
 });
 
-// --- WEBSOCKET LOGIC ---
-
+// WS Logic
 wss.on('connection', (ws, req, user) => {
     const userId = user.id;
-    console.log(`🔌 Connected: ${userId}`);
-
-    // Register
     if (!clients.has(userId)) clients.set(userId, new Set());
     clients.get(userId).add(ws);
 
-    // Heartbeat (Keep-Alive for HF Spaces)
     ws.isAlive = true;
     ws.on('pong', () => { ws.isAlive = true; });
 
@@ -99,15 +63,9 @@ wss.on('connection', (ws, req, user) => {
         try {
             const data = JSON.parse(message.toString());
 
-            // 1. ROUTING: PROMPT -> CORE
             if (data.type === 'prompt') {
                 ws.send(JSON.stringify({ type: 'status', status: 'thinking' }));
 
-                if (!CORE_URL) {
-                    ws.send(JSON.stringify({ type: 'error', message: "Core Server Config Missing" }));
-                    return;
-                }
-
                 // Call Core
                 const response = await fetch(`${CORE_URL}/process`, {
                     method: 'POST',
@@ -116,37 +74,34 @@ wss.on('connection', (ws, req, user) => {
                         userId: userId,
                         projectId: data.projectId,
                         prompt: data.content,
-                        context: data.context
+                        context: data.context,
+                        history: data.history
                     })
                 });
 
                 const result = await response.json();
 
-                // Send Result to Client
+                // Send to Client
                 ws.send(JSON.stringify({
                     type: 'response',
                     text: result.text,
-                    should_reload: result.should_reload
+                    should_reload: result.should_reload, // .NET app listens for this to refresh UI
+                    usage: result.usage
                 }));
             }
-
         } catch (e) {
-            console.error("WS Message Error", e);
-            ws.send(JSON.stringify({ type: 'error', message: "Server Error" }));
+            console.error("WS Error", e);
+            ws.send(JSON.stringify({ type: 'error', message: "Processing Error" }));
         }
     });
 
     ws.on('close', () => {
-        if (clients.has(userId)) {
-            clients.get(userId).delete(ws);
-            if (clients.get(userId).size === 0) clients.delete(userId);
-        }
-        console.log(`❌ Disconnected: ${userId}`);
+        if (clients.has(userId)) clients.get(userId).delete(ws);
     });
 });
 
-// Heartbeat Interval (30s)
-const interval = setInterval(() => {
+// Keep Alive
+setInterval(() => {
     wss.clients.forEach((ws) => {
         if (ws.isAlive === false) return ws.terminate();
         ws.isAlive = false;
@@ -154,8 +109,4 @@ const interval = setInterval(() => {
     });
 }, 30000);
 
-wss.on('close', () => clearInterval(interval));
-
-server.listen(PORT, () => {
-    console.log(`🚀 Front Gateway running on port ${PORT}`);
-});
+server.listen(PORT, () => console.log(`🚀 Gateway on ${PORT}`));