Merge pull request #99 from tintinweb/auto-formatting

Auto formatting

package.json

@@ -168,6 +168,7 @@
         "jquery-mousewheel": "^3.1.13",
         "lodash": "^4.17.21",
         "text-encoding": "^0.7.0",
+        "ts-graphviz": "^0.16.0",
         "vscode-uri": "^3.0.3"
     },
     "scripts": {

src/extension.ts

@@ -10,6 +10,7 @@ import InteractiveWebviewGenerator from "./features/interactiveWebview";
 import PreviewPanel from "./features/previewPanel";
 import ColorProvider from "./language/ColorProvider";
 import DotCompletionItemProvider from "./language/CompletionItemProvider";
+import DotDocumentFormatter from "./language/DocumentFormatter";
 import DotHoverProvider from "./language/HoverProvider";
 import SymbolProvider from "./language/SymbolProvider";
 import * as settings from "./settings";
@@ -117,6 +118,10 @@ function onActivate(context: vscode.ExtensionContext) {
     [settings.languageId],
     symProvider,
   ));
+  context.subscriptions.push(vscode.languages.registerDocumentFormattingEditProvider(
+    [settings.languageId],
+    new DotDocumentFormatter(),
+  ));
 }
 
 /* exports */

src/language/DocumentFormatter.ts

@@ -0,0 +1,42 @@
+import { parse } from "@ts-graphviz/parser";
+import { toDot } from "ts-graphviz";
+import {
+  DocumentFormattingEditProvider,
+  // FormattingOptions,
+  Position,
+  ProviderResult,
+  Range,
+  TextDocument,
+  TextEdit,
+  window,
+} from "vscode";
+
+export default class DotDocumentFormatter implements DocumentFormattingEditProvider {
+  // eslint-disable-next-line class-methods-use-this
+  provideDocumentFormattingEdits(
+    document: TextDocument,
+    // options: FormattingOptions,
+    // token: CancellationToken,
+  ): ProviderResult<TextEdit[]> {
+    try {
+      const d = parse(document.getText());
+      const dot = toDot(d);
+
+      const edit: TextEdit[] = [];
+      edit.push(new TextEdit(
+        new Range(
+          new Position(0, 0),
+          new Position(document.lineCount - 1, document.lineAt(document.lineCount - 1).text.length),
+        ),
+        dot,
+      ));
+      return edit;
+    } catch (e: any) {
+      // (parser error) don't bubble up as a pot. unhandled thenable promise;
+      // explicitly return "no change" instead.
+      // show error message
+      window.showErrorMessage(`${e.name} (@${e.location.start.line}:${e.location.start.column}): ${e.message}`);
+      return undefined;
+    }
+  }
+}

src/language/SymbolProvider.ts

@@ -257,7 +257,10 @@ implements
   }
 
   // eslint-disable-next-line class-methods-use-this
-  private containingSymbol(symbols: DocumentSymbolInformation[], position: Position) {
+  private containingSymbol(
+    symbols: DocumentSymbolInformation[],
+    position: Position,
+  ) {
     let closestSymbol: DocumentSymbolInformation|undefined;
 
     const crawler = (syms: DocumentSymbolInformation[]) => {