WIP, find cursor in broken case

Author: nojaf

analysis/src/CompletionBackEndRevamped.ml

@@ -121,3 +121,4 @@ let processCompletable ~debug ~full ~scope ~env ~pos
            (dec2, doc, maybeInsertText))
     |> List.map mkDecorator
   | CdecoratorPayload _ -> []
+  | Ccase _ -> []

analysis/src/CompletionFrontEndRevamped.ml

@@ -420,16 +420,14 @@ let completionWithParser ~currentFile ~debug ~offset ~path ~posCursor text =
       if expr.pexp_loc |> Loc.hasPos ~pos:posNoWhite && !result = None then (
         setFound ();
         match expr.pexp_desc with
-        | Pexp_match (switchExpr, [{pc_lhs = lhsPat}])
+        (* | Pexp_match (switchExpr, [{pc_lhs = lhsPat}])
           when CompletionPatterns.isPatternHole lhsPat
                && locHasCursor switchExpr.pexp_loc = false ->
-          setResult (Cpattern {kind = Empty; typeLoc = switchExpr.pexp_loc})
+          setResult (Cpattern {kind = Empty; typeLoc = switchExpr.pexp_loc}) *)
         | Pexp_match (switchExpr, cases) ->
           let oldTypeLoc = !currentTypeLoc in
           currentTypeLoc := Some switchExpr.pexp_loc;
-          cases
-          |> List.iter (fun case ->
-                 Ast_iterator.default_iterator.case iterator case);
+          cases |> List.iter (fun case -> iterator.case iterator case);
           currentTypeLoc := oldTypeLoc;
           processed := true
         | Pexp_extension ({txt = "obj"}, PStr [str_item]) ->
@@ -615,6 +613,10 @@ let completionWithParser ~currentFile ~debug ~offset ~path ~posCursor text =
         | _ -> ());
       if not !processed then Ast_iterator.default_iterator.expr iterator expr
   in
+  let case (_iterator : Ast_iterator.iterator) (case : Parsetree.case) =
+    if case.pc_loc |> Loc.hasPos ~pos:posCursor then
+      setResult (Ccase case.pc_loc)
+  in
   let typ (iterator : Ast_iterator.iterator) (core_type : Parsetree.core_type) =
     if core_type.ptyp_loc |> Loc.hasPos ~pos:posNoWhite then (
       found := true;
@@ -781,6 +783,7 @@ let completionWithParser ~currentFile ~debug ~offset ~path ~posCursor text =
       Ast_iterator.default_iterator with
       attribute;
       expr;
+      case;
       location;
       module_expr;
       module_type;
@@ -800,7 +803,8 @@ let completionWithParser ~currentFile ~debug ~offset ~path ~posCursor text =
       Res_driver.parsing_engine.parse_implementation ~for_printer:false
     in
     let {Res_driver.parsetree = str} = parser ~filename:currentFile in
-    iterator.structure iterator str |> ignore;
+    let tree = Res_recovery.map str in
+    iterator.structure iterator tree |> ignore;
     if blankAfterCursor = Some ' ' || blankAfterCursor = Some '\n' then
       scope := !lastScopeBeforeCursor
       (* TODO(revamp) Complete any value *)

analysis/src/SharedTypes.ml

@@ -820,6 +820,7 @@ module CompletableRevamped = struct
     | CextensionNode of string
     | Cdecorator of string
     | CdecoratorPayload of decoratorPayload
+    | Ccase of Location.t
 
   let toString (t : t) =
     match t with
@@ -829,11 +830,13 @@ module CompletableRevamped = struct
     | CextensionNode _ -> "CextensionNode"
     | Cdecorator _ -> "Cdecorator"
     | CdecoratorPayload _ -> "CdecoratorPayload"
+    | Ccase _ -> "Ccase"
 
   let try_loc (t : t) =
     match t with
     | Cexpression {typeLoc; _} -> Some typeLoc
     | Cpattern {typeLoc; _} -> Some typeLoc
+    | Ccase loc -> Some loc
     | _ -> None
 end
 

compiler/syntax/src/res_driver.ml

@@ -139,8 +139,9 @@ let parse_implementation ?(ignore_parse_errors = false) sourcefile =
   in
   if parse_result.invalid then (
     Res_diagnostics.print_report parse_result.diagnostics parse_result.source;
-    if not ignore_parse_errors then exit 1);
-  parse_result.parsetree
+    if not ignore_parse_errors then exit 1;
+    Res_recovery.map parse_result.parsetree)
+  else parse_result.parsetree
 [@@raises exit]
 
 let parse_interface ?(ignore_parse_errors = false) sourcefile =

compiler/syntax/src/res_recovery.ml

