(puppetlabsGH-242) Puppet Facts Endpoint

jpogran · Apr 29, 2020 · f157f3f · f157f3f
1 parent f2e758e
commit f157f3f
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 0 deletions.
diff --git a/lib/lsp/lsp_custom.rb b/lib/lsp/lsp_custom.rb
@@ -35,6 +35,27 @@ def from_h!(value)
     end
   end
 
+  # export interface GetPuppetFactResponse {
+  #   data: string;
+  #   error: string;
+  # }
+  class PuppetFactResponse < LSPBase
+    attr_accessor :facts # type: string
+    attr_accessor :error # type: string
+
+    def initialize(initial_hash = nil)
+      super
+      @optional_method_names = %i[error]
+    end
+
+    def from_h!(value)
+      value = {} if value.nil?
+      self.facts = value['facts']
+      self.error = value['error']
+      self
+    end
+  end
+
   # export interface GetPuppetResourceResponse {
   #   data: string;
   #   error: string;

diff --git a/lib/puppet-languageserver/message_handler.rb b/lib/puppet-languageserver/message_handler.rb
@@ -45,6 +45,16 @@ def request_puppet_getversion(_, _json_rpc_message)
       )
     end
 
+    def request_puppet_getfacts(_, json_rpc_message)
+      begin
+        results = PuppetLanguageServer::PuppetHelper.get_all_facts(documents.store_root_path)
+        LSP::PuppetFactResponse.new('facts' => results)
+      rescue StandardError => e
+        PuppetLanguageServer.log_message(:error, "(puppet/getFacts) Error getting facts. #{e}")
+        LSP::PuppetFactResponse.new('error' => 'An internal error occured while getting facts. Please see the debug log files for more information.')
+      end
+    end
+
     def request_puppet_getresource(_, json_rpc_message)
       type_name = json_rpc_message.params['typename']
       title = json_rpc_message.params['title']

diff --git a/lib/puppet-languageserver/puppet_helper.rb b/lib/puppet-languageserver/puppet_helper.rb
@@ -54,6 +54,15 @@ def self.get_node_graph(content, local_workspace)
       end
     end
 
+    def self.get_all_facts(local_workspace)
+      ap = PuppetLanguageServer::Sidecar::Protocol::ActionParams.new
+
+      args = ['--action-parameters=' + ap.to_json]
+      args << "--local-workspace=#{local_workspace}" unless local_workspace.nil?
+
+      sidecar_queue.execute_sync('facts_all', args, false)
+    end
+
     def self.get_puppet_resource(typename, title, local_workspace)
       ap = PuppetLanguageServer::Sidecar::Protocol::ActionParams.new
       ap['typename'] = typename

diff --git a/lib/puppet-languageserver/sidecar_queue.rb b/lib/puppet-languageserver/sidecar_queue.rb
@@ -118,6 +118,16 @@ def execute_sync(action, additional_args, handle_errors = false)
 
         PuppetLanguageServer::FacterHelper.assert_facts_loaded
 
+      when 'facts_all'
+        list = {}
+        PuppetLanguageServer::Sidecar::Protocol::FactList
+          .new
+          .from_json!(result)
+          .map { |element| list[element.key] = element.value }
+        PuppetLanguageServer.log_message(:debug, "SidecarQueue Thread: facts returned #{list.count} items")
+
+        return list
+
       when 'node_graph'
         return PuppetLanguageServer::Sidecar::Protocol::PuppetNodeGraph.new.from_json!(result)
 

diff --git a/lib/puppet_languageserver_sidecar.rb b/lib/puppet_languageserver_sidecar.rb
@@ -119,6 +119,7 @@ def self.require_gems(options)
     workspace_functions
     workspace_types
     facts
+    facts_all
   ].freeze
 
   class CommandLineParser
@@ -399,6 +400,14 @@ def self.execute(options)
       inject_workspace_as_environment unless injected
       PuppetLanguageServerSidecar::FacterHelper.retrieve_facts(cache)
 
+    when 'facts_all'
+      # Can't cache for facts
+      cache = PuppetLanguageServerSidecar::Cache::Null.new
+      # Inject the workspace etc. if present
+      injected = inject_workspace_as_module
+      inject_workspace_as_environment unless injected
+      PuppetLanguageServerSidecar::FacterHelper.retrieve_facts(cache)
+
     else
       log_message(:error, "Unknown action #{options[:action]}. Expected one of #{ACTION_LIST}")
     end