e2d7f3bb89
We need to use unless value.is_a?(String) || value.is_a?(Array) rather than klass = value.class unless [String, Array].include?(klass) because the klass version enforces type checking which is too strict, and does not allow us to accept objects wich have extended String (or Array). For example, generate() function now returns Puppet::Util::Execution::ProcessOutput which is just a very simple extension of String. While this in it's self was not intentional (PUP-2306) it is not unreasonable to cope with objects which extend Strings
45 lines
1.1 KiB
Ruby
45 lines
1.1 KiB
Ruby
#
|
|
# fqdn_rotate.rb
|
|
#
|
|
|
|
module Puppet::Parser::Functions
|
|
newfunction(:fqdn_rotate, :type => :rvalue, :doc => <<-EOS
|
|
Rotates an array a random number of times based on a nodes fqdn.
|
|
EOS
|
|
) do |arguments|
|
|
|
|
raise(Puppet::ParseError, "fqdn_rotate(): Wrong number of arguments " +
|
|
"given (#{arguments.size} for 1)") if arguments.size < 1
|
|
|
|
value = arguments[0]
|
|
require 'digest/md5'
|
|
|
|
unless value.is_a?(Array) || value.is_a?(String)
|
|
raise(Puppet::ParseError, 'fqdn_rotate(): Requires either ' +
|
|
'array or string to work with')
|
|
end
|
|
|
|
result = value.clone
|
|
|
|
string = value.is_a?(String) ? true : false
|
|
|
|
# Check whether it makes sense to rotate ...
|
|
return result if result.size <= 1
|
|
|
|
# We turn any string value into an array to be able to rotate ...
|
|
result = string ? result.split('') : result
|
|
|
|
elements = result.size
|
|
|
|
srand(Digest::MD5.hexdigest([lookupvar('::fqdn'),arguments].join(':')).hex)
|
|
rand(elements).times {
|
|
result.push result.shift
|
|
}
|
|
|
|
result = string ? result.join : result
|
|
|
|
return result
|
|
end
|
|
end
|
|
|
|
# vim: set ts=2 sw=2 et :
|