From 14b4ca8de2c8802bc599bb45aca5ecbba5ed98c9 Mon Sep 17 00:00:00 2001 From: Krzysztof Wilczynski Date: Fri, 29 Apr 2011 17:30:04 +0100 Subject: Added support for strings to reverse. Signed-off-by: Krzysztof Wilczynski --- reverse.rb | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/reverse.rb b/reverse.rb index 6aa11f4..86dee28 100644 --- a/reverse.rb +++ b/reverse.rb @@ -2,8 +2,6 @@ # reverse.rb # -# TODO(Krzysztof Wilczynski): Support for strings would be nice too ... - module Puppet::Parser::Functions newfunction(:reverse, :type => :rvalue, :doc => <<-EOS EOS @@ -12,13 +10,15 @@ module Puppet::Parser::Functions raise(Puppet::ParseError, "reverse(): Wrong number of arguments " + "given (#{arguments.size} for 1)") if arguments.size < 1 - array = arguments[0] + value = arguments[0] + klass = value.class - if not array.is_a?(Array) - raise(Puppet::ParseError, 'reverse(): Requires an array to work with') + if not [Array, String].include?(klass) + raise(Puppet::ParseError, 'reverse(): Requires either an ' + + 'array or string to work with') end - result = array.reverse + result = value.reverse return result end -- cgit v1.2.3