From 9eb9071aedebd6c282c64c341de13978db370106 Mon Sep 17 00:00:00 2001
From: Sathya Gunasekaran <gsathya.ceg@gmail.com>
Date: Wed, 25 Jan 2023 11:11:16 -0500
Subject: [PATCH] [typer] Add type inference for unary expr

---
 .../forget/src/TypeInference/InferTypes.ts    |  5 ++++
 .../fixtures/hir/unary-expr.expect.md         | 26 +++++--------------
 2 files changed, 11 insertions(+), 20 deletions(-)

diff --git a/compiler/forget/src/TypeInference/InferTypes.ts b/compiler/forget/src/TypeInference/InferTypes.ts
index b27ba32eb3..9f949a5963 100644
--- a/compiler/forget/src/TypeInference/InferTypes.ts
+++ b/compiler/forget/src/TypeInference/InferTypes.ts
@@ -103,6 +103,11 @@ function generateTypeEquation(instr: Instruction): Array<TypeEquation> {
       break;
     }
 
+    case "UnaryExpression": {
+      add(left, { kind: "Primitive" });
+      break;
+    }
+
     case "Identifier": {
       add(left, value.identifier.type);
       break;
diff --git a/compiler/forget/src/__tests__/fixtures/hir/unary-expr.expect.md b/compiler/forget/src/__tests__/fixtures/hir/unary-expr.expect.md
index 1c44af6054..2f2839dc64 100644
--- a/compiler/forget/src/__tests__/fixtures/hir/unary-expr.expect.md
+++ b/compiler/forget/src/__tests__/fixtures/hir/unary-expr.expect.md
@@ -93,28 +93,14 @@ function component(a) {
   } else {
     f = $[15];
   }
-  const c_16 = $[16] !== z;
-  const c_17 = $[17] !== p;
-  const c_18 = $[18] !== q;
-  const c_19 = $[19] !== n;
-  const c_20 = $[20] !== m;
-  const c_21 = $[21] !== e;
-  const c_22 = $[22] !== f;
-  let t23;
-  if (c_16 || c_17 || c_18 || c_19 || c_20 || c_21 || c_22) {
-    t23 = { z: z, p: p, q: q, n: n, m: m, e: e, f: f };
-    $[16] = z;
-    $[17] = p;
-    $[18] = q;
-    $[19] = n;
-    $[20] = m;
-    $[21] = e;
-    $[22] = f;
-    $[23] = t23;
+  let t16;
+  if ($[16] === Symbol.for("react.memo_cache_sentinel")) {
+    t16 = { z: z, p: p, q: q, n: n, m: m, e: e, f: f };
+    $[16] = t16;
   } else {
-    t23 = $[23];
+    t16 = $[16];
   }
-  return t23;
+  return t16;
 }
 
 ```