Update stdlib
[mirror/dsa-puppet.git] / 3rdparty / modules / stdlib / lib / puppet / parser / functions / upcase.rb
1 #
2 #  upcase.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(:upcase, :type => :rvalue, :doc => <<-EOS
8 Converts a string or an array of strings to uppercase.
9
10 *Examples:*
11
12     upcase("abcd")
13
14 Will return:
15
16     ABCD
17   EOS
18   ) do |arguments|
19
20     raise(Puppet::ParseError, "upcase(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size != 1
21
22     value = arguments[0]
23
24     unless value.is_a?(Array) || value.is_a?(Hash) || value.respond_to?(:upcase)
25       raise(Puppet::ParseError, 'upcase(): Requires an array, hash or object that responds to upcase in order to work')
26     end
27
28     if value.is_a?(Array)
29       # Numbers in Puppet are often string-encoded which is troublesome ...
30       result = value.collect { |i| function_upcase([i]) }
31     elsif value.is_a?(Hash)
32       result = {}
33       value.each_pair do |k, v|
34         result[function_upcase([k])] = function_upcase([v])
35       end
36     else
37       result = value.upcase
38     end
39
40     return result
41   end
42 end
43
44 # vim: set ts=2 sw=2 et :