3rdparty/modules/stdlib/lib/puppet/parser/functions/fqdn_rotate.rb

   1 #
   2 # fqdn_rotate.rb
   3 #
   4
   5 module Puppet::Parser::Functions
   6   newfunction(:fqdn_rotate, :type => :rvalue, :doc => <<-EOS
   7 Rotates an array a random number of times based on a nodes fqdn.
   8     EOS
   9   ) do |arguments|
  10
  11     raise(Puppet::ParseError, "fqdn_rotate(): Wrong number of arguments " +
  12       "given (#{arguments.size} for 1)") if arguments.size < 1
  13
  14     value = arguments[0]
  15     require 'digest/md5'
  16
  17     unless value.is_a?(Array) || value.is_a?(String)
  18       raise(Puppet::ParseError, 'fqdn_rotate(): Requires either ' +
  19         'array or string to work with')
  20     end
  21
  22     result = value.clone
  23
  24     string = value.is_a?(String) ? true : false
  25
  26     # Check whether it makes sense to rotate ...
  27     return result if result.size <= 1
  28
  29     # We turn any string value into an array to be able to rotate ...
  30     result = string ? result.split('') : result
  31
  32     elements = result.size
  33
  34     seed = Digest::MD5.hexdigest([lookupvar('::fqdn'),arguments].join(':')).hex
  35     # deterministic_rand() was added in Puppet 3.2.0; reimplement if necessary
  36     if Puppet::Util.respond_to?(:deterministic_rand)
  37       offset = Puppet::Util.deterministic_rand(seed, elements).to_i
  38     else
  39       if defined?(Random) == 'constant' && Random.class == Class
  40         offset = Random.new(seed).rand(elements)
  41       else
  42         srand(seed)
  43         offset = rand(elements)
  44         srand()
  45       end
  46     end
  47     offset.times {
  48        result.push result.shift
  49     }
  50
  51     result = string ? result.join : result
  52
  53     return result
  54   end
  55 end
  56
  57 # vim: set ts=2 sw=2 et :