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

   1 module Puppet::Parser::Functions
   2
   3   newfunction(:validate_slength, :doc => <<-'ENDHEREDOC') do |args|
   4     Validate that the first argument is a string (or an array of strings), and
   5     less/equal to than the length of the second argument. An optional third
   6     parameter can be given a the minimum length. It fails if the first
   7     argument is not a string or array of strings, and if arg 2 and arg 3 are
   8     not convertable to a number.
   9
  10     The following values will pass:
  11
  12       validate_slength("discombobulate",17)
  13       validate_slength(["discombobulate","moo"],17)
  14       validate_slength(["discombobulate","moo"],17,3)
  15
  16     The following valueis will not:
  17
  18       validate_slength("discombobulate",1)
  19       validate_slength(["discombobulate","thermometer"],5)
  20       validate_slength(["discombobulate","moo"],17,10)
  21
  22     ENDHEREDOC
  23
  24     raise Puppet::ParseError, "validate_slength(): Wrong number of arguments (#{args.length}; must be 2 or 3)" unless args.length == 2 or args.length == 3
  25
  26     input, max_length, min_length = *args
  27
  28     begin
  29       max_length = Integer(max_length)
  30       raise ArgumentError if max_length <= 0
  31     rescue ArgumentError, TypeError
  32       raise Puppet::ParseError, "validate_slength(): Expected second argument to be a positive Numeric, got #{max_length}:#{max_length.class}"
  33     end
  34
  35     if min_length
  36       begin
  37         min_length = Integer(min_length)
  38         raise ArgumentError if min_length < 0
  39     rescue ArgumentError, TypeError
  40         raise Puppet::ParseError, "validate_slength(): Expected third argument to be unset or a positive Numeric, got #{min_length}:#{min_length.class}"
  41       end
  42     else
  43       min_length = 0
  44     end
  45
  46     if min_length > max_length
  47       raise Puppet::ParseError, "validate_slength(): Expected second argument to be larger than third argument"
  48     end
  49
  50     validator = lambda do |str|
  51       unless str.length <= max_length and str.length >= min_length
  52         raise Puppet::ParseError, "validate_slength(): Expected length of #{input.inspect} to be between #{min_length} and #{max_length}, was #{input.length}"
  53       end
  54     end
  55
  56     case input
  57     when String
  58       validator.call(input)
  59     when Array
  60       input.each_with_index do |arg, pos|
  61         if arg.is_a? String
  62           validator.call(arg)
  63         else
  64           raise Puppet::ParseError, "validate_slength(): Expected element at array position #{pos} to be a String, got #{arg.class}"
  65         end
  66       end
  67     else
  68       raise Puppet::ParseError, "validate_slength(): Expected first argument to be a String or Array, got #{input.class}"
  69     end
  70   end
  71 end