feat: improve ast parser

originjs · Jul 5, 2021 · 17850c2 · 17850c2
1 parent beec2af
commit 17850c2
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/ast-parse/astParse.ts b/src/ast-parse/astParse.ts
@@ -28,9 +28,13 @@ export type ParsingResultOccurrence = {
   type: ParserType
 }
 
+export type ParsingResult = {
+  [name: string]: ParsingResultOccurrence[]
+}
+
 export function astParseRoot (rootDir: string) {
   const resolvedPaths : string[] = globby.sync(rootDir)
-  const parsingResults: any = {}
+  const parsingResults: ParsingResult = {}
   resolvedPaths.forEach(filePath => {
     // skip files in node_modules
     if (filePath.indexOf('/node_modules/') >= 0) {
@@ -84,15 +88,15 @@ export function astParseRoot (rootDir: string) {
       }
 
       // parse the file
-      const parsingResult = parser.astParse(fileInfo)
+      const parsingResult: ParsingResultOccurrence[] | null = parser.astParse(fileInfo)
       if (!parsingResult) {
         continue
       }
 
       if (!parsingResults[parser.parserType]) {
         parsingResults[parser.parserType] = []
       }
-      parsingResults[parser.parserType].push(parsingResult)
+      parsingResults[parser.parserType].push.apply(parsingResults[parser.parserType], parsingResult)
     }
   })