summaryrefslogtreecommitdiff
path: root/lib/puppet/parser/functions/any2bool.rb
blob: f0f8f83d1dfeb341b1a44ab9b2339e0e33070a56 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#
# any2bool.rb
#

module Puppet::Parser::Functions
  newfunction(:any2bool, :type => :rvalue, :doc => <<-EOS
This converts 'anything' to a boolean. In practise it does the following:

* Strings such as Y,y,1,T,t,TRUE,yes,'true' will return true
* Strings such as 0,F,f,N,n,FALSE,no,'false' will return false
* Booleans will just return their original value
* Number (or a string representation of a number) > 0 will return true, otherwise false
* undef will return false
* Anything else will return true
    EOS
  ) do |arguments|

    raise(Puppet::ParseError, "any2bool(): Wrong number of arguments " +
      "given (#{arguments.size} for 1)") if arguments.size < 1

    # If argument is already Boolean, return it
    if !!arguments[0] == arguments[0]
      return arguments[0]
    end

    arg = arguments[0]

    if arg == nil
      return false
    end

    if arg == :undef
      return false
    end

    valid_float = !!Float(arg) rescue false

    if arg.is_a?(Numeric)
      return function_num2bool( [ arguments[0] ] )
    end

    if arg.is_a?(String)
      if valid_float
        return function_num2bool( [ arguments[0] ] )
      else
        return function_str2bool( [ arguments[0] ] )
      end
    end

    return true

  end
end

# vim: set ts=2 sw=2 et :