fix: reference to class expression in private method

packages/babel-helper-create-class-features-plugin/src/fields.ts

@@ -669,6 +669,16 @@ const thisContextVisitor = traverse.visitors.merge([
   environmentVisitor,
 ]);
 
+const innerReferencesVisitor = {
+  ReferencedIdentifier(path, state) {
+    const { name } = path.node;
+    if (path.scope.bindingIdentifierEquals(name, state.innerBinding)) {
+      state.needsClassRef = true;
+      path.replaceWith(t.cloneNode(state.classRef));
+    }
+  },
+};
+
 function replaceThisContext(
   path,
   ref,
@@ -677,7 +687,8 @@ function replaceThisContext(
   isStaticBlock,
   constantSuper,
 ) {
-  const state = { classRef: ref, needsClassRef: false };
+  const { innerBinding, ...classRef } = ref;
+  const state = { classRef, needsClassRef: false, innerBinding };
 
   const replacer = new ReplaceSupers({
     methodPath: path,
@@ -696,6 +707,11 @@ function replaceThisContext(
   if (isStaticBlock || path.isProperty()) {
     path.traverse(thisContextVisitor, state);
   }
+
+  if (state.classRef?.name !== innerBinding?.name) {
+    path.traverse(innerReferencesVisitor, state);
+  }
+
   return state.needsClassRef;
 }
 

packages/babel-helper-create-class-features-plugin/src/index.ts

@@ -168,15 +168,18 @@ export function createClassFeaturePlugin({
 
         if (!props.length && !isDecorated) return;
 
-        let ref;
-
+        let ref = undefined;
         if (path.isClassExpression() || !path.node.id) {
           nameFunction(path);
           ref = path.scope.generateUidIdentifier("class");
         } else {
           ref = t.cloneNode(path.node.id);
         }
 
+        if (ref !== undefined) {
+          ref.innerBinding = path.node.id;
+        }
+
         // NODE: These three functions don't support decorators yet,
         //       but verifyUsedFeatures throws if there are both
         //       decorators and private fields.

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-field/exec.js

@@ -0,0 +1,19 @@
+function wrapper(wc) {
+  return wc
+}
+
+const f = wrapper(class Foo {
+  static #x = Foo;
+  static y = Foo;
+
+  static extract() {
+    return {
+      x: Foo.#x,
+      y: Foo.y,
+    }
+  }
+});
+
+const { x, y } = f.extract();
+expect(x).toBe(f)
+expect(y).toBe(f)

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-field/input.js

@@ -0,0 +1,8 @@
+function wrapper(wc) {
+  return wc
+}
+
+const f = wrapper(class Foo {
+  static #x = Foo;
+  static y = Foo;
+});

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-field/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["proposal-class-properties", "transform-block-scoping"]
+}

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-field/output.js

@@ -0,0 +1,10 @@
+var _class, _temp, _x;
+
+function wrapper(wc) {
+  return wc;
+}
+
+var f = wrapper((_temp = _class = class Foo {}, _x = {
+  writable: true,
+  value: _class
+}, babelHelpers.defineProperty(_class, "y", _class), _temp));

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-method/exec.js

@@ -0,0 +1,35 @@
+function wrapper(wc) {
+  return wc
+}
+
+const f = wrapper(class Foo {
+  static #bar() {
+    return Foo;
+  }
+
+  static #method() {
+    return function inner() {
+      return Foo;
+    };
+  }
+  static #method_shadowed() {
+    new Foo();
+    return function inner() {
+      let Foo = 3;
+      return Foo;
+    }
+  }
+
+  static extract() {
+    return {
+      bar: Foo.#bar,
+      method: Foo.#method,
+      method_shadowed: Foo.#method_shadowed
+    }
+  }
+});
+
+const { bar, method, method_shadowed } = f.extract();
+expect(bar()).toBe(f)
+expect(method()()).toBe(f)
+expect(method_shadowed()()).toBe(3)

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-method/input.js

@@ -0,0 +1,22 @@
+function wrapper(wc) {
+  return wc
+}
+
+wrapper(class Foo {
+  static #bar() {
+    return Foo;
+  }
+
+  static #method() {
+    return function inner() {
+      return Foo;
+    };
+  }
+  static #method_shadowed() {
+    new Foo();
+    return function inner() {
+      let Foo = 3;
+      return Foo;
+    }
+  }
+});

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-method/options.json

@@ -0,0 +1,3 @@
+{
+  "plugins": ["proposal-class-properties", "transform-block-scoping"]
+}

packages/babel-plugin-proposal-class-properties/test/fixtures/private/static-self-method/output.js

@@ -0,0 +1,25 @@
+var _class;
+
+function wrapper(wc) {
+  return wc;
+}
+
+wrapper(_class = class Foo {});
+
+function _bar() {
+  return _class;
+}
+
+function _method() {
+  return function inner() {
+    return _class;
+  };
+}
+
+function _method_shadowed() {
+  new _class();
+  return function inner() {
+    var Foo = 3;
+    return Foo;
+  };
+}