summaryrefslogtreecommitdiffstats
path: root/lib/puppet/parser/ast/boolean_operator.rb
blob: 8481e4f8de2003c13557b99c842152fd6a71254e (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
require 'puppet'
require 'puppet/parser/ast/branch'

class Puppet::Parser::AST
  class BooleanOperator < AST::Branch

    attr_accessor :operator, :lval, :rval

    # Iterate across all of our children.
    def each
      [@lval,@rval,@operator].each { |child| yield child }
    end

    # Returns a boolean which is the result of the boolean operation
    # of lval and rval operands
    def evaluate(scope)
      # evaluate the first operand, should return a boolean value
      lval = @lval.safeevaluate(scope)

      # return result
      # lazy evaluate right operand
      case @operator
      when "and"
        if Puppet::Parser::Scope.true?(lval)
          rval = @rval.safeevaluate(scope)
          Puppet::Parser::Scope.true?(rval)
        else # false and false == false
          false
        end
      when "or"
        if Puppet::Parser::Scope.true?(lval)
          true
        else
          rval = @rval.safeevaluate(scope)
          Puppet::Parser::Scope.true?(rval)
        end
      end
    end

    def initialize(hash)
      super

      raise ArgumentError, "Invalid boolean operator #{@operator}" unless %w{and or}.include?(@operator)
    end
  end
end