chrome-ai-bridge 1.0.9 → 1.0.10

package/build/src/tools/gemini-image.js

@@ -188,64 +188,71 @@ export const askGeminiImage = defineTool({
                 response.appendResponseLine('⚠️ 送信ボタンが見つかりません (Enterキーを試行)');
             }
             response.appendResponseLine('🎨 画像生成中... (1-2分かかることがあります)');
-            // Wait for image generation using MutationObserver for instant detection
+            // Wait for image generation using MutationObserver + polling hybrid approach
+            // MutationObserver provides instant detection, polling ensures we don't miss state
             const startTime = Date.now();
             const maxWaitTime = 180000; // 3 minutes
-            const imageFound = await page.evaluate((maxWait) => {
-                return new Promise(resolve => {
-                    // Check if image already exists (immediate return)
-                    const checkCompletion = () => {
-                        // Check for generated image
-                        const images = document.querySelectorAll('img[src*="blob:"], img[src*="generated"]');
-                        // Check for download button with various detection methods
-                        const buttons = Array.from(document.querySelectorAll('button, [role="menuitem"]'));
-                        const hasDownload = buttons.some(b => {
-                            const text = b.textContent || '';
-                            const ariaLabel = b.getAttribute('aria-label') || '';
-                            const describedBy = b.getAttribute('aria-describedby');
-                            let desc = '';
-                            if (describedBy) {
-                                const descEl = document.getElementById(describedBy);
-                                desc = descEl?.textContent || '';
-                            }
-                            return (text.includes('ダウンロード') ||
-                                text.includes('Download') ||
-                                text.includes('フルサイズ') ||
-                                ariaLabel.toLowerCase().includes('download') ||
-                                desc.includes('フルサイズ') ||
-                                desc.includes('ダウンロード'));
-                        });
-                        if (images.length > 0 || hasDownload) {
-                            return true;
-                        }
-                        return false;
-                    };
-                    // Initial check
-                    if (checkCompletion()) {
-                        resolve(true);
-                        return;
-                    }
-                    // Set up MutationObserver for instant detection
-                    const observer = new MutationObserver(() => {
-                        if (checkCompletion()) {
-                            observer.disconnect();
-                            clearTimeout(timeoutId);
-                            resolve(true);
+            // Set up MutationObserver in the page (stores result in window object)
+            await page.evaluate(() => {
+                // @ts-expect-error - window property
+                window.__geminiImageFound = false;
+                const checkCompletion = () => {
+                    const images = document.querySelectorAll('img[src*="blob:"], img[src*="generated"]');
+                    const buttons = Array.from(document.querySelectorAll('button, [role="menuitem"]'));
+                    const hasDownload = buttons.some(b => {
+                        const text = b.textContent || '';
+                        const ariaLabel = b.getAttribute('aria-label') || '';
+                        const describedBy = b.getAttribute('aria-describedby');
+                        let desc = '';
+                        if (describedBy) {
+                            const descEl = document.getElementById(describedBy);
+                            desc = descEl?.textContent || '';
                         }
+                        return (text.includes('ダウンロード') ||
+                            text.includes('Download') ||
+                            text.includes('フルサイズ') ||
+                            ariaLabel.toLowerCase().includes('download') ||
+                            desc.includes('フルサイズ') ||
+                            desc.includes('ダウンロード'));
                     });
-                    observer.observe(document.body, {
-                        childList: true,
-                        subtree: true,
-                        attributes: true,
-                        attributeFilter: ['src', 'aria-label', 'aria-describedby'],
-                    });
-                    // Timeout fallback
-                    const timeoutId = setTimeout(() => {
+                    return images.length > 0 || hasDownload;
+                };
+                // Initial check
+                if (checkCompletion()) {
+                    // @ts-expect-error - window property
+                    window.__geminiImageFound = true;
+                    return;
+                }
+                // Set up MutationObserver
+                const observer = new MutationObserver(() => {
+                    if (checkCompletion()) {
+                        // @ts-expect-error - window property
+                        window.__geminiImageFound = true;
                         observer.disconnect();
-                        resolve(false);
-                    }, maxWait);
+                    }
                 });
-            }, maxWaitTime);
+                observer.observe(document.body, {
+                    childList: true,
+                    subtree: true,
+                    attributes: true,
+                    attributeFilter: ['src', 'aria-label', 'aria-describedby'],
+                });
+            });
+            // Poll for the result (short intervals to minimize latency)
+            let imageFound = false;
+            while (Date.now() - startTime < maxWaitTime) {
+                // Check if MutationObserver detected image
+                const found = await page.evaluate(() => {
+                    // @ts-expect-error - window property
+                    return window.__geminiImageFound === true;
+                });
+                if (found) {
+                    imageFound = true;
+                    break;
+                }
+                // Short wait before next check (500ms for responsiveness)
+                await new Promise(resolve => setTimeout(resolve, 500));
+            }
             if (!imageFound) {
                 response.appendResponseLine('❌ 画像生成タイムアウト (3分)');
                 return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chrome-ai-bridge",
-  "version": "1.0.9",
+  "version": "1.0.10",
   "description": "MCP server bridging Chrome browser and AI assistants (ChatGPT, Gemini). Browser automation + AI consultation.",
   "type": "module",
   "bin": "./scripts/cli.mjs",