Update server.js

server.js

@@ -32,8 +32,13 @@ let consoleLogs = [];
 let networkRequests = [];
 let blockedUrls = [];
 
+// --- LIVE STREAM STATE ---
+let latestFrameBuffer = null;
+let isScreencastRunning = false;
+let lastStreamRequestTime = 0;
+
 async function initBrowser() {
-    console.log("Initializing Ultimate God-Mode Browser...");
+    console.log("Initializing Ultimate God-Mode Browser with Live Stream...");
     browser = await puppeteer.launch({
         headless: true,
         args: [
@@ -48,7 +53,7 @@ async function initBrowser() {
     const initialPages = await browser.pages();
     if (initialPages.length > 1) await initialPages[0].close();
     
-    console.log("Browser successfully initialized with Advanced AI Engine!");
+    console.log("Browser successfully initialized with Live Stream Ready!");
 }
 
 async function setupNewPage(newPage) {
@@ -56,7 +61,6 @@ async function setupNewPage(newPage) {
     pages = await browser.pages();
     await page.setViewport({ width: 1920, height: 1080 });
     
-    // --- STEALTH MODE (Bypass Bot Detection) ---
     await page.evaluateOnNewDocument(() => {
         Object.defineProperty(navigator, 'webdriver', { get: () => undefined });
         window.navigator.chrome = { runtime: {} };
@@ -65,7 +69,7 @@ async function setupNewPage(newPage) {
     });
     await page.setUserAgent('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36');
     
-    // --- DEVTOOLS SETUP ---
+    // --- DEVTOOLS & SCREENCAST SETUP ---
     devToolsClient = await page.target().createCDPSession();
     consoleLogs = []; networkRequests = [];
     
@@ -78,8 +82,23 @@ async function setupNewPage(newPage) {
         else interceptedRequest.continue();
     });
     page.on('dialog', async dialog => { await dialog.accept(); });
+
+    // Handle Live Screencast Frames
+    devToolsClient.on('Page.screencastFrame', async (event) => {
+        latestFrameBuffer = Buffer.from(event.data, 'base64');
+        try { await devToolsClient.send('Page.screencastFrameAck', { sessionId: event.sessionId }); } catch(e) {}
+    });
 }
 
+// Auto-stop screencast if no viewers to save server memory
+setInterval(() => {
+    if (isScreencastRunning && Date.now() - lastStreamRequestTime > 5000) {
+        if (devToolsClient) devToolsClient.send('Page.stopScreencast').catch(()=>{});
+        isScreencastRunning = false;
+        console.log("Live stream paused (No active viewers).");
+    }
+}, 2000);
+
 // ==========================================
 // DEFINE 105 TOOLS SCHEMA
 // ==========================================
@@ -208,20 +227,18 @@ const toolSchemas = [
     { name: "emulate_color_scheme", description: "[F12] Dark/Light mode", inputSchema: { type: "object", properties: { theme: { type: "string", enum: ["dark", "light"] } }, required: ["theme"] } },
     { name: "clear_browser_cache", description: "[F12] Clear Cache", inputSchema: { type: "object", properties: {} } },
 
-    // ==========================================
-    // 🔥 ADVANCED AI ENGINE TOOLS (101-105) 🔥
-    // ==========================================
+    // --- ADVANCED AI ENGINE TOOLS (101-105) ---
     { name: "smart_scrape_markdown", description: "[ADVANCED ENGINE] Extract main article content and convert to Clean LLM-readable Markdown", inputSchema: { type: "object", properties: {} } },
     { name: "extract_structured_data", description: "[ADVANCED ENGINE] Extract structured data (Schema.org/JSON-LD)", inputSchema: { type: "object", properties: {} } },
-    { name: "auto_smart_scroll", description: "[ADVANCED ENGINE] Auto-scroll to load dynamic/lazy content before scraping", inputSchema: { type: "object", properties: {} } },
-    { name: "deep_domain_crawler", description: "[ADVANCED ENGINE] Extract all internal links of the same domain for deep crawling", inputSchema: { type: "object", properties: {} } },
-    { name: "clean_page_clutter", description: "[ADVANCED ENGINE] Force remove ads, popups, cookie banners from DOM", inputSchema: { type: "object", properties: {} } }
+    { name: "auto_smart_scroll", description: "[ADVANCED ENGINE] Auto-scroll to load dynamic content", inputSchema: { type: "object", properties: {} } },
+    { name: "deep_domain_crawler", description: "[ADVANCED ENGINE] Extract internal links", inputSchema: { type: "object", properties: {} } },
+    { name: "clean_page_clutter", description: "[ADVANCED ENGINE] Force remove ads, popups", inputSchema: { type: "object", properties: {} } }
 ];
 
 mcpServer.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: toolSchemas }));
 
 // ==========================================
-// TOOL EXECUTION HANDLERS (With Advanced AI Engine Logic)
+// TOOL EXECUTION HANDLERS
 // ==========================================
 const toolHandlers = {
     // Nav
@@ -353,9 +370,7 @@ const toolHandlers = {
     "emulate_color_scheme": async (a) => { await page.emulateMediaFeatures([{ name: 'prefers-color-scheme', value: a.theme }]); return "Theme set"; },
     "clear_browser_cache": async () => { await devToolsClient.send('Network.clearBrowserCache'); return "Cache cleared"; },
 
-    // ==========================================
-    // 🔥 ADVANCED AI ENGINE HANDLERS 🔥
-    // ==========================================
+    // AI Engine
     "smart_scrape_markdown": async () => {
         try {
             await page.addScriptTag({ url: 'https://unpkg.com/@mozilla/readability/Readability.js' });
@@ -387,7 +402,7 @@ const toolHandlers = {
                 }, 100);
             });
         });
-        return "Smart scrolling finished. Dynamic content loaded.";
+        return "Smart scrolling finished.";
     },
     "deep_domain_crawler": async () => {
         const urlObj = new URL(await page.url());
@@ -403,7 +418,7 @@ const toolHandlers = {
             const selectors = ['header', 'footer', 'nav', '.ad', '.ads', '#cookie-banner', '.popup', 'iframe'];
             selectors.forEach(sel => document.querySelectorAll(sel).forEach(el => el.remove()));
         });
