Add endpoint for updating Html documents

Author: davidwengier

src/lsptoolshost/activate.ts

@@ -86,7 +86,7 @@ export async function activateRoslynLanguageServer(
     registerCodeActionFixAllCommands(context, languageServer, _channel);
 
     registerRazorCommands(context, languageServer);
-    registerRazorEndpoints(context, languageServer, razorLogger);
+    registerRazorEndpoints(context, languageServer, razorLogger, platformInfo);
 
     registerUnitTestingCommands(context, languageServer);
 

src/lsptoolshost/razor/htmlUpdateParameters.ts

@@ -0,0 +1,10 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { TextDocumentIdentifier } from 'vscode-languageserver-protocol';
+
+export class HtmlUpdateParameters {
+    constructor(public readonly textDocument: TextDocumentIdentifier, public readonly text: string) {}
+}

src/lsptoolshost/razor/razorEndpoints.ts

@@ -5,16 +5,47 @@
 
 import { RoslynLanguageServer } from '../server/roslynLanguageServer';
 import * as vscode from 'vscode';
-import { LogMessageParams, NotificationType } from 'vscode-languageclient';
+import { LogMessageParams, NotificationType, RequestType } from 'vscode-languageclient';
 import { RazorLogger } from '../../razor/src/razorLogger';
+import { HtmlUpdateParameters } from './htmlUpdateParameters';
+import { UriConverter } from '../utils/uriConverter';
+import { PlatformInformation } from '../../shared/platform';
+import { HtmlDocumentManager } from './htmlDocumentManager';
+import { razorOptions } from '../../shared/options';
 
 export function registerRazorEndpoints(
     context: vscode.ExtensionContext,
     languageServer: RoslynLanguageServer,
-    razorLogger: RazorLogger
+    razorLogger: RazorLogger,
+    platformInfo: PlatformInformation
 ) {
     const logNotificationType = new NotificationType<LogMessageParams>('razor/log');
     languageServer.registerOnNotificationWithParams(logNotificationType, (params) =>
         razorLogger.log(params.message, params.type)
     );
+
+    if (!razorOptions.cohostingEnabled) {
+        return;
+    }
+
+    const documentManager = new HtmlDocumentManager(platformInfo, razorLogger);
+    context.subscriptions.push(documentManager.register());
+
+    registerRequestHandler<HtmlUpdateParameters, void>('razor/updateHtml', async (params) => {
+        const uri = UriConverter.deserialize(params.textDocument.uri);
+        await documentManager.updateDocumentText(uri, params.text);
+    });
+
+    // Helper method that registers a request handler, and logs errors to the Razor logger.
+    function registerRequestHandler<Params, Result>(method: string, invocation: (params: Params) => Promise<Result>) {
+        const requestType = new RequestType<Params, Result, Error>(method);
+        languageServer.registerOnRequest(requestType, async (params) => {
+            try {
+                return await invocation(params);
+            } catch (error) {
+                razorLogger.logError(`Error: ${error}`, error);
+                return undefined;
+            }
+        });
+    }
 }