diff --git a/extension.js b/extension.js index 1771bae..6789a48 100644 --- a/extension.js +++ b/extension.js @@ -2,51 +2,58 @@ const { spawnSync } = require("node:child_process"); const { dirname } = require("node:path"); const vscode = require("vscode"); +const yamlformattedLanguages = [ + "yaml", + "github-actions-workflow" // Provided in https://github.com/github/vscode-github-actions +]; +const provider = { + provideDocumentFormattingEdits(document) { + const workspaceFolder = vscode.workspace.getWorkspaceFolder(document.uri); + const config = vscode.workspace.getConfiguration("", document.uri); -function activate() { - vscode.languages.registerDocumentFormattingEditProvider("yaml", { - provideDocumentFormattingEdits(document) { - const workspaceFolder = vscode.workspace.getWorkspaceFolder(document.uri); - const config = vscode.workspace.getConfiguration("", document.uri); - - const args = config.get("yamlfmt.args", []).filter(arg => arg !== "-in"); - args.push("-in"); - - const result = spawnSync("yamlfmt", args, { - cwd: workspaceFolder ? workspaceFolder.uri.fsPath : dirname(document.uri.fsPath), - input: document.getText(), - }); - - if (result.error) { - console.error(result.error); - const prefix = "spawnSync "; - vscode.window.showErrorMessage( - result.error.message.startsWith(prefix) ? - result.error.message.substring(prefix.length) : - result.error.message - ); - return []; - } - - if (result.stderr.length > 0) { - console.log(result.stderr.toString()); - vscode.window.showErrorMessage(result.stderr.toString()); - return []; - } - - if (result.stdout.length < 1) { - console.warn("yamlfmt's stdout buffer is empty"); - return []; - } - - const range = new vscode.Range( - document.lineAt(0).range.start, - document.lineAt(document.lineCount - 1).range.end, + const args = config.get("yamlfmt.args", []).filter(arg => arg !== "-in"); + args.push("-in"); + + const result = spawnSync("yamlfmt", args, { + cwd: workspaceFolder ? workspaceFolder.uri.fsPath : dirname(document.uri.fsPath), + input: document.getText(), + }); + + if (result.error) { + console.error(result.error); + const prefix = "spawnSync "; + vscode.window.showErrorMessage( + result.error.message.startsWith(prefix) ? + result.error.message.substring(prefix.length) : + result.error.message ); + return []; + } + + if (result.stderr.length > 0) { + console.log(result.stderr.toString()); + vscode.window.showErrorMessage(result.stderr.toString()); + return []; + } - return [vscode.TextEdit.replace(range, result.stdout.toString())]; + if (result.stdout.length < 1) { + console.warn("yamlfmt's stdout buffer is empty"); + return []; } - }); + + const range = new vscode.Range( + document.lineAt(0).range.start, + document.lineAt(document.lineCount - 1).range.end, + ); + + return [vscode.TextEdit.replace(range, result.stdout.toString())]; + } +}; + +function activate() { + for (const lang of yamlformattedLanguages) { + vscode.languages.registerDocumentFormattingEditProvider(lang, provider); + } } function deactivate() { } diff --git a/package.json b/package.json index cb007a2..696e6cb 100644 --- a/package.json +++ b/package.json @@ -23,7 +23,8 @@ "multi-root ready" ], "activationEvents": [ - "onLanguage:yaml" + "onLanguage:yaml", + "onLanguage:github-actions-workflow" ], "main": "./extension.js", "contributes": {