Merge pull request #389 from jpogran/GH-373-convert-to-features

(GH-373) Convert Commands to Features

Author: glennsarti

src/PuppetStatusBar.ts

@@ -8,8 +8,11 @@ export class PuppetStatusBar {
   statusBarItem: vscode.StatusBarItem;
   private logger: ILogger;
 
-  constructor(langID: string, logger: ILogger) {
+  constructor(langID: string, context:vscode.ExtensionContext, logger: ILogger) {
     this.logger = logger;
+    context.subscriptions.push(vscode.commands.registerCommand(PuppetCommandStrings.PuppetShowConnectionMenuCommandId,
+      () => { PuppetStatusBar.showConnectionMenu(); }
+    ));
     this.statusBarItem = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 1);
     this.statusBarItem.command = PuppetCommandStrings.PuppetShowConnectionMenuCommandId;
     this.statusBarItem.show();

src/commands/pdk/pdkNewClassCommand.ts

@@ -9,7 +9,7 @@ import { PuppetStatusBar } from './PuppetStatusBar';
 import { PuppetLanguageClient } from './PuppetLanguageClient';
 import { ConnectionConfiguration } from './configuration';
 import { reporter } from './telemetry/telemetry';
-import { PuppetVersionRequest } from './messages';
+import { PuppetVersionRequest, PuppetCommandStrings } from './messages';
 
 const langID = 'puppet'; // don't change this
 const documentSelector = { scheme: 'file', language: langID };
@@ -44,6 +44,13 @@ export class ConnectionManager implements IConnectionManager {
     this.connectionStatus = ConnectionStatus.NotStarted;
     this.statusBarItem = statusBar;
     this.connectionConfiguration = connectionConfiguration;
+    context.subscriptions.push(vscode.commands.registerCommand(PuppetCommandStrings.PuppetShowConnectionLogsCommandId,
+      () => { this.showLogger(); }
+    ));
+  
+    context.subscriptions.push(vscode.commands.registerCommand(PuppetCommandStrings.PuppetRestartSessionCommandId,
+      () => { this.restartConnection(); }
+    ));
   }
 
   public get status(): ConnectionStatus {