Extend autocompletion to handle nested objects.

Such as `nestedObj["x"]["y"]`.

Author: cristianoc

CHANGELOG.md

@@ -2,7 +2,7 @@
 - Fix issue in JSX autocomplete when the component is declared external.
 - Fix jump-to-definition for uncurried calls.
 - Fix issue where values for autocomplete were pulled from implementations instead of interfaces.
-- Add autocompletion for object access of the form foo["bar"].
+- Add autocompletion for object access of the form `foo["x"]` and `foo["x"]["y"]["z"]`. 
 - Fix issue with autocomplete then punned props are used in JSX. E.g. `<M foo ...>`.
 - Fix issue with JSX autocompletion not working after `foo=#variant`.
 

analysis/src/NewCompletions.ml

@@ -1145,28 +1145,38 @@ let processCompletable ~findItems ~full ~package ~rawOpens
     |> List.filter (fun (name, _t) ->
            Utils.startsWith name prefix && not (List.mem name identsSeen))
     |> List.map mkLabel
-  | Cobj (lhs, prefix) ->
-    let labels =
+  | Cobj (lhs, path, prefix) ->
+    let rec getFields (texp : Types.type_expr) =
+      match texp.desc with
+      | Tfield (name, _, t1, t2) ->
+        let fields = t2 |> getFields in
+        (name, t1) :: fields
+      | Tlink te -> te |> getFields
+      | Tvar None -> []
+      | _ -> []
+    in
+    let rec getObj (t : Types.type_expr) =
+      match t.desc with
+      | Tlink t1 | Tsubst t1 | Tpoly (t1, []) -> getObj t1
+      | Tobject (tObj, _) -> getFields tObj
+      | _ -> []
+    in
+    let fields =
       match [lhs] |> findItems ~exact:true with
-      | {SharedTypes.item = Value typ} :: _ ->
-        let rec getFields (texp : Types.type_expr) =
-          match texp.desc with
-          | Tfield (name, _, t1, t2) ->
-            let fields = t2 |> getFields in
-            (name, t1) :: fields
-          | Tlink te -> te |> getFields
-          | Tvar None -> []
-          | _ -> []
-        in
-        let rec getObj (t : Types.type_expr) =
-          match t.desc with
-          | Tlink t1 | Tsubst t1 -> getObj t1
-          | Tobject (tObj, _) -> getFields tObj
-          | _ -> []
-        in
-        getObj typ
+      | {SharedTypes.item = Value typ} :: _ -> getObj typ
       | _ -> []
     in
+    let rec resolvePath fields path =
+      match path with
+      | name :: restPath -> (
+        match fields |> List.find_opt (fun (n, _) -> n = name) with
+        | Some (_, fieldType) ->
+          let innerFields = getObj fieldType in
+          resolvePath innerFields restPath
+        | None -> [])
+      | [] -> fields
+    in
+    let labels = resolvePath fields path in
     let mkLabel_ name typString =
       mkItem ~name ~kind:4 ~deprecated:None ~detail:typString ~docstring:[]
     in

analysis/src/PartialParser.ml

@@ -190,7 +190,8 @@ type completable =
   | Cpath of string list  (** e.g. ["M", "foo"] for M.foo *)
   | Cjsx of string list * string * string list
       (** E.g. (["M", "Comp"], "id", ["id1", "id2"]) for <M.Comp id1=... id2=... ... id *)
-  | Cobj of string * string  (** e.g. ("foo", "bar") for foo['bar *)
+  | Cobj of string * string list * string
+      (** e.g. ("foo", ["a", "b"], "bar") for foo["a"]["b"]["bar" *)
   | Cpipe of pipe * string  (** E.g. ("x", "foo") for "x->foo" *)
 
 let isLowercaseIdent id =
@@ -234,16 +235,23 @@ let findCompletable text offset =
   in
   let mkObj ~off ~partialName =
     let off = skipWhite text off in
-    let rec loop i =
-      if i < 0 then Some (String.sub text 0 (i - 1))
+    let rec loop off path i =
+      if i < 0 then Some ([], String.sub text 0 (i - 1))
       else
         match text.[i] with
-        | 'a' .. 'z' | 'A' .. 'Z' | '0' .. '9' | '_' -> loop (i - 1)
-        | _ -> Some (String.sub text (i + 1) (off - i))
+        | 'a' .. 'z' | 'A' .. 'Z' | '0' .. '9' | '_' -> loop off path (i - 1)
+        | ']' when i > 1 && text.[i - 1] = '"' ->
+          let i0 = i - 2 in
+          let i1 = startOfLident text i0 in
+          let ident = String.sub text i1 (i0 - i1 + 1) in
+          if ident <> "" && i1 > 1 && text.[i1 - 1] = '"' && text.[i1 - 2] = '['
+          then loop (off - i + i1 - 3) (ident :: path) (i1 - 3)
+          else None
+        | _ -> Some (path, String.sub text (i + 1) (off - i))
     in
-    match loop off with
+    match loop off [] off with
     | None -> None
-    | Some lhs -> Some (Cobj (lhs, partialName))
+    | Some (path, lhs) -> Some (Cobj (lhs, path, partialName))
   in
 
   let suffix i = String.sub text (i + 1) (offset - (i + 1)) in

analysis/tests/src/Completion.res

@@ -3,7 +3,6 @@ module MyList = Belt.List
 //^com Array.
 //^com Array.m
 
-
 module Dep: {
   @ocaml.doc("Some doc comment") @deprecated("Use customDouble instead")
   let customDouble: int => int
@@ -30,20 +29,19 @@ let op = Some(3)
 
 module ForAuto = {
   type t = int
-  let abc = (x:t, _y:int) => x
-  let abd = (x:t, _y:int) => x
+  let abc = (x: t, _y: int) => x
+  let abd = (x: t, _y: int) => x
 }
 
-let fa:ForAuto.t = 34
+let fa: ForAuto.t = 34
 //^com fa->
 
 //^com "hello"->Js.Dict.u
 
 module O = {
   module Comp = {
     @react.component
-    let make = (~first="", ~zoo=3, ~second) =>
-      React.string(first ++ second ++ string_of_int(zoo))
+    let make = (~first="", ~zoo=3, ~second) => React.string(first ++ second ++ string_of_int(zoo))
   }
 }
 
@@ -65,10 +63,17 @@ let zzz = 11
 
 //^com let x = Lib.foo(~age={3+4}, ~
 
-let _ = Lib.foo(//~age,
-//^com ~
-~age=3, ~name="")
+let _ = Lib.foo(
+  //~age,
+  //^com ~
+  ~age=3,
+  ~name="",
+)
 
 let someObj = {"name": "a", "age": 32}
 
 //^com someObj["a
+
+let nestedObj = {"x": {"y": {"name": "a", "age": 32}}}
+
+//^com nestedObj["x"]["y"]["