Fix arity detection for arrows returning nested generics

Author: cknitt

compiler/syntax/src/res_core.ml

@@ -362,8 +362,10 @@ let is_es6_arrow_expression ~in_ternary p =
               (match state.Parser.token with
               (* arrived at `() :typ<` here *)
               | LessThan ->
+                Scanner.set_diamond_mode state.scanner;
                 Parser.next state;
-                go_to_closing GreaterThan state
+                go_to_closing GreaterThan state;
+                Scanner.pop_mode state.scanner Diamond
               | _ -> ());
               match state.Parser.token with
               (* arrived at `() :typ =>` or `() :typ<'a,'b> =>` here *)

tests/syntax_tests/data/parsing/grammar/expressions/arrow.res

@@ -110,4 +110,3 @@ type f = int => unit;
 let a = fn(_ => (): f);
 
 let returnsArrayOption = (): option<array<string>> => Some(["foo"])
-let usesIt = returnsArrayOption()

tests/syntax_tests/data/parsing/grammar/expressions/expected/arrow.res.txt

@@ -74,9 +74,9 @@ type nonrec u = unit
 let un = (() : u)
 type nonrec ('a, 'b) d = ('a * 'b)
 let c [arity:1]() = ((1, 2) : ('a, 'b) d)
-let arr () = ([||] : int nullable array)
+let arr [arity:1]() = ([||] : int nullable array)
 let fn [arity:1]f = f
 type nonrec f = int -> unit (a:1)
 let a = fn (fun [arity:1]_ -> () : f)
-let returnsArrayOption () = (Some [|{js|foo|js}|] : string array option)
-let usesIt = returnsArrayOption ()
+let returnsArrayOption [arity:1]() =
+  (Some [|{js|foo|js}|] : string array option)

tests/tests/src/nested_generic_return.mjs

@@ -0,0 +1,22 @@
+// Generated by ReScript, PLEASE EDIT WITH CARE
+
+
+function foo() {
+  return ["foo"];
+}
+
+let bar = ["foo"];
+
+function nested() {
+  return 1;
+}
+
+let baz = 1;
+
+export {
+  foo,
+  bar,
+  nested,
+  baz,
+}
+/* No side effect */

tests/tests/src/nested_generic_return.res

@@ -0,0 +1,5 @@
+let foo = (): option<array<string>> => Some(["foo"])
+let bar = foo()
+
+let nested = (): option<option<int>> => Some(Some(1))
+let baz = nested()