Fix export bindings not updated by 'for ... in' and 'for ... of'

packages/babel-helper-module-transforms/src/rewrite-live-references.js

@@ -306,4 +306,41 @@ const rewriteReferencesVisitor = {
       }
     },
   },
+  "ForOfStatement|ForInStatement"(path) {
+    const { scope, node } = path;
+    const { left } = node;
+    const { exported, scope: programScope } = this;
+
+    // if it's a variable declaration, such as for (let {foo} of []) {}
+    // then no transformation is needed
+    if (!t.isVariableDeclaration(left)) {
+      let didTransform = false;
+      const bodyPath = path.get("body");
+      const loopBodyScope = bodyPath.scope;
+      for (const name of Object.keys(t.getOuterBindingIdentifiers(left))) {
+        if (
+          exported.get(name) &&
+          programScope.getBinding(name) === scope.getBinding(name)
+        ) {
+          didTransform = true;
+          if (loopBodyScope.hasOwnBinding(name)) {
+            loopBodyScope.rename(name);
+          }
+        }
+      }
+      if (!didTransform) {
+        return;
+      }
+      const newLoopId = scope.generateUidIdentifier();
+      bodyPath.unshiftContainer(
+        "body",
+        t.expressionStatement(t.assignmentExpression("=", left, newLoopId)),
+      );
+      path
+        .get("left")
+        .replaceWith(
+          t.variableDeclaration("let", [t.variableDeclarator(newLoopId)]),
+        );
+    }
+  },
 };

packages/babel-plugin-transform-modules-commonjs/test/fixtures/misc/for-of-in-export/input.mjs

@@ -0,0 +1,22 @@
+export let foo;
+export {foo as bar}
+
+for (foo of []) {}
+for (foo in []) {}
+for (foo of []) {
+    let foo;
+}
+for ({foo} of []) {}
+for ({foo} of []) {
+    let foo;
+}
+for ({test: {foo}} of []) {}
+for ([foo, [...foo]] of []) {}
+for ([foo, [...foo]] of []) {
+    let foo;
+}
+
+{
+    let foo;
+    for(foo of []) {}
+}

packages/babel-plugin-transform-modules-commonjs/test/fixtures/misc/for-of-in-export/output.js

@@ -0,0 +1,65 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.bar = exports.foo = void 0;
+let foo;
+exports.bar = exports.foo = foo;
+
+for (let _temp of []) {
+  exports.bar = exports.foo = foo = _temp;
+}
+
+for (let _temp2 in []) {
+  exports.bar = exports.foo = foo = _temp2;
+}
+
+for (let _temp3 of []) {
+  exports.bar = exports.foo = foo = _temp3;
+
+  let _foo;
+}
+
+for (let _temp4 of []) {
+  ({
+    foo
+  } = _temp4);
+  exports.bar = exports.foo = foo;
+}
+
+for (let _temp5 of []) {
+  ({
+    foo
+  } = _temp5);
+  exports.bar = exports.foo = foo;
+
+  let _foo2;
+}
+
+for (let _temp6 of []) {
+  ({
+    test: {
+      foo
+    }
+  } = _temp6);
+  exports.bar = exports.foo = foo;
+}
+
+for (let _temp7 of []) {
+  [foo, [...foo]] = _temp7;
+  exports.bar = exports.foo = foo;
+}
+
+for (let _temp8 of []) {
+  [foo, [...foo]] = _temp8;
+  exports.bar = exports.foo = foo;
+
+  let _foo3;
+}
+
+{
+  let foo;
+
+  for (foo of []) {}
+}