Avoid re-registering visualize command after restart

Author: xeger

src/Visualize.ts

@@ -16,10 +16,7 @@ class Visualize implements Disposable, TextDocumentContentProvider {
   constructor(languageClient: LanguageClient, outputChannel: OutputChannel) {
     this.languageClient = languageClient;
     this.outputChannel = outputChannel;
-    this.disposables = [
-      commands.registerCommand("syntaxTree.visualize", this.visualize),
-      workspace.registerTextDocumentContentProvider("syntaxTree.visualize", this)
-    ];
+    this.disposables = [];
   }
 
   dispose() {
@@ -40,7 +37,7 @@ class Visualize implements Disposable, TextDocumentContentProvider {
       const doc = await workspace.openTextDocument(uri);
       languages.setTextDocumentLanguage(doc, "plaintext");
 
-			await window.showTextDocument(doc, ViewColumn.Beside, true);
+      await window.showTextDocument(doc, ViewColumn.Beside, true);
     }
   }
 }

src/extension.ts

@@ -11,13 +11,15 @@ import Visualize from "./Visualize";
 const promiseExec = promisify(exec);
 
 export function activate(context: ExtensionContext) {
-  let languageClient: LanguageClient | null = null;
   const outputChannel = window.createOutputChannel("Syntax Tree");
+  let languageClient: LanguageClient | null = null;
+  let visualizer: Visualize | null = null;
 
   context.subscriptions.push(
     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
   );
@@ -35,7 +37,7 @@ export function activate(context: ExtensionContext) {
         await promiseExec("bundle show syntax_tree", { cwd });
         run = { command: "bundle", args: ["exec", "stree", "lsp"], options: { cwd } };
       } catch {
-        outputChannel.appendLine("No bundled syntax_tree, running global stree.");  
+        outputChannel.appendLine("No bundled syntax_tree, running global stree.");
       }
     }