chrome-ai-bridge 1.0.0

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

@@ -0,0 +1,244 @@
+// Copyright 2012 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 SDK from '../../core/sdk/sdk.js';
+import * as TextUtils from '../text_utils/text_utils.js';
+import * as Workspace from '../workspace/workspace.js';
+import { CSSWorkspaceBinding } from './CSSWorkspaceBinding.js';
+import { DebuggerWorkspaceBinding } from './DebuggerWorkspaceBinding.js';
+import { LiveLocationPool, LiveLocationWithPool } from './LiveLocation.js';
+export class PresentationSourceFrameMessageManager {
+    #targetToMessageHelperMap = new WeakMap();
+    constructor() {
+        SDK.TargetManager.TargetManager.instance().observeModels(SDK.DebuggerModel.DebuggerModel, this);
+        SDK.TargetManager.TargetManager.instance().observeModels(SDK.CSSModel.CSSModel, this);
+    }
+    modelAdded(model) {
+        const target = model.target();
+        const helper = this.#targetToMessageHelperMap.get(target) ?? new PresentationSourceFrameMessageHelper();
+        if (model instanceof SDK.DebuggerModel.DebuggerModel) {
+            helper.setDebuggerModel(model);
+        }
+        else {
+            helper.setCSSModel(model);
+        }
+        this.#targetToMessageHelperMap.set(target, helper);
+    }
+    modelRemoved(model) {
+        const target = model.target();
+        const helper = this.#targetToMessageHelperMap.get(target);
+        helper?.clear();
+    }
+    addMessage(message, source, target) {
+        const helper = this.#targetToMessageHelperMap.get(target);
+        void helper?.addMessage(message, source);
+    }
+    clear() {
+        for (const target of SDK.TargetManager.TargetManager.instance().targets()) {
+            const helper = this.#targetToMessageHelperMap.get(target);
+            helper?.clear();
+        }
+    }
+}
+export class PresentationConsoleMessageManager {
+    #sourceFrameMessageManager = new PresentationSourceFrameMessageManager();
+    constructor() {
+        SDK.TargetManager.TargetManager.instance().addModelListener(SDK.ConsoleModel.ConsoleModel, SDK.ConsoleModel.Events.MessageAdded, event => this.consoleMessageAdded(event.data));
+        SDK.ConsoleModel.ConsoleModel.allMessagesUnordered().forEach(this.consoleMessageAdded, this);
+        SDK.TargetManager.TargetManager.instance().addModelListener(SDK.ConsoleModel.ConsoleModel, SDK.ConsoleModel.Events.ConsoleCleared, () => this.#sourceFrameMessageManager.clear());
+    }
+    consoleMessageAdded(consoleMessage) {
+        const runtimeModel = consoleMessage.runtimeModel();
+        if (!consoleMessage.isErrorOrWarning() || !consoleMessage.runtimeModel() ||
+            consoleMessage.source === "violation" /* Protocol.Log.LogEntrySource.Violation */ || !runtimeModel) {
+            return;
+        }
+        const level = consoleMessage.level === "error" /* Protocol.Log.LogEntryLevel.Error */ ?
+            "Error" /* Workspace.UISourceCode.Message.Level.ERROR */ :
+            "Warning" /* Workspace.UISourceCode.Message.Level.WARNING */;
+        this.#sourceFrameMessageManager.addMessage(new Workspace.UISourceCode.Message(level, consoleMessage.messageText), consoleMessage, runtimeModel.target());
+    }
+}
+export class PresentationSourceFrameMessageHelper {
+    #debuggerModel;
+    #cssModel;
+    #presentationMessages = new Map();
+    #locationPool;
+    constructor() {
+        this.#locationPool = new LiveLocationPool();
+        Workspace.Workspace.WorkspaceImpl.instance().addEventListener(Workspace.Workspace.Events.UISourceCodeAdded, this.#uiSourceCodeAdded.bind(this));
+    }
+    setDebuggerModel(debuggerModel) {
+        if (this.#debuggerModel) {
+            throw new Error('Cannot set DebuggerModel twice');
+        }
+        this.#debuggerModel = debuggerModel;
+        // TODO(dgozman): queueMicrotask because we race with DebuggerWorkspaceBinding on ParsedScriptSource event delivery.
+        debuggerModel.addEventListener(SDK.DebuggerModel.Events.ParsedScriptSource, event => {
+            queueMicrotask(() => {
+                this.#parsedScriptSource(event);
+            });
+        });
+        debuggerModel.addEventListener(SDK.DebuggerModel.Events.GlobalObjectCleared, this.#debuggerReset, this);
+    }
+    setCSSModel(cssModel) {
+        if (this.#cssModel) {
+            throw new Error('Cannot set CSSModel twice');
+        }
+        this.#cssModel = cssModel;
+        cssModel.addEventListener(SDK.CSSModel.Events.StyleSheetAdded, event => queueMicrotask(() => this.#styleSheetAdded(event)));
+    }
+    async addMessage(message, source) {
+        const presentation = new PresentationSourceFrameMessage(message, this.#locationPool);
+        const location = this.#rawLocation(source) ?? this.#cssLocation(source) ?? this.#uiLocation(source);
+        if (location) {
+            await presentation.updateLocationSource(location);
+        }
+        if (source.url) {
+            let messages = this.#presentationMessages.get(source.url);
+            if (!messages) {
+                messages = [];
+                this.#presentationMessages.set(source.url, messages);
+            }
+            messages.push({ source, presentation });
+        }
+    }
+    #uiLocation(source) {
+        if (!source.url) {
+            return null;
+        }
+        const uiSourceCode = Workspace.Workspace.WorkspaceImpl.instance().uiSourceCodeForURL(source.url);
+        if (!uiSourceCode) {
+            return null;
+        }
+        return new Workspace.UISourceCode.UILocation(uiSourceCode, source.line, source.column);
+    }
+    #cssLocation(source) {
+        if (!this.#cssModel || !source.url) {
+            return null;
+        }
+        const locations = this.#cssModel.createRawLocationsByURL(source.url, source.line, source.column);
+        return locations[0] ?? null;
+    }
+    #rawLocation(source) {
+        if (!this.#debuggerModel) {
+            return null;
+        }
+        if (source.scriptId) {
+            return this.#debuggerModel.createRawLocationByScriptId(source.scriptId, source.line, source.column);
+        }
+        const callFrame = source.stackTrace?.callFrames ? source.stackTrace.callFrames[0] : null;
+        if (callFrame) {
+            return this.#debuggerModel.createRawLocationByScriptId(callFrame.scriptId, callFrame.lineNumber, callFrame.columnNumber);
+        }
+        if (source.url) {
+            return this.#debuggerModel.createRawLocationByURL(source.url, source.line, source.column);
+        }
+        return null;
+    }
+    #parsedScriptSource(event) {
+        const script = event.data;
+        const messages = this.#presentationMessages.get(script.sourceURL);
+        const promises = [];
+        for (const { presentation, source } of messages ?? []) {
+            const rawLocation = this.#rawLocation(source);
+            if (rawLocation && script.scriptId === rawLocation.scriptId) {
+                promises.push(presentation.updateLocationSource(rawLocation));
+            }
+        }
+        void Promise.all(promises).then(this.parsedScriptSourceForTest.bind(this));
+    }
+    parsedScriptSourceForTest() {
+    }
+    #uiSourceCodeAdded(event) {
+        const uiSourceCode = event.data;
+        const messages = this.#presentationMessages.get(uiSourceCode.url());
+        const promises = [];
+        for (const { presentation, source } of messages ?? []) {
+            promises.push(presentation.updateLocationSource(new Workspace.UISourceCode.UILocation(uiSourceCode, source.line, source.column)));
+        }
+        void Promise.all(promises).then(this.uiSourceCodeAddedForTest.bind(this));
+    }
+    uiSourceCodeAddedForTest() {
+    }
+    #styleSheetAdded(event) {
+        const header = event.data;
+        const messages = this.#presentationMessages.get(header.sourceURL);
+        const promises = [];
+        for (const { source, presentation } of messages ?? []) {
+            if (header.containsLocation(source.line, source.column)) {
+                promises.push(presentation.updateLocationSource(new SDK.CSSModel.CSSLocation(header, source.line, source.column)));
+            }
+        }
+        void Promise.all(promises).then(this.styleSheetAddedForTest.bind(this));
+    }
+    styleSheetAddedForTest() {
+    }
+    clear() {
+        this.#debuggerReset();
+    }
+    #debuggerReset() {
+        const presentations = Array.from(this.#presentationMessages.values()).flat();
+        for (const { presentation } of presentations) {
+            presentation.dispose();
+        }
+        this.#presentationMessages.clear();
+        this.#locationPool.disposeAll();
+    }
+}
+class FrozenLiveLocation extends LiveLocationWithPool {
+    #uiLocation;
+    constructor(uiLocation, updateDelegate, locationPool) {
+        super(updateDelegate, locationPool);
+        this.#uiLocation = uiLocation;
+    }
+    async uiLocation() {
+        return this.#uiLocation;
+    }
+}
+export class PresentationSourceFrameMessage {
+    #uiSourceCode;
+    #liveLocation;
+    #locationPool;
+    #message;
+    constructor(message, locationPool) {
+        this.#message = message;
+        this.#locationPool = locationPool;
+    }
+    async updateLocationSource(source) {
+        if (source instanceof SDK.DebuggerModel.Location) {
+            await DebuggerWorkspaceBinding.instance().createLiveLocation(source, this.#updateLocation.bind(this), this.#locationPool);
+        }
+        else if (source instanceof SDK.CSSModel.CSSLocation) {
+            await CSSWorkspaceBinding.instance().createLiveLocation(source, this.#updateLocation.bind(this), this.#locationPool);
+        }
+        else if (source instanceof Workspace.UISourceCode.UILocation) {
+            if (!this.#liveLocation) { // Don't "downgrade" the location if a debugger or css mapping was already successful
+                this.#liveLocation = new FrozenLiveLocation(source, this.#updateLocation.bind(this), this.#locationPool);
+                await this.#liveLocation.update();
+            }
+        }
+    }
+    async #updateLocation(liveLocation) {
+        if (this.#uiSourceCode) {
+            this.#uiSourceCode.removeMessage(this.#message);
+        }
+        if (liveLocation !== this.#liveLocation) {
+            this.#uiSourceCode?.removeMessage(this.#message);
+            this.#liveLocation?.dispose();
+            this.#liveLocation = liveLocation;
+        }
+        const uiLocation = await liveLocation.uiLocation();
+        if (!uiLocation) {
+            return;
+        }
+        this.#message.range =
+            TextUtils.TextRange.TextRange.createFromLocation(uiLocation.lineNumber, uiLocation.columnNumber || 0);
+        this.#uiSourceCode = uiLocation.uiSourceCode;
+        this.#uiSourceCode.addMessage(this.#message);
+    }
+    dispose() {
+        this.#uiSourceCode?.removeMessage(this.#message);
+        this.#liveLocation?.dispose();
+    }
+}