diff options
Diffstat (limited to 'code/environments/production/modules/stdlib/lib/puppet/parser/functions/num2bool.rb')
-rw-r--r-- | code/environments/production/modules/stdlib/lib/puppet/parser/functions/num2bool.rb | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/code/environments/production/modules/stdlib/lib/puppet/parser/functions/num2bool.rb b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/num2bool.rb new file mode 100644 index 0000000..0957c56 --- /dev/null +++ b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/num2bool.rb @@ -0,0 +1,41 @@ +# +# num2bool.rb +# +module Puppet::Parser::Functions + newfunction(:num2bool, :type => :rvalue, :doc => <<-DOC + This function converts a number or a string representation of a number into a + true boolean. Zero or anything non-numeric becomes false. Numbers higher then 0 + become true. + DOC + ) do |arguments| + + raise(Puppet::ParseError, "num2bool(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size != 1 + + number = arguments[0] + + case number + when Numeric # rubocop:disable Lint/EmptyWhen : Required for the module to work + # Yay, it's a number + when String + begin + number = Float(number) + rescue ArgumentError => ex + raise(Puppet::ParseError, "num2bool(): '#{number}' does not look like a number: #{ex.message}") + end + else + begin + number = number.to_s + rescue NoMethodError => ex + raise(Puppet::ParseError, "num2bool(): Unable to parse argument: #{ex.message}") + end + end + + # Truncate Floats + number = number.to_i + + # Return true for any positive number and false otherwise + return number > 0 + end +end + +# vim: set ts=2 sw=2 et : |