Refactor CodeActions to fit the new model

Author: Kapil Borle

src/features/CodeActions.ts

@@ -1,25 +1,37 @@
 import vscode = require('vscode');
 import { LanguageClient, RequestType, NotificationType } from 'vscode-languageclient';
 import Window = vscode.window;
+import { IFeature } from '../feature';
 
-export function registerCodeActionCommands(client: LanguageClient): void {
-    vscode.commands.registerCommand('PowerShell.ApplyCodeActionEdits', (edit: any) => {
-        console.log("Applying edits");
-        console.log(edit);
-        var editor = Window.activeTextEditor;
-        var filePath = editor.document.fileName;
-        var workspaceEdit = new vscode.WorkspaceEdit();
-        workspaceEdit.set(
-            vscode.Uri.file(filePath),
-            [
-                new vscode.TextEdit(
-                    new vscode.Range(
-                        edit.StartLineNumber - 1,
-                        edit.StartColumnNumber - 1,
-                        edit.EndLineNumber - 1,
-                        edit.EndColumnNumber - 1),
-                    edit.Text)
-            ]);
-        vscode.workspace.applyEdit(workspaceEdit);
-    });
+export class CodeActionsFeature implements IFeature {
+    private command: vscode.Disposable;
+    private languageClient: LanguageClient;
+
+    constructor() {
+        this.command = vscode.commands.registerCommand('PowerShell.ApplyCodeActionEdits', (edit: any) => {
+            var editor = Window.activeTextEditor;
+            var filePath = editor.document.fileName;
+            var workspaceEdit = new vscode.WorkspaceEdit();
+            workspaceEdit.set(
+                vscode.Uri.file(filePath),
+                [
+                    new vscode.TextEdit(
+                        new vscode.Range(
+                            edit.StartLineNumber - 1,
+                            edit.StartColumnNumber - 1,
+                            edit.EndLineNumber - 1,
+                            edit.EndColumnNumber - 1),
+                        edit.Text)
+                ]);
+            vscode.workspace.applyEdit(workspaceEdit);
+        });
+    }
+
+    public setLanguageClient(languageclient: LanguageClient) {
+        this.languageClient = languageclient;
+    }
+
+    public dispose() {
+        this.command.dispose();
+    }
 }

src/main.ts

@@ -15,6 +15,7 @@ import { ExpandAliasFeature } from './features/ExpandAlias';
 import { ShowHelpFeature } from './features/ShowOnlineHelp';
 import { FindModuleFeature } from './features/PowerShellFindModule';
 import { ExtensionCommandsFeature } from './features/ExtensionCommands';
+import { CodeActionsFeature } from './features/CodeActions';
 
 // NOTE: We will need to find a better way to deal with the required
 //       PS Editor Services version...
@@ -74,7 +75,8 @@ export function activate(context: vscode.ExtensionContext): void {
         new ExpandAliasFeature(),
         new ShowHelpFeature(),
         new FindModuleFeature(),
-        new ExtensionCommandsFeature()
+        new ExtensionCommandsFeature(),
+        new CodeActionsFeature()
     ];
 
     sessionManager =