(GH-136) Add UI to extension client to show load status of server

The client will now query the language server periodically after connecting.
This query is for the load status which is shown in the UI of the extension
status bar item, until it is fully loaded.  Once loaded, the puppet version is
displayed.

Author: glennsarti

client/src/commands/puppetcommands.ts

@@ -19,7 +19,11 @@ export function setupPuppetCommands(langID:string, connManager:IConnectionManage
   ctx.subscriptions.push(vscode.commands.registerCommand(messages.PuppetCommandStrings.PuppetNodeGraphToTheSideCommandId,
     uri => showNodeGraph(uri, true))
   );
-  
+
+  ctx.subscriptions.push(vscode.commands.registerCommand(messages.PuppetCommandStrings.PuppetShowConnectionMenuCommandId,
+    () => { connManager.showConnectionMenu(); }
+  ));
+
   const contentProvider = new PuppetNodeGraphContentProvider(ctx, connManager);
   const contentProviderRegistration = vscode.workspace.registerTextDocumentContentProvider(langID, contentProvider);
 

client/src/connection.ts

@@ -325,9 +325,11 @@ export class ConnectionManager implements IConnectionManager {
     var languageServerClient = new LanguageClient(title, serverOptions, clientOptions)
     languageServerClient.onReady().then(() => {
       langClient.logger.debug('Language server client started, setting puppet version')
-      languageServerClient.sendRequest(messages.PuppetVersionRequest.type).then((versionDetails) => {
-        this.setConnectionStatus(versionDetails.puppetVersion, ConnectionStatus.Running);
-        if (reporter) {
+      this.setConnectionStatus("Loading Puppet", ConnectionStatus.Starting);
+      this.queryLanguageServerStatus();
+      // Send telemetry
+      if (reporter) {
+        languageServerClient.sendRequest(messages.PuppetVersionRequest.type).then((versionDetails) => {
           reporter.sendTelemetryEvent('puppetVersion' +versionDetails.puppetVersion);
           reporter.sendTelemetryEvent('facterVersion' + versionDetails.facterVersion);
           reporter.sendTelemetryEvent('languageServerVersion' + versionDetails.languageServerVersion);
@@ -336,15 +338,54 @@ export class ConnectionManager implements IConnectionManager {
             facterVersion: versionDetails.facterVersion,
             languageServerVersion: versionDetails.languageServerVersion,
           });
-        }
-      });
+        });
+      }
     }, (reason) => {
       this.setSessionFailure("Could not start language service: ", reason);
     });
 
     return languageServerClient;
   }
 
+  private queryLanguageServerStatus() {
+    let connectionManager = this;
+
+    return new Promise((resolve, reject) => {
+      let count = 0;
+      let lastVersionResponse = null;
+      let handle = setInterval(() => {
+        count++;
+
+        // After 30 seonds timeout the progress
+        if (count >= 30 || connectionManager.languageClient == undefined) {
+          clearInterval(handle);
+          connectionManager.setConnectionStatus(lastVersionResponse.puppetVersion, ConnectionStatus.Running);
+          resolve();
+        }
+
+        connectionManager.languageClient.sendRequest(messages.PuppetVersionRequest.type).then((versionDetails) => {
+          lastVersionResponse = versionDetails
+          if (versionDetails.factsLoaded && versionDetails.functionsLoaded && versionDetails.typesLoaded) {
+            clearInterval(handle);
+            connectionManager.setConnectionStatus(lastVersionResponse.puppetVersion, ConnectionStatus.Running);
+            resolve();
+          } else {
+            let progress = 0;
+
+            if (versionDetails.factsLoaded) { progress++; }
+            if (versionDetails.functionsLoaded) { progress++; }
+            if (versionDetails.typesLoaded) { progress++; }
+            progress = Math.round(progress / 3.0 * 100);
+
+            this.setConnectionStatus("Loading Puppet (" + progress.toString() + "%)", ConnectionStatus.Starting);
+          }
+        });
+
+      }, 1000);
+    });
+  }
+
+
   private restartConnection(connectionConfig?: IConnectionConfiguration) {
       this.stop();
       this.start(connectionConfig);