(FACT-932) Allow use of fact() on other hashes

Because sometimes people want to use an alternative data set, but treat
it like it's a set of facts.

1 files changed, 17 insertions, 6 deletions

diff --git a/lib/puppet/functions/fact.rb b/lib/puppet/functions/fact.rb
index dfb048b..48d0d3c 100644
--- a/lib/puppet/functions/fact.rb
+++ b/lib/puppet/functions/fact.rb
@@ -18,22 +18,27 @@ Puppet::Functions.create_function(:fact) do
     param 'String', :fact_name
   end
 
-  def to_dot_syntax(array_path)
-    array_path.map do |string|
-      string.include?('.') ? %Q{"#{string}"} : string
-    end.join('.')
+  dispatch :alternative do
+    param 'Hash',   :fact_hash
+    param 'String', :fact_name
   end
 
   def fact(fact_name)
-    facts = closure_scope['facts']
+    dot_dig(closure_scope['facts'], fact_name)
+  end
+
+  def alternative(alternative_hash, fact_name)
+    dot_dig(alternative_hash, fact_name)
+  end
 
+  def dot_dig(fact_hash, fact_name)
     # Transform the dot-notation string into an array of paths to walk. Make
     # sure to correctly extract double-quoted values containing dots as single
     # elements in the path.
     path = fact_name.scan(/([^."]+)|(?:")([^"]+)(?:")/).map {|x| x.compact.first }
 
     walked_path = []
-    path.reduce(facts) do |d, k|
+    path.reduce(fact_hash) do |d, k|
       return nil if d.nil? || k.nil?
 
       case
@@ -55,4 +60,10 @@ Puppet::Functions.create_function(:fact) do
       result
     end
   end
+
+  def to_dot_syntax(array_path)
+    array_path.map do |string|
+      string.include?('.') ? %Q{"#{string}"} : string
+    end.join('.')
+  end
 end