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

class Puppet::Parser::AST
  class ComparisonOperator < 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 operands, should return a boolean value
      lval = @lval.safeevaluate(scope)

      case @operator
      when "==","!="
        @rval.evaluate_match(lval, scope) ? @operator == '==' : @operator == '!='
      else
        rval = @rval.safeevaluate(scope)
        rval = Puppet::Parser::Scope.number?(rval) || rval
        lval = Puppet::Parser::Scope.number?(lval) || lval

        lval.send(@operator,rval)
      end
    end

    def initialize(hash)
      super

      raise ArgumentError, "Invalid comparison operator #{@operator}" unless %w{== != < > <= >=}.include?(@operator)
    end
  end
end