feat: Support filtering by request type in list_network_requests

Author: OzTamir

src/McpResponse.ts

@@ -15,8 +15,9 @@ import { formatA11ySnapshot } from './formatters/snapshotFormatter.js';
 import { formatConsoleEvent } from './formatters/consoleFormatter.js';
 import {
   paginateNetworkRequests,
-  type NetworkPaginationOptions,
-} from './utils/networkPagination.js';
+  type NetworkRequestsListingOptions,
+  sanitizeRequestTypeFilter,
+} from './utils/networkUtils.js';
 
 export class McpResponse implements Response {
   #includePages: boolean = false;
@@ -27,7 +28,7 @@ export class McpResponse implements Response {
   #textResponseLines: string[] = [];
   #formattedConsoleData?: string[];
   #images: ImageContentData[] = [];
-  #networkRequestsPaginationOptions?: NetworkPaginationOptions;
+  #networkRequestsPaginationOptions?: NetworkRequestsListingOptions;
 
   setIncludePages(value: boolean): void {
     this.#includePages = value;
@@ -39,7 +40,7 @@ export class McpResponse implements Response {
 
   setIncludeNetworkRequests(
     value: boolean,
-    options?: { pageSize?: number; pageToken?: string | null },
+    options?: NetworkRequestsListingOptions,
   ): void {
     this.#includeNetworkRequests = value;
     if (!value) {
@@ -52,13 +53,21 @@ export class McpResponse implements Response {
       return;
     }
 
-    const sanitizedOptions: NetworkPaginationOptions = {};
+    const sanitizedOptions: NetworkRequestsListingOptions = {};
     if (options.pageSize !== undefined) {
       sanitizedOptions.pageSize = options.pageSize;
     }
     if (options.pageToken !== undefined) {
       sanitizedOptions.pageToken = options.pageToken ?? undefined;
     }
+    if (options.requestType !== undefined) {
+      const sanitizedRequestType = sanitizeRequestTypeFilter(
+        options.requestType,
+      );
+      if (sanitizedRequestType !== undefined) {
+        sanitizedOptions.requestType = sanitizedRequestType;
+      }
+    }
 
     this.#networkRequestsPaginationOptions =
       Object.keys(sanitizedOptions).length > 0 ? sanitizedOptions : undefined;
@@ -193,18 +202,33 @@ Call browser_handle_dialog to handle it before continuing.`);
     if (this.#includeNetworkRequests) {
       const requests = context.getNetworkRequests();
       response.push('## Network requests');
-      if (requests.length) {
-        const paginationResult = paginateNetworkRequests(
-          requests,
-          this.#networkRequestsPaginationOptions,
-        );
-        if (paginationResult.invalidToken) {
-          response.push('Invalid page token provided. Showing first page.');
+      const paginationOptions = this.#networkRequestsPaginationOptions;
+      const paginationResult = paginateNetworkRequests(
+        requests,
+        paginationOptions,
+      );
+
+      if (paginationResult.appliedRequestType) {
+        const summary = Array.isArray(paginationResult.appliedRequestType)
+          ? paginationResult.appliedRequestType.join(', ')
+          : paginationResult.appliedRequestType;
+        response.push(`Filtered by type: ${summary}`);
+      }
+
+      if (paginationResult.invalidToken) {
+        response.push('Invalid page token provided. Showing first page.');
+      }
+
+      const { startIndex, endIndex, total } = paginationResult;
+      if (total === 0) {
+        if (paginationOptions?.requestType) {
+          response.push('No requests found for the selected type(s).');
+        } else {
+          response.push('No requests found.');
         }
-        const { startIndex, endIndex } = paginationResult;
-        response.push(
-          `Showing ${startIndex + 1}-${endIndex} of ${requests.length}.`,
-        );
+      }
+      else {
+        response.push(`Showing ${startIndex + 1}-${endIndex} of ${total}.`);
         for (const request of paginationResult.requests) {
           response.push(getShortDescriptionForRequest(request));
         }
@@ -214,8 +238,6 @@ Call browser_handle_dialog to handle it before continuing.`);
         if (paginationResult.previousPageToken) {
           response.push(`Prev: ${paginationResult.previousPageToken}`);
         }
-      } else {
-        response.push('No requests found.');
       }
     }
 

src/tools/ToolDefinition.ts

@@ -6,6 +6,7 @@
 
 import z from 'zod';
 import { Dialog, ElementHandle, Page } from 'puppeteer-core';
+import type { FilterableResourceType } from '../utils/networkUtils.js';
 import { ToolCategories } from './categories.js';
 import { TraceResult } from '../trace-processing/parse.js';
 
@@ -39,12 +40,18 @@ export type ImageContentData = {
   mimeType: string;
 };
 
+export type NetworkRequestsOptions = {
+  pageSize?: number;
+  pageToken?: string | null;
+  requestType?: FilterableResourceType | FilterableResourceType[] | null;
+};
+
 export interface Response {
   appendResponseLine(value: string): void;
   setIncludePages(value: boolean): void;
   setIncludeNetworkRequests(
     value: boolean,
-    options?: { pageSize?: number; pageToken?: string | null },
+    options?: NetworkRequestsOptions,
   ): void;
   setIncludeConsoleData(value: boolean): void;
   setIncludeSnapshot(value: boolean): void;

src/tools/network.ts

@@ -5,6 +5,7 @@
  */
 
 import z from 'zod';
+import { FILTERABLE_RESOURCE_TYPES } from '../utils/networkUtils.js';
 import { defineTool } from './ToolDefinition.js';
 import { ToolCategories } from './categories.js';
 
@@ -31,11 +32,18 @@ export const listNetworkRequests = defineTool({
       .describe(
         'Opaque token representing the next page. Use the token returned by a previous call.',
       ),
+    requestType: z
+      .enum(FILTERABLE_RESOURCE_TYPES)
+      .optional()
+      .describe(
+        `Type of request to return. When omitted, returns all requests. Available types are: ${FILTERABLE_RESOURCE_TYPES.join(', ')}.`,
+      ),
   },
   handler: async (request, response) => {
     response.setIncludeNetworkRequests(true, {
       pageSize: request.params.pageSize,
       pageToken: request.params.pageToken ?? null,
+      requestType: request.params.requestType ?? null,
     });
   },
 });