sentienceapi 0.90.16__py3-none-any.whl → 0.98.0__py3-none-any.whl

sentience/extension/content.js

@@ -1,298 +1,161 @@
-// content.js - ISOLATED WORLD (Bridge between Main World and Background)
-console.log('[Sentience Bridge] Loaded.');
-
-// Detect if we're in a child frame (for iframe support)
-const isChildFrame = window !== window.top;
-if (isChildFrame) {
-    console.log('[Sentience Bridge] Running in child frame:', window.location.href);
-}
-
-// 1. Pass Extension ID to Main World (So API knows where to find resources)
-document.documentElement.dataset.sentienceExtensionId = chrome.runtime.id;
-
-// 2. Message Router - Handles all communication between page and background
-window.addEventListener('message', (event) => {
-    // Security check: only accept messages from same window
-    if (event.source !== window) return;
-
-    // Route different message types
-    switch (event.data.type) {
-        case 'SENTIENCE_SCREENSHOT_REQUEST':
-            handleScreenshotRequest(event.data);
-            break;
-
-        case 'SENTIENCE_SNAPSHOT_REQUEST':
-            handleSnapshotRequest(event.data);
+!function() {
+    "use strict";
+    window, window.top;
+    document.documentElement.dataset.sentienceExtensionId = chrome.runtime.id, window.addEventListener("message", event => {
+        var data;
+        if (event.source === window) switch (event.data.type) {
+          case "SENTIENCE_SCREENSHOT_REQUEST":
+            data = event.data, chrome.runtime.sendMessage({
+                action: "captureScreenshot",
+                options: data.options
+            }, response => {
+                window.postMessage({
+                    type: "SENTIENCE_SCREENSHOT_RESULT",
+                    requestId: data.requestId,
+                    screenshot: response?.success ? response.screenshot : null,
+                    error: response?.error
+                }, "*");
+            });
             break;
 
-        case 'SENTIENCE_SHOW_OVERLAY':
-            handleShowOverlay(event.data);
+          case "SENTIENCE_SNAPSHOT_REQUEST":
+            !function(data) {
+                const startTime = performance.now();
+                let responded = !1;
+                const timeoutId = setTimeout(() => {
+                    if (!responded) {
+                        responded = !0;
+                        const duration = performance.now() - startTime;
+                        window.postMessage({
+                            type: "SENTIENCE_SNAPSHOT_RESULT",
+                            requestId: data.requestId,
+                            error: "WASM processing timeout - background script may be unresponsive",
+                            duration: duration
+                        }, "*");
+                    }
+                }, 2e4);
+                try {
+                    chrome.runtime.sendMessage({
+                        action: "processSnapshot",
+                        rawData: data.rawData,
+                        options: data.options
+                    }, response => {
+                        if (responded) return;
+                        responded = !0, clearTimeout(timeoutId);
+                        const duration = performance.now() - startTime;
+                        chrome.runtime.lastError ? window.postMessage({
+                            type: "SENTIENCE_SNAPSHOT_RESULT",
+                            requestId: data.requestId,
+                            error: `Chrome runtime error: ${chrome.runtime.lastError.message}`,
+                            duration: duration
+                        }, "*") : response?.success ? window.postMessage({
+                            type: "SENTIENCE_SNAPSHOT_RESULT",
+                            requestId: data.requestId,
+                            elements: response.result.elements,
+                            raw_elements: response.result.raw_elements,
+                            duration: duration
+                        }, "*") : window.postMessage({
+                            type: "SENTIENCE_SNAPSHOT_RESULT",
+                            requestId: data.requestId,
+                            error: response?.error || "Processing failed",
+                            duration: duration
+                        }, "*");
+                    });
+                } catch (error) {
+                    if (!responded) {
+                        responded = !0, clearTimeout(timeoutId);
+                        const duration = performance.now() - startTime;
+                        window.postMessage({
+                            type: "SENTIENCE_SNAPSHOT_RESULT",
+                            requestId: data.requestId,
+                            error: `Failed to send message: ${error.message}`,
+                            duration: duration
+                        }, "*");
+                    }
+                }
+            }(event.data);
             break;
 
-        case 'SENTIENCE_CLEAR_OVERLAY':
-            handleClearOverlay();
+          case "SENTIENCE_SHOW_OVERLAY":
+            !function(data) {
+                const {elements: elements, targetElementId: targetElementId} = data;
+                if (!elements || !Array.isArray(elements)) return;
+                removeOverlay();
+                const host = document.createElement("div");
+                host.id = OVERLAY_HOST_ID, host.style.cssText = "\n        position: fixed !important;\n        top: 0 !important;\n        left: 0 !important;\n        width: 100vw !important;\n        height: 100vh !important;\n        pointer-events: none !important;\n        z-index: 2147483647 !important;\n        margin: 0 !important;\n        padding: 0 !important;\n    ", 
+                document.body.appendChild(host);
+                const shadow = host.attachShadow({
+                    mode: "closed"
+                }), maxImportance = Math.max(...elements.map(e => e.importance || 0), 1);
+                elements.forEach(element => {
+                    const bbox = element.bbox;
+                    if (!bbox) return;
+                    const isTarget = element.id === targetElementId, isPrimary = element.visual_cues?.is_primary || !1, importance = element.importance || 0;
+                    let color;
+                    color = isTarget ? "#FF0000" : isPrimary ? "#0066FF" : "#00FF00";
+                    const importanceRatio = maxImportance > 0 ? importance / maxImportance : .5, borderOpacity = isTarget ? 1 : isPrimary ? .9 : Math.max(.4, .5 + .5 * importanceRatio), fillOpacity = .2 * borderOpacity, borderWidth = isTarget ? 2 : isPrimary ? 1.5 : Math.max(.5, Math.round(2 * importanceRatio)), hexOpacity = Math.round(255 * fillOpacity).toString(16).padStart(2, "0"), box = document.createElement("div");
+                    if (box.style.cssText = `\n            position: absolute;\n            left: ${bbox.x}px;\n            top: ${bbox.y}px;\n            width: ${bbox.width}px;\n            height: ${bbox.height}px;\n            border: ${borderWidth}px solid ${color};\n            background-color: ${color}${hexOpacity};\n            box-sizing: border-box;\n            opacity: ${borderOpacity};\n            pointer-events: none;\n        `, 
+                    importance > 0 || isPrimary) {
+                        const badge = document.createElement("span");
+                        badge.textContent = isPrimary ? `⭐${importance}` : `${importance}`, badge.style.cssText = `\n                position: absolute;\n                top: -18px;\n                left: 0;\n                background: ${color};\n                color: white;\n                font-size: 11px;\n                font-weight: bold;\n                padding: 2px 6px;\n                font-family: Arial, sans-serif;\n                border-radius: 3px;\n                opacity: 0.95;\n                white-space: nowrap;\n                pointer-events: none;\n            `, 
+                        box.appendChild(badge);
+                    }
+                    if (isTarget) {
+                        const targetIndicator = document.createElement("span");
+                        targetIndicator.textContent = "🎯", targetIndicator.style.cssText = "\n                position: absolute;\n                top: -18px;\n                right: 0;\n                font-size: 16px;\n                pointer-events: none;\n            ", 
+                        box.appendChild(targetIndicator);
+                    }
+                    shadow.appendChild(box);
+                }), overlayTimeout = setTimeout(() => {
+                    removeOverlay();
+                }, 5e3);
+            }(event.data);
             break;
 
-        default:
-            // Ignore unknown message types
+          case "SENTIENCE_CLEAR_OVERLAY":
+            removeOverlay();
             break;
-    }
-});
-
-/**
- * Handle screenshot requests (existing functionality)
- */
-function handleScreenshotRequest(data) {
-    chrome.runtime.sendMessage(
-        { action: 'captureScreenshot', options: data.options },
-        (response) => {
-            window.postMessage({
-                type: 'SENTIENCE_SCREENSHOT_RESULT',
-                requestId: data.requestId,
-                screenshot: response?.success ? response.screenshot : null,
-                error: response?.error
-            }, '*');
-        }
-    );
-}
-
-/**
- * Handle snapshot processing requests (NEW!)
- * Sends raw DOM data to background worker for WASM processing
- * Includes timeout protection to prevent extension crashes
- */
-function handleSnapshotRequest(data) {
-    const startTime = performance.now();
-    const TIMEOUT_MS = 20000; // 20 seconds (longer than injected_api timeout)
-    let responded = false;
-
-    // Timeout protection: if background doesn't respond, send error
-    const timeoutId = setTimeout(() => {
-        if (!responded) {
-            responded = true;
-            const duration = performance.now() - startTime;
-            console.error(`[Sentience Bridge] ⚠️ WASM processing timeout after ${duration.toFixed(1)}ms`);
-            window.postMessage({
-                type: 'SENTIENCE_SNAPSHOT_RESULT',
-                requestId: data.requestId,
-                error: 'WASM processing timeout - background script may be unresponsive',
-                duration: duration
-            }, '*');
-        }
-    }, TIMEOUT_MS);
-
-    try {
-        chrome.runtime.sendMessage(
-            {
-                action: 'processSnapshot',
-                rawData: data.rawData,
-                options: data.options
-            },
-            (response) => {
-                if (responded) return; // Already responded via timeout
-                responded = true;
-                clearTimeout(timeoutId);
-
-                const duration = performance.now() - startTime;
-
-                // Handle Chrome extension errors (e.g., background script crashed)
-                if (chrome.runtime.lastError) {
-                    console.error('[Sentience Bridge] Chrome runtime error:', chrome.runtime.lastError.message);
-                    window.postMessage({
-                        type: 'SENTIENCE_SNAPSHOT_RESULT',
-                        requestId: data.requestId,
-                        error: `Chrome runtime error: ${chrome.runtime.lastError.message}`,
-                        duration: duration
-                    }, '*');
-                    return;
-                }
-
-                if (response?.success) {
-                    console.log(`[Sentience Bridge] ✓ WASM processing complete in ${duration.toFixed(1)}ms`);
-                    window.postMessage({
-                        type: 'SENTIENCE_SNAPSHOT_RESULT',
-                        requestId: data.requestId,
-                        elements: response.result.elements,
-                        raw_elements: response.result.raw_elements,
-                        duration: duration
-                    }, '*');
-                } else {
-                    console.error('[Sentience Bridge] WASM processing failed:', response?.error);
-                    window.postMessage({
-                        type: 'SENTIENCE_SNAPSHOT_RESULT',
-                        requestId: data.requestId,
-                        error: response?.error || 'Processing failed',
-                        duration: duration
-                    }, '*');
-                }
-            }
-        );
-    } catch (error) {
-        if (!responded) {
-            responded = true;
-            clearTimeout(timeoutId);
-            const duration = performance.now() - startTime;
-            console.error('[Sentience Bridge] Exception sending message:', error);
-            window.postMessage({
-                type: 'SENTIENCE_SNAPSHOT_RESULT',
-                requestId: data.requestId,
-                error: `Failed to send message: ${error.message}`,
-                duration: duration
-            }, '*');
-        }
-    }
-}
-
-// ============================================================================
-// Visual Overlay - Shadow DOM Implementation
-// ============================================================================
 
-const OVERLAY_HOST_ID = 'sentience-overlay-host';
-let overlayTimeout = null;
-
-/**
- * Show visual overlay highlighting elements using Shadow DOM
- * @param {Object} data - Message data with elements and targetElementId
- */
-function handleShowOverlay(data) {
-    const { elements, targetElementId } = data;
-
-    if (!elements || !Array.isArray(elements)) {
-        console.warn('[Sentience Bridge] showOverlay: elements must be an array');
-        return;
-    }
-
-    removeOverlay();
-
-    // Create host with Shadow DOM for CSS isolation
-    const host = document.createElement('div');
-    host.id = OVERLAY_HOST_ID;
-    host.style.cssText = `
-        position: fixed !important;
-        top: 0 !important;
-        left: 0 !important;
-        width: 100vw !important;
-        height: 100vh !important;
-        pointer-events: none !important;
-        z-index: 2147483647 !important;
-        margin: 0 !important;
-        padding: 0 !important;
-    `;
-    document.body.appendChild(host);
-
-    // Attach shadow root (closed mode for security and CSS isolation)
-    const shadow = host.attachShadow({ mode: 'closed' });
-
-    // Calculate max importance for scaling
-    const maxImportance = Math.max(...elements.map(e => e.importance || 0), 1);
-
-    elements.forEach((element) => {
-        const bbox = element.bbox;
-        if (!bbox) return;
-
-        const isTarget = element.id === targetElementId;
-        const isPrimary = element.visual_cues?.is_primary || false;
-        const importance = element.importance || 0;
-
-        // Color: Red (target), Blue (primary), Green (regular)
-        let color;
-        if (isTarget) color = '#FF0000';
-        else if (isPrimary) color = '#0066FF';
-        else color = '#00FF00';
-
-        // Scale opacity and border width based on importance
-        const importanceRatio = maxImportance > 0 ? importance / maxImportance : 0.5;
-        const borderOpacity = isTarget ? 1.0 : (isPrimary ? 0.9 : Math.max(0.4, 0.5 + importanceRatio * 0.5));
-        const fillOpacity = borderOpacity * 0.2;
-        const borderWidth = isTarget ? 2 : (isPrimary ? 1.5 : Math.max(0.5, Math.round(importanceRatio * 2)));
-
-        // Convert fill opacity to hex for background-color
-        const hexOpacity = Math.round(fillOpacity * 255).toString(16).padStart(2, '0');
-
-        // Create box with semi-transparent fill
-        const box = document.createElement('div');
-        box.style.cssText = `
-            position: absolute;
-            left: ${bbox.x}px;
-            top: ${bbox.y}px;
-            width: ${bbox.width}px;
-            height: ${bbox.height}px;
-            border: ${borderWidth}px solid ${color};
-            background-color: ${color}${hexOpacity};
-            box-sizing: border-box;
-            opacity: ${borderOpacity};
-            pointer-events: none;
-        `;
-
-        // Add badge showing importance score
-        if (importance > 0 || isPrimary) {
-            const badge = document.createElement('span');
-            badge.textContent = isPrimary ? `⭐${importance}` : `${importance}`;
-            badge.style.cssText = `
-                position: absolute;
-                top: -18px;
-                left: 0;
-                background: ${color};
-                color: white;
-                font-size: 11px;
-                font-weight: bold;
-                padding: 2px 6px;
-                font-family: Arial, sans-serif;
-                border-radius: 3px;
-                opacity: 0.95;
-                white-space: nowrap;
-                pointer-events: none;
-            `;
-            box.appendChild(badge);
-        }
-
-        // Add target emoji for target element
-        if (isTarget) {
-            const targetIndicator = document.createElement('span');
-            targetIndicator.textContent = '🎯';
-            targetIndicator.style.cssText = `
-                position: absolute;
-                top: -18px;
-                right: 0;
-                font-size: 16px;
-                pointer-events: none;
-            `;
-            box.appendChild(targetIndicator);
+          case "SENTIENCE_SHOW_GRID_OVERLAY":
+            !function(data) {
+                const {grids: grids, targetGridId: targetGridId} = data;
+                if (!grids || !Array.isArray(grids)) return;
+                removeOverlay();
+                const host = document.createElement("div");
+                host.id = OVERLAY_HOST_ID, host.style.cssText = "\n        position: fixed !important;\n        top: 0 !important;\n        left: 0 !important;\n        width: 100vw !important;\n        height: 100vh !important;\n        pointer-events: none !important;\n        z-index: 2147483647 !important;\n        margin: 0 !important;\n        padding: 0 !important;\n    ", 
+                document.body.appendChild(host);
+                const shadow = host.attachShadow({
+                    mode: "closed"
+                });
+                grids.forEach(grid => {
+                    const bbox = grid.bbox;
+                    if (!bbox) return;
+                    const isTarget = grid.grid_id === targetGridId, isDominant = !0 === grid.is_dominant;
+                    let color = "#9B59B6";
+                    isTarget ? color = "#FF0000" : isDominant && (color = "#FF8C00");
+                    const borderStyle = isTarget ? "solid" : "dashed", borderWidth = isTarget ? 3 : isDominant ? 2.5 : 2, opacity = isTarget ? 1 : isDominant ? .9 : .8, fillOpacity = .1 * opacity, hexOpacity = Math.round(255 * fillOpacity).toString(16).padStart(2, "0"), box = document.createElement("div");
+                    box.style.cssText = `\n            position: absolute;\n            left: ${bbox.x}px;\n            top: ${bbox.y}px;\n            width: ${bbox.width}px;\n            height: ${bbox.height}px;\n            border: ${borderWidth}px ${borderStyle} ${color};\n            background-color: ${color}${hexOpacity};\n            box-sizing: border-box;\n            opacity: ${opacity};\n            pointer-events: none;\n        `;
+                    let labelText = grid.label ? `Grid ${grid.grid_id}: ${grid.label}` : `Grid ${grid.grid_id}`;
+                    grid.is_dominant && (labelText = `⭐ ${labelText} (dominant)`);
+                    const badge = document.createElement("span");
+                    if (badge.textContent = labelText, badge.style.cssText = `\n                position: absolute;\n                top: -18px;\n                left: 0;\n                background: ${color};\n                color: white;\n                font-size: 11px;\n                font-weight: bold;\n                padding: 2px 6px;\n                font-family: Arial, sans-serif;\n                border-radius: 3px;\n                opacity: 0.95;\n                white-space: nowrap;\n                pointer-events: none;\n            `, 
+                    box.appendChild(badge), isTarget) {
+                        const targetIndicator = document.createElement("span");
+                        targetIndicator.textContent = "🎯", targetIndicator.style.cssText = "\n                position: absolute;\n                top: -18px;\n                right: 0;\n                font-size: 16px;\n                pointer-events: none;\n            ", 
+                        box.appendChild(targetIndicator);
+                    }
+                    shadow.appendChild(box);
+                }), overlayTimeout = setTimeout(() => {
+                    removeOverlay();
+                }, 5e3);
+            }(event.data);
         }
-
-        shadow.appendChild(box);
     });
-
-    console.log(`[Sentience Bridge] Overlay shown for ${elements.length} elements`);
-
-    // Auto-remove after 5 seconds
-    overlayTimeout = setTimeout(() => {
-        removeOverlay();
-        console.log('[Sentience Bridge] Overlay auto-cleared after 5 seconds');
-    }, 5000);
-}
-
-/**
- * Clear overlay manually
- */
-function handleClearOverlay() {
-    removeOverlay();
-    console.log('[Sentience Bridge] Overlay cleared manually');
-}
-
-/**
- * Remove overlay from DOM
- */
-function removeOverlay() {
-    const existing = document.getElementById(OVERLAY_HOST_ID);
-    if (existing) {
-        existing.remove();
+    const OVERLAY_HOST_ID = "sentience-overlay-host";
+    let overlayTimeout = null;
+    function removeOverlay() {
+        const existing = document.getElementById(OVERLAY_HOST_ID);
+        existing && existing.remove(), overlayTimeout && (clearTimeout(overlayTimeout), 
+        overlayTimeout = null);
     }
-
-    if (overlayTimeout) {
-        clearTimeout(overlayTimeout);
-        overlayTimeout = null;
-    }
-}
-
-// console.log('[Sentience Bridge] Ready - Extension ID:', chrome.runtime.id);
+}();