Add generator support

nicolo-ribaudo · Aug 2, 2023 · 83b601c · 83b601c
1 parent cf0ad76
commit 83b601c
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 0 deletions.
diff --git a/packages/babel-generator/src/generators/modules.ts b/packages/babel-generator/src/generators/modules.ts
@@ -248,6 +248,10 @@ export function ImportDeclaration(this: Printer, node: t.ImportDeclaration) {
     this.noIndentInnerCommentsHere();
     this.word("module");
     this.space();
+  } else if (node.phase) {
+    this.noIndentInnerCommentsHere();
+    this.word(node.phase);
+    this.space();
   }
 
   const specifiers = node.specifiers.slice(0);
@@ -315,6 +319,10 @@ export function ImportNamespaceSpecifier(
 
 export function ImportExpression(this: Printer, node: t.ImportExpression) {
   this.word("import");
+  if (node.phase) {
+    this.token(".");
+    this.word(node.phase);
+  }
   this.token("(");
   this.print(node.source, node);
   if (node.options != null) {

diff --git a/packages/babel-generator/test/fixtures/import-phases/options.json b/packages/babel-generator/test/fixtures/import-phases/options.json
@@ -0,0 +1,4 @@
+{
+  "plugins": ["sourcePhaseImports"],
+  "parserOpts": { "createImportExpression": true }
+}
diff --git a/packages/babel-generator/test/fixtures/import-phases/source-declaration/input.js b/packages/babel-generator/test/fixtures/import-phases/source-declaration/input.js
@@ -0,0 +1 @@
+import source x from "y";
diff --git a/packages/babel-generator/test/fixtures/import-phases/source-declaration/output.js b/packages/babel-generator/test/fixtures/import-phases/source-declaration/output.js
@@ -0,0 +1 @@
+import source x from "y";
diff --git a/packages/babel-generator/test/fixtures/import-phases/source-expression/input.js b/packages/babel-generator/test/fixtures/import-phases/source-expression/input.js
@@ -0,0 +1 @@
+import.source("x");
diff --git a/packages/babel-generator/test/fixtures/import-phases/source-expression/output.js b/packages/babel-generator/test/fixtures/import-phases/source-expression/output.js
@@ -0,0 +1 @@
+import.source("x");