chore: get tabid tool

Author: OrKoN

src/cli.ts

@@ -163,6 +163,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

@@ -131,6 +131,12 @@ function registerTool(tool: ToolDefinition): void {
   ) {
     return;
   }
+  if (
+    tool.annotations.conditions?.includes('experimentalInterop') &&
+    !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: ['experimentalInterop'],
+  },
+  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,13 @@ describe('e2e', () => {
             if (maybeTool.annotations?.conditions?.includes('computerVision')) {
               continue;
             }
+            if (
+              maybeTool.annotations?.conditions?.includes(
+                'experimentalInteropTools',
+              )
+            ) {
+              continue;
+            }
             definedNames.push(maybeTool.name);
           }
         }

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'));
+      });
+    });
+  });
 });