blob: 7fd8a576a10405161cacf47cdd6734b33a63066d (
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
|
require 'puppet/parser/ast/branch'
class Puppet::Parser::AST
# A basic 'if/elsif/else' statement.
class IfStatement < AST::Branch
associates_doc
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(scope)
level = scope.ephemeral_level
value = @test.safeevaluate(scope)
# let's emulate a new scope for each branches
begin
if Puppet::Parser::Scope.true?(value)
return @statements.safeevaluate(scope)
else
return defined?(@else) ? @else.safeevaluate(scope) : nil
end
ensure
scope.unset_ephemeral_var(level)
end
end
end
end
|
