Find references in both .res and .resi files.

Author: cristianoc

analysis/src/Cmt.ml

@@ -24,12 +24,12 @@ let fullFromUri ~uri =
       prerr_endline ("can't find module " ^ moduleName);
       None)
 
-let fullFromModule ~package ~moduleName =
+let fullsFromModule ~package ~moduleName =
   if Hashtbl.mem package.pathsForModule moduleName then
     let paths = Hashtbl.find package.pathsForModule moduleName in
-    let uri = getUri paths in
-    fullFromUri ~uri
-  else None
+    let uris = getUris paths in
+    uris |> List.filter_map (fun uri -> fullFromUri ~uri)
+  else []
 
 let loadFullCmtFromPath ~path =
   let uri = Uri.fromPath path in

analysis/src/Commands.ml

@@ -172,11 +172,9 @@ let references ~path ~pos ~debug =
                  | Some loc -> loc
                  | None -> Uri.toTopLevelLoc uri2
                in
-               if loc.loc_ghost then acc
-               else
-                 Protocol.stringifyLocation
-                   {uri = Uri.toString uri2; range = Utils.cmtLocToRange loc}
-                 :: acc)
+               Protocol.stringifyLocation
+                 {uri = Uri.toString uri2; range = Utils.cmtLocToRange loc}
+               :: acc)
              [])
   in
   print_endline

analysis/src/References.ml

@@ -506,28 +506,24 @@ let forLocalStamp ~full:{file; extra; package} stamp (tip : Tip.t) =
                 package.projectFiles |> FileSet.elements
                 |> List.filter (fun name -> name <> file.moduleName)
                 |> List.map (fun moduleName ->
-                       match ProcessCmt.fileForModule ~package moduleName with
-                       | None -> []
-                       | Some file -> (
-                         match Cmt.fullFromModule ~package ~moduleName with
-                         | None -> []
-                         | Some {extra} -> (
-                           match
-                             Hashtbl.find_opt extra.externalReferences
-                               thisModuleName
-                           with
-                           | None -> []
-                           | Some refs ->
-                             let locs =
-                               refs
-                               |> Utils.filterMap (fun (p, t, locs) ->
-                                      if p = path && t = tip then Some locs
-                                      else None)
-                             in
-                             locs
-                             |> List.map (fun loc ->
-                                    {uri = file.uri; locOpt = Some loc}))))
-                |> List.concat
+                       Cmt.fullsFromModule ~package ~moduleName
+                       |> List.map (fun {file; extra} ->
+                              match
+                                Hashtbl.find_opt extra.externalReferences
+                                  thisModuleName
+                              with
+                              | None -> []
+                              | Some refs ->
+                                let locs =
+                                  refs
+                                  |> Utils.filterMap (fun (p, t, locs) ->
+                                         if p = path && t = tip then Some locs
+                                         else None)
+                                in
+                                locs
+                                |> List.map (fun loc ->
+                                       {uri = file.uri; locOpt = Some loc})))
+                |> List.concat |> List.concat
               in
               alternativeReferences @ externals)
           else (

analysis/src/SharedTypes.ml

@@ -295,6 +295,12 @@ let getUri p =
   | Namespace {cmt} -> Uri.fromPath cmt
   | IntfAndImpl {resi} -> Uri.fromPath resi
 
+let getUris p =
+  match p with
+  | Impl {res} -> [Uri.fromPath res]
+  | Namespace {cmt} -> [Uri.fromPath cmt]
+  | IntfAndImpl {res; resi} -> [Uri.fromPath res; Uri.fromPath resi]
+
 let getCmtPath ~uri p =
   match p with
   | Impl {cmt} -> cmt

analysis/tests/src/References.res

@@ -20,4 +20,6 @@ let bb = M.aa
 let cc = bb
 let dd = M.aa
 //          ^ref
-let _ = <ComponentInner/>
+
+let _ = <ComponentInner/>
+//             ^ref

analysis/tests/src/expected/References.res.txt

@@ -21,3 +21,13 @@ References src/References.res 20:12
 {"uri": "References.res", "range": {"start": {"line": 20, "character": 11}, "end": {"line": 20, "character": 13}}}
 ]
 
+References src/References.res 23:15
+getLocItem #4: heuristic for </Comp> within fragments: take make as makeProps does not work
+the type is not great but jump to definition works
+[
+{"uri": "ReferencesInner.res", "range": {"start": {"line": 1, "character": 28}, "end": {"line": 1, "character": 32}}},
+{"uri": "References.res", "range": {"start": {"line": 23, "character": 19}, "end": {"line": 23, "character": 23}}},
+{"uri": "ComponentInner.res", "range": {"start": {"line": 1, "character": 4}, "end": {"line": 1, "character": 8}}},
+{"uri": "ComponentInner.resi", "range": {"start": {"line": 1, "character": 4}, "end": {"line": 1, "character": 8}}}
+]
+