Expose sendRequestWithProgress as experimental API (#8074)

Author: genlu

src/activateRoslyn.ts

@@ -94,6 +94,8 @@ export function activateRoslyn(
         determineBrowserType: BlazorDebugConfigurationProvider.determineBrowserType,
         experimental: {
             sendServerRequest: async (t, p, ct) => await languageServerExport.sendRequest(t, p, ct),
+            sendServerRequestWithProgress: async (t, p, pr, ct) =>
+                await languageServerExport.sendRequestWithProgress(t, p, pr, ct),
             languageServerEvents: roslynLanguageServerEvents,
         },
         getComponentFolder: (componentName) => {

src/csharpExtensionExports.ts

@@ -8,7 +8,7 @@ import { Advisor } from './omnisharp/features/diagnosticsProvider';
 import { EventStream } from './eventStream';
 import TestManager from './omnisharp/features/dotnetTest';
 import { GlobalBrokeredServiceContainer } from '@microsoft/servicehub-framework';
-import { RequestType } from 'vscode-languageclient/node';
+import { PartialResultParams, ProtocolRequestType, RequestType } from 'vscode-languageclient/node';
 import { LanguageServerEvents } from './lsptoolshost/server/languageServerEvents';
 
 export interface OmnisharpExtensionExports {
@@ -34,5 +34,17 @@ export interface CSharpExtensionExperimentalExports {
         params: Params,
         token: vscode.CancellationToken
     ) => Promise<Response>;
+    sendServerRequestWithProgress<
+        Params extends PartialResultParams,
+        Response,
+        PartialResult,
+        Error,
+        RegistrationOptions
+    >(
+        type: ProtocolRequestType<Params, Response, PartialResult, Error, RegistrationOptions>,
+        params: Params,
+        onProgress: (p: PartialResult) => Promise<any>,
+        token?: vscode.CancellationToken
+    ): Promise<Response>;
     languageServerEvents: LanguageServerEvents;
 }

src/lsptoolshost/extensions/roslynLanguageServerExportChannel.ts

@@ -5,6 +5,7 @@
 import * as vscode from 'vscode';
 import { RequestType } from 'vscode-jsonrpc';
 import { RoslynLanguageServer } from '../server/roslynLanguageServer';
+import { PartialResultParams, ProtocolRequestType } from 'vscode-languageserver-protocol';
 
 export class RoslynLanguageServerExport {
     constructor(private _serverInitialized: Promise<RoslynLanguageServer>) {}
@@ -20,4 +21,25 @@ export class RoslynLanguageServerExport {
         const newType = new RequestType<Params, Response, Error>(type.method);
         return await server.sendRequest(newType, params, token);
     }
+
+    public async sendRequestWithProgress<
+        Params extends PartialResultParams,
+        Response,
+        PartialResult,
+        Error,
+        RegistrationOptions
+    >(
+        type: ProtocolRequestType<Params, Response, PartialResult, Error, RegistrationOptions>,
+        params: Params,
+        onProgress: (p: PartialResult) => Promise<any>,
+        token?: vscode.CancellationToken
+    ): Promise<Response> {
+        const server = await this._serverInitialized;
+        // We need to recreate the type parameter to ensure that the prototypes line up. The `ProtocolRequestType` we receive could have been
+        // from a different version.
+        const newType = new ProtocolRequestType<Params, Response, PartialResult, Error, RegistrationOptions>(
+            type.method
+        );
+        return await server.sendRequestWithProgress(newType, params, onProgress, token);
+    }
 }