Provide CodeActionKind for code actions

VSCode side of OmniSharp/omnisharp-roslyn#2430. Fixes dotnet#5335.
333fred · Aug 11, 2022 · 64ba64d · 64ba64d
1 parent a675b5f
commit 64ba64d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/src/features/codeActionProvider.ts b/src/features/codeActionProvider.ts
@@ -62,6 +62,7 @@ export default class CodeActionProvider extends AbstractProvider implements vsco
 
                 return {
                     title: codeAction.Name,
+                    kind: this.mapOmniSharpCodeActionKindToVSCodeCodeActionKind(codeAction.CodeActionKind),
                     command: {
                         title: codeAction.Name,
                         command: this._commandId,
@@ -74,6 +75,21 @@ export default class CodeActionProvider extends AbstractProvider implements vsco
         }
     }
 
+    private mapOmniSharpCodeActionKindToVSCodeCodeActionKind(kind: string): vscode.CodeActionKind {
+        switch (kind) {
+            case 'QuickFix':
+                return vscode.CodeActionKind.QuickFix;
+            case 'Refactor':
+                return vscode.CodeActionKind.Refactor;
+            case 'RefactorExtract':
+                return vscode.CodeActionKind.RefactorExtract;
+            case 'RefactorInline':
+                return vscode.CodeActionKind.RefactorInline;
+            default:
+                return vscode.CodeActionKind.Empty;
+        }
+    }
+
     private async _runCodeAction(req: protocol.V2.RunCodeActionRequest, token: vscode.CancellationToken): Promise<boolean | string | {}> {
         try {
             const response = await serverUtils.runCodeAction(this._server, req);

diff --git a/src/omnisharp/protocol.ts b/src/omnisharp/protocol.ts
@@ -682,6 +682,7 @@ export namespace V2 {
     export interface OmniSharpCodeAction {
         Identifier: string;
         Name: string;
+        CodeActionKind?: string;
     }
 
     export interface GetCodeActionsResponse {