puppet custom functions

rand.rb

# Return a random number
Puppet::Functions.create_function(:rand) do
  # @param [Integer] max
  #     maximum random number
  # @return [Variant[Integer,Float]]
  #     random number
  # @example Calling the function
  #   rand(10)
  dispatch :generate_random_number do
    optional_param 'Integer', :max
    return_type 'Variant[Integer,Float]'
  end

  def generate_random_number(max = nil)
    rand(max)
  end
end