require 'puppet/parser/ast/branch'

class Puppet::Parser::AST
    # The basic container class.  This object behaves almost identically
    # to a normal array except at initialization time.  Note that its name
    # is 'AST::ASTArray', rather than plain 'AST::Array'; I had too many
    # bugs when it was just 'AST::Array', because things like
    # 'object.is_a?(Array)' never behaved as I expected.
    class ASTArray < Branch
        include Enumerable

        # Return a child by index.  Probably never used.
        def [](index)
            @children[index]
        end

        # Evaluate our children.
        def evaluate(scope)
            rets = nil
            # We basically always operate declaratively, and when we
            # do we need to evaluate the settor-like statements first.  This
            # is basically variable and type-default declarations.
            if scope.declarative?
                test = [
                    AST::VarDef, AST::TypeDefaults
                ]

                settors = []
                others = []
                @children.each { |child|
                    if test.include?(child.class)
                        settors.push child
                    else
                        others.push child
                    end
                }
                rets = [settors,others].flatten.collect { |child|
                    child.safeevaluate(scope)
                }
            else
                # If we're not declarative, just do everything in order.
                rets = @children.collect { |item|
                    item.safeevaluate(scope)
                }
            end

            rets = rets.reject { |obj| obj.nil? }

            return rets
        end

        def push(*ary)
            ary.each { |child|
                #Puppet.debug "adding %s(%s) of type %s to %s" %
                #    [child, child.object_id, child.class.to_s.sub(/.+::/,''),
                #    self.object_id]
                @children.push(child)
            }

            return self
        end

        # Convert to a string.  Only used for printing the parse tree.
        def to_s
            return "[" + @children.collect { |child|
                child.to_s
            }.join(", ") + "]"
        end

        # Print the parse tree.
        def tree(indent = 0)
            #puts((AST.indent * indent) + self.pin)
            self.collect { |child|
                child.tree(indent)
            }.join("\n" + (AST.midline * (indent+1)) + "\n")
        end
    end

    # A simple container class, containing the parameters for an object.
    # Used for abstracting the grammar declarations.  Basically unnecessary
    # except that I kept finding bugs because I had too many arrays that
    # meant completely different things.
    class ObjectInst < ASTArray; end

    # Another simple container class to make sure we can correctly arrayfy
    # things.
    class CompArgument < ASTArray; end
end