Merge pull request #5 from xeger/config

Allow users to configure the extension

Author: kddnewton

package.json

@@ -49,6 +49,30 @@
         "title": "Syntax Tree: Visualize"
       }
     ],
+    "configuration": {
+      "type": "object",
+      "title": "Syntax Tree",
+      "properties": {
+        "syntaxTree.singleQuotes": {
+          "default": false,
+          "markdownDescription": "Uses single-quoted strings when possible.",
+          "type": "boolean"
+        },
+        "syntaxTree.trailingComma": {
+          "default": false,
+          "markdownDescription": "Adds an extra comma after the last item of arrays, hashes and parameters.",
+          "type": "boolean"
+        },
+        "syntaxTree.additionalPlugins": {
+          "default": [],
+          "markdownDescription": "Registers [extra behaviors](https://github.com/ruby-syntax-tree/syntax_tree#plugins) with the language server.",
+          "items": {
+            "type": "string"
+          },
+          "type": "array"
+        }
+      }
+    },
     "colors": [
       {
         "id": "syntaxTree.inlayHints",

src/extension.ts

@@ -1,6 +1,6 @@
 "use strict";
 
-import { ExtensionContext, commands, window, workspace } from "vscode";
+import { ConfigurationChangeEvent, ExtensionContext, commands, window, workspace } from "vscode";
 import { LanguageClient, ServerOptions } from "vscode-languageclient/node";
 import { promisify } from "util";
 import { exec } from "child_process";
@@ -16,19 +16,42 @@ export function activate(context: ExtensionContext) {
   let visualizer: Visualize | null = null;
 
   context.subscriptions.push(
+    outputChannel,
     commands.registerCommand("syntaxTree.start", startLanguageServer),
     commands.registerCommand("syntaxTree.stop", stopLanguageServer),
     commands.registerCommand("syntaxTree.restart", restartLanguageServer),
     commands.registerCommand("syntaxTree.visualize", () => visualizer?.visualize()),
     commands.registerCommand("syntaxTree.showOutputChannel", () => outputChannel.show()),
-    outputChannel
+    workspace.onDidChangeConfiguration(event =>
+      event.affectsConfiguration("syntaxTree") &&
+      restartLanguageServer())
   );
 
   return startLanguageServer();
 
   async function startLanguageServer() {
-    outputChannel.appendLine("Starting language server...");
-    let run: ServerOptions = { command: "stree", args: ["lsp"] };
+    const config = workspace.getConfiguration("syntaxTree");
+    const addlPlugins = config.get<string[]>("additionalPlugins") || [];
+    const singleQuotes = config.get<boolean>("singleQuotes");
+    const trailingComma = config.get<boolean>("trailingComma");
+
+    const args = ["lsp"];
+
+    const plugins = new Set<string>();
+    if (singleQuotes) {
+      plugins.add("plugin/single_quotes");
+    }
+    if (trailingComma) {
+      plugins.add("plugin/trailing_comma");
+    }
+    addlPlugins.forEach(plugins.add);
+
+    if (plugins.size) {
+      args.push(`--plugins=${Array.from(plugins).join(",")}`);
+    }
+
+    outputChannel.appendLine(`Starting language server with ${plugins.size} plugin(s)...`);
+    let run: ServerOptions = { command: "stree", args };
 
     if (workspace.workspaceFolders) {
       const cwd = workspace.workspaceFolders![0].uri.fsPath;