chrome-devtools-frontend 1.0.1662965 → 1.0.1664496

package/front_end/core/sdk/CSSStyleSheetHeader.ts

@@ -3,10 +3,10 @@
 // found in the LICENSE file.
 
 import type * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as Common from '../common/common.js';
 import * as i18n from '../i18n/i18n.js';
 import * as Platform from '../platform/platform.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {CSSModel} from './CSSModel.js';
 import {DeferredDOMNode} from './DOMModel.js';

package/front_end/core/sdk/CSSSupports.ts

@@ -3,7 +3,7 @@
 // found in the LICENSE file.
 
 import type * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {CSSModel} from './CSSModel.js';
 import {CSSQuery} from './CSSQuery.js';

package/front_end/core/sdk/CompilerSourceMappingContentProvider.ts

@@ -2,10 +2,10 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import type * as Common from '../common/common.js';
 import * as i18n from '../i18n/i18n.js';
 import type * as Platform from '../platform/platform.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {PageResourceLoader, PageResourceLoadInitiator} from './PageResourceLoader.js';
 

package/front_end/core/sdk/NetworkManager.ts

@@ -4,11 +4,11 @@
 
 import type * as ProtocolProxyApi from '../../generated/protocol-proxy-api.js';
 import * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as Common from '../common/common.js';
 import * as i18n from '../i18n/i18n.js';
 import * as Platform from '../platform/platform.js';
 import * as Root from '../root/root.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import {Cookie} from './Cookie.js';
 import {

package/front_end/core/sdk/NetworkRequest.ts

@@ -3,10 +3,10 @@
 // found in the LICENSE file.
 
 import * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as Common from '../common/common.js';
 import * as i18n from '../i18n/i18n.js';
 import * as Platform from '../platform/platform.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import {Attribute, type Cookie} from './Cookie.js';
 import {CookieModel} from './CookieModel.js';
@@ -33,11 +33,11 @@ const UIStrings = {
   /**
    * @description Tooltip to explain why a cookie was blocked
    */
-  notOnPath: 'This cookie was blocked because its path was not an exact match for or a superdirectory of the request url\'s path.',
+  notOnPath: 'This cookie was blocked because its path was not an exact match for or a superdirectory of the request url’s path.',
   /**
    * @description Tooltip to explain why a cookie was blocked
    */
-  domainMismatch: 'This cookie was blocked because neither did the request URL\'s domain exactly match the cookie\'s domain, nor was the request URL\'s domain a subdomain of the cookie\'s Domain attribute value.',
+  domainMismatch: 'This cookie was blocked because neither did the request URL’s domain exactly match the cookie’s domain, nor was the request URL’s domain a subdomain of the cookie’s Domain attribute value.',
   /**
    * @description Tooltip to explain why a cookie was blocked
    */
@@ -49,7 +49,7 @@ const UIStrings = {
   /**
    * @description Tooltip to explain why a cookie was blocked
    */
-  sameSiteUnspecifiedTreatedAsLax: 'This cookie didn\'t specify a "`SameSite`" attribute when it was stored and was defaulted to "SameSite=Lax," and was blocked because the request was made from a different site and was not initiated by a top-level navigation. The cookie had to have been set with "`SameSite=None`" to enable cross-site usage.',
+  sameSiteUnspecifiedTreatedAsLax: 'This cookie didn’t specify a "`SameSite`" attribute when it was stored and was defaulted to "SameSite=Lax," and was blocked because the request was made from a different site and was not initiated by a top-level navigation. The cookie had to have been set with "`SameSite=None`" to enable cross-site usage.',
   /**
    * @description Tooltip to explain why a cookie was blocked
    */
@@ -77,7 +77,7 @@ const UIStrings = {
   /**
    * @description Tooltip to explain why a cookie was blocked due to Schemeful Same-Site
    */
-  schemefulSameSiteUnspecifiedTreatedAsLax: 'This cookie didn\'t specify a "`SameSite`" attribute when it was stored, was defaulted to "`SameSite=Lax"`, and was blocked because the request was cross-site and was not initiated by a top-level navigation. This request is considered cross-site because the URL has a different scheme than the current site.',
+  schemefulSameSiteUnspecifiedTreatedAsLax: 'This cookie didn’t specify a "`SameSite`" attribute when it was stored, was defaulted to "`SameSite=Lax"`, and was blocked because the request was cross-site and was not initiated by a top-level navigation. This request is considered cross-site because the URL has a different scheme than the current site.',
   /**
    * @description Tooltip to explain why a cookie was blocked due to exceeding the maximum size
    */
@@ -114,7 +114,7 @@ const UIStrings = {
   /**
    * @description Tooltip to explain why a cookie was blocked due to Schemeful Same-Site
    */
-  thisSetcookieDidntSpecifyASamesite: 'This `Set-Cookie` header didn\'t specify a "`SameSite`" attribute, was defaulted to "`SameSite=Lax"`, and was blocked because it came from a cross-site response which was not the response to a top-level navigation. This response is considered cross-site because the URL has a different scheme than the current site.',
+  thisSetcookieDidntSpecifyASamesite: 'This `Set-Cookie` header didn’t specify a "`SameSite`" attribute, was defaulted to "`SameSite=Lax"`, and was blocked because it came from a cross-site response which was not the response to a top-level navigation. This response is considered cross-site because the URL has a different scheme than the current site.',
   /**
    * @description Tooltip to explain why an attempt to set a cookie via a `Set-Cookie` HTTP header on a request's response was blocked.
    */
@@ -127,7 +127,7 @@ const UIStrings = {
   /**
    * @description Tooltip to explain why an attempt to set a cookie via a `Set-Cookie` HTTP header on a request's response was blocked.
    */
-  blockedReasonSameSiteUnspecifiedTreatedAsLax: 'This `Set-Cookie` header didn\'t specify a "`SameSite`" attribute and was defaulted to "`SameSite=Lax,`" and was blocked because it came from a cross-site response which was not the response to a top-level navigation. The `Set-Cookie` had to have been set with "`SameSite=None`" to enable cross-site usage.',
+  blockedReasonSameSiteUnspecifiedTreatedAsLax: 'This `Set-Cookie` header didn’t specify a "`SameSite`" attribute and was defaulted to "`SameSite=Lax,`" and was blocked because it came from a cross-site response which was not the response to a top-level navigation. The `Set-Cookie` had to have been set with "`SameSite=None`" to enable cross-site usage.',
   /**
    * @description Tooltip to explain why an attempt to set a cookie via a `Set-Cookie` HTTP header on a request's response was blocked.
    */

package/front_end/core/sdk/Resource.ts

@@ -31,9 +31,9 @@
  */
 
 import type * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as Common from '../common/common.js';
 import * as Platform from '../platform/platform.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {NetworkRequest} from './NetworkRequest.js';
 import type {ResourceTreeFrame, ResourceTreeModel} from './ResourceTreeModel.js';

package/front_end/core/sdk/ScopeTreeCache.ts

@@ -3,7 +3,7 @@
 // found in the LICENSE file.
 
 import * as Formatter from '../../models/formatter/formatter.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {Script} from './Script.js';
 

package/front_end/core/sdk/Script.ts

@@ -29,9 +29,9 @@
 
 import * as Platform from '../../core/platform/platform.js';
 import * as Protocol from '../../generated/protocol.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as Common from '../common/common.js';
 import * as i18n from '../i18n/i18n.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import {
   COND_BREAKPOINT_SOURCE_URL,

package/front_end/core/sdk/ServerSentEvents.ts

@@ -2,7 +2,7 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import * as TextUtils from '../../models/text_utils/text_utils.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import {Events, type EventSourceMessage, type NetworkRequest} from './NetworkRequest.js';
 import {ServerSentEventsParser} from './ServerSentEventsProtocol.js';

package/front_end/core/sdk/SourceMap.ts

@@ -2,10 +2,10 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-import * as TextUtils from '../../models/text_utils/text_utils.js';
 import * as ScopesCodec from '../../third_party/source-map-scopes-codec/source-map-scopes-codec.js';
 import * as Common from '../common/common.js';
 import * as Platform from '../platform/platform.js';
+import * as TextUtils from '../text_utils/text_utils.js';
 
 import type {CallFrame, ScopeChainEntry} from './DebuggerModel.js';
 import {scopeTreeForScript} from './ScopeTreeCache.js';

package/front_end/core/sdk/SourceMapScopesInfo.ts

@@ -4,9 +4,9 @@
 
 import * as Protocol from '../../generated/protocol.js';
 import * as Formatter from '../../models/formatter/formatter.js';
-import type * as TextUtils from '../../models/text_utils/text_utils.js';
 import type * as ScopesCodec from '../../third_party/source-map-scopes-codec/source-map-scopes-codec.js';
 import type * as Platform from '../platform/platform.js';
+import type * as TextUtils from '../text_utils/text_utils.js';
 
 import type {CallFrame, ScopeChainEntry} from './DebuggerModel.js';
 import type {SourceMap} from './SourceMap.js';

package/front_end/core/sdk/sdk-meta.ts

@@ -9,11 +9,11 @@ const UIStrings = {
   /**
    * @description Title of a setting under the Console category that can be invoked through the Command Menu
    */
-  preserveLogUponNavigation: 'Preserve log upon navigation',
+  preserveLogUponNavigation: 'Keep log on navigation',
   /**
    * @description Title of a setting under the Console category that can be invoked through the Command Menu
    */
-  doNotPreserveLogUponNavigation: 'Do not preserve log upon navigation',
+  doNotPreserveLogUponNavigation: 'Don’t keep log on navigation',
   /**
    * @description Text for pausing the debugger on exceptions
    */
@@ -53,7 +53,7 @@ const UIStrings = {
   /**
    * @description Text of a setting that do turn off the measuring rulers when hover over a target
    */
-  doNotShowRulersOnHover: 'Don\'t show rulers on hover',
+  doNotShowRulersOnHover: 'Don’t show rulers on hover',
   /**
    * @description Title of a setting that turns on grid area name labels
    */

package/front_end/{models → core}/text_utils/ContentProvider.ts

@@ -17,9 +17,8 @@ export interface ContentProvider {
 }
 
 export class SearchMatch {
-  constructor(
-      readonly lineNumber: number, readonly lineContent: string, readonly columnNumber: number,
-      readonly matchLength: number) {
+  constructor(readonly lineNumber: number, readonly lineContent: string, readonly columnNumber: number,
+              readonly matchLength: number) {
   }
 
   static comparator(a: SearchMatch, b: SearchMatch): number {
@@ -27,9 +26,8 @@ export class SearchMatch {
   }
 }
 
-export const contentAsDataURL = function(
-    content: string|null, mimeType: string, contentEncoded: boolean, charset?: string|null,
-    limitSize = true): string|null {
+export const contentAsDataURL = function(content: string|null, mimeType: string, contentEncoded: boolean,
+                                         charset?: string|null, limitSize = true): string|null {
   const maxDataUrlSize = 1024 * 1024;
   if (content === undefined || content === null || (limitSize && content.length > maxDataUrlSize)) {
     return null;

package/front_end/{models → core}/text_utils/StaticContentProvider.ts

@@ -14,17 +14,15 @@ export class StaticContentProvider implements ContentProvider {
   readonly #contentType: Common.ResourceType.ResourceType;
   readonly #lazyContent: () => Promise<ContentDataOrError>;
 
-  constructor(
-      contentURL: Platform.DevToolsPath.UrlString, contentType: Common.ResourceType.ResourceType,
-      lazyContent: () => Promise<ContentDataOrError>) {
+  constructor(contentURL: Platform.DevToolsPath.UrlString, contentType: Common.ResourceType.ResourceType,
+              lazyContent: () => Promise<ContentDataOrError>) {
     this.#contentURL = contentURL;
     this.#contentType = contentType;
     this.#lazyContent = lazyContent;
   }
 
-  static fromString(
-      contentURL: Platform.DevToolsPath.UrlString, contentType: Common.ResourceType.ResourceType,
-      content: string): StaticContentProvider {
+  static fromString(contentURL: Platform.DevToolsPath.UrlString, contentType: Common.ResourceType.ResourceType,
+                    content: string): StaticContentProvider {
     const lazyContent = (): Promise<ContentData> =>
         Promise.resolve(new ContentData(content, /* isBase64 */ false, contentType.canonicalMimeType()));
     return new StaticContentProvider(contentURL, contentType, lazyContent);

package/front_end/{models → core}/text_utils/StreamingContentData.ts

@@ -86,8 +86,7 @@ export class StreamingContentData extends Common.ObjectWrapper.ObjectWrapper<Eve
 export type StreamingContentDataOrError = StreamingContentData|{error: string};
 
 export const isError = function(contentDataOrError: StreamingContentDataOrError): contentDataOrError is {
-error:
-  string,
+  error: string,
 } {
   return 'error' in contentDataOrError;
 };

package/front_end/{models → core}/text_utils/TextRange.ts

@@ -26,9 +26,8 @@ export class TextRange {
   }
 
   static fromObject(serializedTextRange: SerializedTextRange): TextRange {
-    return new TextRange(
-        serializedTextRange.startLine, serializedTextRange.startColumn, serializedTextRange.endLine,
-        serializedTextRange.endColumn);
+    return new TextRange(serializedTextRange.startLine, serializedTextRange.startColumn, serializedTextRange.endLine,
+                         serializedTextRange.endColumn);
   }
 
   static comparator(range1: TextRange, range2: TextRange): number {
@@ -263,4 +262,8 @@ export class TextRange {
 export class SourceRange {
   constructor(public offset: number, public length: number) {
   }
+
+  cssValue(): string {
+    return `${this.offset},${this.length}`;
+  }
 }

package/front_end/{models → core}/text_utils/TextUtils.ts

@@ -310,8 +310,8 @@ export const isMinified = function(text: string): boolean {
  *
  * @returns empty search matches if `contentData` is an error or not text content.
  */
-export const performSearchInContentData = function(
-    contentData: ContentDataOrError, query: string, caseSensitive: boolean, isRegex: boolean): SearchMatch[] {
+export const performSearchInContentData = function(contentData: ContentDataOrError, query: string,
+                                                   caseSensitive: boolean, isRegex: boolean): SearchMatch[] {
   if (ContentData.isError(contentData) || !contentData.isTextContent) {
     return [];
   }
@@ -322,8 +322,8 @@ export const performSearchInContentData = function(
  * @returns One {@link SearchMatch} per match. Multiple matches on the same line each
  * result in their own `SearchMatchExact` instance.
  */
-export const performSearchInContent = function(
-    text: Text, query: string, caseSensitive: boolean, isRegex: boolean): SearchMatch[] {
+export const performSearchInContent = function(text: Text, query: string, caseSensitive: boolean,
+                                               isRegex: boolean): SearchMatch[] {
   const regex = Platform.StringUtilities.createSearchRegex(query, caseSensitive, isRegex);
 
   const result = [];
@@ -343,9 +343,9 @@ export const performSearchInContent = function(
  * @param matches is deliberatedly typed as an object literal so we can pass the
  *                CDP search result type.
  */
-export const performSearchInSearchMatches = function(
-    matches: Array<{lineNumber: number, lineContent: string}>, query: string, caseSensitive: boolean,
-    isRegex: boolean): SearchMatch[] {
+export const performSearchInSearchMatches = function(matches: Array<{lineNumber: number, lineContent: string}>,
+                                                     query: string, caseSensitive: boolean,
+                                                     isRegex: boolean): SearchMatch[] {
   const regex = Platform.StringUtilities.createSearchRegex(query, caseSensitive, isRegex);
   const result = [];
 

package/front_end/{models → core}/text_utils/WasmDisassembly.ts

@@ -51,8 +51,8 @@ export class WasmDisassembly extends ContentData {
   }
 
   bytecodeOffsetToLineNumber(bytecodeOffset: number): number {
-    return Platform.ArrayUtilities.upperBound(
-               this.#offsets, bytecodeOffset, Platform.ArrayUtilities.DEFAULT_COMPARATOR) -
+    return Platform.ArrayUtilities.upperBound(this.#offsets, bytecodeOffset,
+                                              Platform.ArrayUtilities.DEFAULT_COMPARATOR) -
         1;
   }
 

package/front_end/core/text_utils/text_utils.ts

@@ -0,0 +1,27 @@
+// Copyright 2019 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 CodeMirrorUtils from './CodeMirrorUtils.js';
+import * as ContentData from './ContentData.js';
+import * as ContentProvider from './ContentProvider.js';
+import * as StaticContentProvider from './StaticContentProvider.js';
+import * as StreamingContentData from './StreamingContentData.js';
+import * as Text from './Text.js';
+import * as TextCursor from './TextCursor.js';
+import * as TextRange from './TextRange.js';
+import * as TextUtils from './TextUtils.js';
+import * as WasmDisassembly from './WasmDisassembly.js';
+
+export {
+  CodeMirrorUtils,
+  ContentData,
+  ContentProvider,
+  StaticContentProvider,
+  StreamingContentData,
+  Text,
+  TextCursor,
+  TextRange,
+  TextUtils,
+  WasmDisassembly,
+};

package/front_end/entrypoints/formatter_worker/AcornTokenizer.ts

@@ -3,7 +3,7 @@
 // found in the LICENSE file.
 
 import * as Platform from '../../core/platform/platform.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
+import * as TextUtils from '../../core/text_utils/text_utils.js';
 import * as Acorn from '../../third_party/acorn/acorn.js';
 
 export type TokenOrComment = Acorn.Token|Acorn.Comment;

package/front_end/entrypoints/heap_snapshot_worker/HeapSnapshotLoader.ts

@@ -4,7 +4,7 @@
 
 import type * as PlatformApi from '../../core/platform/api/api.js';
 import * as Platform from '../../core/platform/platform.js';
-import * as TextUtils from '../../models/text_utils/text_utils.js';
+import * as TextUtils from '../../core/text_utils/text_utils.js';
 
 import {type HeapSnapshotHeader, HeapSnapshotProgress, JSHeapSnapshot, type Profile} from './HeapSnapshot.js';
 import type {HeapSnapshotWorkerDispatcher} from './HeapSnapshotWorkerDispatcher.js';

package/front_end/entrypoints/main/MainImpl.ts

@@ -460,7 +460,7 @@ export class MainImpl {
     targetManager.addEventListener(
         SDK.TargetManager.Events.SUSPEND_STATE_CHANGED, this.#onSuspendStateChanged.bind(this));
 
-    new Bindings.PresentationConsoleMessageHelper.PresentationConsoleMessageManager();
+    new Bindings.PresentationConsoleMessageHelper.PresentationConsoleMessageManager(targetManager);
     targetManager.setScopeTarget(targetManager.primaryPageTarget());
     UI.Context.Context.instance().addFlavorChangeListener(SDK.Target.Target, ({data}) => {
       const outermostTarget = data?.outermostTarget();
@@ -488,6 +488,7 @@ export class MainImpl {
         () => Snackbar.Snackbar.Snackbar.show({message: i18nString(UIStrings.aiModelDownloaded)}));
 
     new PauseListener();
+    new ConsoleProfileFinishedListener();
 
     const actionRegistryInstance = UI.ActionRegistry.ActionRegistry.instance({forceNew: true});
     // Required for legacy a11y layout tests
@@ -961,6 +962,18 @@ export class PauseListener {
   }
 }
 
+export class ConsoleProfileFinishedListener {
+  constructor() {
+    SDK.TargetManager.TargetManager.instance().addModelListener(SDK.CPUProfilerModel.CPUProfilerModel,
+                                                                SDK.CPUProfilerModel.Events.CONSOLE_PROFILE_FINISHED,
+                                                                this.#consoleProfileFinished, this);
+  }
+
+  #consoleProfileFinished(event: Common.EventTarget.EventTargetEvent<SDK.CPUProfilerModel.ProfileFinishedData>): void {
+    void Common.Revealer.reveal(event.data);
+  }
+}
+
 /** Unused but mentioned at https://chromedevtools.github.io/devtools-protocol/#:~:text=use%20Main.MainImpl.-,sendOverProtocol,-()%20in%20the **/
 export function sendOverProtocol(
     method: ProtocolClient.InspectorBackend.QualifiedName, params: Object|null): Promise<unknown[]|null> {

package/front_end/foundation/Universe.ts

@@ -188,7 +188,7 @@ export class Universe {
     const liveMetrics = new LiveMetrics.LiveMetrics(targetManager, deviceModeModel);
     context.set(LiveMetrics.LiveMetrics, liveMetrics);
 
-    const userBadges = new Badges.UserBadges(settings, gdpClient);
+    const userBadges = new Badges.UserBadges(settings, gdpClient, options.inspectorFrontendHost);
     context.set(Badges.UserBadges, userBadges);
 
     const aiHistoryStorage = new AiAssistance.AiHistoryStorage.AiHistoryStorage(settings);

package/front_end/models/ai_assistance/AgentProject.ts

@@ -2,9 +2,9 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+import * as TextUtils from '../../core/text_utils/text_utils.js';
 import * as Diff from '../../third_party/diff/diff.js';
 import * as Persistence from '../persistence/persistence.js';
-import * as TextUtils from '../text_utils/text_utils.js';
 import type * as Workspace from '../workspace/workspace.js';
 
 import {debugLog} from './debug.js';

package/front_end/models/ai_assistance/agents/AccessibilityAgent.ts

@@ -125,7 +125,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   }
 
   protected override async preRun(): Promise<void> {
-    const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
+    const target = this.targetManager.primaryPageTarget();
     const domModel = target?.model(SDK.DOMModel.DOMModel);
     // We need to ensure the document is requested so that #getDocumentBodyNode()
     // can return a valid node for the JavaScript execution context.
@@ -144,10 +144,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
    * so that the AI has a valid $0 to start with.
    */
   #getDocumentBodyNode(): SDK.DOMModel.DOMNode|null {
-    const document = SDK.TargetManager.TargetManager.instance()
-                         .primaryPageTarget()
-                         ?.model(SDK.DOMModel.DOMModel)
-                         ?.existingDocument();
+    const document = this.targetManager.primaryPageTarget()?.model(SDK.DOMModel.DOMModel)?.existingDocument();
     return document?.body ?? document ?? null;
   }
 
@@ -168,7 +165,7 @@ export class AccessibilityAgent extends AiAgent<LHModel.ReporterTypes.ReportJSON
   }
 
   async #resolvePathToNode(path: string): Promise<SDK.DOMModel.DOMNode|null> {
-    const target = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
+    const target = this.targetManager.primaryPageTarget();
     if (!target) {
       return null;
     }

package/front_end/models/ai_assistance/agents/AiAgent.ts

@@ -7,9 +7,9 @@ import type {UrlString} from '../../../core/platform/DevToolsPath.js';
 import type * as Platform from '../../../core/platform/platform.js';
 import * as Root from '../../../core/root/root.js';
 import * as SDK from '../../../core/sdk/sdk.js';
+import type * as TextUtils from '../../../core/text_utils/text_utils.js';
 import type * as Protocol from '../../../generated/protocol.js';
 import type * as LHModel from '../../lighthouse/lighthouse.js';
-import type * as TextUtils from '../../text_utils/text_utils.js';
 import type * as Trace from '../../trace/trace.js';
 import type * as Workspace from '../../workspace/workspace.js';
 import {areOriginsEquivalent, extractContextOrigin, isOpaqueOrigin} from '../AiOrigins.js';

package/front_end/models/ai_assistance/agents/ContextSelectionAgent.ts

@@ -74,6 +74,14 @@ Your role is to understand the user's query, identify the appropriate specialize
 * The only available types are \`#req\` for network request and \`#file\` for source files. Only use ID inside the link, never ask about user selecting by ID.
 `;
 
+export interface ContextSelectionAgentOptions extends AgentOptions {
+  performanceRecordAndReload?: () => Promise<Trace.TraceModel.ParsedTrace>;
+  onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>;
+  networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator;
+  networkLog?: Logs.NetworkLog.NetworkLog;
+  workspace?: Workspace.Workspace.WorkspaceImpl;
+}
+
 /**
  * One agent instance handles one conversation. Create a new agent
  * instance for a new conversation.
@@ -101,13 +109,13 @@ export class ContextSelectionAgent extends AiAgent<never> {
   readonly #lighthouseRecording?:
       (overrides?: LHModel.RunTypes.RunOverrides) => Promise<LHModel.ReporterTypes.ReportJSON|null>;
   #allowedOrigin: () => AllowedOriginResult;
+  readonly #networkLog: Logs.NetworkLog.NetworkLog;
+  readonly #workspace: Workspace.Workspace.WorkspaceImpl;
 
-  constructor(opts: AgentOptions&{
-    performanceRecordAndReload?: () => Promise<Trace.TraceModel.ParsedTrace>,
-    onInspectElement?: () => Promise<SDK.DOMModel.DOMNode|null>,
-    networkTimeCalculator?: NetworkTimeCalculator.NetworkTransferTimeCalculator,
-  }) {
+  constructor(opts: ContextSelectionAgentOptions) {
     super(opts);
+    this.#networkLog = opts.networkLog ?? Logs.NetworkLog.NetworkLog.instance();
+    this.#workspace = opts.workspace ?? Workspace.Workspace.WorkspaceImpl.instance();
     this.#performanceRecordAndReload = opts.performanceRecordAndReload;
     this.#lighthouseRecording = opts.lighthouseRecording;
     this.#onInspectElement = opts.onInspectElement;
@@ -146,7 +154,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
 
         let hasCrossOriginRequest = false;
         const requestsToShow: NetworkRequest[] = [];
-        for (const request of Logs.NetworkLog.NetworkLog.instance().requests()) {
+        for (const request of this.#networkLog.requests()) {
           const requestOrigin = getRequestContextOrigin(request);
           /**
            * NOTE: this origin check does not ensure that all the requests are
@@ -226,7 +234,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
             error: 'No request found',
           };
         }
-        const request = Logs.NetworkLog.NetworkLog.instance().requests().find(req => {
+        const request = this.#networkLog.requests().find(req => {
           if (req.requestId() !== id) {
             return false;
           }
@@ -281,7 +289,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
 
         const files: Array<{file: string, id: number | undefined}> = [];
         const uiSourceCodes: Workspace.UISourceCode.UISourceCode[] = [];
-        for (const file of ContextSelectionAgent.getUISourceCodes()) {
+        for (const file of ContextSelectionAgent.getUISourceCodes(this.#workspace)) {
           const fileUrl = file.url();
           const fileOrigin = Common.ParsedURL.ParsedURL.extractOrigin(fileUrl);
 
@@ -339,7 +347,7 @@ export class ContextSelectionAgent extends AiAgent<never> {
         }
         const origin = allowedOriginResult.origin;
 
-        const file = ContextSelectionAgent.getUISourceCodes().find(file => {
+        const file = ContextSelectionAgent.getUISourceCodes(this.#workspace).find(file => {
           if (ContextSelectionAgent.uiSourceCodeId.get(file) !== params.id) {
             return false;
           }
@@ -554,8 +562,8 @@ export class ContextSelectionAgent extends AiAgent<never> {
    * coming from SourceMaps (usually only one) as that has simple code and
    * usually is what the user authored.
    */
-  static getUISourceCodes(): Workspace.UISourceCode.UISourceCode[] {
-    const workspace = Workspace.Workspace.WorkspaceImpl.instance();
+  static getUISourceCodes(workspace: Workspace.Workspace.WorkspaceImpl = Workspace.Workspace.WorkspaceImpl.instance()):
+      Workspace.UISourceCode.UISourceCode[] {
     const projects =
         workspace.projects().filter(project => project.type() === Workspace.Workspace.projectTypes.Network);
     const uiSourceCodes = new Map<string, Workspace.UISourceCode.UISourceCode>();