Spawn multiple language servers to make extension work in Multi Root workspace

Author: vojvodics

client/client.ts

@@ -9,10 +9,22 @@ import {
 import { resolve } from 'path';
 import { existsSync } from 'fs';
 
-export function initializeLanguageClient(vlsModulePath: string, globalSnippetDir: string): LanguageClient {
+export function initializeLanguageClient(
+  vlsModulePath: string,
+  globalSnippetDir: string,
+  workspaceFolder?: vscode.WorkspaceFolder
+): LanguageClient {
   const debugOptions = { execArgv: ['--nolazy', '--inspect=6005'] };
 
-  const documentSelector = ['vue'];
+  const documentSelector = workspaceFolder
+    ? [
+        {
+          language: 'vue',
+          scheme: 'file',
+          pattern: `${workspaceFolder.uri.fsPath}/**/*`
+        }
+      ]
+    : ['vue'];
   const config = vscode.workspace.getConfiguration();
 
   let serverPath;
@@ -56,7 +68,8 @@ export function initializeLanguageClient(vlsModulePath: string, globalSnippetDir
       config,
       globalSnippetDir
     },
-    revealOutputChannelOn: RevealOutputChannelOn.Never
+    revealOutputChannelOn: RevealOutputChannelOn.Never,
+    workspaceFolder
   };
 
   return new LanguageClient('vetur', 'Vue Language Server', serverOptions, clientOptions);

client/vueMain.ts

@@ -1,5 +1,5 @@
 import * as vscode from 'vscode';
-import { LanguageClient, WorkspaceEdit } from 'vscode-languageclient';
+import { LanguageClient, WorkspaceEdit, WorkspaceFolder } from 'vscode-languageclient';
 import { generateGrammarCommandHandler } from './commands/generateGrammarCommand';
 import { registerLanguageConfigurations } from './languages';
 import { initializeLanguageClient } from './client';
@@ -12,6 +12,11 @@ import {
 import { getGlobalSnippetDir } from './userSnippetDir';
 import { generateOpenUserScaffoldSnippetFolderCommand } from './commands/openUserScaffoldSnippetFolderCommand';
 
+/**
+ * Map of workspace folder URI -> LanguageClient
+ */
+const clients = new Map<string, LanguageClient>();
+
 export async function activate(context: vscode.ExtensionContext) {
   const isInsiders = vscode.env.appName.includes('Insiders');
   const globalSnippetDir = getGlobalSnippetDir(isInsiders);
@@ -28,6 +33,15 @@ export async function activate(context: vscode.ExtensionContext) {
     vscode.commands.registerCommand('vetur.generateGrammar', generateGrammarCommandHandler(context.extensionPath))
   );
 
+  registerLanguageConfigurations();
+
+  /**
+   * Vue Language Server path
+   * To be able to work with multiple workspaces, we initialize one language service per workspace folder
+   */
+
+  const serverModule = context.asAbsolutePath(join('server', 'dist', 'vueServerMain.js'));
+
   /**
    * Open custom snippet folder
    */

server/src/vueServerMain.ts

@@ -11,7 +11,7 @@ connection.onInitialize(
   async (params: InitializeParams): Promise<InitializeResult> => {
     await vls.init(params);
 
-    console.log('Vetur initialized');
+    console.log(`Vetur initialized for workspace folder: ${params.rootPath}`);
 
     return {
       capabilities: vls.capabilities