Distribute some notifications over all remote servers.

Author: toinehartman

rascal-lsp/src/main/java/org/rascalmpl/vscode/lsp/LanguageServerRouter.java

@@ -48,6 +48,7 @@
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.stream.Stream;
 import org.apache.commons.lang3.tuple.Triple;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
@@ -128,6 +129,11 @@ public CompletableFuture<IBaseLanguageServerExtensions> languageByName(String la
         return service;
     }
 
+    @Override
+    public Stream<CompletableFuture<IBaseLanguageServerExtensions>> allRoutes() {
+        return languageServers.values().stream();
+    }
+
     @Override
     public CompletableFuture<IBaseLanguageServerExtensions> route(ISourceLocation file) {
         return route(extension(file));

rascal-lsp/src/main/java/org/rascalmpl/vscode/lsp/parametric/routing/RoutingTextDocumentService.java

@@ -33,6 +33,7 @@
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
@@ -131,6 +132,13 @@ public void setServer(LanguageServerRouter server) {
         this.server = server;
     }
 
+    @Override
+    public Stream<CompletableFuture<IBaseTextDocumentService>> allRoutes() {
+        return availableServer()
+            .allRoutes()
+            .map(server -> server.thenApply(IBaseLanguageServerExtensions::getIBaseTextDocumentService));
+    }
+
     @Override
     public CompletableFuture<IBaseTextDocumentService> route(ISourceLocation loc) {
         return availableServer()
@@ -242,14 +250,12 @@ public void unregisterLanguage(LanguageParameter lang) {
 
     @Override
     public void projectAdded(String name, ISourceLocation projectRoot) {
-        // TODO Auto-generated method stub
-        throw new UnsupportedOperationException("Unimplemented method 'projectAdded'");
+        callAccept(allRoutes(), IBaseTextDocumentService::projectAdded, name, projectRoot);
     }
 
     @Override
     public void projectRemoved(String name, ISourceLocation projectRoot) {
-        // TODO Auto-generated method stub
-        throw new UnsupportedOperationException("Unimplemented method 'projectRemoved'");
+        callAccept(allRoutes(), IBaseTextDocumentService::projectRemoved, name, projectRoot);
     }
 
     @Override
@@ -307,8 +313,8 @@ public void didDeleteFiles(DeleteFilesParams params) {
 
     @Override
     public void cancelProgress(String progressId) {
-        // TODO Auto-generated method stub
-        throw new UnsupportedOperationException("Unimplemented method 'cancelProgress'");
+        // Note: floating futures
+        callAccept(allRoutes(), IBaseTextDocumentService::cancelProgress, progressId);
     }
 
     @Override

rascal-lsp/src/main/java/org/rascalmpl/vscode/lsp/util/DocumentRouter.java

@@ -26,6 +26,7 @@
  */
 package org.rascalmpl.vscode.lsp.util;
 
+import java.util.stream.Stream;
 import org.eclipse.lsp4j.TextDocumentIdentifier;
 import org.eclipse.lsp4j.TextDocumentItem;
 import org.eclipse.lsp4j.VersionedTextDocumentIdentifier;
@@ -58,4 +59,6 @@ default T route(TextDocumentIdentifier id) {
         return route(Locations.toLoc(id.getUri()));
     }
 
+    Stream<T> allRoutes();
+
 }

rascal-lsp/src/main/java/org/rascalmpl/vscode/lsp/util/FutureCaller.java

@@ -29,6 +29,8 @@
 import java.util.concurrent.CompletableFuture;
 import java.util.function.BiConsumer;
 import java.util.function.BiFunction;
+import java.util.stream.Stream;
+import org.apache.commons.lang3.function.TriConsumer;
 
 /**
  * A helper interface to support calling functions on values in futures.
@@ -47,4 +49,16 @@ default <A, R> CompletableFuture<R> callApply(CompletableFuture<T> t, BiFunction
     default <A> CompletableFuture<Void> callAccept(CompletableFuture<T> t, BiConsumer<T, A> func, A arg) {
         return t.thenAccept(actualT -> func.accept(actualT, arg));
     }
+
+    default <A, B> CompletableFuture<Void> callAccept(CompletableFuture<T> t, TriConsumer<T, A, B> func, A arg1, B arg2) {
+        return t.thenAccept(actualT -> func.accept(actualT, arg1, arg2));
+    }
+
+    default <A> Stream<CompletableFuture<Void>> callAccept(Stream<CompletableFuture<T>> ts, BiConsumer<T, A> func, A arg) {
+        return ts.map(t -> callAccept(t, func, arg));
+    }
+
+    default <A, B> Stream<CompletableFuture<Void>> callAccept(Stream<CompletableFuture<T>> ts, TriConsumer<T, A, B> func, A arg1, B arg2) {
+        return ts.map(t -> callAccept(t, func, arg1, arg2));
+    }
 }