summaryrefslogtreecommitdiffstats
path: root/lib/puppet/parser/ast/ifstatement.rb
blob: b5cb63cdfccee971c2d3b8f865612c602378def1 (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
class Puppet::Parser::AST
    # A basic 'if/elsif/else' statement.
    class IfStatement < AST::Branch
        attr_accessor :test, :else, :statements

        def each
            [@test,@else,@statements].each { |child| yield child }
        end

        # Short-curcuit evaluation.  If we're true, evaluate our statements,
        # else if there's an 'else' setting, evaluate it.
        # the first option that matches.
        def evaluate(hash)
            scope = hash[:scope]
            value = @test.safeevaluate(:scope => scope)

            if Puppet::Parser::Scope.true?(value)
                return @statements.safeevaluate(:scope => scope)
            else
                if defined? @else
                    return @else.safeevaluate(:scope => scope)
                else
                    return nil
                end
            end
        end

        def tree(indent = 0)
            rettree = [
                @test.tree(indent + 1),
                ((@@indline * indent) + self.typewrap(self.pin)),
                @statements.tree(indent + 1),
                @else.tree(indent + 1)
            ]

            return rettree.flatten.join("\n")
        end
    end
end

# $Id$