require 'puppet/parser/ast/branch'

class Puppet::Parser::AST
  # Each individual option in a case statement.
  class CaseOpt < AST::Branch
    attr_accessor :value, :statements

    # CaseOpt is a bit special -- we just want the value first,
    # so that CaseStatement can compare, and then it will selectively
    # decide whether to fully evaluate this option

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

    # Are we the default option?
    def default?
      # Cache the @default value.
      return @default if defined?(@default)

      @value.each { |subval|
        if subval.is_a?(AST::Default)
          @default = true
          break
        end
      }

      @default ||= false

      @default
    end

    # You can specify a list of values; return each in turn.
    def eachvalue(scope)
      @value.each { |subval|
        yield subval.safeevaluate(scope)
      }
    end

    def eachopt
      @value.each { |subval|
        yield subval
      }
    end

    # Evaluate the actual statements; this only gets called if
    # our option matched.
    def evaluate(scope)
      @statements.safeevaluate(scope)
    end
  end
end