chore: introduce modal states (#204)

src/tools/files.ts

@@ -34,16 +34,20 @@ const uploadFile: ToolFactory = captureSnapshot => ({
     const validatedParams = uploadFileSchema.parse(params);
     const tab = context.currentTab();
     return await tab.runAndWait(async () => {
-      await tab.submitFileChooser(validatedParams.paths);
+      const modalState = context.modalStates().find(state => state.type === 'fileChooser');
+      if (!modalState)
+        throw new Error('No file chooser visible');
+      await modalState.fileChooser.setFiles(validatedParams.paths);
+      context.clearModalState(modalState);
       const code = [
         `// <internal code to chose files ${validatedParams.paths.join(', ')}`,
       ];
       return { code };
     }, {
       captureSnapshot,
-      noClearFileChooser: true,
     });
   },
+  clearsModalState: 'fileChooser',
 });
 
 export default (captureSnapshot: boolean) => [

src/tools/tool.ts

@@ -17,7 +17,7 @@
 import type { ImageContent, TextContent } from '@modelcontextprotocol/sdk/types';
 import type { JsonSchema7Type } from 'zod-to-json-schema';
 import type { Context } from '../context';
-
+import type * as playwright from 'playwright';
 export type ToolCapability = 'core' | 'tabs' | 'pdf' | 'history' | 'wait' | 'files' | 'install';
 
 export type ToolSchema = {
@@ -26,6 +26,14 @@ export type ToolSchema = {
   inputSchema: JsonSchema7Type;
 };
 
+export type FileUploadModalState = {
+  type: 'fileChooser';
+  description: string;
+  fileChooser: playwright.FileChooser;
+};
+
+export type ModalState = FileUploadModalState;
+
 export type ToolResult = {
   content: (ImageContent | TextContent)[];
   isError?: boolean;
@@ -34,6 +42,7 @@ export type ToolResult = {
 export type Tool = {
   capability: ToolCapability;
   schema: ToolSchema;
+  clearsModalState?: ModalState['type'];
   handle: (context: Context, params?: Record<string, any>) => Promise<ToolResult>;
 };