fix: clone comments in cloneNode (#13178)

babel · Apr 20, 2021 · ab06cca · ab06cca
1 parent 86d9cf5
commit ab06cca
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 16 deletions.
diff --git a/packages/babel-types/src/clone/cloneNode.ts b/packages/babel-types/src/clone/cloneNode.ts
@@ -102,25 +102,18 @@ export default function cloneNode<T extends t.Node>(
   return newNode;
 }
 
-function cloneCommentsWithoutLoc<T extends t.Comment>(
-  comments: ReadonlyArray<T>,
-): T[] {
-  return comments.map(
-    ({ type, value }) =>
-      ({
-        type,
-        value,
-        loc: null,
-      } as T),
-  );
-}
-
 function maybeCloneComments<T extends t.Comment>(
   comments: ReadonlyArray<T> | null,
   deep: boolean,
   withoutLoc: boolean,
 ): ReadonlyArray<T> | null {
-  return deep && withoutLoc && comments
-    ? cloneCommentsWithoutLoc(comments)
-    : comments;
+  if (!comments || !deep) {
+    return comments;
+  }
+  return comments.map(({ type, value, loc }) => {
+    if (withoutLoc) {
+      return { type, value, loc: null } as T;
+    }
+    return { type, value, loc } as T;
+  });
 }
diff --git a/packages/babel-types/test/cloning.js b/packages/babel-types/test/cloning.js
@@ -120,6 +120,9 @@ describe("cloneNode", function () {
     node.declarations[0].id.loc = {};
 
     const cloned = t.cloneNode(node, /* deep */ true, /* withoutLoc */ false);
+    expect(cloned.declarations[0].id.leadingComments[0]).not.toBe(
+      node.declarations[0].id.leadingComments[0],
+    );
     expect(cloned.declarations[0].id.leadingComments[0].loc).toBe(
       node.declarations[0].id.leadingComments[0].loc,
     );