add formatters

Author: Rusty Key

package.json

@@ -86,6 +86,11 @@
           "description":
             "The path to the `env` command which prints the language server environment for debugging editor issues."
         },
+        "reason.path.ocamlformat": {
+          "type": "string",
+          "default": "ocamlformat",
+          "description": "The path to the `ocamlformat` binary."
+        },
         "reason.path.ocamlmerlin": {
           "type": "string",
           "default": "ocamlmerlin",
@@ -96,11 +101,6 @@
           "default": null,
           "description": "The path to the `ocamlmerlin-lsp` binary."
         },
-        "reason.path.ocpindent": {
-          "type": "string",
-          "default": "ocp-indent",
-          "description": "The path to the `ocp-indent` binary."
-        },
         "reason.path.opam": {
           "type": "string",
           "default": "opam",

src/extension.ts

@@ -2,6 +2,8 @@
 
 import * as vscode from "vscode";
 import * as client from "./client";
+import { register as registerOcamlForamtter } from "./formatters/ocaml";
+import { register as registerReasonForamtter } from "./formatters/reason";
 
 const reasonConfiguration = {
   indentationRules: {
@@ -93,6 +95,8 @@ const reasonConfiguration = {
 export async function activate(context: vscode.ExtensionContext) {
   context.subscriptions.push(vscode.languages.setLanguageConfiguration("reason", reasonConfiguration));
   await client.launch(context);
+  registerOcamlForamtter();
+  registerReasonForamtter();
 }
 
 export function deactivate() {

src/formatters/ocaml.ts

@@ -0,0 +1,30 @@
+import { execSync } from "child_process";
+import * as path from "path";
+import * as vscode from "vscode";
+import { getFullTextRange } from "./utils";
+
+export function register() {
+  const configuration = vscode.workspace.getConfiguration("ocaml-reason-format");
+  const rootPath = vscode.workspace.rootPath || "";
+
+  vscode.languages.registerDocumentFormattingEditProvider("ocaml", {
+    provideDocumentFormattingEdits(_document: vscode.TextDocument): vscode.TextEdit[] {
+      const formatterPath = configuration.get<string | undefined>("ocamlformat");
+      const formatter = formatterPath ? path.resolve(rootPath, formatterPath) : "ocamlformat";
+      const textEditor = vscode.window.activeTextEditor;
+
+      if (textEditor) {
+        const text = textEditor.document.getText();
+        const filePath = textEditor.document.fileName;
+        const formattedText = execSync(`cd ${rootPath} && ${formatter} --name=${filePath} /dev/stdin <<<'${text}'`, {
+          cwd: rootPath,
+        }).toString();
+        const textRange = getFullTextRange(textEditor);
+
+        return [vscode.TextEdit.replace(textRange, formattedText)];
+      } else {
+        return [];
+      }
+    },
+  });
+}

src/formatters/reason.ts

@@ -0,0 +1,27 @@
+import { execSync } from "child_process";
+import * as path from "path";
+import * as vscode from "vscode";
+import { getFullTextRange } from "./utils";
+
+export function register() {
+  const configuration = vscode.workspace.getConfiguration("ocaml-reason-format");
+  const rootPath = vscode.workspace.rootPath || "";
+
+  vscode.languages.registerDocumentFormattingEditProvider("reason", {
+    provideDocumentFormattingEdits(_document: vscode.TextDocument): vscode.TextEdit[] {
+      const formatterPath = configuration.get<string | undefined>("refmt");
+      const formatter = formatterPath ? path.resolve(rootPath, formatterPath) : "refmt";
+      const textEditor = vscode.window.activeTextEditor;
+
+      if (textEditor) {
+        const text = textEditor.document.getText();
+        const formattedText = execSync(`${formatter} <<<'${text}'`).toString();
+        const textRange = getFullTextRange(textEditor);
+
+        return [vscode.TextEdit.replace(textRange, formattedText)];
+      } else {
+        return [];
+      }
+    },
+  });
+}

src/formatters/utils.ts

@@ -0,0 +1,13 @@
+import * as vscode from "vscode";
+
+export function getFullTextRange(textEditor: vscode.TextEditor) {
+  const firstLine = textEditor.document.lineAt(0);
+  const lastLine = textEditor.document.lineAt(textEditor.document.lineCount - 1);
+
+  return new vscode.Range(
+    0,
+    firstLine.range.start.character,
+    textEditor.document.lineCount - 1,
+    lastLine.range.end.character,
+  );
+}