Merge pull request #74 from PowerShell/spike-dcf-psgallery

Add support for finding and installing modules from the PowerShell Gallery

Author: daviwil

package.json

@@ -63,7 +63,13 @@
                 "command": "PowerShell.OpenInISE",
                 "key": "ctrl+alt+i",
                 "when": "editorTextFocus && editorLangId == 'powershell'"
+            },
+            {
+                "command": "PowerShell.PowerShellFindModule",
+                "key": "ctrl+K ctrl+f",
+                "when": "editorTextFocus && editorLangId == 'powershell'"
             }
+
         ],
         "commands": [
             {
@@ -85,6 +91,11 @@
                 "command": "PowerShell.OpenInISE",
                 "title": "Open current file in PowerShell ISE",
                 "category": "PowerShell"
+            },
+            {
+                "command": "PowerShell.PowerShellFindModule",
+                "title": "Find/Install PowerShell modules from the gallery",
+                "category": "PowerShell"
             }
         ],
         "snippets": [

src/features/PowerShellFindModule.ts

@@ -0,0 +1,51 @@
+import vscode = require('vscode');
+import { LanguageClient, RequestType, NotificationType } from 'vscode-languageclient';
+import Window = vscode.window;
+import QuickPickItem = vscode.QuickPickItem;
+
+export namespace FindModuleRequest {
+    export const type: RequestType<any, any, void> = { get method() { return 'powerShell/findModule'; } };
+}
+
+export namespace InstallModuleRequest {
+    export const type: RequestType<string, void, void> = { get method() { return 'powerShell/installModule'; } };
+}
+
+function GetCurrentTime() {
+
+    var timeNow = new Date();
+    var hours   = timeNow.getHours();
+    var minutes = timeNow.getMinutes();
+    var seconds = timeNow.getSeconds();
+
+    var timeString = "" + ((hours > 12) ? hours - 12 : hours);
+    timeString  += ((minutes < 10) ? ":0" : ":") + minutes;
+    timeString  += ((seconds < 10) ? ":0" : ":") + seconds;
+    timeString  += (hours >= 12) ? " PM" : " AM";
+
+    return timeString;
+}
+
+export function registerPowerShellFindModuleCommand(client: LanguageClient): void {
+    var disposable = vscode.commands.registerCommand('PowerShell.PowerShellFindModule', () => {
+        var items: QuickPickItem[] = [];
+
+        vscode.window.setStatusBarMessage(GetCurrentTime() + " Initializing...");
+        client.sendRequest(FindModuleRequest.type, null).then((modules) => {
+            for(var item in modules) {
+                items.push({ label: modules[item].name, description: modules[item].description });
+            };
+
+            vscode.window.setStatusBarMessage("");
+            Window.showQuickPick(items,{placeHolder: "Results: (" + modules.length + ")"}).then((selection) => {
+            	if (!selection) { return; }
+            	switch (selection.label) {
+            	    default :
+            	        var moduleName = selection.label;
+            	        //vscode.window.setStatusBarMessage("Installing PowerShell Module " + moduleName, 1500);
+            	        client.sendRequest(InstallModuleRequest.type, moduleName);
+            	    }
+            	});
+            });
+    });
+}

src/main.ts

@@ -12,6 +12,7 @@ import { LanguageClient, LanguageClientOptions, Executable, RequestType, Notific
 import { registerExpandAliasCommand } from './features/ExpandAlias';
 import { registerShowHelpCommand } from './features/ShowOnlineHelp';
 import { registerOpenInISECommand } from './features/OpenInISE';
+import { registerPowerShellFindModuleCommand } from './features/PowerShellFindModule';
 import { registerConsoleCommands } from './features/Console';
 
 var languageServerClient: LanguageClient = undefined;
@@ -101,6 +102,7 @@ export function activate(context: vscode.ExtensionContext): void {
     registerShowHelpCommand(languageServerClient);
     registerConsoleCommands(languageServerClient);
     registerOpenInISECommand();
+    registerPowerShellFindModuleCommand(languageServerClient);
 }
 
 export function deactivate(): void {