chore: Add execute_script to code generator

Author: Winify

src/recording/code-generator.ts

@@ -48,7 +48,7 @@ function generateStep(step: RecordedStep, history: SessionHistory): string {
       return `const browser = await remote(${indentJson(config)});`;
     }
     case 'close_session':
-      return 'await browser.deleteSession();';
+      return '// Session closed';
     case 'navigate':
       return `await browser.url('${escapeStr(p.url)}');`;
     case 'click_element':
@@ -71,12 +71,22 @@ function generateStep(step: RecordedStep, history: SessionHistory): string {
         return `await browser.$('${escapeStr(p.sourceSelector)}').dragAndDrop(browser.$('${escapeStr(p.targetSelector)}'));`;
       }
       return `await browser.$('${escapeStr(p.sourceSelector)}').dragAndDrop({ x: ${p.x}, y: ${p.y} });`;
+    case 'execute_script': {
+      const scriptCode = `'${escapeStr(p.script)}'`;
+      const scriptArgs = (p.args as unknown[])?.length ? `, ${indentJson(p.args)}` : '';
+      return `await browser.execute(${scriptCode}${scriptArgs});`;
+    }
     default:
       return `// [unknown tool] ${step.tool}`;
   }
 }
 
 export function generateCode(history: SessionHistory): string {
-  const steps = history.steps.map(step => generateStep(step, history)).join('\n');
-  return `import { remote } from 'webdriverio';\n\n${steps}`;
+  const steps = history.steps
+    .map(step => generateStep(step, history))
+    .join('\n')
+    .split('\n')
+    .map(line => `  ${line}`)
+    .join('\n');
+  return `import { remote } from 'webdriverio';\n\ntry {\n${steps}\n} finally {\n  await browser.deleteSession();\n}`;
 }

src/server.ts

@@ -130,7 +130,7 @@ registerTool(rotateDeviceToolDefinition, rotateDeviceTool);
 registerTool(hideKeyboardToolDefinition, hideKeyboardTool);
 registerTool(setGeolocationToolDefinition, setGeolocationTool);
 
-registerTool(executeScriptToolDefinition, executeScriptTool);
+registerTool(executeScriptToolDefinition, withRecording('execute_script', executeScriptTool));
 registerTool(getElementsToolDefinition, getElementsTool);
 
 registerResource(sessionsIndexResource);

tests/recording/code-generator.test.ts

@@ -35,10 +35,14 @@ function makeHistory(steps: Partial<RecordedStep>[]): SessionHistory {
 }
 
 describe('generateCode - header', () => {
-  it('wraps output in remote() setup', () => {
+  it('wraps output in remote() setup with try/finally and deleteSession', () => {
     const code = generateCode(makeHistory([]));
     expect(code).toContain("import { remote } from 'webdriverio';");
     expect(code).toContain('browserName');
+    expect(code).toContain('try {');
+    expect(code).toContain('} finally {');
+    expect(code).toContain('  await browser.deleteSession();');
+    expect(code).toContain('}');
   });
 
   it('generates start_session (browser) using history.capabilities', () => {
@@ -177,6 +181,31 @@ describe('generateCode - tool mappings', () => {
     expect(code).toContain("await browser.execute('mobile: swipe', { direction: 'up' });");
   });
 
+  it('execute_script → browser.execute with single-quoted script string', () => {
+    const code = generateCode(makeHistory([{
+      tool: 'execute_script',
+      params: { script: 'return document.title' },
+    }]));
+    expect(code).toContain("await browser.execute('return document.title');");
+  });
+
+  it('execute_script escapes backslashes in script string', () => {
+    const code = generateCode(makeHistory([{
+      tool: 'execute_script',
+      params: { script: "return document.querySelector('[data-\\\\x]')" },
+    }]));
+    expect(code).toContain("await browser.execute('return document.querySelector(\\'[data-\\\\\\\\x]\\')');");
+  });
+
+  it('execute_script with args → browser.execute with args array', () => {
+    const code = generateCode(makeHistory([{
+      tool: 'execute_script',
+      params: { script: 'arguments[0].click()', args: ['#btn'] },
+    }]));
+    expect(code).toContain("await browser.execute('arguments[0].click()',");
+    expect(code).toContain('"#btn"');
+  });
+
   it('drag_and_drop (selector form) → $().dragAndDrop($())', () => {
     const code = generateCode(makeHistory([{
       tool: 'drag_and_drop',