use edit

Author: TylerLeonhardt

src/PowerShellEditorServices/Services/TextDocument/Handlers/CodeActionHandler.cs

@@ -91,12 +91,18 @@ public async Task<CommandOrCodeActionContainer> Handle(CodeActionParams request,
                     {
                         Title = correction.Name,
                         Kind = CodeActionKind.QuickFix,
-                        Diagnostics = new Container<Diagnostic>(diagnostic),
-                        Command = new Command
+                        Edit = new WorkspaceEdit
                         {
-                            Title = correction.Name,
-                            Name = "PowerShell.ApplyCodeActionEdits",
-                            Arguments = JArray.FromObject(correction.Edits)
+                            DocumentChanges = new Container<WorkspaceEditDocumentChange>(
+                                new WorkspaceEditDocumentChange(
+                                    new TextDocumentEdit()
+                                    {
+                                        TextDocument = new VersionedTextDocumentIdentifier()
+                                        {
+                                            Uri = request.TextDocument.Uri
+                                        },
+                                        Edits = new Container<TextEdit>(correction.Edits.Select(ScriptRegion.ToTextEdit))
+                                    }))
                         }
                     });
                 }

src/PowerShellEditorServices/Services/TextDocument/ScriptRegion.cs

@@ -5,6 +5,7 @@
 
 using System;
 using System.Management.Automation.Language;
+using OmniSharp.Extensions.LanguageServer.Protocol.Models;
 
 namespace Microsoft.PowerShell.EditorServices.Services.TextDocument
 {
@@ -49,6 +50,27 @@ public static ScriptRegion Create(IScriptExtent scriptExtent)
                 scriptExtent.EndOffset);
         }
 
+        internal static TextEdit ToTextEdit(ScriptRegion scriptRegion)
+        {
+            return new TextEdit
+            {
+                NewText = scriptRegion.Text,
+                Range = new Range
+                {
+                    Start = new Position
+                    {
+                        Line = scriptRegion.StartLineNumber - 1,
+                        Character = scriptRegion.StartColumnNumber - 1,
+                    },
+                    End = new Position
+                    {
+                        Line = scriptRegion.EndLineNumber - 1,
+                        Character = scriptRegion.EndColumnNumber - 1,
+                    }
+                }
+            };
+        }
+
         #endregion
 
         #region Constructors