resolveCdpElementId fixes

Author: wolfib

src/McpPage.ts

@@ -267,7 +267,7 @@ export class McpPage implements ContextPage {
         cdpElementIds.push(`stashed-${i}`);
         continue;
       }
-      const cdpElementId = context.resolveCdpElementId(this, backendNodeId);
+      const cdpElementId = this.resolveCdpElementId(backendNodeId);
       if (!cdpElementId) {
         logger(`Could not get cdpElementId for backend node ${backendNodeId}`);
         cdpElementIds.push(`stashed-${i}`);

src/TextSnapshot.ts

@@ -134,10 +134,7 @@ export class TextSnapshot {
     const data = options.devtoolsData ?? (await context.getDevToolsData(page));
     if (data?.cdpBackendNodeId) {
       snapshot.hasSelectedElement = true;
-      snapshot.selectedElementUid = context.resolveCdpElementId(
-        page,
-        data.cdpBackendNodeId,
-      );
+      snapshot.selectedElementUid = page.resolveCdpElementId(data.cdpBackendNodeId);
     }
 
     // Clean up unique IDs that we did not see anymore.

src/tools/ToolDefinition.ts

@@ -221,11 +221,6 @@ export type Context = Readonly<{
   triggerExtensionAction(id: string): Promise<void>;
   listExtensions(): Promise<Map<string, Extension>>;
   getExtension(id: string): Promise<Extension | undefined>;
-  resolveCdpElementId(
-    page: ContextPage,
-    cdpBackendNodeId: number,
-  ): string | undefined;
-
   getSelectedMcpPage(): McpPage;
   getExtensionServiceWorkers(): ExtensionServiceWorker[];
   getExtensionServiceWorkerId(

tests/tools/inPage.test.ts

@@ -623,7 +623,7 @@ describe('inPage', () => {
           });
 
           const stub = sinon
-            .stub(context, 'resolveCdpElementId')
+            .stub(page, 'resolveCdpElementId')
             .returns('mock-uid');
 
           await executeInPageTool.handler(