summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/parser/functions')
-rw-r--r--lib/puppet/parser/functions/num2bool.rb29
1 files changed, 24 insertions, 5 deletions
diff --git a/lib/puppet/parser/functions/num2bool.rb b/lib/puppet/parser/functions/num2bool.rb
index 15dd560..cf98f80 100644
--- a/lib/puppet/parser/functions/num2bool.rb
+++ b/lib/puppet/parser/functions/num2bool.rb
@@ -13,13 +13,32 @@ become true.
raise(Puppet::ParseError, "num2bool(): Wrong number of arguments " +
"given (#{arguments.size} for 1)") if arguments.size != 1
- # we need to get an Integer out of this
- begin
- number = arguments[0].to_i
- rescue NoMethodError
- raise(Puppet::ParseError, 'num2bool(): Unable to parse number: ' + $!)
+ number = arguments[0]
+
+ case number
+ when Numeric
+ # Yay, it's a number
+ when String
+ # Deal with strings later
+ else
+ begin
+ number = number.to_s
+ rescue NoMethodError
+ raise(Puppet::ParseError, 'num2bool(): Unable to parse argument: ' + $!)
+ end
+ end
+
+ case number
+ when String
+ # Only accept strings that look somewhat like numbers
+ unless number =~ /^-?\d+/
+ raise(Puppet::ParseError, "num2bool(): '#{number}' does not look like a number")
+ end
end
+ # Truncate floats
+ number = number.to_i
+
# Return true for any positive number and false otherwise
return number > 0 ? true : false
end