tree and commit

1 files changed, 40 insertions, 0 deletions

diff --git a/repository.go b/repository.go
index 2dadcb6..c732d0f 100644
--- a/repository.go
+++ b/repository.go
@@ -1,6 +1,7 @@
 package git
 
 import (
+	"errors"
 	"fmt"
 
 	"gopkg.in/src-d/go-git.v2/clients/common"
@@ -8,6 +9,10 @@ import (
 	"gopkg.in/src-d/go-git.v2/internal"
 )
 
+var (
+	ObjectNotFoundErr = errors.New("object not found")
+)
+
 const (
 	DefaultRemoteName = "origin"
 )
@@ -66,3 +71,38 @@ func (r *Repository) Pull(remoteName, branch string) error {
 
 	return nil
 }
+
+// Commit return the commit with the given hash
+func (r *Repository) Commit(h internal.Hash) (*Commit, error) {
+	obj, ok := r.Storage.Get(h)
+	if !ok {
+		return nil, ObjectNotFoundErr
+	}
+
+	commit := &Commit{r: r}
+	return commit, commit.Decode(obj)
+}
+
+// Commits decode the objects into commits
+func (r *Repository) Commits() *CommitIter {
+	i := NewCommitIter(r)
+	go func() {
+		defer i.Close()
+		for _, obj := range r.Storage.Commits {
+			i.Add(obj)
+		}
+	}()
+
+	return i
+}
+
+// Tree return the tree with the given hash
+func (r *Repository) Tree(h internal.Hash) (*Tree, error) {
+	obj, ok := r.Storage.Get(h)
+	if !ok {
+		return nil, ObjectNotFoundErr
+	}
+
+	tree := &Tree{r: r}
+	return tree, tree.Decode(obj)
+}