Code gen: improve pattern matching of optional fields.

CHANGELOG.md

@@ -56,6 +56,7 @@
 - Improve code generation or pattern matching of untagged variants. https://github.com/rescript-lang/rescript-compiler/pull/7128
 - Improve negation handling in combination with and/or to simplify generated code (especially coming out of pattern matching). https://github.com/rescript-lang/rescript-compiler/pull/7138
 - optimize JavaScript code generation by using x == null checks and improving type-based optimizations for string/number literals. https://github.com/rescript-lang/rescript-compiler/pull/7141
+- Improve pattern matching on optional fields. https://github.com/rescript-lang/rescript-compiler/pull/7143
 
 
 #### :house: Internal

compiler/ml/parmatch.ml

@@ -524,7 +524,25 @@ let all_record_args lbls =
         (fun lbl -> (mknoloc (Longident.Lident "?temp?"), lbl, omega))
         lbl_all
     in
-    List.iter (fun ((_, lbl, _) as x) -> t.(lbl.lbl_pos) <- x) lbls;
+    List.iter
+      (fun ((id, lbl, pat) as x) ->
+        let lbl_is_optional () =
+          match lbl.lbl_repres with
+          | Record_optional_labels labels -> List.mem lbl.lbl_name labels
+          | _ -> false
+        in
+        let x =
+          match pat.pat_desc with
+          | Tpat_construct
+              ( {txt = Longident.Ldot (Longident.Lident "*predef*", "Some")},
+                _,
+                [({pat_desc = Tpat_constant _} as c)] )
+            when lbl_is_optional () ->
+            (id, lbl, c)
+          | _ -> x
+        in
+        t.(lbl.lbl_pos) <- x)
+      lbls;
     Array.to_list t
   | _ -> fatal_error "Parmatch.all_record_args"
 

tests/tests/src/DictTests.mjs

@@ -17,33 +17,22 @@ let intDict = {
 };
 
 function inferDictByPattern(dict) {
-  let match = dict.one;
-  if (match === 1) {
-    let match$1 = dict.three;
-    if (match$1 === 3) {
-      let match$2 = dict.four;
-      if (match$2 === 4) {
-        dict["five"] = 5;
-        return;
-      }
-
-    }
-
+  if (dict.one === 1 && dict.three === 3 && dict.four === 4) {
+    dict["five"] = 5;
+    return;
   }
-  let match$3 = dict.two;
-  if (match$3 === 1) {
-    console.log("two");
-  } else {
+  if (dict.two !== 1) {
     console.log("not one");
+  } else {
+    console.log("two");
   }
 }
 
 function constrainedAsDict(dict) {
-  let match = dict.one;
-  if (match === 1) {
-    console.log("one");
-  } else {
+  if (dict.one !== 1) {
     console.log("not one");
+  } else {
+    console.log("one");
   }
 }