Add support for nuget restore commands

CHANGELOG.md

@@ -5,6 +5,11 @@
 - [O# Parity] Nuget restore [#5725](https://github.com/dotnet/vscode-csharp/issues/5725)
 - Debug from .csproj and .sln [#5876](https://github.com/dotnet/vscode-csharp/issues/5876)
 
+## Next (Replace once current release version is generated)
+* Update Roslyn to 4.9.0-1.23530.4 (PR: [#6603](https://github.com/dotnet/vscode-csharp/pull/6603))
+  * Enable NuGet restore commands `dotnet.restore.all` and `dotnet.restore.project` (PR: [#70588](https://github.com/dotnet/roslyn/pull/70588))
+  * Fix issue where server did not reload projects after NuGet restore (PR: [#70602](https://github.com/dotnet/roslyn/pull/70602))
+
 ## Latest
 * Update Roslyn to 4.9.0-1.23525.5 (PR: [#6595](https://github.com/dotnet/vscode-csharp/pull/6595))
   * Fix some project loading issues caused by evaluation failures (PR: [#70496](https://github.com/dotnet/roslyn/pull/70496))

l10n/bundle.l10n.json

@@ -127,6 +127,9 @@
   "Choose": "Choose",
   "Choose and set default": "Choose and set default",
   "Do not load any": "Do not load any",
+  "Restore already in progress": "Restore already in progress",
+  "Restore": "Restore",
+  "Sending request": "Sending request",
   "C# configuration has changed. Would you like to reload the window to apply your changes?": "C# configuration has changed. Would you like to reload the window to apply your changes?",
   "Pick a fix all scope": "Pick a fix all scope",
   "Fix All Code Action": "Fix All Code Action",

package.json

@@ -37,7 +37,7 @@
     }
   },
   "defaults": {
-    "roslyn": "4.9.0-1.23526.14",
+    "roslyn": "4.9.0-1.23530.4",
     "omniSharp": "1.39.10",
     "razor": "7.0.0-preview.23528.1",
     "razorOmnisharp": "7.0.0-preview.23363.1",
@@ -2037,13 +2037,13 @@
         "command": "dotnet.restore.project",
         "title": "%command.dotnet.restore.project%",
         "category": ".NET",
-        "enablement": "config.dotnet.server.useOmnisharp"
+        "enablement": "dotnet.server.activatedStandalone"
       },
       {
         "command": "dotnet.restore.all",
         "title": "%command.dotnet.restore.all%",
         "category": ".NET",
-        "enablement": "config.dotnet.server.useOmnisharp"
+        "enablement": "dotnet.server.activatedStandalone"
       },
       {
         "command": "csharp.downloadDebugger",

src/lsptoolshost/restore.ts

@@ -0,0 +1,108 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { RoslynLanguageServer } from './roslynLanguageServer';
+import { RestorableProjects, RestoreParams, RestorePartialResult, RestoreRequest } from './roslynProtocol';
+import path = require('path');
+
+let _restoreInProgress = false;
+
+export function registerRestoreCommands(
+    context: vscode.ExtensionContext,
+    languageServer: RoslynLanguageServer,
+    restoreChannel: vscode.OutputChannel
+) {
+    context.subscriptions.push(
+        vscode.commands.registerCommand('dotnet.restore.project', async (_request): Promise<void> => {
+            return chooseProjectToRestore(languageServer, restoreChannel);
+        })
+    );
+    context.subscriptions.push(
+        vscode.commands.registerCommand('dotnet.restore.all', async (): Promise<void> => {
+            return restore(languageServer, restoreChannel);
+        })
+    );
+}
+async function chooseProjectToRestore(
+    languageServer: RoslynLanguageServer,
+    restoreChannel: vscode.OutputChannel
+): Promise<void> {
+    const projects = await languageServer.sendRequest0(
+        RestorableProjects.type,
+        new vscode.CancellationTokenSource().token
+    );
+
+    const items = projects.map((p) => {
+        const projectName = path.basename(p);
+        const item: vscode.QuickPickItem = {
+            label: vscode.l10n.t(`Restore {0}`, projectName),
+            description: p,
+        };
+        return item;
+    });
+
+    const pickedItem = await vscode.window.showQuickPick(items);
+    if (!pickedItem) {
+        return;
+    }
+
+    await restore(languageServer, restoreChannel, pickedItem.description);
+}
+
+async function restore(
+    languageServer: RoslynLanguageServer,
+    restoreChannel: vscode.OutputChannel,
+    projectFile?: string
+): Promise<void> {
+    if (_restoreInProgress) {
+        vscode.window.showErrorMessage(vscode.l10n.t('Restore already in progress'));
+        return;
+    }
+    _restoreInProgress = true;
+    restoreChannel.show(true);
+
+    const request: RestoreParams = { projectFilePath: projectFile };
+    await vscode.window
+        .withProgress(
+            {
+                location: vscode.ProgressLocation.Notification,
+                title: vscode.l10n.t('Restore'),
+                cancellable: true,
+            },
+            async (progress, token) => {
+                const writeOutput = (output: RestorePartialResult) => {
+                    if (output.message) {
+                        restoreChannel.appendLine(output.message);
+                    }
+
+                    progress.report({ message: output.stage });
+                };
+
+                progress.report({ message: vscode.l10n.t('Sending request') });
+                const responsePromise = languageServer.sendRequestWithProgress(
+                    RestoreRequest.type,
+                    request,
+                    async (p) => writeOutput(p),
+                    token
+                );
+
+                await responsePromise.then(
+                    (result) => result.forEach((r) => writeOutput(r)),
+                    (err) => restoreChannel.appendLine(err)
+                );
+            }
+        )
+        .then(
+            () => {
+                _restoreInProgress = false;
+            },
+            () => {
+                _restoreInProgress = false;
+            }
+        );
+
+    return;
+}

src/lsptoolshost/roslynLanguageServer.ts

@@ -56,6 +56,7 @@ import { registerCodeActionFixAllCommands } from './fixAllCodeAction';
 import { commonOptions, languageServerOptions, omnisharpOptions } from '../shared/options';
 import { NamedPipeInformation } from './roslynProtocol';
 import { IDisposable } from '../disposable';
+import { registerRestoreCommands } from './restore';
 
 let _channel: vscode.OutputChannel;
 let _traceChannel: vscode.OutputChannel;
@@ -835,6 +836,7 @@ export async function activateRoslynLanguageServer(
     optionObservable: Observable<void>,
     outputChannel: vscode.OutputChannel,
     dotnetTestChannel: vscode.OutputChannel,
+    dotnetChannel: vscode.OutputChannel,
     reporter: TelemetryReporter,
     languageServerEvents: RoslynLanguageServerEvents
 ): Promise<RoslynLanguageServer> {
@@ -872,6 +874,8 @@ export async function activateRoslynLanguageServer(
     // Register any needed debugger components that need to communicate with the language server.
     registerDebugger(context, languageServer, languageServerEvents, platformInfo, _channel);
 
+    registerRestoreCommands(context, languageServer, dotnetChannel);
+
     registerOnAutoInsert(languageServer);
 
     context.subscriptions.push(registerLanguageServerOptionChanges(optionObservable));

src/lsptoolshost/roslynProtocol.ts

@@ -150,6 +150,19 @@ export interface NamedPipeInformation {
     pipeName: string;
 }
 
+export interface RestoreParams extends lsp.WorkDoneProgressParams, lsp.PartialResultParams {
+    /**
+     * An optional file path to restore.
+     * If none is specified, the solution (or all loaded projects) are restored.
+     */
+    projectFilePath?: string;
+}
+
+export interface RestorePartialResult {
+    stage: string;
+    message: string;
+}
+
 export namespace WorkspaceDebugConfigurationRequest {
     export const method = 'workspace/debugConfiguration';
     export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
@@ -229,3 +242,21 @@ export namespace CodeActionFixAllResolveRequest {
     export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
     export const type = new lsp.RequestType<RoslynFixAllCodeAction, RoslynFixAllCodeAction, void>(method);
 }
+
+export namespace RestoreRequest {
+    export const method = 'workspace/_roslyn_restore';
+    export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
+    export const type = new lsp.ProtocolRequestType<
+        RestoreParams,
+        RestorePartialResult[],
+        RestorePartialResult,
+        void,
+        void
+    >(method);
+}
+
+export namespace RestorableProjects {
+    export const method = 'workspace/_roslyn_restorableProjects';
+    export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
+    export const type = new lsp.RequestType0<string[], void>(method);
+}

src/main.ts

@@ -82,6 +82,7 @@ export async function activate(
 
     const csharpChannel = vscode.window.createOutputChannel('C#');
     const dotnetTestChannel = vscode.window.createOutputChannel('.NET Test Log');
+    const dotnetChannel = vscode.window.createOutputChannel('.NET');
     const csharpchannelObserver = new CsharpChannelObserver(csharpChannel);
     const csharpLogObserver = new CsharpLoggerObserver(csharpChannel);
     eventStream.subscribe(csharpchannelObserver.post);
@@ -175,11 +176,11 @@ export async function activate(
             optionStream,
             csharpChannel,
             dotnetTestChannel,
+            dotnetChannel,
             reporter,
             roslynLanguageServerEvents
         );
     } else {
-        const dotnetChannel = vscode.window.createOutputChannel('.NET');
         const dotnetChannelObserver = new DotNetChannelObserver(dotnetChannel);
         const dotnetLoggerObserver = new DotnetLoggerObserver(dotnetChannel);
         eventStream.subscribe(dotnetChannelObserver.post);