temporary command for extracting docs

Author: zth

client/src/commands.ts

@@ -6,6 +6,7 @@ import {
 } from "./commands/code_analysis";
 
 export { createInterface } from "./commands/create_interface";
+export { extractDocs } from "./commands/extract_docs";
 export { openCompiled } from "./commands/open_compiled";
 export { switchImplIntf } from "./commands/switch_impl_intf";
 

client/src/commands/extract_docs.ts

@@ -0,0 +1,50 @@
+import * as p from "vscode-languageserver-protocol";
+import { LanguageClient, RequestType } from "vscode-languageclient/node";
+import { Position, Uri, window, workspace, WorkspaceEdit } from "vscode";
+import path = require("path");
+
+export const extractDocsRequest = new RequestType<
+  p.TextDocumentIdentifier,
+  string,
+  void
+>("textDocument/extractDocs");
+
+export const extractDocs = async (client: LanguageClient) => {
+  if (!client) {
+    return window.showInformationMessage("Language server not running");
+  }
+
+  const editor = window.activeTextEditor;
+
+  if (!editor) {
+    return window.showInformationMessage("No active editor");
+  }
+
+  try {
+    const docUri = editor.document.uri.toString();
+    const res = await client.sendRequest(extractDocsRequest, {
+      uri: docUri,
+    });
+
+    const newFile = Uri.parse(
+      "untitled:" +
+        path.join(
+          workspace.workspaceFolders[0].uri.fsPath,
+          `${path.basename(docUri)}.json`
+        )
+    );
+    workspace.openTextDocument(newFile).then((document) => {
+      const edit = new WorkspaceEdit();
+      edit.insert(newFile, new Position(0, 0), JSON.stringify(res, null, 2));
+      return workspace.applyEdit(edit).then((success) => {
+        if (success) {
+          window.showTextDocument(document);
+        } else {
+          window.showInformationMessage("Error!");
+        }
+      });
+    });
+  } catch (e) {
+    console.error("failed", e);
+  }
+};

client/src/extension.ts

@@ -201,6 +201,10 @@ export function activate(context: ExtensionContext) {
     customCommands.openCompiled(client);
   });
 
+  commands.registerCommand("rescript-vscode.extract_docs", () => {
+    customCommands.extractDocs(client);
+  });
+
   commands.registerCommand(
     "rescript-vscode.go_to_location",
     async (fileUri: string, startLine: number, startCol: number) => {

package.json

@@ -49,6 +49,10 @@
 				"command": "rescript-vscode.create_interface",
 				"title": "ReScript: Create an interface file for this implementation file"
 			},
+			{
+				"command": "rescript-vscode.extract_docs",
+				"title": "ReScript: Extract documentation as JSON for file."
+			},
 			{
 				"command": "rescript-vscode.open_compiled",
 				"category": "ReScript",

server/src/server.ts

@@ -152,6 +152,12 @@ let openCompiledFileRequest = new v.RequestType<
   void
 >("textDocument/openCompiled");
 
+let extractDocsRequest = new v.RequestType<
+  p.TextDocumentIdentifier,
+  p.TextDocumentIdentifier,
+  void
+>("textDocument/extractDocs");
+
 let getCurrentCompilerDiagnosticsForFile = (
   fileUri: string
 ): p.Diagnostic[] => {
@@ -965,6 +971,24 @@ function createInterface(msg: p.RequestMessage): p.Message {
   }
 }
 
+function extractDocs(msg: p.RequestMessage): p.Message {
+  let params = msg.params as p.TextDocumentIdentifier;
+  let filePath = fileURLToPath(params.uri);
+
+  let response = utils.runAnalysisCommand(
+    filePath,
+    ["extractDocs", filePath],
+    msg
+  );
+
+  let res: p.ResponseMessage = {
+    jsonrpc: c.jsonrpcVersion,
+    id: msg.id,
+    result: response.result,
+  };
+  return res;
+}
+
 function openCompiledFile(msg: p.RequestMessage): p.Message {
   let params = msg.params as p.TextDocumentIdentifier;
   let filePath = fileURLToPath(params.uri);
@@ -1230,6 +1254,8 @@ function onMessage(msg: p.Message) {
       send(createInterface(msg));
     } else if (msg.method === openCompiledFileRequest.method) {
       send(openCompiledFile(msg));
+    } else if (msg.method === extractDocsRequest.method) {
+      send(extractDocs(msg));
     } else if (msg.method === p.InlayHintRequest.method) {
       let params = msg.params as InlayHintParams;
       let extName = path.extname(params.textDocument.uri);