summaryrefslogtreecommitdiffstats
path: root/lib/puppet/parser
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/parser')
-rw-r--r--lib/puppet/parser/functions/lookup_hiera_hash.rb22
1 files changed, 22 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/lookup_hiera_hash.rb b/lib/puppet/parser/functions/lookup_hiera_hash.rb
new file mode 100644
index 0000000..d96d65f
--- /dev/null
+++ b/lib/puppet/parser/functions/lookup_hiera_hash.rb
@@ -0,0 +1,22 @@
+module Puppet::Parser::Functions
+ newfunction(:lookup_hiera_hash, :arity => 2, :type => :rvalue,
+ :doc => "Lookup a key->value from a Hiera hash") do |args|
+ hash_name = args[0]
+ key_name = args[1]
+ unless hash_name.is_a?(String) and key_name.is_a?(String)
+ raise Puppet::ParseError, "The hash name and the key name must be given as strings."
+ end
+ if defined? call_function
+ hash = call_function('hiera', [hash_name])
+ else
+ hash = function_hiera([hash_name])
+ end
+ unless hash.is_a?(Hash)
+ raise Puppet::ParseError, "The value Hiera returned for #{hash_name} is not a Hash."
+ end
+ unless hash.key?(key_name)
+ raise Puppet::ParseError, "The Hiera hash #{hash_name} does not contain key #{key_name}."
+ end
+ return hash[key_name]
+ end
+end