Add support for printing ImportAttribute

babel · May 29, 2020 · 0eb8e6d · 0eb8e6d
1 parent 8e41f26
commit 0eb8e6d
Show file tree

Hide file tree

Showing 8 changed files with 53 additions and 0 deletions.
diff --git a/packages/babel-generator/src/generators/modules.js b/packages/babel-generator/src/generators/modules.js
@@ -179,9 +179,24 @@ export function ImportDeclaration(node: Object) {
   }
 
   this.print(node.source, node);
+
+  if (node.attributes?.length) {
+    this.space();
+    this.word("with");
+    this.space();
+    this.printList(node.attributes, node);
+  }
+
   this.semicolon();
 }
 
+export function ImportAttribute(node: Object) {
+  this.print(node.key);
+  this.token(":");
+  this.space();
+  this.print(node.value);
+}
+
 export function ImportNamespaceSpecifier(node: Object) {
   this.token("*");
   this.space();

diff --git a/packages/babel-generator/test/fixtures/types/ImportAttribute/input.js b/packages/babel-generator/test/fixtures/types/ImportAttribute/input.js
@@ -0,0 +1 @@
+import foo from "foo.json" with type: "json";
diff --git a/packages/babel-generator/test/fixtures/types/ImportAttribute/options.json b/packages/babel-generator/test/fixtures/types/ImportAttribute/options.json
@@ -0,0 +1,11 @@
+{
+  "plugins": [
+    [
+      "moduleAttributes",
+      {
+        "version": "may-2020"
+      }
+    ]
+  ],
+  "sourceType": "module"
+}
diff --git a/packages/babel-generator/test/fixtures/types/ImportAttribute/output.js b/packages/babel-generator/test/fixtures/types/ImportAttribute/output.js
@@ -0,0 +1 @@
+import foo from "foo.json" with type: "json";
diff --git a/packages/babel-types/src/asserts/generated/index.js b/packages/babel-types/src/asserts/generated/index.js
@@ -767,6 +767,9 @@ export function assertClassPrivateMethod(
 export function assertImport(node: Object, opts?: Object = {}): void {
   assert("Import", node, opts);
 }
+export function assertImportAttribute(node: Object, opts?: Object = {}): void {
+  assert("ImportAttribute", node, opts);
+}
 export function assertDecorator(node: Object, opts?: Object = {}): void {
   assert("Decorator", node, opts);
 }

diff --git a/packages/babel-types/src/builders/generated/index.js b/packages/babel-types/src/builders/generated/index.js
@@ -692,6 +692,10 @@ export function Import(...args: Array<any>): Object {
   return builder("Import", ...args);
 }
 export { Import as import };
+export function ImportAttribute(...args: Array<any>): Object {
+  return builder("ImportAttribute", ...args);
+}
+export { ImportAttribute as importAttribute };
 export function Decorator(...args: Array<any>): Object {
   return builder("Decorator", ...args);
 }

diff --git a/packages/babel-types/src/definitions/experimental.js b/packages/babel-types/src/definitions/experimental.js
@@ -224,6 +224,10 @@ defineType("Import", {
   aliases: ["Expression"],
 });
 
+defineType("ImportAttribute", {
+  visitor: ["key", "value"],
+});
+
 defineType("Decorator", {
   visitor: ["expression"],
   fields: {

diff --git a/packages/babel-types/src/validators/generated/index.js b/packages/babel-types/src/validators/generated/index.js
@@ -2441,6 +2441,20 @@ export function isImport(node: ?Object, opts?: Object): boolean {
 
   return false;
 }
+export function isImportAttribute(node: ?Object, opts?: Object): boolean {
+  if (!node) return false;
+
+  const nodeType = node.type;
+  if (nodeType === "ImportAttribute") {
+    if (typeof opts === "undefined") {
+      return true;
+    } else {
+      return shallowEqual(node, opts);
+    }
+  }
+
+  return false;
+}
 export function isDecorator(node: ?Object, opts?: Object): boolean {
   if (!node) return false;