66 lines
2.3 KiB
TypeScript
66 lines
2.3 KiB
TypeScript
/**
|
|
* Copyright (c) Microsoft Corporation.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { z } from '../mcp/bundle.js';
|
|
import { defineTool } from './tool';
|
|
|
|
const wait = defineTool({
|
|
capability: 'core',
|
|
|
|
schema: {
|
|
name: 'browser_wait_for',
|
|
title: 'Wait for',
|
|
description: 'Wait for text to appear or disappear or a specified time to pass',
|
|
inputSchema: z.object({
|
|
time: z.number().optional().describe('The time to wait in seconds'),
|
|
text: z.string().optional().describe('The text to wait for'),
|
|
textGone: z.string().optional().describe('The text to wait for to disappear'),
|
|
}),
|
|
type: 'readOnly',
|
|
},
|
|
|
|
handle: async (context, params, response) => {
|
|
if (!params.text && !params.textGone && !params.time)
|
|
throw new Error('Either time, text or textGone must be provided');
|
|
|
|
if (params.time) {
|
|
response.addCode(`await new Promise(f => setTimeout(f, ${params.time!} * 1000));`);
|
|
await new Promise(f => setTimeout(f, Math.min(30000, params.time! * 1000)));
|
|
}
|
|
|
|
const tab = context.currentTabOrDie();
|
|
const locator = params.text ? tab.page.getByText(params.text).first() : undefined;
|
|
const goneLocator = params.textGone ? tab.page.getByText(params.textGone).first() : undefined;
|
|
|
|
if (goneLocator) {
|
|
response.addCode(`await page.getByText(${JSON.stringify(params.textGone)}).first().waitFor({ state: 'hidden' });`);
|
|
await goneLocator.waitFor({ state: 'hidden' });
|
|
}
|
|
|
|
if (locator) {
|
|
response.addCode(`await page.getByText(${JSON.stringify(params.text)}).first().waitFor({ state: 'visible' });`);
|
|
await locator.waitFor({ state: 'visible' });
|
|
}
|
|
|
|
response.addResult(`Waited for ${params.text || params.textGone || params.time}`);
|
|
response.setIncludeSnapshot();
|
|
},
|
|
});
|
|
|
|
export default [
|
|
wait,
|
|
];
|