Improve code for pipe autocomplete.

Use a dedicate code path that takes care of pipe and is separate from normal identifiers.

examples/example-project/src/ZZ.res

@@ -79,25 +79,25 @@ let testRecordFields = (gr: gr) => {
 }
 
 @ocaml.doc("vr docstring")
-type vr = | V1 | V2
+type vr = V1 | V2
 
 let v1 = V1
 
 module DoubleNested = ModuleWithDocComment.Nested.NestedAgain
 
-let uncurried = (. x) => x+1;
+let uncurried = (. x) => x + 1
 
 module Inner = {
-  type tInner = int;
+  type tInner = int
   let vInner = 34
 }
 
-type typeInner = Inner.tInner;
+type typeInner = Inner.tInner
 
-let valueInner = Inner.vInner;
+let valueInner = Inner.vInner
 
 @ocaml.doc("Doc comment for functionWithTypeAnnotation")
-let functionWithTypeAnnotation : unit => int = () => 1
+let functionWithTypeAnnotation: unit => int = () => 1
 
 module HoverInsideModuleWithComponent = {
   let x = 2 // check that hover on x works
@@ -111,14 +111,12 @@ module Lib = {
   let next = (~number=0, ~year) => number + year
 }
 
-@ocaml.doc("This module is commented")
-@deprecated("This module is deprecated")
-module Dep : {
-  @ocaml.doc("Some doc comment")
-  @deprecated("Use customDouble instead")
-  let customDouble : int => int
+@ocaml.doc("This module is commented") @deprecated("This module is deprecated")
+module Dep: {
+  @ocaml.doc("Some doc comment") @deprecated("Use customDouble instead")
+  let customDouble: int => int
 
-  let customDouble2 : int => int
+  let customDouble2: int => int
 } = {
   let customDouble = foo => foo * 2
   let customDouble2 = foo => foo * 2
@@ -129,8 +127,12 @@ let cc = Dep.customDouble(11)
 module O = {
   module Comp = {
     @react.component
-    let make = (~first="", ~kas=11, ~foo=3, ~second, ~v) => React.string(first ++ second ++ string_of_int(foo))
+    let make = (~first="", ~kas=11, ~foo=3, ~second, ~v) =>
+      React.string(first ++ second ++ string_of_int(foo))
   }
 }
 
-let comp = <O.Comp key="12" second="abcc" v=12 />
+let comp = <O.Comp key="12" second="abcc" v=12 />
+
+let lll = List.make(3, 4)
+

src/NewCompletions.ml

@@ -514,7 +514,7 @@ let processCompletable ~findItems ~full ~package ~pos ~rawOpens
          ->
            mkItem ~name ~kind:(kindToInt item) ~deprecated
              ~detail:(detail name item) ~docstring ~uri ~pos_lnum)
-  | Cpipe s -> (
+  | Cpipe (lhs, partialName) -> (
     let getModulePath path =
       let rec loop (path : Path.t) =
         match path with
@@ -536,14 +536,7 @@ let processCompletable ~findItems ~full ~package ~pos ~rawOpens
         Some (modulePath, partialName)
       | _ -> None
     in
-    let lhsPath =
-      match Str.split (Str.regexp_string "->") s with
-      | [lhs] -> getLhsPath ~lhs ~partialName:""
-      | [lhs; partialName] -> getLhsPath ~lhs ~partialName
-      | _ ->
-        (* Only allow one -> *)
-        None
-    in
+    let lhsPath = getLhsPath ~lhs ~partialName in
     let removePackageOpens modulePath =
       match modulePath with
       | toplevel :: rest when package.TopTypes.opens |> List.mem toplevel ->

src/PartialParser.ml

@@ -116,7 +116,7 @@ type completable =
   | Cpath of string list  (** e.g. ["M", "foo"] for M.foo *)
   | Cjsx of string list * string
       (** E.g. (["M", "Comp"], "id") for <M.Comp ... id *)
-  | Cpipe of string  (** E.g. "x->foo" *)
+  | Cpipe of string * string  (** E.g. ("x", "foo") for "x->foo" *)
 
 let isLowercaseIdent id =
   let rec loop i =
@@ -132,24 +132,32 @@ let isLowercaseIdent id =
 let findCompletable text offset =
   let mkPath s =
     let len = String.length s in
-    let pipeParts = Str.split (Str.regexp_string "->") s in
-    if
-      (len > 1 && s.[len - 2] = '-' && s.[len - 1] = '>')
-      || List.length pipeParts > 1
-    then Cpipe s
-    else
-      let parts = Str.split (Str.regexp_string ".") s in
-      let parts =
-        match s.[len - 1] = '.' with true -> parts @ [""] | false -> parts
-      in
-      match parts with
-      | [id] when String.lowercase_ascii id = id -> (
-        match findJsxContext text (offset - len - 1) with
-        | None -> Cpath parts
-        | Some componentName ->
-          Cjsx (Str.split (Str.regexp_string ".") componentName, id))
-      | _ -> Cpath parts
+    let parts = Str.split (Str.regexp_string ".") s in
+    let parts =
+      match s.[len - 1] = '.' with true -> parts @ [""] | false -> parts
+    in
+    match parts with
+    | [id] when String.lowercase_ascii id = id -> (
+      match findJsxContext text (offset - len - 1) with
+      | None -> Cpath parts
+      | Some componentName ->
+        Cjsx (Str.split (Str.regexp_string ".") componentName, id))
+    | _ -> Cpath parts
   in
+  let mkPipe off partialName =
+    let rec loop i =
+      match i < 0 with
+      | true -> Some (String.sub text 0 (i - 1))
+      | false -> (
+        match text.[i] with
+        | 'a' .. 'z' | 'A' .. 'Z' | '0' .. '9' | '.' | '_' -> loop (i - 1)
+        | _ -> Some (String.sub text (i + 1) (off - i)))
+    in
+    match loop off with
+    | None -> None
+    | Some lhs -> Some (Cpipe (lhs, partialName))
+  in
+
   let suffix i = String.sub text (i + 1) (offset - (i + 1)) in
   let rec loop i =
     match i < 0 with
@@ -158,7 +166,8 @@ let findCompletable text offset =
       match text.[i] with
       | '>' when i > 0 && text.[i - 1] = '-' ->
         let rest = suffix i in
-        if isLowercaseIdent rest then loop (i - 2) else Some (mkPath rest)
+        if isLowercaseIdent rest then mkPipe (i - 2) rest
+        else Some (mkPath rest)
       | '~' ->
         let labelPrefix = suffix i in
         let funPath = findCallFromArgument text (i - 1) in