Handles arbitrary module identifiers

sidshank · sidshank · commit aa787a9cb1a2 · 2024-09-11T00:27:42.000Z
This commit enables support for arbitrary names for identifier and namespace imports and exports
diff --git a/javascript/extractor/src/com/semmle/ts/extractor/TypeScriptASTConverter.java b/javascript/extractor/src/com/semmle/ts/extractor/TypeScriptASTConverter.java
@@ -1203,27 +1203,35 @@ private Node convertExportDeclaration(JsonObject node, SourceLocation loc) throw
     Literal source = tryConvertChild(node, "moduleSpecifier", Literal.class);
     Expression attributes = convertChild(node, "attributes");
     if (hasChild(node, "exportClause")) {
+      JsonElement exportClauseNode = node.get("exportClause");
       boolean hasTypeKeyword = node.get("isTypeOnly").getAsBoolean();
+      boolean isNamespaceExportNode = hasKind(exportClauseNode, "NamespaceExport");
       List<ExportSpecifier> specifiers =
-          hasKind(node.get("exportClause"), "NamespaceExport")
+          isNamespaceExportNode
               ? Collections.singletonList(convertChild(node, "exportClause"))
-              : convertChildren(node.get("exportClause").getAsJsonObject(), "elements");
+              : convertChildren(exportClauseNode.getAsJsonObject(), "elements");
       return new ExportNamedDeclaration(loc, null, specifiers, source, attributes, hasTypeKeyword);
     } else {
       return new ExportAllDeclaration(loc, source, attributes);
     }
   }
 
   private Node convertExportSpecifier(JsonObject node, SourceLocation loc) throws ParseError {
+    Identifier local = convertChild(node, hasChild(node, "propertyName") ? "propertyName" : "name");
+    JsonObject exportedToken = node.get("name").getAsJsonObject();
+    Identifier exported = convertNodeAsIdentifier(exportedToken);
+
     return new ExportSpecifier(
         loc,
-        convertChild(node, hasChild(node, "propertyName") ? "propertyName" : "name"),
-        convertChild(node, "name"));
+        local,
+        exported);
   }
 
   private Node convertNamespaceExport(JsonObject node, SourceLocation loc) throws ParseError {
     // Convert the "* as ns" from an export declaration.
-    return new ExportNamespaceSpecifier(loc, convertChild(node, "name"));
+    JsonObject exportedNamespaceToken = node.get("name").getAsJsonObject();
+    Identifier exportedNamespaceIdentifier = convertNodeAsIdentifier(exportedNamespaceToken);
+    return new ExportNamespaceSpecifier(loc, exportedNamespaceIdentifier);
   }
 
   private Node convertExpressionStatement(JsonObject node, SourceLocation loc) throws ParseError {
@@ -1431,7 +1439,8 @@ private Node convertImportKeyword(SourceLocation loc) {
 
   private Node convertImportSpecifier(JsonObject node, SourceLocation loc) throws ParseError {
     boolean hasImported = hasChild(node, "propertyName");
-    Identifier imported = convertChild(node, hasImported ? "propertyName" : "name");
+    JsonObject importedToken = node.get(hasImported? "propertyName" : "name").getAsJsonObject();
+    Identifier imported = convertNodeAsIdentifier(importedToken);
     Identifier local = convertChild(node, "name");
     boolean isTypeOnly = node.get("isTypeOnly").getAsBoolean() == true;
     return new ImportSpecifier(loc, imported, local, isTypeOnly);
