feat: add extensions filter for ast transformations

originjs · Jul 2, 2021 · a8a7380 · a8a7380
1 parent 83d9494
commit a8a7380
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 7 deletions.
diff --git a/src/ast-parse/astParse.ts b/src/ast-parse/astParse.ts
@@ -3,8 +3,6 @@ import { vueSfcAstParser } from '@originjs/vue-sfc-ast-parser'
 import * as globby from 'globby'
 import fs from 'fs'
 
-const extensions = ['.vue']
-
 export function astParseRoot (rootDir: string) {
   const resolvedPaths : string[] = globby.sync(rootDir)
   resolvedPaths.forEach(filePath => {
@@ -14,9 +12,6 @@ export function astParseRoot (rootDir: string) {
     }
 
     const extension = (/\.([^.]*)$/.exec(filePath) || [])[0]
-    if (!extensions.includes(extension)) {
-      return
-    }
 
     let fileChanged: boolean = false
     let context = parseVueSfc(filePath)
@@ -26,6 +21,14 @@ export function astParseRoot (rootDir: string) {
     // iter all transformations
     for (const key in transformationMap) {
       const transformation = transformationMap[key]
+
+      // filter by file extension
+      const extensions: string[] = transformation.extensions
+      if (!extensions.includes(extension)) {
+        continue
+      }
+
+      // execute the transformation
       tempTransformationResult = transformation.transformAST(context)
       if (tempTransformationResult == null) {
         continue

diff --git a/src/ast-parse/transformations/addJsxTransformation.ts b/src/ast-parse/transformations/addJsxTransformation.ts
@@ -13,4 +13,6 @@ export const transformAST:ASTTransformation = (context: Context) => {
   return stringifyDescriptor(descriptor);
 }
 
-export const needReparse : boolean = false
+export const needReparse: boolean = false
+
+export const extensions: string[] = ['vue']
diff --git a/src/ast-parse/transformations/index.ts b/src/ast-parse/transformations/index.ts
@@ -19,7 +19,8 @@ export type ASTTransformation<Params = void> = {
 export const transformationMap: {
   [name: string]: {
     transformAST: ASTTransformation,
-    needReparse: boolean
+    needReparse: boolean,
+    extensions: string[]
   }
 } = {
   addJsxTransformation: require('./addJsxTransformation'),

diff --git a/src/ast-parse/transformations/removeHtmlLangInTemplateTransformation.ts b/src/ast-parse/transformations/removeHtmlLangInTemplateTransformation.ts
@@ -14,3 +14,5 @@ export const transformAST:ASTTransformation = (context: Context) => {
 }
 
 export const needReparse : boolean = false
+
+export const extensions: string[] = ['vue']