-        return "Clutter removed from DOM.";
+        return "Clutter removed.";
     }
 };
 
@@ -416,10 +431,65 @@ mcpServer.setRequestHandler(CallToolRequestSchema, async (req) => {
 });
 
 // ==========================================
-// SSE SERVER CONNECTION LOGIC
+// EXPRESS ROUTES (INCLUDING LIVE STREAM)
 // ==========================================
 const transports = new Map();
 
+// Endpoint to fetch the raw image buffer for live streaming
+app.get('/latest-frame', async (req, res) => {
+    lastStreamRequestTime = Date.now();
+    if (!isScreencastRunning && devToolsClient) {
+        try {
+            await devToolsClient.send('Page.startScreencast', { format: 'jpeg', quality: 60, everyNthFrame: 1 });
+            isScreencastRunning = true;
+            console.log("Live stream started by viewer.");
+        } catch(e) {}
+    }
+    
+    if (latestFrameBuffer) {
+        res.set('Content-Type', 'image/jpeg');
+        res.send(latestFrameBuffer);
+    } else {
+        res.status(404).send('Waiting for AI to start working...');
+    }
+});
+
+// The actual HTML page you will view or embed
+app.get('/live', (req, res) => {
+    res.send(`
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>AI God-Mode Live Tracker</title>
+        <style>
+            body { margin: 0; background-color: #0f172a; display: flex; justify-content: center; align-items: center; height: 100vh; overflow: hidden; font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; }
+            .container { position: relative; width: 100%; height: 100%; display: flex; flex-direction: column; justify-content: center; align-items: center; background-image: radial-gradient(#1e293b 1px, transparent 1px); background-size: 20px 20px; }
+            img { max-width: 95%; max-height: 90%; object-fit: contain; border-radius: 12px; box-shadow: 0 10px 40px rgba(16, 185, 129, 0.2); border: 2px solid #10b981; transition: opacity 0.2s; }
+            .status-bar { position: absolute; top: 20px; left: 20px; color: #10b981; font-weight: bold; font-size: 15px; background: rgba(15, 23, 42, 0.8); padding: 8px 15px; border-radius: 8px; border: 1px solid #1e293b; display: flex; align-items: center; gap: 10px; backdrop-filter: blur(5px); }
+            .dot { width: 12px; height: 12px; background-color: #10b981; border-radius: 50%; animation: pulse 1.5s infinite; }
+            @keyframes pulse { 0% { transform: scale(0.95); box-shadow: 0 0 0 0 rgba(16, 185, 129, 0.7); } 70% { transform: scale(1); box-shadow: 0 0 0 10px rgba(16, 185, 129, 0); } 100% { transform: scale(0.95); box-shadow: 0 0 0 0 rgba(16, 185, 129, 0); } }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <div class="status-bar"><div class="dot"></div> LIVE AI TRACKING</div>
+            <img id="stream" src="" alt="Connecting to AI Browser..." />
+        </div>
+        <script>
+            const img = document.getElementById('stream');
+            // Fetch a fresh frame from the browser every 500ms
+            setInterval(() => {
+                img.src = '/latest-frame?t=' + new Date().getTime();
+            }, 500);
+        </script>
+    </body>
+    </html>
+    `);
+});
+
+// SSE MCP Endpoint
 app.get('/sse', async (req, res) => {
     try {
         globalProtocol = req.headers['x-forwarded-proto'] || req.protocol;
@@ -441,6 +511,6 @@ app.post('/messages', async (req, res) => {
 });
 
 app.listen(PORT, async () => {
-    console.log(`God-Mode MCP Server running on ${PORT}`);
+    console.log(`God-Mode MCP Server running on ${PORT} with Live Streaming enabled!`);
     await initBrowser();
 });