Pass name of modified export to runSetters

Author: zodern

lib/assignment-visitor.js

@@ -96,23 +96,30 @@ function assignmentHelper(visitor, path, childName) {
       visitor.exportedLocalNames[name] === true &&
       inModuleScope(name)
     ) {
-      wrap(visitor, path);
+      wrap(visitor, path, name);
       break;
     }
   }
 }
 
-function wrap(visitor, path) {
+function wrap(visitor, path, name) {
   const value = path.getValue();
 
   if (visitor.magicString !== null) {
+    let end = name ? `,"${name}")` : ')';
+
     visitor.magicString.prependRight(
       value.start,
       visitor.moduleAlias + ".runSetters("
-    ).appendLeft(value.end, ")");
+    ).appendLeft(value.end, end);
   }
 
   if (visitor.modifyAST) {
+    let args = name ? [value, {
+      type: "StringLiteral",
+      value: name
+    }] : [value];
+
     path.replace({
       type: "CallExpression",
       callee: {
@@ -126,7 +133,7 @@ function wrap(visitor, path) {
           name: "runSetters"
         }
       },
-      arguments: [value]
+      arguments: args
     });
   }
 }

lib/runtime/index.js

@@ -117,8 +117,12 @@ function moduleExportAs(name) {
 // the module system is about to return module.exports from require. This
 // might happen more than once per module, in case of dependency cycles,
 // so we want Module.prototype.runSetters to run each time.
-function runSetters(valueToPassThrough) {
-  Entry.getOrCreate(this.id, this).runSetters();
+function runSetters(valueToPassThrough, names) {
+  if (typeof names === 'string') {
+    names = [names];
+  }
+
+  Entry.getOrCreate(this.id, this).runSetters(names, true);
 
   // Assignments to exported local variables get wrapped with calls to
   // module.runSetters, so module.runSetters returns the

test/output/declarations-basic/expected.js

@@ -7,4 +7,4 @@ function d() {
   return b;
 };
 
-module.runSetters(c = "c");
+module.runSetters(c = "c","c");

test/output/live/expected.js

@@ -1,9 +1,9 @@
 "use strict";module.export({value:()=>value,reset:()=>reset,add:()=>add});var value = reset();
 
 function reset() {
-  return module.runSetters(value = 0);
+  return module.runSetters(value = 0,"value");
 }
 
 function add(x) {
-  module.runSetters(value += x);
+  module.runSetters(value += x,"value");
 };