3rdparty/modules/stdlib/lib/puppet/parser/functions/num2bool.rb

   1 #
   2 # num2bool.rb
   3 #
   4
   5 module Puppet::Parser::Functions
   6   newfunction(:num2bool, :type => :rvalue, :doc => <<-EOS
   7 This function converts a number or a string representation of a number into a
   8 true boolean. Zero or anything non-numeric becomes false. Numbers higher then 0
   9 become true.
  10     EOS
  11   ) do |arguments|
  12
  13     raise(Puppet::ParseError, "num2bool(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size != 1
  14
  15     number = arguments[0]
  16
  17     case number
  18     when Numeric
  19       # Yay, it's a number
  20     when String
  21       begin
  22         number = Float(number)
  23       rescue ArgumentError => ex
  24         raise(Puppet::ParseError, "num2bool(): '#{number}' does not look like a number: #{ex.message}")
  25       end
  26     else
  27       begin
  28         number = number.to_s
  29       rescue NoMethodError => ex
  30         raise(Puppet::ParseError, "num2bool(): Unable to parse argument: #{ex.message}")
  31       end
  32     end
  33
  34     # Truncate Floats
  35     number = number.to_i
  36
  37     # Return true for any positive number and false otherwise
  38     return number > 0
  39   end
  40 end
  41
  42 # vim: set ts=2 sw=2 et :