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
|
class Puppet::Pops::Parser::CodeMerger
# Concatenates the logic in the array of parse results into one parse result.
# @return Puppet::Parser::AST::BlockExpression
#
def concatenate(parse_results)
# this is a bit brute force as the result is already 3x ast with wrapped 4x content
# this could be combined in a more elegant way, but it is only used to process a handful of files
# at the beginning of a puppet run. TODO: Revisit for Puppet 4x when there is no 3x ast at the top.
# PUP-5299, some sites have thousands of entries, and run out of stack when evaluating - the logic
# below maps the logic as flatly as possible.
#
children = parse_results.select {|x| !x.nil? && x.code}.flat_map do |parsed_class|
case parsed_class.code
when Puppet::Parser::AST::BlockExpression
# the BlockExpression wraps a single 4x instruction that is most likely wrapped in a Factory
parsed_class.code.children.map {|c| c.is_a?(Puppet::Pops::Model::Factory) ? c.model : c }
when Puppet::Pops::Model::Factory
# If it is a 4x instruction wrapped in a Factory
parsed_class.code.model
else
# It is the instruction directly
parsed_class.code
end
end
Puppet::Parser::AST::BlockExpression.new(:children => children)
end
end
|
