chrome-ai-bridge 1.0.4 → 1.0.5

package/build/node_modules/chrome-devtools-frontend/front_end/models/annotations/AnnotationRepository.js

@@ -0,0 +1,163 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+import * as Common from '../../core/common/common.js';
+import * as GreenDev from '../greendev/greendev.js';
+import { AnnotationType } from './AnnotationType.js';
+export class AnnotationRepository {
+    static #instance = null;
+    static #hasRepliedGreenDevDisabled = false;
+    static #hasShownFlagWarning = false;
+    #events = new Common.ObjectWrapper.ObjectWrapper();
+    #annotationData = [];
+    #nextId = 0;
+    static instance() {
+        if (!AnnotationRepository.#instance) {
+            AnnotationRepository.#instance = new AnnotationRepository();
+        }
+        return AnnotationRepository.#instance;
+    }
+    static annotationsEnabled() {
+        const enabled = GreenDev.Prototypes.instance().isEnabled('aiAnnotations');
+        // TODO(finnur): Fix race when Repository is created before feature flags have been set properly.
+        if (!enabled) {
+            this.#hasRepliedGreenDevDisabled = true;
+        }
+        else if (this.#hasRepliedGreenDevDisabled && !this.#hasShownFlagWarning) {
+            console.warn('Flag controlling GreenDev has flipped from false to true. ' +
+                'Only some callers will expect GreenDev to be enabled, which can lead to unexpected results.');
+            this.#hasShownFlagWarning = true;
+        }
+        return Boolean(enabled);
+    }
+    addEventListener(eventType, listener, thisObject) {
+        if (!AnnotationRepository.annotationsEnabled()) {
+            console.warn('Received request to add event listener with annotations disabled');
+        }
+        return this.#events.addEventListener(eventType, listener, thisObject);
+    }
+    getAnnotationDataByType(type) {
+        if (!AnnotationRepository.annotationsEnabled()) {
+            console.warn('Received query for annotation types with annotations disabled');
+            return [];
+        }
+        const annotations = this.#annotationData.filter(annotation => annotation.type === type);
+        return annotations;
+    }
+    getAnnotationDataById(id) {
+        if (!AnnotationRepository.annotationsEnabled()) {
+            console.warn('Received query for annotation type with annotations disabled');
+            return undefined;
+        }
+        return this.#annotationData.find(annotation => annotation.id === id);
+    }
+    #getExistingAnnotation(type, anchor) {
+        const annotations = this.getAnnotationDataByType(type);
+        const annotation = annotations.find(annotation => {
+            if (typeof anchor === 'string') {
+                return annotation.lookupId === anchor;
+            }
+            switch (type) {
+                case AnnotationType.ELEMENT_NODE: {
+                    const elementAnnotation = annotation;
+                    return elementAnnotation.anchor === anchor;
+                }
+                case AnnotationType.NETWORK_REQUEST_SUBPANEL_HEADERS: {
+                    const networkRequestDetailsAnnotation = annotation;
+                    return networkRequestDetailsAnnotation.anchor === anchor;
+                }
+                default:
+                    console.warn('[AnnotationRepository] Unknown AnnotationType', type);
+                    return false;
+            }
+        });
+        return annotation;
+    }
+    #updateExistingAnnotationLabel(label, type, anchor) {
+        const annotation = this.#getExistingAnnotation(type, anchor);
+        if (annotation) {
+            // TODO(finnur): This should work for annotations that have not been displayed yet,
+            // but we need to also notify the AnnotationManager for those that have been shown.
+            annotation.message = label;
+            return true;
+        }
+        return false;
+    }
+    addElementsAnnotation(label, anchor, anchorToString) {
+        if (!AnnotationRepository.annotationsEnabled()) {
+            console.warn('Received annotation registration with annotations disabled');
+            return;
+        }
+        if (this.#updateExistingAnnotationLabel(label, AnnotationType.ELEMENT_NODE, anchor)) {
+            return;
+        }
+        const annotationData = {
+            id: this.#nextId++,
+            type: AnnotationType.ELEMENT_NODE,
+            message: label,
+            lookupId: typeof anchor === 'string' ? anchor : '',
+            anchor: typeof anchor !== 'string' ? anchor : undefined,
+            anchorToString,
+        };
+        this.#annotationData.push(annotationData);
+        // eslint-disable-next-line no-console
+        console.log('[AnnotationRepository] Added element annotation:', label, {
+            annotationData,
+            annotations: this.#annotationData.length,
+        });
+        this.#events.dispatchEventToListeners("AnnotationAdded" /* Events.ANNOTATION_ADDED */, annotationData);
+    }
+    addNetworkRequestAnnotation(label, anchor, anchorToString) {
+        if (!AnnotationRepository.annotationsEnabled()) {
+            console.warn('Received annotation registration with annotations disabled');
+            return;
+        }
+        // We only need to update the NETWORK_REQUEST_SUBPANEL_HEADERS because the
+        // NETWORK_REQUEST Annotation has no meaningful label.
+        if (this.#updateExistingAnnotationLabel(label, AnnotationType.NETWORK_REQUEST_SUBPANEL_HEADERS, anchor)) {
+            return;
+        }
+        const annotationData = {
+            id: this.#nextId++,
+            type: AnnotationType.NETWORK_REQUEST,
+            message: '',
+            lookupId: typeof anchor === 'string' ? anchor : '',
+            anchor: typeof anchor !== 'string' ? anchor : undefined,
+            anchorToString,
+        };
+        this.#annotationData.push(annotationData);
+        // eslint-disable-next-line no-console
+        console.log('[AnnotationRepository] Added annotation:', label, {
+            annotationData,
+            annotations: this.#annotationData.length,
+        });
+        this.#events.dispatchEventToListeners("AnnotationAdded" /* Events.ANNOTATION_ADDED */, annotationData);
+        const annotationDetailsData = {
+            id: this.#nextId++,
+            type: AnnotationType.NETWORK_REQUEST_SUBPANEL_HEADERS,
+            message: label,
+            lookupId: typeof anchor === 'string' ? anchor : '',
+            anchor: typeof anchor !== 'string' ? anchor : undefined,
+            anchorToString,
+        };
+        this.#annotationData.push(annotationDetailsData);
+        this.#events.dispatchEventToListeners("AnnotationAdded" /* Events.ANNOTATION_ADDED */, annotationDetailsData);
+    }
+    deleteAllAnnotations() {
+        this.#annotationData = [];
+        this.#events.dispatchEventToListeners("AllAnnotationsDeleted" /* Events.ALL_ANNOTATIONS_DELETED */);
+        // eslint-disable-next-line no-console
+        console.log('[AnnotationRepository] Deleting all annotations');
+    }
+    deleteAnnotation(id) {
+        const index = this.#annotationData.findIndex(annotation => annotation.id === id);
+        if (index === -1) {
+            console.warn(`[AnnotationRepository] Could not find annotation with id ${id}`);
+            return;
+        }
+        this.#annotationData.splice(index, 1);
+        this.#events.dispatchEventToListeners("AnnotationDeleted" /* Events.ANNOTATION_DELETED */, { id });
+        // eslint-disable-next-line no-console
+        console.log(`[AnnotationRepository] Deleted annotation with id ${id}`);
+    }
+}

