Auto formatting

hexalang · Jun 21, 2024 · 8676064 · 8676064
1 parent bc002f1
commit 8676064
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/extension/extension.js b/extension/extension.js
@@ -255,6 +255,33 @@ exports.activate = function (context) {
         { label: 'Hexa' }
     )
 
+    // Formatting
+    vscode.languages.registerDocumentFormattingEditProvider('hexa', {
+        provideDocumentFormattingEdits(document/*: vscode.TextDocument*/, options, token)/*: vscode.TextEdit[]*/ {
+            const fullText = document.getText()
+            const fullFile = new vscode.Range(document.lineAt(0).range.start, document.lineAt(document.lineCount - 1).range.end)
+
+            return new Promise((resolve, reject) => {
+                const commands = [linter.wholeDocumentFormatting(fullText)]
+
+                linter.request(commands,
+                    onJSON => {
+                        const code = onJSON[0][0].content
+                        if (code && code.length > 0) {
+                            const edits = [vscode.TextEdit.replace(fullFile, code)]
+                            resolve(edits)
+                        } else {
+                            reject()
+                        }
+                    },
+                    onError => {
+                        reject()
+                    }
+                )
+            })
+        }
+    })
+
     /** @implements {HoverProvider} */
     class HexaHoverProvider {
         /**

diff --git a/extension/hexa-linter.js b/extension/hexa-linter.js
@@ -271,6 +271,13 @@ class HexaLinter {
         }
     }
 
+    wholeDocumentFormatting(content) {
+        return {
+            kind: 'WholeDocumentFormatting',
+            payload: content
+        }
+    }
+
     request(commands, onJSON, onError) {
         const req = http.request(options, res => {
             const chunks = []