From 8e3720e227900ee53465aa33e1abb626a57ac83d Mon Sep 17 00:00:00 2001
From: Dirk Baeumer <dirkb@microsoft.com>
Date: Fri, 3 Nov 2017 08:58:15 +0100
Subject: [PATCH] cpp multi server sample

---
 .../client/src/extension.ts                   | 69 +++++++++++++++----
 lsp-multi-server-sample/server/src/server.ts  | 10 ++-
 2 files changed, 64 insertions(+), 15 deletions(-)

diff --git a/lsp-multi-server-sample/client/src/extension.ts b/lsp-multi-server-sample/client/src/extension.ts
index 1975c4a7..7c3586a2 100644
--- a/lsp-multi-server-sample/client/src/extension.ts
+++ b/lsp-multi-server-sample/client/src/extension.ts
@@ -55,34 +55,61 @@ export function activate(context: ExtensionContext) {
 		}
 
 		let uri = document.uri;
-		// Untitled files go to a default client.
-		if (uri.scheme === 'untitled' && !defaultClient) {
+		let folder = Workspace.getWorkspaceFolder(uri);
+		// Files outside a folder go to the default client
+		if (!folder && !defaultClient) {
 			let debugOptions = { execArgv: ["--nolazy", "--inspect=6010"] };
 			let serverOptions = {
 				run: { module, transport: TransportKind.ipc },
 				debug: { module, transport: TransportKind.ipc, options: debugOptions}
 			};
+			let syncedDocuments: Set<string> = new Set(); 
 			let clientOptions: LanguageClientOptions = {
 				documentSelector: [
-					{ scheme: 'untitled', language: 'plaintext' }
+					{ scheme: 'untitled', language: 'plaintext' },
+					{ scheme: 'file', language: 'plaintext' }
 				],
 				diagnosticCollectionName: 'multi-lsp',
-				outputChannel: outputChannel
+				outputChannel: outputChannel,
+				initializationOptions: {
+					genericServer: true
+				},
+				middleware: {
+					didOpen: (document, next) => {
+						let uri: Uri = document.uri;
+						// The file is part of a know workspace folder.
+						if (Workspace.getWorkspaceFolder(uri)) {
+							return;
+						}
+						syncedDocuments.add(uri.toString());
+						next(document);
+					},
+					didChange: (event, next) => {
+						if (syncedDocuments.has(event.document.uri.toString())) {
+							next(event);
+						}
+					},
+					didSave: (document, next) => {
+						if (syncedDocuments.has(document.uri.toString())) {
+							next(document);
+						}
+					},
+					didClose: (document, next) => {
+						if (syncedDocuments.has(document.uri.toString())) {
+							syncedDocuments.delete(document.uri.toString());
+							next(document);
+						}
+					}
+				}
 			}
 			defaultClient = new LanguageClient('lsp-multi-server-example', 'LSP Multi Server Example', serverOptions, clientOptions);
 			defaultClient.registerProposedFeatures();
 			defaultClient.start();
 			return;
 		}
-		let folder = Workspace.getWorkspaceFolder(uri);
-		// Files outside a folder can't be handled. This might depend on the language.
-		// Single file languages like JSON might handle files outside the workspace folders.
-		if (!folder) {
-			return;
-		}
+
 		// If we have nested workspace folders we only start a server on the outer most workspace folder.
 		folder = getOuterMostWorkspaceFolder(folder);
-		
 		if (!clients.has(folder.uri.toString())) {
 			let debugOptions = { execArgv: ["--nolazy", `--inspect=${6011 + clients.size}`] };
 			let serverOptions = {
@@ -95,7 +122,25 @@ export function activate(context: ExtensionContext) {
 				],
 				diagnosticCollectionName: 'lsp-multi-server-example',
 				workspaceFolder: folder,
-				outputChannel: outputChannel
+				outputChannel: outputChannel,
+				middleware: {
+					didOpen: (document, next) => {
+						console.log(`didOpen middleware called for server serving: ${folder.uri.toString()}`);
+						next(document);
+					},
+					didChange: (event, next) => {
+						console.log(`didChange middleware called for server serving: ${folder.uri.toString()}`);
+						next(event);
+					},
+					didSave: (document, next) => {
+						console.log(`didSave middleware called for server serving: ${folder.uri.toString()}`);
+						next(document);
+					},
+					didClose: (document, next) => {
+						console.log(`didClose middleware called for server serving: ${folder.uri.toString()}`);
+						next(document);
+					}
+				}
 			}
 			let client = new LanguageClient('lsp-multi-server-example', 'LSP Multi Server Example', serverOptions, clientOptions);
 			client.registerProposedFeatures();
diff --git a/lsp-multi-server-sample/server/src/server.ts b/lsp-multi-server-sample/server/src/server.ts
index 0f5cd0a5..40cdb373 100644
--- a/lsp-multi-server-sample/server/src/server.ts
+++ b/lsp-multi-server-sample/server/src/server.ts
@@ -17,13 +17,17 @@ let documents = new TextDocuments();
 let workspaceFolder: string;
 
 documents.onDidOpen((event) => {
-	connection.console.log(`[Server(${process.pid}) ${workspaceFolder}] Document opened: ${event.document.uri}`);
+	connection.console.log(`[Server(${process.pid}) ${workspaceFolder ? workspaceFolder : 'generic'}] Document opened: ${event.document.uri}`);
 })
 documents.listen(connection);
 
 connection.onInitialize((params) => {
-	workspaceFolder = params.rootUri;
-	connection.console.log(`[Server(${process.pid}) ${workspaceFolder}] Started and initialize received`);
+	if (params.initializationOptions && params.initializationOptions.genericServer) {
+		connection.console.log(`[Server(${process.pid}) generic] Started and initialize received`);
+	} else {
+		workspaceFolder = params.rootUri;
+		connection.console.log(`[Server(${process.pid}) ${workspaceFolder}] Started and initialize received`);
+	}
 	return {
 		capabilities: {
 			textDocumentSync: {