module Grit
  
  class Tree
    include Lazy
    
    lazy_reader :contents
    attr_reader :id
    attr_reader :mode
    attr_reader :name
    
    def initialize
      @contents = nil
      @__baked__ = nil
    end
    
    # Construct the contents of the tree
    #   +repo+ is the Repo
    #   +treeish+ is the reference
    #   +paths+ is an optional Array of directory paths to restrict the tree
    #
    # Returns Grit::Tree (baked)
    def self.construct(repo, treeish, paths = [])
      output = repo.git.ls_tree({}, treeish, *paths)
      
      self.allocate.construct_initialize(repo, treeish, output)
    end
    
    def construct_initialize(repo, id, text)
      @repo = repo
      @id = id
      @contents = []
      @__baked__ = nil
      
      text.split("\n").each do |line|
        @contents << content_from_string(repo, line)
      end
      @contents.compact!
      __baked__
      self
    end
    
    def __bake__
      temp = Tree.construct(@repo, @id, [])
      @contents = temp.contents
    end
    
    # Create an unbaked Tree containing just the specified attributes
    #   +repo+ is the Repo
    #   +atts+ is a Hash of instance variable data
    #
    # Returns Grit::Tree (unbaked)
    def self.create(repo, atts)
      self.allocate.create_initialize(repo, atts)
    end
    
    # Initializer for Tree.create
    #   +repo+ is the Repo
    #   +atts+ is a Hash of instance variable data
    #
    # Returns Grit::Tree (unbaked)
    def create_initialize(repo, atts)
      @repo = repo
      @contents = nil
      @__baked__ = nil
      
      atts.each do |k, v|
        instance_variable_set("@#{k}".to_sym, v)
      end
      self
    end
    
    # Parse a content item and create the appropriate object
    #   +repo+ is the Repo
    #   +text+ is the single line containing the items data in `git ls-tree` format
    #
    # Returns Grit::Blob or Grit::Tree
    def content_from_string(repo, text)
      mode, type, id, name = text.split(" ", 4)
      case type
        when "tree"
          Tree.create(repo, :id => id, :mode => mode, :name => name)
        when "blob"
          Blob.create(repo, :id => id, :mode => mode, :name => name)
        when "commit"
          nil
        else
          raise "Invalid type: #{type}"
      end
    end
    
    # Find the named object in this tree's contents
    #
    # Examples
    #   Repo.new('/path/to/grit').tree/'lib'
    #   # => #<Grit::Tree "6cc23ee138be09ff8c28b07162720018b244e95e">
    #   Repo.new('/path/to/grit').tree/'README.txt'
    #   # => #<Grit::Blob "8b1e02c0fb554eed2ce2ef737a68bb369d7527df">
    #
    # Returns Grit::Blob or Grit::Tree or nil if not found
    def /(file)
      self.contents.select { |c| c.name == file }.first
    end
    
    # Pretty object inspection
    def inspect
      %Q{#<Grit::Tree "#{@id}">}
    end
  end # Tree
  
end # Grit