fix: snapshot does not capture Iframe content

docs/tool-reference.md

@@ -281,15 +281,16 @@ so returned values have to JSON-serializable.
 
 - **args** (array) _(optional)_: An optional list of arguments to pass to the function.
 - **function** (string) **(required)**: A JavaScript function to run in the currently selected page.
-  Example without arguments: `() => {
+Example without arguments: `() => {
   return document.title
 }` or `async () => {
   return await fetch("example.com")
 }`.
-  Example with arguments: `(el) => {
+Example with arguments: `(el) => {
   return el.innerText;
 }`
 
+
 ---
 
 ### `list_console_messages`

src/McpContext.ts

@@ -302,7 +302,9 @@ export class McpContext implements Context {
    */
   async createTextSnapshot(): Promise<void> {
     const page = this.getSelectedPage();
-    const rootNode = await page.accessibility.snapshot();
+    const rootNode = await page.accessibility.snapshot({
+      includeIframes: true,
+    });
     if (!rootNode) {
       return;
     }

src/tools/snapshot.ts

@@ -36,11 +36,14 @@ export const waitFor = defineTool({
   },
   handler: async (request, response, context) => {
     const page = context.getSelectedPage();
+    const frames = page.frames();
 
-    await Locator.race([
-      page.locator(`aria/${request.params.text}`),
-      page.locator(`text/${request.params.text}`),
-    ]).wait();
+    const locators = frames.flatMap(frame => [
+      frame.locator(`aria/${request.params.text}`),
+      frame.locator(`text/${request.params.text}`),
+    ]);
+
+    await Locator.race(locators).wait();
 
     response.appendResponseLine(
       `Element with text "${request.params.text}" found.`,

tests/tools/snapshot.test.ts

@@ -95,5 +95,32 @@ describe('snapshot', () => {
         assert.ok(response.includeSnapshot);
       });
     });
+
+    it('should work with iframe content', async () => {
+      await withBrowser(async (response, context) => {
+        const page = await context.getSelectedPage();
+
+        await page.setContent(
+          html`<h1>Top level</h1>
+            <iframe srcdoc="<p>Hello iframe</p>"></iframe>`,
+        );
+
+        await waitFor.handler(
+          {
+            params: {
+              text: 'Hello iframe',
+            },
+          },
+          response,
+          context,
+        );
+
+        assert.equal(
+          response.responseLines[0],
+          'Element with text "Hello iframe" found.',
+        );
+        assert.ok(response.includeSnapshot);
+      });
+    });
   });
 });