package/build/node_modules/chrome-devtools-frontend/front_end/models/annotations/AnnotationType.js

@@ -0,0 +1,10 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+export var AnnotationType;
+(function (AnnotationType) {
+    AnnotationType[AnnotationType["ELEMENT_NODE"] = 0] = "ELEMENT_NODE";
+    AnnotationType[AnnotationType["STYLE_RULE"] = 1] = "STYLE_RULE";
+    AnnotationType[AnnotationType["NETWORK_REQUEST"] = 2] = "NETWORK_REQUEST";
+    AnnotationType[AnnotationType["NETWORK_REQUEST_SUBPANEL_HEADERS"] = 3] = "NETWORK_REQUEST_SUBPANEL_HEADERS";
+})(AnnotationType || (AnnotationType = {}));

package/build/node_modules/chrome-devtools-frontend/front_end/models/annotations/annotations.js

@@ -0,0 +1,5 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+export * from './AnnotationRepository.js';
+export * from './AnnotationType.js';

package/build/node_modules/chrome-devtools-frontend/front_end/models/greendev/Prototypes.js

@@ -0,0 +1,33 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+import * as Common from '../../core/common/common.js';
+import * as Root from '../../core/root/root.js';
+let instance = null;
+export class Prototypes {
+    constructor() {
+    }
+    static instance() {
+        if (instance) {
+            return instance;
+        }
+        instance = new Prototypes();
+        return instance;
+    }
+    /**
+     * Returns true if the specific setting is turned on AND the GreenDev flag is enabled
+     */
+    isEnabled(setting) {
+        const greendevFlagEnabled = Boolean(Root.Runtime.hostConfig.devToolsGreenDevUi?.enabled);
+        return greendevFlagEnabled && this.settings()[setting].get();
+    }
+    settings() {
+        const settings = Common.Settings.Settings.instance();
+        const inDevToolsFloaty = settings.createSetting('greendev-in-devtools-floaty-enabled', false, "Local" /* Common.Settings.SettingStorageType.LOCAL */);
+        const inlineWidgets = settings.createSetting('greendev-inline-widgets-enabled', false, "Local" /* Common.Settings.SettingStorageType.LOCAL */);
+        const aiAnnotations = settings.createSetting('greendev-ai-annotations-enabled', false, "Local" /* Common.Settings.SettingStorageType.LOCAL */);
+        const artifactViewer = settings.createSetting('greendev-artifact-viewer-enabled', false, "Local" /* Common.Settings.SettingStorageType.LOCAL */);
+        const copyToGemini = settings.createSetting('greendev-copy-to-gemini-enabled', false, "Local" /* Common.Settings.SettingStorageType.LOCAL */);
+        return { inDevToolsFloaty, inlineWidgets, aiAnnotations, artifactViewer, copyToGemini };
+    }
+}

