compiler: fix interleaved hook chained-call memoization

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-arg-in-hook.expect.md

@@ -0,0 +1,52 @@
+
+## Input
+
+```javascript
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+
+  const {data} = useBaz(value);
+  const a = result(data, value);
+
+  return a;
+}
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+export function useFoo(id) {
+  const $ = _c(5);
+  const bar = useBar();
+  let t0;
+  if ($[0] !== bar) {
+    t0 = func(bar);
+    $[0] = bar;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const value = t0;
+
+  const { data } = useBaz(value);
+  let t1;
+  if ($[2] !== data || $[3] !== value) {
+    t1 = result(data, value);
+    $[2] = data;
+    $[3] = value;
+    $[4] = t1;
+  } else {
+    t1 = $[4];
+  }
+  const a = t1;
+
+  return a;
+}
+
+```
+
+### Eval output
+(kind: exception) Fixture not implemented

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-arg-in-hook.ts

@@ -0,0 +1,9 @@
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+
+  const {data} = useBaz(value);
+  const a = result(data, value);
+
+  return a;
+}

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-arg-used-after.expect.md

@@ -0,0 +1,50 @@
+
+## Input
+
+```javascript
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return [a, value];
+}
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+export function useFoo(id) {
+  const $ = _c(5);
+  const bar = useBar();
+  const value = func(bar);
+  let t0;
+  if ($[0] !== id) {
+    t0 = options(id);
+    $[0] = id;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const { data } = useBaz(t0);
+  const a = result(data, value);
+  let t1;
+  if ($[2] !== a || $[3] !== value) {
+    t1 = [a, value];
+    $[2] = a;
+    $[3] = value;
+    $[4] = t1;
+  } else {
+    t1 = $[4];
+  }
+  return t1;
+}
+
+```
+
+### Eval output
+(kind: exception) Fixture not implemented

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-arg-used-after.ts

@@ -0,0 +1,9 @@
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return [a, value];
+}

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-non-call-arg.expect.md

@@ -0,0 +1,46 @@
+
+## Input
+
+```javascript
+export function useFoo(id: string, value: number) {
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return a;
+}
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+export function useFoo(id, value) {
+  const $ = _c(5);
+  let t0;
+  if ($[0] !== id) {
+    t0 = options(id);
+    $[0] = id;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const { data } = useBaz(t0);
+  let t1;
+  if ($[2] !== data || $[3] !== value) {
+    t1 = result(data, value);
+    $[2] = data;
+    $[3] = value;
+    $[4] = t1;
+  } else {
+    t1 = $[4];
+  }
+  const a = t1;
+
+  return a;
+}
+
+```
+
+### Eval output
+(kind: exception) Fixture not implemented

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-non-call-arg.ts

@@ -0,0 +1,6 @@
+export function useFoo(id: string, value: number) {
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return a;
+}

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-non-call-use.expect.md

@@ -0,0 +1,77 @@
+
+## Input
+
+```javascript
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+  const arr = [value];
+
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return [a, arr];
+}
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+export function useFoo(id) {
+  const $ = _c(12);
+  const bar = useBar();
+  let t0;
+  if ($[0] !== bar) {
+    t0 = func(bar);
+    $[0] = bar;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const value = t0;
+  let t1;
+  if ($[2] !== value) {
+    t1 = [value];
+    $[2] = value;
+    $[3] = t1;
+  } else {
+    t1 = $[3];
+  }
+  const arr = t1;
+  let t2;
+  if ($[4] !== id) {
+    t2 = options(id);
+    $[4] = id;
+    $[5] = t2;
+  } else {
+    t2 = $[5];
+  }
+  const { data } = useBaz(t2);
+  let t3;
+  if ($[6] !== data || $[7] !== value) {
+    t3 = result(data, value);
+    $[6] = data;
+    $[7] = value;
+    $[8] = t3;
+  } else {
+    t3 = $[8];
+  }
+  const a = t3;
+  let t4;
+  if ($[9] !== a || $[10] !== arr) {
+    t4 = [a, arr];
+    $[9] = a;
+    $[10] = arr;
+    $[11] = t4;
+  } else {
+    t4 = $[11];
+  }
+  return t4;
+}
+
+```
+
+### Eval output
+(kind: exception) Fixture not implemented

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-non-call-use.ts

@@ -0,0 +1,10 @@
+export function useFoo(id: string) {
+  const bar = useBar();
+  const value = func(bar);
+  const arr = [value];
+
+  const {data} = useBaz(options(id));
+  const a = result(data, value);
+
+  return [a, arr];
+}

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-triple-chain.expect.md

@@ -0,0 +1,73 @@
+
+## Input
+
+```javascript
+export function useFoo(id: string) {
+  const bar = useBar();
+  const b = transform(bar);
+
+  const {data} = useBaz(options(id));
+  const c = transform2(b, data);
+
+  useQux();
+  const d = transform3(c);
+
+  return d;
+}
+
+```
+
+## Code
+
+```javascript
+import { c as _c } from "react/compiler-runtime";
+export function useFoo(id) {
+  const $ = _c(9);
+  const bar = useBar();
+  let t0;
+  if ($[0] !== bar) {
+    t0 = transform(bar);
+    $[0] = bar;
+    $[1] = t0;
+  } else {
+    t0 = $[1];
+  }
+  const b = t0;
+  let t1;
+  if ($[2] !== id) {
+    t1 = options(id);
+    $[2] = id;
+    $[3] = t1;
+  } else {
+    t1 = $[3];
+  }
+  const { data } = useBaz(t1);
+  let t2;
+  if ($[4] !== b || $[5] !== data) {
+    t2 = transform2(b, data);
+    $[4] = b;
+    $[5] = data;
+    $[6] = t2;
+  } else {
+    t2 = $[6];
+  }
+  const c = t2;
+
+  useQux();
+  let t3;
+  if ($[7] !== c) {
+    t3 = transform3(c);
+    $[7] = c;
+    $[8] = t3;
+  } else {
+    t3 = $[8];
+  }
+  const d = t3;
+
+  return d;
+}
+
+```
+
+### Eval output
+(kind: exception) Fixture not implemented

compiler/packages/babel-plugin-react-compiler/src/__tests__/fixtures/compiler/repro-hook-interleaved-chained-call-memoization-triple-chain.ts

@@ -0,0 +1,12 @@
+export function useFoo(id: string) {
+  const bar = useBar();
+  const b = transform(bar);
+
+  const {data} = useBaz(options(id));
+  const c = transform2(b, data);
+
+  useQux();
+  const d = transform3(c);
+
+  return d;
+}