chore: don't sanitize file extension away (#327)

src/tools/utils.ts

@@ -70,6 +70,13 @@ export async function waitForCompletion<R>(context: Context, page: playwright.Pa
   }
 }
 
-export function sanitizeForFilePath(s: string) {
+function sanitize(s: string) {
   return s.replace(/[\x00-\x2C\x2E-\x2F\x3A-\x40\x5B-\x60\x7B-\x7F]+/g, '-');
 }
+
+export function sanitizeForFilePath(s: string) {
+  const separator = s.lastIndexOf('.');
+  if (separator === -1)
+    return sanitize(s);
+  return sanitize(s.substring(0, separator)) + '.' + sanitize(s.substring(separator + 1));
+}

tests/files.spec.ts

@@ -110,14 +110,14 @@ test('clicking on download link emits download', async ({ startClient }, testInf
       url: 'data:text/html,<a href="data:text/plain,Hello world!" download="test.txt">Download</a>',
     },
   })).toContainTextContent('- link "Download" [ref=s1e3]');
-
+  await client.callTool({
-  await expect.poll(() => client.callTool({
     name: 'browser_click',
     arguments: {
       element: 'Download link',
       ref: 's1e3',
     },
-  })).toContainTextContent(`
+  });
+  await expect.poll(() => client.callTool({ name: 'browser_snapshot', arguments: {} })).toContainTextContent(`
 ### Downloads
-- Downloaded file test.txt to ${path.join(outputDir, 'test-txt')}`);
+- Downloaded file test.txt to ${path.join(outputDir, 'test.txt')}`);
 });