Поддержка нескольких workspace в LSP

src/main/java/com/github/_1c_syntax/bsl/languageserver/BSLLanguageServer.java

@@ -23,6 +23,7 @@
 
 import com.github._1c_syntax.bsl.languageserver.configuration.LanguageServerConfiguration;
 import com.github._1c_syntax.bsl.languageserver.context.ServerContext;
+import com.github._1c_syntax.bsl.languageserver.context.ServerContextProvider;
 import com.github._1c_syntax.bsl.languageserver.jsonrpc.DiagnosticParams;
 import com.github._1c_syntax.bsl.languageserver.jsonrpc.Diagnostics;
 import com.github._1c_syntax.bsl.languageserver.jsonrpc.ProtocolExtension;
@@ -62,6 +63,8 @@
 import org.eclipse.lsp4j.TextDocumentClientCapabilities;
 import org.eclipse.lsp4j.TextDocumentSyncKind;
 import org.eclipse.lsp4j.TextDocumentSyncOptions;
+import org.eclipse.lsp4j.WorkspaceFoldersOptions;
+import org.eclipse.lsp4j.WorkspaceServerCapabilities;
 import org.eclipse.lsp4j.WorkspaceSymbolOptions;
 import org.eclipse.lsp4j.jsonrpc.messages.Either;
 import org.eclipse.lsp4j.services.LanguageServer;
@@ -97,7 +100,7 @@ public class BSLLanguageServer implements LanguageServer, ProtocolExtension {
   private final BSLWorkspaceService workspaceService;
   private final CommandProvider commandProvider;
   private final ClientCapabilitiesHolder clientCapabilitiesHolder;
-  private final ServerContext context;
+  private final ServerContextProvider serverContextProvider;
   private final ServerInfo serverInfo;
   private final SemanticTokensLegend legend;
 
@@ -131,6 +134,7 @@ public CompletableFuture<InitializeResult> initialize(InitializeParams params) {
     capabilities.setExecuteCommandProvider(getExecuteCommandProvider());
     capabilities.setDiagnosticProvider(getDiagnosticProvider());
     capabilities.setSemanticTokensProvider(getSemanticTokensProvider());
+    capabilities.setWorkspace(getWorkspaceCapabilities());
 
     var result = new InitializeResult(capabilities, serverInfo);
 
@@ -143,31 +147,29 @@ private void setConfigurationRoot(InitializeParams params) {
       return;
     }
 
-    var rootUri = workspaceFolders.get(0).getUri();
-    Path rootPath;
-    try {
-      rootPath = new File(new URI(rootUri).getPath()).getCanonicalFile().toPath();
-    } catch (URISyntaxException | IOException e) {
-      LOGGER.error("Can't read root URI from initialization params.", e);
-      return;
-    }
-
-    var configurationRoot = LanguageServerConfiguration.getCustomConfigurationRoot(
-      configuration,
-      rootPath);
-    context.setConfigurationRoot(configurationRoot);
+    // Добавляем все workspace folders
+    workspaceFolders.forEach(serverContextProvider::addWorkspace);
   }
 
   @Override
   public void initialized(InitializedParams params) {
     var factory = new NamedForkJoinWorkerThreadFactory("populate-context-");
     var executorService = new ForkJoinPool(ForkJoinPool.getCommonPoolParallelism(), factory, null, true);
-    CompletableFuture
-      .runAsync(context::populateContext, executorService)
+
+    // Populate all workspace contexts
+    var allContexts = serverContextProvider.getAllContexts();
+    var tasks = allContexts.stream()
+      .map(serverContext -> CompletableFuture.runAsync(
+        serverContext::populateContext,
+        executorService
+      ))
+      .toArray(CompletableFuture[]::new);
+
+    CompletableFuture.allOf(tasks)
       .whenComplete((Void unused, @Nullable Throwable throwable) -> {
         executorService.shutdown();
         if (throwable != null) {
-          LOGGER.error("Error populating context", throwable);
+          LOGGER.error("Error populating workspace contexts", throwable);
         }
       });
   }
@@ -176,7 +178,7 @@ public void initialized(InitializedParams params) {
   public CompletableFuture<Object> shutdown() {
     shutdownWasCalled = true;
     textDocumentService.reset();
-    context.clear();
+    serverContextProvider.clear();
     return CompletableFuture.completedFuture(Boolean.TRUE);
   }
 
@@ -384,4 +386,15 @@ private SemanticTokensWithRegistrationOptions getSemanticTokensProvider() {
     return semanticTokensProvider;
   }
 
+  private static WorkspaceServerCapabilities getWorkspaceCapabilities() {
+    var workspaceCapabilities = new WorkspaceServerCapabilities();
+
+    var workspaceFoldersOptions = new WorkspaceFoldersOptions();
+    workspaceFoldersOptions.setSupported(Boolean.TRUE);
+    workspaceFoldersOptions.setChangeNotifications(Boolean.TRUE);
+
+    workspaceCapabilities.setWorkspaceFolders(workspaceFoldersOptions);
+    return workspaceCapabilities;
+  }
+
 }