import { z } from 'zod';
import { expectationSchema } from '../schemas/expectation.js';
import { elementSelectorSchema } from '../types/selectors.js';
import { quote } from '../utils/codegen.js';
import { generateKeyPressCode } from '../utils/common-formatters.js';
import {
handleSnapshotExpectation,
resolveFirstElement,
} from './shared-element-utils.js';
import { defineTabTool } from './tool.js';
import { generateLocator } from './utils.js';
const pressKey = defineTabTool({
capability: 'core',
schema: {
name: 'browser_press_key',
title: 'Press a key',
description: 'Press a key on the keyboard',
inputSchema: z.object({
key: z.string().describe('Key to press'),
expectation: expectationSchema.describe(
'Page state config. Use batch_execute for multiple keys'
),
}),
type: 'destructive',
},
handle: async (tab, params, response) => {
response.addCode(`// Press ${params.key}`);
response.addCode(generateKeyPressCode(params.key));
await tab.waitForCompletion(async () => {
await tab.page.keyboard.press(params.key);
});
// If expectation includes snapshot, capture it now after navigation
await handleSnapshotExpectation(tab, params.expectation, response);
},
});
// Enhanced selector schema for browser tools
const selectorsSchema = z
.array(elementSelectorSchema)
.min(1)
.max(5)
.describe(
'Array of element selectors (max 5) supporting ref, role, CSS, or text-based selection'
);
const typeSchema = z.object({
selectors: selectorsSchema,
text: z.string().describe('Text to type into the element'),
submit: z.boolean().optional().describe('Press Enter after typing if true'),
slowly: z
.boolean()
.optional()
.describe('Type slowly for auto-complete if true'),
expectation: expectationSchema.describe(
'Page state config. Use batch_execute for forms'
),
});
const type = defineTabTool({
capability: 'core',
schema: {
name: 'browser_type',
title: 'Type text',
description: 'Type text into editable element',
inputSchema: typeSchema,
type: 'destructive',
},
handle: async (tab, params, response) => {
const { locator } = await resolveFirstElement(tab, params.selectors);
await tab.waitForCompletion(async () => {
if (params.slowly) {
response.addCode(
`await page.${await generateLocator(locator)}.pressSequentially(${quote(params.text)});`
);
await locator.pressSequentially(params.text);
} else {
response.addCode(
`await page.${await generateLocator(locator)}.fill(${quote(params.text)});`
);
await locator.fill(params.text);
}
if (params.submit) {
response.addCode(
`await page.${await generateLocator(locator)}.press('Enter');`
);
await locator.press('Enter');
}
});
await handleSnapshotExpectation(tab, params.expectation, response);
},
});
export default [pressKey, type];