package/build/node_modules/chrome-devtools-frontend/front_end/models/greendev/greendev.js

@@ -0,0 +1,4 @@
+// Copyright 2025 The Chromium Authors
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+export * from './Prototypes.js';

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

@@ -304,6 +304,10 @@ export const askGeminiWeb = defineTool({
                 return;
             }
             await new Promise(resolve => setTimeout(resolve, 500));
+            // 質問送信前に model-response 要素数を記録（ChatGPTと同じカウント方式）
+            const initialModelResponseCount = await page.evaluate(() => {
+                return document.querySelectorAll('model-response').length;
+            });
             // Click send button - look for "プロンプトを送信" or similar
             const sent = await page.evaluate(() => {
                 const buttons = Array.from(document.querySelectorAll('button'));
@@ -405,9 +409,17 @@ export const askGeminiWeb = defineTool({
                     });
                     return !!stopButton;
                 });
-                // Stop button/icon disappeared = generation complete
+                // Stop button/icon disappeared = check if new message appeared
                 if (!hasStopIndicator) {
-                    break;
+                    // 追加: model-response 要素数が増えたか確認（ChatGPTと同じ方式）
+                    const currentModelResponseCount = await page.evaluate(() => {
+                        return document.querySelectorAll('model-response').length;
+                    });
+                    if (currentModelResponseCount > initialModelResponseCount) {
+                        // ストップボタン消滅 AND 新規メッセージ出現で完了
+                        break;
+                    }
+                    // メッセージ数が増えていなければ、まだ待機続行
                 }
                 if (Date.now() - startTime > 180000) {
                     // 3 mins timeout
@@ -415,19 +427,24 @@ export const askGeminiWeb = defineTool({
                     break;
                 }
             }
-            // Get the final response content
-            const responseText = await page.evaluate(() => {
+            // Get the final response content (新規に追加された model-response のみを取得)
+            const responseText = await page.evaluate(initialCount => {
                 // Get content from model-response elements
                 const modelResponses = Array.from(document.querySelectorAll('model-response'));
+                if (modelResponses.length > initialCount) {
+                    // 新規に追加された model-response を取得（ChatGPTと同じ方式）
+                    const newResponse = modelResponses[initialCount];
+                    return newResponse.textContent?.trim() || '';
+                }
+                // Fallback: get the last model response if any
                 if (modelResponses.length > 0) {
-                    // Get the last model response
                     const lastResponse = modelResponses[modelResponses.length - 1];
                     return lastResponse.textContent?.trim() || '';
                 }
                 // Fallback: get text from main area
                 const main = document.querySelector('main');
                 return main?.innerText.slice(-5000) || '';
-            });
+            }, initialModelResponseCount);
             response.appendResponseLine('✅ 回答完了');
             // Always save/update session (not just for new chats)
             const chatUrl = page.url();

package/package.json

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