chore: get tabid tool

Author: OrKoN

src/cli.ts

@@ -168,6 +168,11 @@ export const cliOptions = {
       'Whether to include all kinds of pages such as webviews or background pages as pages.',
     hidden: true,
   },
+  experimentalInteropTools: {
+    type: 'boolean',
+    describe: 'Whether to enable interoperability tools',
+    hidden: true,
+  },
   chromeArg: {
     type: 'array',
     describe:

src/main.ts

@@ -139,6 +139,12 @@ function registerTool(tool: ToolDefinition): void {
   ) {
     return;
   }
+  if (
+    tool.annotations.conditions?.includes('experimentalInteropTools') &&
+    !args.experimentalInteropTools
+  ) {
+    return;
+  }
   server.registerTool(
     tool.name,
     {

src/tools/pages.ts

@@ -269,3 +269,26 @@ export const handleDialog = defineTool({
     response.setIncludePages(true);
   },
 });
+
+export const getTabId = defineTool({
+  name: 'get_tab_id',
+  description: `Get the tab ID of the page`,
+  annotations: {
+    category: ToolCategory.NAVIGATION,
+    readOnlyHint: true,
+    conditions: ['experimentalInteropTools'],
+  },
+  schema: {
+    pageId: zod
+      .number()
+      .describe(
+        `The ID of the page to get the tab ID for. Call ${listPages.name} to get available pages.`,
+      ),
+  },
+  handler: async (request, response, context) => {
+    const page = context.getPageById(request.params.pageId);
+    // @ts-expect-error _tabId is internal.
+    const tabId = page._tabId;
+    response.appendResponseLine(tabId);
+  },
+});

tests/index.test.ts

@@ -101,6 +101,11 @@ describe('e2e', () => {
             if (maybeTool.annotations?.conditions?.includes('computerVision')) {
               continue;
             }
+            if (
+              maybeTool.annotations?.conditions?.includes('experimentalInteropTools')
+            ) {
+              continue;
+            }
             definedNames.push(maybeTool.name);
           }
         }
@@ -120,4 +125,16 @@ describe('e2e', () => {
       ['--experimental-vision'],
     );
   });
+
+
+  it('has experimental interop tools', async () => {
+    await withClient(
+      async client => {
+        const {tools} = await client.listTools();
+        const getTabId = tools.find(t => t.name === 'get_tab_id');
+        assert.ok(getTabId);
+      },
+      ['--experimental-interop-tools'],
+    );
+  });
 });

tests/tools/pages.test.ts

@@ -17,6 +17,7 @@ import {
   navigatePage,
   resizePage,
   handleDialog,
+  getTabId,
 } from '../../src/tools/pages.js';
 import {withMcpContext} from '../utils.js';
 
@@ -322,4 +323,16 @@ describe('pages', () => {
       });
     });
   });
+
+  describe('get_tab_id', () => {
+    it('returns the tab id', async () => {
+      await withMcpContext(async (response, context) => {
+        const page = context.getSelectedPage();
+        // @ts-expect-error _tabId is internal.
+        page._tabId = 'test-tab-id';
+        await getTabId.handler({params: {pageId: 1}}, response, context);
+        assert.ok(response.responseLines.includes('test-tab-id'));
+      });
+    });
+  });
 });