1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
# Add brackets if the argument is an IPv6 address.
# Returns the argument untouched otherwise.
#
# CAUTION: this code "fails" when the user is passing
# an IPv6 address with the port in it without the
# brackets: 2001::1:8080, to specify address 2001::1
# and port 8080. This code will change it to
# [2001::1:8080] as it's a valid ip address. This
# shouldn't be an issue in most cases.
#
# If an array is given, each member will be normalized to
# a valid IPv6 address with brackets when needed.
Puppet::Functions.create_function(:normalize_ip_for_uri) do
def normalize_ip_for_uri(*args)
require 'ipaddr'
result = []
args = args[0] if args[0].kind_of?(Array)
args = [args] unless args.kind_of?(Array)
args.each do |ip|
begin
if IPAddr.new(ip).ipv6?
unless ip.match(/\[.+\]/)
Puppet.debug("IP #{ip} is changed to [#{ip}]")
ip = "[#{ip}]"
end
end
rescue ArgumentError
# ignore it
end
result << ip
end
return result[0] if args.size == 1
result
end
end
|
