compiler: Allow global mutation in jsx props

[ghstack-poisoned]

compiler/packages/babel-plugin-react-compiler/src/Inference/InferReferenceEffects.ts

@@ -1103,13 +1103,56 @@ function inferBlock(
         break;
       }
       case "JsxExpression": {
-        valueKind = {
+        if (instrValue.tag.kind === "Identifier") {
+          state.referenceAndRecordEffects(
+            instrValue.tag,
+            Effect.Freeze,
+            ValueReason.JsxCaptured,
+            functionEffects
+          );
+        }
+        if (instrValue.children !== null) {
+          for (const child of instrValue.children) {
+            state.referenceAndRecordEffects(
+              child,
+              Effect.Freeze,
+              ValueReason.JsxCaptured,
+              functionEffects
+            );
+          }
+        }
+        for (const attr of instrValue.props) {
+          if (attr.kind === "JsxSpreadAttribute") {
+            state.referenceAndRecordEffects(
+              attr.argument,
+              Effect.Freeze,
+              ValueReason.JsxCaptured,
+              functionEffects
+            );
+          } else {
+            const propEffects: Array<FunctionEffect> = [];
+            state.referenceAndRecordEffects(
+              attr.place,
+              Effect.Freeze,
+              ValueReason.JsxCaptured,
+              propEffects
+            );
+            functionEffects.push(
+              ...propEffects.filter(
+                (propEffect) => propEffect.kind !== "GlobalMutation"
+              )
+            );
+          }
+        }
+
+        state.initialize(instrValue, {
           kind: ValueKind.Frozen,
           reason: new Set([ValueReason.Other]),
           context: new Set(),
-        };
-        effect = { kind: Effect.Freeze, reason: ValueReason.JsxCaptured };
-        break;
+        });
+        state.define(instr.lvalue, instrValue);
+        instr.lvalue.effect = Effect.ConditionallyMutate;
+        continue;
       }
       case "JsxFragment": {
         valueKind = {

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/allow-modify-global-in-callback-jsx.expect.md

@@ -0,0 +1,86 @@
+
+## Input
+
+```javascript
+import { useMemo } from "react";
+
+const someGlobal = { value: 0 };
+
+function Component({ value }) {
+  const onClick = () => {
+    someGlobal.value = value;
+  };
+  return useMemo(() => {
+    return <div onClick={onClick}>{someGlobal.value}</div>;
+  }, []);
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{ value: 0 }],
+  sequentialRenders: [
+    { value: 1 },
+    { value: 1 },
+    { value: 42 },
+    { value: 42 },
+    { value: 0 },
+  ],
+};
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+import { useMemo } from "react";
+
+const someGlobal = { value: 0 };
+
+function Component(t0) {
+  const $ = _c(4);
+  const { value } = t0;
+  let t1;
+  if ($[0] !== value) {
+    t1 = () => {
+      someGlobal.value = value;
+    };
+    $[0] = value;
+    $[1] = t1;
+  } else {
+    t1 = $[1];
+  }
+  const onClick = t1;
+  let t2;
+  let t3;
+  if ($[2] !== onClick) {
+    t3 = <div onClick={onClick}>{someGlobal.value}</div>;
+    $[2] = onClick;
+    $[3] = t3;
+  } else {
+    t3 = $[3];
+  }
+  t2 = t3;
+  return t2;
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{ value: 0 }],
+  sequentialRenders: [
+    { value: 1 },
+    { value: 1 },
+    { value: 42 },
+    { value: 42 },
+    { value: 0 },
+  ],
+};
+
+```
+
+### Eval output
+(kind: ok) <div>0</div>
+<div>0</div>
+<div>0</div>
+<div>0</div>
+<div>0</div>

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/allow-modify-global-in-callback-jsx.js

@@ -0,0 +1,24 @@
+import { useMemo } from "react";
+
+const someGlobal = { value: 0 };
+
+function Component({ value }) {
+  const onClick = () => {
+    someGlobal.value = value;
+  };
+  return useMemo(() => {
+    return <div onClick={onClick}>{someGlobal.value}</div>;
+  }, []);
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{ value: 0 }],
+  sequentialRenders: [
+    { value: 1 },
+    { value: 1 },
+    { value: 42 },
+    { value: 42 },
+    { value: 0 },
+  ],
+};

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/allow-reassignment-to-global-function-jsx-prop.expect.md

@@ -0,0 +1,53 @@
+
+## Input
+
+```javascript
+function Component() {
+  const onClick = () => {
+    // Cannot assign to globals
+    someUnknownGlobal = true;
+    moduleLocal = true;
+  };
+  // It's possible that this could be an event handler / effect function,
+  // but we don't know that and optimistically assume it will only be
+  // called by an event handler or effect, where it is allowed to modify globals
+  return <div onClick={onClick} />;
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{}],
+};
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+function Component() {
+  const $ = _c(1);
+  let t0;
+  if ($[0] === Symbol.for("react.memo_cache_sentinel")) {
+    const onClick = () => {
+      someUnknownGlobal = true;
+      moduleLocal = true;
+    };
+
+    t0 = <div onClick={onClick} />;
+    $[0] = t0;
+  } else {
+    t0 = $[0];
+  }
+  return t0;
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{}],
+};
+
+```
+
+### Eval output
+(kind: ok) <div></div>

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/allow-reassignment-to-global-function-jsx-prop.js

@@ -0,0 +1,16 @@
+function Component() {
+  const onClick = () => {
+    // Cannot assign to globals
+    someUnknownGlobal = true;
+    moduleLocal = true;
+  };
+  // It's possible that this could be an event handler / effect function,
+  // but we don't know that and optimistically assume it will only be
+  // called by an event handler or effect, where it is allowed to modify globals
+  return <div onClick={onClick} />;
+}
+
+export const FIXTURE_ENTRYPOINT = {
+  fn: Component,
+  params: [{}],
+};