@@ -0,0 +1,44 @@
+let map_expr (mapper : Ast_mapper.mapper) (expr : Parsetree.expression) =
+  match expr.pexp_desc with
+  | Pexp_match (e, cases) ->
+    let mapped_e = mapper.expr mapper e in
+    let match_end_loc = expr.pexp_loc.loc_end in
+
+    let is_ghost_case case =
+      let open Parsetree in
+      case.pc_lhs.ppat_loc.loc_ghost && case.pc_rhs.pexp_loc.loc_ghost
+    in
+
+    let rec process_cases mapped_cases cases =
+      match cases with
+      | [] -> mapped_cases
+      | [last_case] when is_ghost_case last_case ->
+        prerr_endline "last case";
+        let mapped =
+          mapper.case mapper
+            {
+              last_case with
+              pc_loc = {last_case.pc_loc with loc_end = match_end_loc};
+            }
+        in
+        process_cases (mapped :: mapped_cases) []
+      | current :: (next :: _ as rest) when is_ghost_case current ->
+        let mapped =
+          mapper.case mapper
+            {
+              current with
+              pc_loc =
+                {current.pc_loc with loc_end = next.pc_lhs.ppat_loc.loc_start};
+            }
+        in
+        process_cases (mapped :: mapped_cases) rest
+      | c :: rest -> process_cases (mapper.case mapper c :: mapped_cases) rest
+    in
+
+    let adjusted_cases = process_cases [] cases in
+    {expr with pexp_desc = Pexp_match (mapped_e, adjusted_cases)}
+  | _ -> Ast_mapper.default_mapper.expr mapper expr
+
+let map (tree : Parsetree.structure) =
+  let mapper = {Ast_mapper.default_mapper with expr = map_expr} in
+  mapper.structure mapper tree

tests/analysis_new_tests/tests/test_files/__snapshots__/SwitchCaseCompletions.res_Empty_case_array.snap

@@ -1,20 +1,15 @@
-Found Completable: Cpattern at type loc: [3:15->3:28]
+case _ vamp [4:2->6:1], has cursor true
+Found Completable: Ccase
 
-  2 │ let someStringArr = ["hello"]
   3 │ 
   4 │ let x = switch someStringArr {
-    │                ‾‾‾‾‾‾‾‾‾‾‾‾‾
   5 │   | 
+    │   ‾‾
   6 │ //  ^com
+    │ ‾‾‾‾‾‾‾‾
+  7 │ }
+    │ ‾
+  8 │ 
 
-[{
-    "label": "[]",
-    "kind": 12,
-    "tags": [],
-    "detail": "array<string>",
-    "documentation": null,
-    "sortText": "A",
-    "insertText": "[$0]",
-    "insertTextFormat": 2
-  }]
+[]
 

tests/analysis_new_tests/tests/test_files/__snapshots__/SwitchCaseCompletions.res_Empty_case_bool.snap

@@ -1,28 +1,15 @@
-Found Completable: Cpattern at type loc: [1:15->1:19]
+case _ vamp [2:2->4:1], has cursor true
+Found Completable: Ccase
 
   1 │ // Empty case, bool
   2 │ let x = switch true {
-    │                ‾‾‾‾
   3 │   | 
+    │   ‾‾
   4 │ //  ^com
+    │ ‾‾‾‾‾‾‾‾
+  5 │ }
+    │ ‾
+  6 │ 
 
-[{
-    "label": "true",
-    "kind": 12,
-    "tags": [],
-    "detail": "bool",
-    "documentation": null,
-    "sortText": "A",
-    "insertText": "true",
-    "insertTextFormat": 2
-  }, {
-    "label": "false",
-    "kind": 12,
-    "tags": [],
-    "detail": "bool",
-    "documentation": null,
-    "sortText": "A",
-    "insertText": "false",
-    "insertTextFormat": 2
-  }]
+[]
 

tests/analysis_new_tests/tests/test_files/__snapshots__/SwitchCaseCompletions.res_Empty_case_record.snap

@@ -1,19 +1,15 @@
-Found Completable: Cpattern at type loc: [1:15->1:44]
+case _ vamp [2:2->4:1], has cursor true
+Found Completable: Ccase
 
   1 │ // Empty case, record
   2 │ let x = switch TestTypeDefs.nestedTestRecord {
-    │                ‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾
   3 │   | 
+    │   ‾‾
   4 │ //  ^com
+    │ ‾‾‾‾‾‾‾‾
+  5 │ }
+    │ ‾
+  6 │ 
 
-[{
-    "label": "{}",
-    "kind": 12,
-    "tags": [],
-    "detail": "TestTypeDefs.nestedTestRecord",
-    "documentation": {"kind": "markdown", "value": "```rescript\ntype nestedTestRecord = {\n  test: bool,\n  nested: {name: string, oneMoreLevel: {here: bool}},\n}\n```"},
-    "sortText": "A",
-    "insertText": "{$0}",
-    "insertTextFormat": 2
-  }]
+[]
 

tests/analysis_new_tests/tests/test_files/__snapshots__/SwitchCaseCompletions.res_Empty_case_string.snap

@@ -1,20 +1,15 @@
-Found Completable: Cpattern at type loc: [2:15->2:18]
+case _ vamp [3:2->5:1], has cursor true
+Found Completable: Ccase
 
-  1 │ // Empty case, string
   2 │ let str = "hello"
   3 │ let x = switch str {
-    │                ‾‾‾
   4 │   | 
+    │   ‾‾
   5 │ //  ^com
+    │ ‾‾‾‾‾‾‾‾
+  6 │ }
+    │ ‾
+  7 │ 
 
-[{
-    "label": "\"\"",
-    "kind": 12,
-    "tags": [],
-    "detail": "string",
-    "documentation": null,
-    "sortText": "A",
-    "insertText": "\"$0\"",
-    "insertTextFormat": 2
-  }]
+[]