Update stdlib
[mirror/dsa-puppet.git] / 3rdparty / modules / stdlib / lib / puppet / parser / functions / swapcase.rb
1 #
2 #  swapcase.rb
3 #  Please note: This function is an implementation of a Ruby class and as such may not be entirely UTF8 compatible. To ensure compatibility please use this function with Ruby 2.4.0 or greater - https://bugs.ruby-lang.org/issues/10085.
4 #
5
6 module Puppet::Parser::Functions
7   newfunction(:swapcase, :type => :rvalue, :doc => <<-EOS
8 This function will swap the existing case of a string.
9
10 *Examples:*
11
12     swapcase("aBcD")
13
14 Would result in: "AbCd"
15     EOS
16   ) do |arguments|
17
18     raise(Puppet::ParseError, "swapcase(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size < 1
19
20     value = arguments[0]
21
22     unless value.is_a?(Array) || value.is_a?(String)
23       raise(Puppet::ParseError, 'swapcase(): Requires either array or string to work with')
24     end
25
26     if value.is_a?(Array)
27       # Numbers in Puppet are often string-encoded which is troublesome ...
28       result = value.collect { |i| i.is_a?(String) ? i.swapcase : i }
29     else
30       result = value.swapcase
31     end
32
33     return result
34   end
35 end
36
37 # vim: set ts=2 sw=2 et :