Update formatting

Author: gernberg

docs/tool-reference.md

@@ -160,6 +160,7 @@
 - **timeout** (integer) _(optional)_: Maximum wait time in milliseconds. If set to 0, the default timeout will be used.
 - **type** (enum: "url", "back", "forward", "reload") _(optional)_: Navigate the page by URL, back or forward in history, or reload.
 - **url** (string) _(optional)_: Target URL (only type=url)
+- **initScript** (string) _(optional)_: Script to execute whenever a page is navigated to.
 
 ---
 

src/tools/pages.ts

@@ -136,6 +136,12 @@ export const navigatePage = defineTool({
     initScript: zod.string().optional().describe(
       `(optional) A JavaScript function to be executed by the tool on new document for every page load before any other scripts.
       `),
+    initScript: zod
+      .string()
+      .optional()
+      .describe(
+        '(optional) A JavaScript function to be executed by the tool on new document for every page load before any other scripts.',
+      ),
     ...timeoutSchema,
   },
   handler: async (request, response, context) => {
@@ -164,8 +170,66 @@ export const navigatePage = defineTool({
         }
         // We are not going to report the dialog like regular dialogs.
         context.clearDialog();
+    if (request.params.initScript) {
+      await page.evaluateOnNewDocument(request.params.initScript);
+    }
+
+    await context.waitForEventsAfterAction(async () => {
+      switch (request.params.type) {
+        case 'url':
+          if (!request.params.url) {
+            throw new Error('A URL is required for navigation of type=url.');
+          }
+          try {
+            await page.goto(request.params.url, options);
+            response.appendResponseLine(
+              `Successfully navigated to ${request.params.url}.`,
+            );
+          } catch (error) {
+            response.appendResponseLine(
+              `Unable to navigate in the  selected page: ${error.message}.`,
+            );
+          }
+          break;
+        case 'back':
+          try {
+            await page.goBack(options);
+            response.appendResponseLine(
+              `Successfully navigated back to ${page.url()}.`,
+            );
+          } catch (error) {
+            response.appendResponseLine(
+              `Unable to navigate back in the selected page: ${error.message}.`,
+            );
+          }
+          break;
+        case 'forward':
+          try {
+            await page.goForward(options);
+            response.appendResponseLine(
+              `Successfully navigated forward to ${page.url()}.`,
+            );
+          } catch (error) {
+            response.appendResponseLine(
+              `Unable to navigate forward in the selected page: ${error.message}.`,
+            );
+          }
+          break;
+        case 'reload':
+          try {
+            await page.reload({
+              ...options,
+              ignoreCache: request.params.ignoreCache,
+            });
+            response.appendResponseLine(`Successfully reloaded the page.`);
+          } catch (error) {
+            response.appendResponseLine(
+              `Unable to reload the selected page: ${error.message}.`,
+            );
+          }
+          break;
       }
-    };
+    });
 
     if (request.params.initScript) {
       await page.evaluateOnNewDocument(request.params.initScript);

tests/tools/pages.test.ts

@@ -300,14 +300,21 @@ describe('pages', () => {
     it('navigates to correct page with initScript', async () => {
       await withBrowser(async (response, context) => {
         await navigatePage.handler(
-          {params: {url: 'data:text/html,<div>Hello MCP</div>', initScript: 'window.initScript = "completed"'}},
+          {
+            params: {
+              url: 'data:text/html,<div>Hello MCP</div>',
+              initScript: 'window.initScript = "completed"',
+            },
+          },
           response,
           context,
         );
         const page = context.getSelectedPage();
 
         // wait for up to 1s for the global variable to set by the initScript to exist
-        await page.waitForFunction("window.initScript==='completed'", {timeout: 1000});
+        await page.waitForFunction("window.initScript==='completed'", {
+          timeout: 1000,
+        });
 
         assert.ok(response.includePages);
       });