babel-plugin-react-compiler 0.0.0-experimental-7f3c872-20250318 → 0.0.0-experimental-794db7b-20250319

package/dist/index.js

@@ -139766,7 +139766,8 @@ function collectHoistablePropertyLoads(fn, temporaries, hoistableFromOptionals)
     knownImmutableIdentifiers,
     hoistableFromOptionals,
     registry,
-    nestedFnImmutableContext: null
+    nestedFnImmutableContext: null,
+    assumedInvokedFns: fn.env.config.enableTreatFunctionDepsAsConditional ? /* @__PURE__ */ new Set() : getAssumedInvokedFunctions(fn)
   });
 }
 function collectHoistablePropertyLoadsImpl(fn, context) {
@@ -139904,23 +139905,25 @@ function collectNonNullsInBlocks(fn, context) {
       if (maybeNonNull != null && isImmutableAtInstr(maybeNonNull.fullPath.identifier, instr.id, context)) {
         assumedNonNullObjects.add(maybeNonNull);
       }
-      if ((instr.value.kind === "FunctionExpression" || instr.value.kind === "ObjectMethod") && !fn.env.config.enableTreatFunctionDepsAsConditional) {
+      if (instr.value.kind === "FunctionExpression") {
         const innerFn = instr.value.loweredFunc;
-        const innerHoistableMap = collectHoistablePropertyLoadsImpl(
-          innerFn.func,
-          __spreadProps(__spreadValues({}, context), {
-            nestedFnImmutableContext: (_a = context.nestedFnImmutableContext) != null ? _a : new Set(
-              innerFn.func.context.filter(
-                (place) => isImmutableAtInstr(place.identifier, instr.id, context)
-              ).map((place) => place.identifier.id)
-            )
-          })
-        );
-        const innerHoistables = assertNonNull(
-          innerHoistableMap.get(innerFn.func.body.entry)
-        );
-        for (const entry of innerHoistables.assumedNonNullObjects) {
-          assumedNonNullObjects.add(entry);
+        if (context.assumedInvokedFns.has(innerFn)) {
+          const innerHoistableMap = collectHoistablePropertyLoadsImpl(
+            innerFn.func,
+            __spreadProps(__spreadValues({}, context), {
+              nestedFnImmutableContext: (_a = context.nestedFnImmutableContext) != null ? _a : new Set(
+                innerFn.func.context.filter(
+                  (place) => isImmutableAtInstr(place.identifier, instr.id, context)
+                ).map((place) => place.identifier.id)
+              )
+            })
+          );
+          const innerHoistables = assertNonNull(
+            innerHoistableMap.get(innerFn.func.body.entry)
+          );
+          for (const entry of innerHoistables.assumedNonNullObjects) {
+            assumedNonNullObjects.add(entry);
+          }
         }
       }
     }
@@ -140048,6 +140051,96 @@ function reduceMaybeOptionalChains(nodes, registry) {
     }
   } while (changed);
 }
+function getAssumedInvokedFunctions(fn, temporaries = /* @__PURE__ */ new Map()) {
+  var _a;
+  const hoistableFunctions = /* @__PURE__ */ new Set();
+  for (const block of fn.body.blocks.values()) {
+    for (const { lvalue, value } of block.instructions) {
+      if (value.kind === "FunctionExpression") {
+        temporaries.set(lvalue.identifier.id, {
+          fn: value.loweredFunc,
+          mayInvoke: /* @__PURE__ */ new Set()
+        });
+      } else if (value.kind === "StoreLocal") {
+        const lvalue2 = value.lvalue.place.identifier;
+        const maybeLoweredFunc = temporaries.get(value.value.identifier.id);
+        if (maybeLoweredFunc != null) {
+          temporaries.set(lvalue2.id, maybeLoweredFunc);
+        }
+      } else if (value.kind === "LoadLocal") {
+        const maybeLoweredFunc = temporaries.get(value.place.identifier.id);
+        if (maybeLoweredFunc != null) {
+          temporaries.set(lvalue.identifier.id, maybeLoweredFunc);
+        }
+      }
+    }
+  }
+  for (const block of fn.body.blocks.values()) {
+    for (const { lvalue, value } of block.instructions) {
+      if (value.kind === "CallExpression") {
+        const callee = value.callee;
+        const maybeHook = getHookKind(fn.env, callee.identifier);
+        const maybeLoweredFunc = temporaries.get(callee.identifier.id);
+        if (maybeLoweredFunc != null) {
+          hoistableFunctions.add(maybeLoweredFunc.fn);
+        } else if (maybeHook != null) {
+          for (const arg of value.args) {
+            if (arg.kind === "Identifier") {
+              const maybeLoweredFunc2 = temporaries.get(arg.identifier.id);
+              if (maybeLoweredFunc2 != null) {
+                hoistableFunctions.add(maybeLoweredFunc2.fn);
+              }
+            }
+          }
+        }
+      } else if (value.kind === "JsxExpression") {
+        for (const attr of value.props) {
+          if (attr.kind === "JsxSpreadAttribute") {
+            continue;
+          }
+          const maybeLoweredFunc = temporaries.get(attr.place.identifier.id);
+          if (maybeLoweredFunc != null) {
+            hoistableFunctions.add(maybeLoweredFunc.fn);
+          }
+        }
+        for (const child of (_a = value.children) != null ? _a : []) {
+          const maybeLoweredFunc = temporaries.get(child.identifier.id);
+          if (maybeLoweredFunc != null) {
+            hoistableFunctions.add(maybeLoweredFunc.fn);
+          }
+        }
+      } else if (value.kind === "FunctionExpression") {
+        const loweredFunc = value.loweredFunc.func;
+        const lambdasCalled = getAssumedInvokedFunctions(
+          loweredFunc,
+          temporaries
+        );
+        const maybeLoweredFunc = temporaries.get(lvalue.identifier.id);
+        if (maybeLoweredFunc != null) {
+          for (const called of lambdasCalled) {
+            maybeLoweredFunc.mayInvoke.add(called);
+          }
+        }
+      }
+    }
+    if (block.terminal.kind === "return") {
+      const maybeLoweredFunc = temporaries.get(
+        block.terminal.value.identifier.id
+      );
+      if (maybeLoweredFunc != null) {
+        hoistableFunctions.add(maybeLoweredFunc.fn);
+      }
+    }
+  }
+  for (const [_, { fn: fn2, mayInvoke }] of temporaries) {
+    if (hoistableFunctions.has(fn2)) {
+      for (const called of mayInvoke) {
+        hoistableFunctions.add(called);
+      }
+    }
+  }
+  return hoistableFunctions;
+}
 
 // src/Utils/Stack.ts
 function empty() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "babel-plugin-react-compiler",
-  "version": "0.0.0-experimental-7f3c872-20250318",
+  "version": "0.0.0-experimental-794db7b-20250319",
   "description": "Babel plugin for React Compiler.",
   "main": "dist/index.js",
   "license": "MIT",