feat(typescript): add exports to name support

phodal · Apr 9, 2022 · ec6fafb · ec6fafb
1 parent 2bdd161
commit ec6fafb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/chapi-ast-typescript/src/main/kotlin/chapi/ast/typescriptast/TypeScriptFullIdentListener.kt b/chapi-ast-typescript/src/main/kotlin/chapi/ast/typescriptast/TypeScriptFullIdentListener.kt
@@ -963,6 +963,16 @@ class TypeScriptFullIdentListener(node: TSIdentify) : TypeScriptAstListener() {
         return args
     }
 
+    override fun enterExportDefaultDeclaration(ctx: TypeScriptParser.ExportDefaultDeclarationContext?) {
+        val singleExpr = ctx!!.singleExpression()
+        if (singleExpr != null) {
+            if (!singleExpr.text.contains("(")) {
+                currentNode.Exports += CodeExport(singleExpr.text)
+                defaultNode.Exports += CodeExport(singleExpr.text)
+            }
+        }
+    }
+
     fun getNodeInfo(): CodeContainer {
         for (entry in nodeMap) {
             codeContainer.DataStructures += entry.value

diff --git a/chapi-ast-typescript/src/test/kotlin/chapi/ast/typescriptast/TypeScriptAnalyserTest.kt b/chapi-ast-typescript/src/test/kotlin/chapi/ast/typescriptast/TypeScriptAnalyserTest.kt
@@ -57,6 +57,7 @@ internal class TypeScriptAnalyserTest {
         val codeFile = TypeScriptAnalyser().analysis(content, "BadSmellThreshold.tsx")
 
         assertEquals(codeFile.DataStructures.size, 1)
+        assertEquals(codeFile.DataStructures[0].Exports[0].Name, "BadSmellThreshold")
     }
 
     @Test
-Original file line number
+Diff line change
@@ Expand Up / @@ -57,6 +57,7 @@ internal class TypeScriptAnalyserTest { @@
             val codeFile = TypeScriptAnalyser().analysis(content, "BadSmellThreshold.tsx")
             assertEquals(codeFile.DataStructures.size, 1)
+            assertEquals(codeFile.DataStructures[0].Exports[0].Name, "BadSmellThreshold")
         }
         @Test
@@ Expand Down @@