Support "Show Razor" commands in cohosting

Author: davidwengier

package.json

@@ -1896,13 +1896,13 @@
         "command": "extension.showRazorCSharpWindow",
         "title": "%command.extension.showRazorCSharpWindow%",
         "category": "Razor",
-        "enablement": "isWorkspaceTrusted && razor.mode == 'lsp'"
+        "enablement": "isWorkspaceTrusted"
       },
       {
         "command": "extension.showRazorHtmlWindow",
         "title": "%command.extension.showRazorHtmlWindow%",
         "category": "Razor",
-        "enablement": "isWorkspaceTrusted && razor.mode == 'lsp'"
+        "enablement": "isWorkspaceTrusted"
       },
       {
         "command": "razor.reportIssue",
@@ -5631,4 +5631,4 @@
       }
     }
   }
-}
+}

src/lsptoolshost/razor/documentContentsRequest.ts

@@ -0,0 +1,11 @@
+/*---------------------------------------------------------------------------------------------
+ *  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-types';
+import { GeneratedDocumentKind } from './generatedDocumentKind';
+
+export class DocumentContentsRequest {
+    constructor(public readonly textDocument: TextDocumentIdentifier, public readonly kind: GeneratedDocumentKind) {}
+}

src/lsptoolshost/razor/generatedDocumentKind.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.
+ *--------------------------------------------------------------------------------------------*/
+
+export enum GeneratedDocumentKind {
+    CSharp,
+    Html,
+    Formatting,
+}

src/lsptoolshost/razor/razorEndpoints.ts

@@ -53,6 +53,7 @@ import { ReportIssueCommand } from '../../razor/src/diagnostics/reportIssueComma
 import { HtmlDocument } from './htmlDocument';
 import { HtmlForwardedRequest } from './htmlForwardedRequest';
 import { BlazorDebugConfigurationProvider } from '../../razor/src/blazorDebug/blazorDebugConfigurationProvider';
+import { ShowGeneratedDocumentCommand } from './showGeneratedDocumentCommand';
 
 export function registerRazorEndpoints(
     context: vscode.ExtensionContext,
@@ -70,6 +71,7 @@ export function registerRazorEndpoints(
         registerCohostingEndpoints();
 
         context.subscriptions.push(BlazorDebugConfigurationProvider.register(razorLogger, vscode));
+        context.subscriptions.push(ShowGeneratedDocumentCommand.register(roslynLanguageServer));
     } else {
         vscode.commands.executeCommand('setContext', 'razor.mode', 'lsp');
         registerNonCohostingEndpoints();
@@ -82,7 +84,13 @@ export function registerRazorEndpoints(
     //
     function registerCohostingEndpoints() {
         const documentManager = new HtmlDocumentManager(platformInfo, roslynLanguageServer, razorLogger);
-        const reportIssueCommand = new ReportIssueCommand(vscode, undefined, documentManager, razorLogger);
+        const reportIssueCommand = new ReportIssueCommand(
+            vscode,
+            undefined,
+            documentManager,
+            roslynLanguageServer,
+            razorLogger
+        );
         context.subscriptions.push(documentManager.register());
         context.subscriptions.push(reportIssueCommand.register());
 

src/lsptoolshost/razor/showGeneratedDocumentCommand.ts

@@ -0,0 +1,91 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import { RazorLanguage } from '../../razor/src/razorLanguage';
+import { getUriPath } from '../../razor/src/uriPaths';
+import { RoslynLanguageServer } from '../server/roslynLanguageServer';
+import { DocumentContentsRequest } from './documentContentsRequest';
+import { CancellationToken, RequestType, TextDocumentIdentifier } from 'vscode-languageserver-protocol';
+import { UriConverter } from '../utils/uriConverter';
+import { GeneratedDocumentKind } from './generatedDocumentKind';
+
+export class ShowGeneratedDocumentCommand {
+    private static requestType = new RequestType<DocumentContentsRequest, string, void>(
+        'razor/generatedDocumentContents'
+    );
+
+    public static register(roslynLanguageServer: RoslynLanguageServer) {
+        return vscode.Disposable.from(
+            vscode.commands.registerCommand('extension.showRazorCSharpWindow', async () =>
+                this.show(roslynLanguageServer, GeneratedDocumentKind.CSharp, '.g.cs')
+            ),
+            vscode.commands.registerCommand('extension.showRazorHtmlWindow', async () =>
+                this.show(roslynLanguageServer, GeneratedDocumentKind.Html, '.g.html')
+            )
+        );
+    }
+
+    private static async show(
+        roslynLanguageServer: RoslynLanguageServer,
+        kind: GeneratedDocumentKind,
+        extension: string
+    ) {
+        const uri = await this.getActiveDocumentUri();
+        if (!uri) {
+            return;
+        }
+
+        const title = `${getUriPath(uri)}${extension}`;
+        const panel = vscode.window.createWebviewPanel('razorGeneratedDocument', title, vscode.ViewColumn.Two, {
+            enableScripts: false,
+            // Disallow any remote sources
+            localResourceRoots: [],
+        });
+
+        const content = await this.getGeneratedDocumentContent(uri, kind, roslynLanguageServer);
+
+        panel.webview.html = await this.getWebViewContent(content, title);
+    }
+
+    private static async getActiveDocumentUri() {
+        if (!vscode.window.activeTextEditor) {
+            return null;
+        }
+
+        if (vscode.window.activeTextEditor.document.languageId !== RazorLanguage.id) {
+            return null;
+        }
+
+        return vscode.window.activeTextEditor.document.uri;
+    }
+
+    public static async getGeneratedDocumentContent(
+        uri: vscode.Uri,
+        kind: GeneratedDocumentKind,
+        roslynLanguageServer: RoslynLanguageServer
+    ) {
+        return roslynLanguageServer.sendRequest(
+            ShowGeneratedDocumentCommand.requestType,
+            new DocumentContentsRequest(TextDocumentIdentifier.create(UriConverter.serialize(uri)), kind),
+            CancellationToken.None
+        );
+    }
+
+    private static async getWebViewContent(content: string, title: string) {
+        return `<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta http-equiv="Content-Security-Policy">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>${title}</title>
+</head>
+<body>
+    <pre>${content.replace(/</g, '&lt;').replace(/</g, '&gt;')}</pre>
+</body>
+</html>`;
+    }
+}