add transformer for untested file "all" options

cenfun · Sep 15, 2024 · 5f60e4e · 5f60e4e
1 parent 3cc04f3
commit 5f60e4e
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 12 deletions.
diff --git a/lib/converter/untested.js b/lib/converter/untested.js
@@ -64,12 +64,14 @@ const resolveAllOptions = (input) => {
 
     let dir;
     let filter;
+    let transformer;
 
     if (typeof input === 'string') {
         const obj = Util.strToObj(input);
         if (obj) {
             dir = obj.dir;
             filter = obj.filter;
+            transformer = obj.transformer;
         } else {
             dir = input;
         }
@@ -78,6 +80,7 @@ const resolveAllOptions = (input) => {
     } else {
         dir = input.dir;
         filter = input.filter;
+        transformer = input.transformer;
     }
 
     const dirList = resolveAllDirList(dir);
@@ -86,10 +89,12 @@ const resolveAllOptions = (input) => {
     }
 
     const fileFilter = resolveAllFilter(filter);
+    const fileTransformer = typeof transformer === 'function' ? transformer : () => {};
 
     return {
         dirList,
-        fileFilter
+        fileFilter,
+        fileTransformer
     };
 };
 
@@ -216,12 +221,7 @@ const getUntestedCoverageData = async (emptyList, options, coverageType) => {
     return emptyCoverageList;
 };
 
-const getEmptyCoverages = async (fileList, options, coverageType) => {
-
-    let onEmpty = options.onEmpty;
-    if (typeof onEmpty !== 'function') {
-        onEmpty = () => {};
-    }
+const getEmptyCoverages = async (fileList, options, coverageType, fileTransformer) => {
 
     const emptyList = [];
 
@@ -241,9 +241,9 @@ const getEmptyCoverages = async (fileList, options, coverageType) => {
             source
         };
 
-        await onEmpty(entryFile, coverageType);
+        await fileTransformer(entryFile, coverageType);
 
-        // after onEmpty
+        // after transformer
         entryFile.id = Util.calculateSha1(entryFile.sourcePath + entryFile.source);
 
         emptyList.push(entryFile);
@@ -259,7 +259,9 @@ const getUntestedList = (testedMap, options, coverageType = 'v8') => {
     if (!allOptions) {
         return;
     }
-    const { dirList, fileFilter } = allOptions;
+    const {
+        dirList, fileFilter, fileTransformer
+    } = allOptions;
     const fileList = [];
     dirList.forEach((dir) => {
         Util.forEachFile(dir, [], (fileName, fileDir) => {
@@ -283,7 +285,7 @@ const getUntestedList = (testedMap, options, coverageType = 'v8') => {
         return;
     }
 
-    return getEmptyCoverages(fileList, options, coverageType);
+    return getEmptyCoverages(fileList, options, coverageType, fileTransformer);
 
 };
 

diff --git a/lib/default/options.js b/lib/default/options.js
@@ -67,7 +67,8 @@ module.exports = {
     // options for adding empty coverage for all files
     // all: {
     //     dir: ['src'],
-    //     filter: (sourcePath) => true
+    //     filter: (sourcePath) => true,
+    //     transformer: (entry) => {}
     // },
     all: null,
 

diff --git a/lib/index.d.ts b/lib/index.d.ts
@@ -368,6 +368,14 @@ declare namespace MCR {
             filter?: string | {
                 [pattern: string]: "js" | "css" | boolean;
             } | ((filePath: string) => "js" | "css" | boolean);
+
+            /**
+             * the file transformer for source and sourceMap
+             * some of untested files like .ts/.jsx/.vue can not be parsed to AST directly by acorn
+             * so this is the function which can transform the original source to generated source and sourceMap
+             */
+            transformer?: (entry: any) => Promise<void>;
+
         };
 
         /** (V8 only) {boolean} Enable/Disable ignoring uncovered codes with the special comments: v8 ignore next/next N/start/stop */