diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/puppet/parser/functions/validate_nonemptyhash.rb | 36 | ||||
-rw-r--r-- | lib/puppet/parser/functions/validate_nonemptyhash_msg.rb | 30 |
2 files changed, 66 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/validate_nonemptyhash.rb b/lib/puppet/parser/functions/validate_nonemptyhash.rb new file mode 100644 index 0000000..9a7b1f6 --- /dev/null +++ b/lib/puppet/parser/functions/validate_nonemptyhash.rb @@ -0,0 +1,36 @@ +module Puppet::Parser::Functions + + newfunction(:validate_nonemptyhash, :doc => <<-'ENDHEREDOC') do |args| + Validate that all passed values are hash data structures. Abort catalog + compilation if any value fails this check. Also abort if any of the hashes + is empty. + + The following values will pass: + + $my_hash = { 'one' => 'two' } + validate_nonemptyhash($my_array) + + The following values will fail, causing compilation to abort: + + validate_nonemptyhash(true) + validate_nonemptyhash([]) + validate_nonemptyhash('some_string') + $undefined = undef + validate_nonemptyhash($undefined) + + ENDHEREDOC + + unless args.length > 0 then + raise Puppet::ParseError, ("validate_nonemptyhash(): wrong number of arguments (#{args.length}; must be > 0)") + end + + args.each do |arg| + unless arg.is_a?(Hash) + raise Puppet::ParseError, ("#{arg.inspect} is not an Hash. It looks to be a #{arg.class}") + end + unless arg.keys.length > 0 + raise Puppet::ParseError, ("#{arg.inspect} is an empty Hash.") + end + end + end +end diff --git a/lib/puppet/parser/functions/validate_nonemptyhash_msg.rb b/lib/puppet/parser/functions/validate_nonemptyhash_msg.rb new file mode 100644 index 0000000..f402b92 --- /dev/null +++ b/lib/puppet/parser/functions/validate_nonemptyhash_msg.rb @@ -0,0 +1,30 @@ +module Puppet::Parser::Functions + + newfunction(:validate_nonemptyhash_msg, :doc => <<-'ENDHEREDOC') do |args| + Validate that the passed value is an hash data structure. Abort catalog + compilation if any value fails this check. Also abort if any of the hashs + is empty. On failure it prints the message given as the second argument. + + The following values will pass: + + $my_hash = { 'one', 'two' } + validate_nonemptyhash_msg($my_hash, 'foo') + + The following values will fail, causing compilation to abort with the given msg: + + validate_nonemptyhash_msg(true, 'Not a nonempty hash') + validate_nonemptyhash_msg([], 'Not a nonempty hash') + + ENDHEREDOC + + unless args.length == 2 then + raise Puppet::ParseError, ("validate_nonemptyhash_msg(): wrong number of arguments (#{args.length}; must be == 2)") + end + + arg = args[0] + msg = args[1] + unless arg.is_a?(Hash) and arg.keys.length > 0 + raise Puppet::ParseError, (msg) + end + end +end |