(GH-166) Add class loading to UI

This commit adds class loading the client UI so instead of steps 1 to 3, there
are now 4 steps (Functions, Classes, Types and Facts).

Author: glennsarti

client/src/connection.ts

@@ -287,7 +287,10 @@ export class ConnectionManager implements IConnectionManager {
 
         connectionManager.languageClient.sendRequest(messages.PuppetVersionRequest.type).then((versionDetails) => {
           lastVersionResponse = versionDetails
-          if (versionDetails.factsLoaded && versionDetails.functionsLoaded && versionDetails.typesLoaded) {
+          if (versionDetails.factsLoaded &&
+              versionDetails.functionsLoaded &&
+              versionDetails.typesLoaded &&
+              versionDetails.classesLoaded) {
             clearInterval(handle);
             connectionManager.setConnectionStatus(lastVersionResponse.puppetVersion, ConnectionStatus.Running);
             resolve();
@@ -297,7 +300,8 @@ export class ConnectionManager implements IConnectionManager {
             if (versionDetails.factsLoaded) { progress++; }
             if (versionDetails.functionsLoaded) { progress++; }
             if (versionDetails.typesLoaded) { progress++; }
-            progress = Math.round(progress / 3.0 * 100);
+            if (versionDetails.classesLoaded) { progress++; }
+            progress = Math.round(progress / 4.0 * 100);
 
             this.setConnectionStatus("Loading Puppet (" + progress.toString() + "%)", ConnectionStatus.Starting);
           }

client/src/messages.ts

@@ -11,6 +11,7 @@ export interface PuppetVersionDetails {
   factsLoaded: boolean;
   functionsLoaded: boolean;
   typesLoaded: boolean;
+  classesLoaded: boolean;
 }
 
 export interface PuppetResourceRequestParams {

server/lib/languageserver/puppet_version.rb

@@ -7,6 +7,7 @@ module PuppetVersion
     #   factsLoaded: boolean;
     #   functionsLoaded: boolean;
     #   typesLoaded: boolean;
+    #   classesLoaded: boolean;
     # }
 
     def self.create(options)
@@ -20,7 +21,8 @@ def self.create(options)
       result['factsLoaded']     = options['factsLoaded'] unless options['factsLoaded'].nil?
       result['functionsLoaded'] = options['functionsLoaded'] unless options['functionsLoaded'].nil?
       result['typesLoaded']     = options['typesLoaded'] unless options['typesLoaded'].nil?
-
+      result['classesLoaded']   = options['classesLoaded'] unless options['classesLoaded'].nil?
+      
       result['languageServerVersion'] = PuppetVSCode.version
 
       result

server/lib/puppet-languageserver/message_router.rb

@@ -101,7 +101,8 @@ def receive_request(request)
                                                                   'facterVersion'   => Facter.version,
                                                                   'factsLoaded'     => PuppetLanguageServer::FacterHelper.facts_loaded?,
                                                                   'functionsLoaded' => PuppetLanguageServer::PuppetHelper.functions_loaded?,
-                                                                  'typesLoaded'     => PuppetLanguageServer::PuppetHelper.types_loaded?))
+                                                                  'typesLoaded'     => PuppetLanguageServer::PuppetHelper.types_loaded?,
+                                                                  'classesLoaded'   => PuppetLanguageServer::PuppetHelper.classes_loaded?))
 
       when 'puppet/getResource'
         type_name = request.params['typename']