Use a resolver for table conversion (#12)

Author: rpaul-stripe

server/package-lock.json

@@ -6,7 +6,7 @@
   "author": "Ryan Paul",
   "license": "MIT",
   "devDependencies": {
-    "@markdoc/markdoc": "^0.3.2",
+    "@markdoc/markdoc": "^0.3.3",
     "@types/picomatch": "^2.3.0",
     "picomatch": "^2.3.1",
     "typescript": "^5.0.4",

server/package.json

@@ -10,23 +10,38 @@ export default class CodeActionProvider {
     protected services: ServiceInstances
   ) {
     connection.onCodeAction(this.onCodeAction.bind(this));
+    connection.onCodeActionResolve(this.onCodeActionResolve.bind(this));
   }
 
   register(registration: LSP.BulkRegistration) {
-    registration.add(LSP.CodeActionRequest.type, {documentSelector: null});
+    registration.add(LSP.CodeActionRequest.type, {documentSelector: null, resolveProvider: true});
   }
 
-  findTables(ast: Markdoc.Node, params: LSP.CodeActionParams): LSP.Command[] {
-    const output: LSP.Command[] = []; 
+  convertTable(uri: string, line: number): LSP.WorkspaceEdit | void {
+    const ast = this.services.Documents.ast(uri);
+    if (!ast) return;
+
+    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);
+        return {changes: {[uri]: [{range, newText}]}};
+      }
+    }
+  }
+
+  findTables(ast: Markdoc.Node, params: LSP.CodeActionParams): LSP.CodeAction[] {
+    const output: LSP.CodeAction[] = []; 
     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',
+          data: {type: 'convertTable', uri, line},
           title: 'Convert to Markdoc Table',
-          arguments: [uri, line]
         });
 
     return output;
@@ -36,4 +51,14 @@ export default class CodeActionProvider {
     const ast = this.services.Documents.ast(params.textDocument.uri);
     return ast ? this.findTables(ast, params) : [];
   }
+
+  onCodeActionResolve(action: LSP.CodeAction): LSP.CodeAction {
+    if (action.data.type === 'convertTable') {
+      const {uri, line} = action.data;
+      const edit = this.convertTable(uri, line);
+      if (edit) return {...action, edit};
+    }
+
+    return action;
+  }
 }

server/plugins/codeaction.ts

@@ -15,7 +15,6 @@ 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) {

server/plugins/formatting.ts

@@ -78,7 +78,15 @@ export default class ValidationProvider {
       if (Scanner.has(part.attributes.file))
         partials[part.attributes.file] = true;
 
-    return { ...Schema?.get(uri), partials };
+    const schema = Schema?.get(uri);
+    return {
+      ...schema,
+      partials,
+      validation: {
+        environment: 'language-server',
+        ...schema?.validation
+      }
+    };
   }
 
   onDidSave({ document: { uri } }: TextChangeEvent) {

server/plugins/validation.ts

@@ -53,6 +53,7 @@ export type MarkdocConfig = {
   typographer?: boolean;
   allowIndentation?: boolean;
   allowComments?: boolean;
+  validateFunctions?: boolean;
 };
 
 export type RoutingConfig = {