Await language server shutdown on deactivate (#58)

Signed-off-by: Ben Sherman <[email protected]>
nextflow-io · Jan 13, 2025 · b45ed8c · b45ed8c
1 parent 337344d
commit b45ed8c
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/extension.ts b/src/extension.ts
@@ -142,11 +142,11 @@ function restartLanguageServer() {
   );
 }
 
-function stopLanguageServer() {
+function stopLanguageServer(): Thenable<void> | undefined {
   if (!languageClient) {
-    return;
+    return undefined;
   }
-  languageClient.stop();
+  return languageClient.stop();
 }
 
 function onDidChangeConfiguration(event: vscode.ConfigurationChangeEvent) {
@@ -174,7 +174,6 @@ export function activate(context: vscode.ExtensionContext) {
   startLanguageServer();
 }
 
-export function deactivate() {
-  stopLanguageServer();
-  extensionContext = null;
+export function deactivate(): Thenable<void> | undefined {
+  return stopLanguageServer();
 }