Adds table converter code action (#9)

Author: rpaul-stripe

server/plugins/codeaction.ts

@@ -0,0 +1,39 @@
+import * as LSP from "vscode-languageserver/node";
+import * as Markdoc from "@markdoc/markdoc";
+
+import type { Config, ServiceInstances } from "../types";
+
+export default class CodeActionProvider {
+  constructor(
+    protected config: Config,
+    protected connection: LSP.Connection,
+    protected services: ServiceInstances
+  ) {
+    connection.onCodeAction(this.onCodeAction.bind(this));
+  }
+  
+  register(registration: LSP.BulkRegistration) {
+    registration.add(LSP.CodeActionRequest.type, {documentSelector: null});
+  }
+
+  findTables(ast: Markdoc.Node, params: LSP.CodeActionParams): LSP.Command[] {
+    const output: LSP.Command[] = []; 
+    const {line} = params.range.start;
+    const {uri} = params.textDocument;
+    
+    for (const node of ast.walk())
+      if (node.type === 'table' && node.lines.includes(line))
+        output.push({
+          command: 'markdoc.convertTable',
+          title: 'Convert to Markdoc Table',
+          arguments: [uri, line]
+        });
+
+    return output;
+  }
+
+  onCodeAction(params: LSP.CodeActionParams): (LSP.CodeAction | LSP.Command)[] {
+    const ast = this.services.Documents.ast(params.textDocument.uri);
+    return ast ? this.findTables(ast, params) : [];
+  }
+}

server/plugins/formatting.test.ts

@@ -31,11 +31,10 @@ function mockDoc(content: string) {
 
 test("formatting provider", async (t) => {
   // @ts-expect-error
-  const provider = new FormattingProvider({}, mockConnection, {});
+  const provider = new FormattingProvider({}, mockConnection, {Commands: {add(...args: any) {}}});
 
   await t.test("simple full-text formatting", () => {
     const doc = mockDoc(example1);
-    // @ts-expect-error
     const [output] = provider.formatRange(doc);
     assert.strictEqual(output.newText.trim(), example1Formatted.trim());
   });

server/plugins/formatting.ts

@@ -15,6 +15,7 @@ export default class FormattingProvider {
     this.tokenizer = new Markdoc.Tokenizer(config.markdoc ?? {});
     connection.onDocumentFormatting(this.onDocumentFormatting.bind(this));
     connection.onDocumentRangeFormatting(this.onRangeFormatting.bind(this));
+    services.Commands.add('markdoc.convertTable', this.convertTable.bind(this));
   }
 
   register(registration: LSP.BulkRegistration) {
@@ -26,8 +27,26 @@ export default class FormattingProvider {
       documentSelector: null,
     });
   }
+  
+  convertTable(uri: string, line: number) {
+    const ast = this.services.Documents.ast(uri);
+    if (!ast) return;
 
-  protected formatRange(doc: TextDocument, range?: LSP.Range) {
+    for (const node of ast.walk())
+      if (node.type === 'table' && node.lines.includes(line)) {
+        const content = new Markdoc.Ast.Node('tag', {}, [node], 'table');
+        const newText = Markdoc.format(content);
+        const [start, end] = node.lines;
+        const range = LSP.Range.create(start, 0, end + 1, 0);
+
+        const wschange = new LSP.WorkspaceChange();
+        const edit = wschange.getTextEditChange(uri);
+        edit.replace(range, newText);
+        this.connection.workspace.applyEdit(wschange.edit);
+      }
+  }
+
+  formatRange(doc: TextDocument, range?: LSP.Range) {
     const actualRange = range
       ? LSP.Range.create(range.start.line, 0, range.end.line + 1, 0)
       : LSP.Range.create(0, 0, doc.lineCount, 0);

server/plugins/index.ts

@@ -1,3 +1,4 @@
+import CodeActionProvider from "./codeaction";
 import CompletionProvider from "./completion";
 import DefinitionProvider from "./definition";
 import DependenciesProvider from "./dependencies";
@@ -10,6 +11,7 @@ import ValidationProvider from "./validation";
 import Watch from "./watch";
 
 export {
+  CodeActionProvider,
   CompletionProvider,
   DefinitionProvider,
   DependenciesProvider,