Update src/utils/files.ts

Co-authored-by: Nikolay Vitkov <34244704+Lightning00Blade@users.noreply.github.com>

Author: OrKoN

src/tools/memory.ts

@@ -5,6 +5,7 @@
  */
 
 import {zod} from '../third_party/index.js';
+import {ensureExtension} from '../utils/files.js';
 
 import {ToolCategory} from './categories.js';
 import {definePageTool} from './ToolDefinition.js';
@@ -25,7 +26,7 @@ export const takeMemorySnapshot = definePageTool({
     const page = request.page;
 
     await page.pptrPage.captureHeapSnapshot({
-      path: request.params.filePath,
+      path: ensureExtension(request.params.filePath, '.heapsnapshot'),
     });
 
     response.appendResponseLine(

src/utils/files.ts

@@ -25,10 +25,10 @@ export async function saveTemporaryFile(
   }
 }
 
-export function ensureExtension(filepath: string, extension: string): string {
-  const normalizedExtension = extension.startsWith('.')
-    ? extension
-    : `.${extension}`;
+export function ensureExtension(
+  filepath: string,
+  extension: `.${string}`,
+): string {
   const ext = path.extname(filepath);
-  return filepath.slice(0, filepath.length - ext.length) + normalizedExtension;
+  return filepath.slice(0, filepath.length - ext.length) + extension;
 }

tests/utils/files.test.ts

@@ -19,7 +19,7 @@ describe('ensureExtension', () => {
   });
 
   it('should handle extension without a leading dot', () => {
-    assert.strictEqual(ensureExtension('filename', 'txt'), 'filename.txt');
+    assert.strictEqual(ensureExtension('filename', '.txt'), 'filename.txt');
   });
 
   it('should not add a second dot if already present', () => {