mirror of
https://github.com/microsoft/playwright-mcp.git
synced 2025-07-26 08:32:26 +08:00
90 lines
2.9 KiB
TypeScript
90 lines
2.9 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 type { ImageContent, TextContent } from '@modelcontextprotocol/sdk/types.js';
|
|
import type { z } from 'zod';
|
|
import type { Context } from '../context.js';
|
|
import type * as playwright from 'playwright';
|
|
import type { ToolCapability } from '../../config.js';
|
|
import type { Tab } from '../tab.js';
|
|
|
|
export type ToolSchema<Input extends InputType> = {
|
|
name: string;
|
|
title: string;
|
|
description: string;
|
|
inputSchema: Input;
|
|
type: 'readOnly' | 'destructive';
|
|
};
|
|
|
|
type InputType = z.Schema;
|
|
|
|
export type FileUploadModalState = {
|
|
type: 'fileChooser';
|
|
description: string;
|
|
fileChooser: playwright.FileChooser;
|
|
};
|
|
|
|
export type DialogModalState = {
|
|
type: 'dialog';
|
|
description: string;
|
|
dialog: playwright.Dialog;
|
|
};
|
|
|
|
export type ModalState = FileUploadModalState | DialogModalState;
|
|
|
|
export type ToolActionResult = { content?: (ImageContent | TextContent)[] } | undefined | void;
|
|
|
|
export type ToolResult = {
|
|
code: string[];
|
|
action?: () => Promise<ToolActionResult>;
|
|
captureSnapshot: boolean;
|
|
waitForNetwork: boolean;
|
|
resultOverride?: ToolActionResult;
|
|
};
|
|
|
|
export type Tool<Input extends InputType = InputType> = {
|
|
capability: ToolCapability;
|
|
schema: ToolSchema<Input>;
|
|
clearsModalState?: ModalState['type'];
|
|
handle: (context: Context, params: z.output<Input>) => Promise<ToolResult>;
|
|
};
|
|
|
|
export function defineTool<Input extends InputType>(tool: Tool<Input>): Tool<Input> {
|
|
return tool;
|
|
}
|
|
|
|
export type TabTool<Input extends InputType = InputType> = {
|
|
capability: ToolCapability;
|
|
schema: ToolSchema<Input>;
|
|
clearsModalState?: ModalState['type'];
|
|
handle: (tab: Tab, params: z.output<Input>) => Promise<ToolResult>;
|
|
};
|
|
|
|
export function defineTabTool<Input extends InputType>(tool: TabTool<Input>): Tool<Input> {
|
|
return {
|
|
...tool,
|
|
handle: async (context, params) => {
|
|
const tab = context.currentTabOrDie();
|
|
const modalStates = tab.modalStates().map(state => state.type);
|
|
if (tool.clearsModalState && !modalStates.includes(tool.clearsModalState))
|
|
throw new Error(`The tool "${tool.schema.name}" can only be used when there is related modal state present.\n` + tab.modalStatesMarkdown().join('\n'));
|
|
if (!tool.clearsModalState && modalStates.length)
|
|
throw new Error(`Tool "${tool.schema.name}" does not handle the modal state.\n` + tab.modalStatesMarkdown().join('\n'));
|
|
return tool.handle(tab, params);
|
|
},
|
|
};
|
|
}
|