From a718b9b4a4eaf36ee68462573b9ca1537067ae9d Mon Sep 17 00:00:00 2001 From: Krzysztof Wilczynski Date: Tue, 26 Apr 2011 00:13:31 +0100 Subject: First version. Simple count function to use within Puppet DSL. Signed-off-by: Krzysztof Wilczynski --- count.rb | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 count.rb diff --git a/count.rb b/count.rb new file mode 100644 index 0000000..c4a5977 --- /dev/null +++ b/count.rb @@ -0,0 +1,30 @@ +# +# count.rb +# + +# TODO(Krzysztof Wilczynski): We need to add support for regular expression ... + +module Puppet::Parser::Functions + newfunction(:count, :type => :rvalue, :doc => <<-EOS + EOS + ) do |arguments| + + # Technically we support two arguments but only first is mandatory ... + raise(Puppet::ParseError, "count(): Wrong number of arguments " + + "given (#{arguments.size} for 1)") if arguments.size < 1 + + array = arguments[0] + + if not array.is_a?(Array) + raise(Puppet::ParseError, 'count(): Requires an array to work with') + end + + item = arguments[1] if arguments[1] + + result = item ? array.count(item) : array.count + + return result + end +end + +# vim: set ts=2 sw=2 et : -- cgit v1.2.3