Fix missing inner comments in class expressions (#13821)

babel · Oct 6, 2021 · 58a65e9 · 58a65e9
1 parent 7acc68a
commit 58a65e9
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/packages/babel-generator/src/generators/classes.ts b/packages/babel-generator/src/generators/classes.ts
@@ -31,6 +31,7 @@ export function ClassDeclaration(
   }
 
   this.word("class");
+  this.printInnerComments(node);
 
   if (node.id) {
     this.space();

diff --git a/packages/babel-generator/test/fixtures/comments/class-comments/input.js b/packages/babel-generator/test/fixtures/comments/class-comments/input.js
@@ -0,0 +1,7 @@
+const A = class /* before body */ { /* inside body */ };
+const B = class /* before name */ X /* before body */ { /* inside body */ };
+const C = class /* before extends */ extends /* after extends */ X /* before body */ {/* inside body */};
+const D = class /* before name */ X /* before extends */ extends /* after extends */ X /* before body */ {/* inside body */};
+
+class /* before name */ E /* before body */ { /* inside body */ }
+class /* before name */ F /* before extends */ extends /* after extends */ X /* before body */ {/* inside body */}
diff --git a/packages/babel-generator/test/fixtures/comments/class-comments/options.json b/packages/babel-generator/test/fixtures/comments/class-comments/options.json
@@ -0,0 +1,3 @@
+{
+  "retainLines": true
+}
diff --git a/packages/babel-generator/test/fixtures/comments/class-comments/output.js b/packages/babel-generator/test/fixtures/comments/class-comments/output.js
@@ -0,0 +1,7 @@
+const A = class /* before body */{/* inside body */};
+const B = class /* before name */X /* before body */ {/* inside body */};
+const C = class /* before extends */ extends /* after extends */X /* before body */ {/* inside body */};
+const D = class /* before name */X /* before extends */ extends /* after extends */X /* before body */ {/* inside body */};
+
+class /* before name */E /* before body */ {/* inside body */}
+class /* before name */F /* before extends */ extends /* after extends */X /* before body */ {/* inside body */}