Code Actions - Add fix all support

l10n/bundle.l10n.json

@@ -127,6 +127,7 @@
   "Choose and set default": "Choose and set default",
   "Do not load any": "Do not load any",
   "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",
   "pipeArgs must be a string or a string array type": "pipeArgs must be a string or a string array type",
   "Name not defined in current configuration.": "Name not defined in current configuration.",
   "Configuration \"{0}\" in launch.json does not have a {1} argument with {2} for remote process listing.": "Configuration \"{0}\" in launch.json does not have a {1} argument with {2} for remote process listing.",

src/lsptoolshost/fixAllCodeAction.ts

@@ -0,0 +1,67 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 * as RoslynProtocol from './roslynProtocol';
+import { LSPAny } from 'vscode-languageserver-protocol';
+import { RoslynLanguageServer } from './roslynLanguageServer';
+import { URIConverter, createConverter } from 'vscode-languageclient/lib/common/protocolConverter';
+import { UriConverter } from './uriConverter';
+
+export function registerCodeActionFixAllCommands(
+    context: vscode.ExtensionContext,
+    languageServer: RoslynLanguageServer
+) {
+    context.subscriptions.push(
+        vscode.commands.registerCommand(
+            'roslyn.client.fixAllCodeAction',
+            async (request): Promise<void> => registerFixAllResolveCodeAction(languageServer, request)
+        )
+    );
+}
+
+async function registerFixAllResolveCodeAction(languageServer: RoslynLanguageServer, codeActionData: any) {
+    if (codeActionData) {
+        const data = <LSPAny>codeActionData;
+        const result = await vscode.window.showQuickPick(data.FixAllFlavors, {
+            placeHolder: vscode.l10n.t('Pick a fix all scope'),
+        });
+
+        let fixAllEdit = new vscode.WorkspaceEdit();
+        await vscode.window.withProgress(
+            {
+                location: vscode.ProgressLocation.Notification,
+                title: 'Fix All Code Action',
+                cancellable: true,
+            },
+            async (_, token) => {
+                if (result) {
+                    const fixAllCodeAction: RoslynProtocol.RoslynFixAllCodeAction = {
+                        title: data.UniqueIdentifier,
+                        data: data,
+                        scope: result,
+                    };
+
+                    const response = await languageServer.sendRequest(
+                        RoslynProtocol.CodeActionFixAllResolveRequest.type,
+                        fixAllCodeAction,
+                        token
+                    );
+
+                    if (response.edit) {
+                        const uriConverter: URIConverter = (value: string): vscode.Uri =>
+                            UriConverter.deserialize(value);
+                        const protocolConverter = createConverter(uriConverter, true, true);
+                        fixAllEdit = await protocolConverter.asWorkspaceEdit(response.edit);
+                    }
+                }
+            }
+        );
+
+        if (!(await vscode.workspace.applyEdit(fixAllEdit))) {
+            throw new Error('Tried to insert multiple code action edits, but an error occurred.');
+        }
+    }
+}

src/lsptoolshost/roslynLanguageServer.ts

@@ -49,6 +49,7 @@ import { RoslynLanguageServerEvents } from './languageServerEvents';
 import { registerShowToastNotification } from './showToastNotification';
 import { registerRazorCommands } from './razorCommands';
 import { registerOnAutoInsert } from './onAutoInsert';
+import { registerCodeActionFixAllCommands } from './fixAllCodeAction';
 
 let _channel: vscode.OutputChannel;
 let _traceChannel: vscode.OutputChannel;
@@ -202,6 +203,46 @@ export class RoslynLanguageServer {
                 workspace: {
                     configuration: (params) => readConfigurations(params),
                 },
+                // resolveCodeAction: async (codeAction, token, next) => {
+                //     const lspCodeAction = <CodeAction>codeAction;
+                //     const data = lspCodeAction.data;
+                //     if (data.FixAllFlavors) {
+                //         const result = await vscode.window.showQuickPick(data.FixAllFlavors, {
+                //             placeHolder: 'Pick a fix all scope',
+                //         });
+
+                //         if (result) {
+                //             const fixAllCodeAction: RoslynProtocol.RoslynFixAllCodeAction = {
+                //                 title: lspCodeAction.title,
+                //                 edit: lspCodeAction.edit,
+                //                 data: data,
+                //                 scope: result,
+                //             };
+
+                //             const response = await _languageServer.sendRequest(
+                //                 RoslynProtocol.CodeActionFixAllResolveRequest.type,
+                //                 fixAllCodeAction,
+                //                 token
+                //             );
+
+                //             if (response.edit) {
+                //                 const uriConverter: URIConverter = (value: string): vscode.Uri =>
+                //                     UriConverter.deserialize(value);
+                //                 const protocolConverter = createConverter(uriConverter, true, true);
+                //                 const result = await protocolConverter.asWorkspaceEdit(response.edit);
+
+                //                 if (!(await vscode.workspace.applyEdit(result))) {
+                //                     throw new Error(
+                //                         'Tried to insert multiple code action edits, but an error occurred.'
+                //                     );
+                //                 }
+                //             }
+                //             console.log(response);
+                //         }
+                //     } else {
+                //         return await next(codeAction, token);
+                //     }
+                // },
             },
         };
 
@@ -745,7 +786,7 @@ export async function activateRoslynLanguageServer(
 
     // Register any commands that need to be handled by the extension.
     registerCommands(context, languageServer, optionProvider, hostExecutableResolver, _channel);
-
+    registerCodeActionFixAllCommands(context, languageServer);
     registerRazorCommands(context, languageServer);
 
     registerUnitTestingCommands(context, languageServer, dotnetTestChannel);

src/lsptoolshost/roslynProtocol.ts

@@ -5,6 +5,7 @@
 
 import { Command } from 'vscode';
 import * as lsp from 'vscode-languageserver-protocol';
+import { CodeAction } from 'vscode-languageserver-protocol';
 import { ProjectConfigurationMessage } from '../shared/projectConfiguration';
 
 export interface WorkspaceDebugConfigurationParams {
@@ -136,6 +137,10 @@ export interface BuildOnlyDiagnosticIdsResult {
     ids: string[];
 }
 
+export interface RoslynFixAllCodeAction extends CodeAction {
+    scope: string;
+}
+
 export namespace WorkspaceDebugConfigurationRequest {
     export const method = 'workspace/debugConfiguration';
     export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
@@ -209,3 +214,9 @@ export namespace BuildOnlyDiagnosticIdsRequest {
     export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
     export const type = new lsp.RequestType0<BuildOnlyDiagnosticIdsResult, void>(method);
 }
+
+export namespace CodeActionFixAllResolveRequest {
+    export const method = 'codeAction/resolveFixAll';
+    export const messageDirection: lsp.MessageDirection = lsp.MessageDirection.clientToServer;
+    export const type = new lsp.RequestType<RoslynFixAllCodeAction, RoslynFixAllCodeAction, void>(method);
+}