chrome-devtools-frontend/front_end/models/ai_assistance/tools/ListPageOrigins.ts

Chrome DevTools UI

// Copyright 2026 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 Host from '../../../core/host/host.js';
import * as i18n from '../../../core/i18n/i18n.js';
import * as SDK from '../../../core/sdk/sdk.js';
import {areOriginsEquivalent, isOpaqueOrigin} from '../AiOrigins.js';
import {isSamePageOrigin} from '../contexts/StorageContext.js';

import {
  type BaseToolCapability,
  type DataHandlerResult,
  type DataTool,
  type OriginLockCapability,
  ToolName,
} from './Tool.js';

const lockedString = i18n.i18n.lockedString;

export class ListPageOriginsTool implements
    DataTool<Record<string, never>, {origins: string[]}, BaseToolCapability&OriginLockCapability> {
  readonly name = ToolName.LIST_PAGE_ORIGINS;
  readonly description =
      'Lists all active, non-empty frame origins loaded by the page. Use this first when generic category context is active to discover all page origins, then pass them to listCookies or listStorageKeys, unless the user\'s explicit request hints at focusing only on the primary page.';

  readonly parameters: Host.AidaClient.FunctionObjectParam<never> = {
    type: Host.AidaClient.ParametersTypes.OBJECT,
    description: '',
    nullable: false,
    properties: {},
    required: [],
  };

  displayInfoFromArgs(): {
    title: string,
    action: string,
  } {
    return {
      title: lockedString('Listing page origins'),
      action: 'listPageOrigins()',
    };
  }

  async handler(
      _args: Record<string, never>,
      context: BaseToolCapability&OriginLockCapability,
      ): Promise<DataHandlerResult<{origins: string[]}>> {
    // eslint-disable-next-line @devtools/no-instance-of-migrated-singletons
    const targetManager = SDK.TargetManager.TargetManager.instance();
    const primaryPageTarget = targetManager.primaryPageTarget();

    const allowedOrigin = context.getEstablishedOrigin();
    if (!allowedOrigin || isOpaqueOrigin(allowedOrigin)) {
      return {error: 'No origin available or not allowed.'};
    }

    const pageOrigin =
        primaryPageTarget ? Common.ParsedURL.ParsedURL.extractOrigin(primaryPageTarget.inspectedURL()) : '';
    const isAllowed = pageOrigin !== '' && areOriginsEquivalent(pageOrigin, allowedOrigin);

    if (!isAllowed) {
      return {error: 'No origin available or not allowed.'};
    }

    const origins = new Set<string>();
    for (const frame of SDK.ResourceTreeModel.ResourceTreeModel.frames(targetManager)) {
      if (!isSamePageOrigin(frame.resourceTreeModel().target().outermostTarget(), allowedOrigin)) {
        continue;
      }
      const origin = frame.securityOrigin;
      if (!origin || isOpaqueOrigin(origin) || origins.has(origin)) {
        continue;
      }
      origins.add(origin);
    }

    return {result: {origins: Array.from(origins)}};
  }
}