code: Ensure lsp-devtools can be relaunched when closed

swyddfa · Sep 12, 2023 · 598bf4c · 598bf4c
1 parent c0552bd
commit 598bf4c
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/code/src/node/client.ts b/code/src/node/client.ts
@@ -69,9 +69,19 @@ export class EsbonioClient {
   ) {
     this.handlers = new Map()
 
+    // Restart server implementation
     context.subscriptions.push(
       vscode.commands.registerCommand(Commands.RESTART_SERVER, async () => await this.restartServer())
     )
+
+    // Unset devtools task when it finishes.
+    context.subscriptions.push(
+      vscode.tasks.onDidEndTask((event) => {
+        if (event.execution === this.devtools) {
+          this.devtools = undefined
+        }
+      })
+    )
   }
 
   public addHandler(event: string, handler: any) {
@@ -146,8 +156,14 @@ export class EsbonioClient {
     command.push("-S")
 
     let config = vscode.workspace.getConfiguration("esbonio")
-    if (config.get<boolean>('server.enableDevTools')) {
+    let serverDevtools = config.get<boolean>('server.enableDevTools')
+    let sphinxDevtools = config.get<boolean>('sphinx.enableDevTools')
+
+    if (serverDevtools || sphinxDevtools) {
       await this.startDevtools(command[0], ...command.slice(1), "-m", "lsp_devtools", "tui")
+    }
+
+    if (serverDevtools) {
       command.push("-m", "lsp_devtools", "agent", "--", ...command)
     }