module CassetteRack
  module Tree
    class Branch < CassetteRack::Tree::Leaf
      attr_reader :entries

      def initialize(path, level=0, trunk=nil)
        super
        @entries = []
        node
      end

      def leaf?
        false
      end

      def each(&block)
        entries.each do |entry|
          block.call(entry)
          entry.each(&block) if entry.is_a?(CassetteRack::Tree::Branch)
        end
      end

      private
        def node
          Dir[File.join(path, '*')].each do |path|
            if File.directory?(path)
              entries << CassetteRack::Tree::Branch.new(path, level, trunk)
            else
              entries << CassetteRack::Tree::Leaf.new(path, level, trunk)
            end
          end
        end
      # end private
    end
  end
end