chore: move conditional service worker logic inside setIncludePages

Author: nroscino

src/McpResponse.ts

@@ -75,8 +75,12 @@ export class McpResponse implements Response {
     this.#tabId = tabId;
   }
 
-  setIncludePages(value: boolean): void {
+  setIncludePages(value: boolean, includeServiceWorkers?: boolean): void {
     this.#includePages = value;
+
+    if (includeServiceWorkers) {
+      this.#includeExtensionServiceWorkers = value;
+    }
   }
 
   includeSnapshot(params?: SnapshotParams): void {
@@ -89,10 +93,6 @@ export class McpResponse implements Response {
     this.#listExtensions = true;
   }
 
-  setIncludeExtensionServiceWorkers(value: boolean): void {
-    this.#includeExtensionServiceWorkers = value;
-  }
-
   setIncludeNetworkRequests(
     value: boolean,
     options?: PaginationOptions & {

src/tools/ToolDefinition.ts

@@ -63,7 +63,7 @@ export interface DevToolsData {
 
 export interface Response {
   appendResponseLine(value: string): void;
-  setIncludePages(value: boolean): void;
+  setIncludePages(value: boolean, shouldIncludeExtension?: boolean): void;
   setIncludeNetworkRequests(
     value: boolean,
     options?: PaginationOptions & {
@@ -96,7 +96,6 @@ export interface Response {
     insightName: InsightName,
   ): void;
   setListExtensions(): void;
-  setIncludeExtensionServiceWorkers(value: boolean): void;
 }
 
 /**

src/tools/pages.ts

@@ -21,10 +21,7 @@ export const listPages = defineTool(args => {
     },
     schema: {},
     handler: async (_request, response) => {
-      response.setIncludePages(true);
-      if (args?.categoryExtensions) {
-        response.setIncludeExtensionServiceWorkers(true);
-      }
+      response.setIncludePages(true, args?.categoryExtensions);
     },
   };
 });