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

   1 #
   2 # to_bytes.rb
   3 #
   4 module Puppet::Parser::Functions
   5   newfunction(:to_bytes, :type => :rvalue, :doc => <<-DOC
   6     Converts the argument into bytes, for example 4 kB becomes 4096.
   7     Takes a single string value as an argument.
   8     These conversions reflect a layperson's understanding of
   9     1 MB = 1024 KB, when in fact 1 MB = 1000 KB, and 1 MiB = 1024 KiB.
  10     DOC
  11              ) do |arguments|
  12
  13     raise(Puppet::ParseError, "to_bytes(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size != 1
  14
  15     arg = arguments[0]
  16
  17     return arg if arg.is_a? Numeric
  18
  19     value, prefix = *%r{([0-9.e+-]*)\s*([^bB]?)}.match(arg)[1, 2]
  20
  21     value = value.to_f
  22     case prefix
  23     when '' then return value.to_i
  24     when 'k' then return (value * (1 << 10)).to_i
  25     when 'M' then return (value * (1 << 20)).to_i
  26     when 'G' then return (value * (1 << 30)).to_i
  27     when 'T' then return (value * (1 << 40)).to_i
  28     when 'P' then return (value * (1 << 50)).to_i
  29     when 'E' then return (value * (1 << 60)).to_i
  30     else raise Puppet::ParseError, "to_bytes(): Unknown prefix #{prefix}"
  31     end
  32   end
  33 end