add workspaceSymbols

Author: aspeddro

.gitignore

@@ -2,9 +2,9 @@
 out
 node_modules
 examples/*/lib
-analysis/tests/lib
-analysis/tests/.bsb.lock
+analysis/tests/**/*/lib
+analysis/tests/*/.bsb.lock
 analysis/_build
-analysis/tests/.merlin
+analysis/tests/**/*/.merlin
 analysis/rescript-editor-analysis.exe
 

CHANGELOG.md

@@ -7,6 +7,7 @@ API examples:
   ./rescript-editor-analysis.exe definition src/MyFile.res 9 3
   ./rescript-editor-analysis.exe typeDefinition src/MyFile.res 9 3
   ./rescript-editor-analysis.exe documentSymbol src/Foo.res
+  ./rescript-editor-analysis.exe workspaceSymbols src/
   ./rescript-editor-analysis.exe hover src/MyFile.res 10 2
   ./rescript-editor-analysis.exe references src/MyFile.res 10 2
   ./rescript-editor-analysis.exe rename src/MyFile.res 10 2 foo
@@ -38,6 +39,10 @@ Options:
 
     ./rescript-editor-analysis.exe documentSymbol src/MyFile.res
 
+  workspaceSymbols: get all symbols in directory src/
+
+    ./rescript-editor-analysis.exe workspaceSymbols src/
+
   hover: get inferred type for MyFile.res at line 10 column 2:
 
     ./rescript-editor-analysis.exe hover src/MyFile.res 10 2
@@ -89,7 +94,8 @@ let main () =
     Commands.typeDefinition ~path
       ~pos:(int_of_string line, int_of_string col)
       ~debug:false
-  | [_; "documentSymbol"; path] -> DocumentSymbol.command ~path
+  | [_; "documentSymbol"; path] -> Commands.documentSymbol ~path
+  | [_; "workspaceSymbols"; dir] -> Commands.workspaceSymbols ~dir
   | [_; "hover"; path; line; col; currentFile] ->
     Commands.hover ~path
       ~pos:(int_of_string line, int_of_string col)

README.md

@@ -264,10 +264,27 @@ let format ~path =
 let diagnosticSyntax ~path =
   print_endline (Diagnostics.document_syntax ~path |> Protocol.array)
 
+let documentSymbol ~path =
+  Symbols.document ~path
+  |> List.map Protocol.stringifyDocumentSymbolItem
+  |> Protocol.array |> print_endline
+
+let workspaceSymbols ~dir =
+  print_endline
+    (match Symbols.workspace ~dir with
+    | Some symbol -> symbol
+    | None -> Protocol.null)
+
 let test ~path =
   Uri.stripPath := true;
   match Files.readFile path with
-  | None -> assert false
+  | None -> (
+    (* Test for workspaces/directory *)
+    let nameTest = Filename.basename(path) in
+    match nameTest with
+    | "workspaceSymbols" -> workspaceSymbols ~dir:path
+    | _ -> assert false
+  )
   | Some text ->
     let lines = text |> String.split_on_char '\n' in
     let processLine i line =
@@ -323,7 +340,7 @@ let test ~path =
             DceCommand.command ()
           | "doc" ->
             print_endline ("DocumentSymbol " ^ path);
-            DocumentSymbol.command ~path
+            documentSymbol ~path
           | "hig" ->
             print_endline ("Highlight " ^ path);
             SemanticTokens.command ~debug:true

analysis/src/Cli.ml

@@ -24,7 +24,7 @@ let kindNumber = function
   | EnumMember -> 22
   | TypeParameter -> 26
 
-let command ~path =
+let document ~path =
   let symbols = ref [] in
   let rec exprKind (exp : Parsetree.expression) =
     match exp.pexp_desc with
@@ -134,19 +134,42 @@ let command ~path =
     let parser = Res_driver.parsingEngine.parseInterface ~forPrinter:false in
     let {Res_driver.parsetree = signature} = parser ~filename:path in
     iterator.signature iterator signature |> ignore);
-  let result =
-    !symbols
-    |> List.rev_map (fun (name, loc, kind) ->
-           Protocol.stringifyDocumentSymbolItem
-             {
-               name;
-               location =
-                 {
-                   uri = Uri.toString (Uri.fromPath path);
-                   range = Utils.cmtLocToRange loc;
-                 };
-               kind = kindNumber kind;
-             })
-    |> String.concat ",\n"
-  in
-  print_endline ("[\n" ^ result ^ "\n]")
+  !symbols
+  |> List.rev_map (fun (name, loc, kind) ->
+         let symbol : Protocol.documentSymbolItem =
+           {
+             name;
+             location =
+               {
+                 uri = Uri.toString (Uri.fromPath path);
+                 range = Utils.cmtLocToRange loc;
+               };
+             kind = kindNumber kind;
+           }
+         in
+         symbol)
+
+let workspace ~dir =
+  let open FindFiles in
+  let bsconfig = dir /+ "bsconfig.json" in
+  match Files.readFile bsconfig with
+  | None -> None
+  | Some text -> (
+    match Json.parse text with
+    | None -> None
+    | Some inner ->
+      let sourceDirectories =
+        getSourceDirectories ~includeDev:false ~baseDir:dir inner
+      in
+      let result =
+        sourceDirectories
+        |> List.map (fun srcDir ->
+               Files.readDirectory (dir /+ srcDir)
+               |> List.map (fun path -> dir /+ srcDir /+ path)
+               |> List.filter isSourceFile |> filterDuplicates)
+        |> List.flatten
+        |> List.map (fun path ->
+               document ~path |> List.map Protocol.stringifyDocumentSymbolItem)
+        |> List.flatten |> Protocol.array
+      in
+      Some result)

analysis/src/Commands.ml

@@ -1,16 +1,15 @@
 SHELL = /bin/bash
 
-node_modules/.bin/rescript:
-	npm install
-
-build: node_modules/.bin/rescript
-	node_modules/.bin/rescript
+build:
+	make -C document build
+	make -C workspaces build
 
 test: build
 	./test.sh
 
 clean:
-	rm -r node_modules lib
+	make -C document clean
+	make -C workspaces clean
 
 .DEFAULT_GOAL := test
 

analysis/src/DocumentSymbol.ml analysis/src/Symbols.mlanalysis/src/DocumentSymbol.ml renamed to analysis/src/Symbols.ml

@@ -0,0 +1,12 @@
+SHELL = /bin/bash
+
+node_modules/.bin/rescript:
+	npm install
+
+build: node_modules/.bin/rescript
+	node_modules/.bin/rescript
+
+clean:
+	rm -r node_modules lib
+
+.PHONY: clean