plumbing: object, add APIs for traversing over commit graphs

Signed-off-by: Filip Navara <filip.navara@gmail.com>

1 files changed, 310 insertions, 0 deletions

diff --git a/plumbing/object/commitnode.go b/plumbing/object/commitnode.go
new file mode 100644
index 0000000..a613eb4
--- /dev/null
+++ b/plumbing/object/commitnode.go
@@ -0,0 +1,310 @@
+package object

+

+import (

+	"fmt"

+	"io"

+	"time"

+

+	"gopkg.in/src-d/go-git.v4/plumbing"

+	"gopkg.in/src-d/go-git.v4/plumbing/format/commitgraph"

+	"gopkg.in/src-d/go-git.v4/plumbing/storer"

+)

+

+// CommitNode is generic interface encapsulating either Commit object or

+// graphCommitNode object

+type CommitNode interface {

+	ID() plumbing.Hash

+	Tree() (*Tree, error)

+	CommitTime() time.Time

+}

+

+// CommitNodeIndex is generic interface encapsulating an index of CommitNode objects

+// and accessor methods for walking it as a directed graph

+type CommitNodeIndex interface {

+	NumParents(node CommitNode) int

+	ParentNodes(node CommitNode) CommitNodeIter

+	ParentNode(node CommitNode, i int) (CommitNode, error)

+	ParentHashes(node CommitNode) []plumbing.Hash

+

+	Get(hash plumbing.Hash) (CommitNode, error)

+

+	// Commit returns the full commit object from the node

+	Commit(node CommitNode) (*Commit, error)

+}

+

+// CommitNodeIter is a generic closable interface for iterating over commit nodes.

+type CommitNodeIter interface {

+	Next() (CommitNode, error)

+	ForEach(func(CommitNode) error) error

+	Close()

+}

+

+// graphCommitNode is a reduced representation of Commit as presented in the commit

+// graph file (commitgraph.Node). It is merely useful as an optimization for walking

+// the commit graphs.

+//

+// graphCommitNode implements the CommitNode interface.

+type graphCommitNode struct {

+	// Hash for the Commit object

+	hash plumbing.Hash

+	// Index of the node in the commit graph file

+	index int

+

+	node *commitgraph.Node

+	gci  *graphCommitNodeIndex

+}

+

+// graphCommitNodeIndex is an index that can load CommitNode objects from both the commit

+// graph files and the object store.

+//

+// graphCommitNodeIndex implements the CommitNodeIndex interface

+type graphCommitNodeIndex struct {

+	commitGraph commitgraph.Index

+	s           storer.EncodedObjectStorer

+}

+

+// objectCommitNode is a representation of Commit as presented in the GIT object format.

+//

+// objectCommitNode implements the CommitNode interface.

+type objectCommitNode struct {

+	commit *Commit

+}

+

+// objectCommitNodeIndex is an index that can load CommitNode objects only from the

+// object store.

+//

+// objectCommitNodeIndex implements the CommitNodeIndex interface

+type objectCommitNodeIndex struct {

+	s storer.EncodedObjectStorer

+}

+

+// ID returns the Commit object id referenced by the commit graph node.

+func (c *graphCommitNode) ID() plumbing.Hash {

+	return c.hash

+}

+

+// Tree returns the Tree referenced by the commit graph node.

+func (c *graphCommitNode) Tree() (*Tree, error) {

+	return GetTree(c.gci.s, c.node.TreeHash)

+}

+

+// CommitTime returns the Commiter.When time of the Commit referenced by the commit graph node.

+func (c *graphCommitNode) CommitTime() time.Time {

+	return c.node.When

+}

+

+func (c *graphCommitNode) String() string {

+	return fmt.Sprintf(

+		"%s %s\nDate:   %s",

+		plumbing.CommitObject, c.ID(),

+		c.CommitTime().Format(DateFormat),

+	)

+}

+

+func NewGraphCommitNodeIndex(commitGraph commitgraph.Index, s storer.EncodedObjectStorer) CommitNodeIndex {

+	return &graphCommitNodeIndex{commitGraph, s}

+}

+

+// NumParents returns the number of parents in a commit.

+func (gci *graphCommitNodeIndex) NumParents(node CommitNode) int {

+	if cgn, ok := node.(*graphCommitNode); ok {

+		return len(cgn.node.ParentIndexes)

+	}

+	co := node.(*objectCommitNode)

+	return co.commit.NumParents()

+}

+

+// ParentNodes return a CommitNodeIter for parents of specified node.

+func (gci *graphCommitNodeIndex) ParentNodes(node CommitNode) CommitNodeIter {

+	return newParentgraphCommitNodeIter(gci, node)

+}

+

+// ParentNode returns the ith parent of a commit.

+func (gci *graphCommitNodeIndex) ParentNode(node CommitNode, i int) (CommitNode, error) {

+	if cgn, ok := node.(*graphCommitNode); ok {

+		if len(cgn.node.ParentIndexes) == 0 || i >= len(cgn.node.ParentIndexes) {

+			return nil, ErrParentNotFound

+		}

+

+		parent, err := gci.commitGraph.GetNodeByIndex(cgn.node.ParentIndexes[i])

+		if err != nil {

+			return nil, err

+		}

+

+		return &graphCommitNode{

+			hash:  cgn.node.ParentHashes[i],

+			index: cgn.node.ParentIndexes[i],

+			node:  parent,

+			gci:   gci,

+		}, nil

+	}

+

+	co := node.(*objectCommitNode)

+	if len(co.commit.ParentHashes) == 0 || i >= len(co.commit.ParentHashes) {

+		return nil, ErrParentNotFound

+	}

+

+	parentHash := co.commit.ParentHashes[i]

+	return gci.Get(parentHash)

+}

+

+// ParentHashes returns hashes of the parent commits for a specified node

+func (gci *graphCommitNodeIndex) ParentHashes(node CommitNode) []plumbing.Hash {

+	if cgn, ok := node.(*graphCommitNode); ok {

+		return cgn.node.ParentHashes

+	}

+	co := node.(*objectCommitNode)

+	return co.commit.ParentHashes

+}

+

+// NodeFromHash looks up a commit node by it's object hash

+func (gci *graphCommitNodeIndex) Get(hash plumbing.Hash) (CommitNode, error) {

+	// Check the commit graph first

+	parentIndex, err := gci.commitGraph.GetIndexByHash(hash)

+	if err == nil {

+		parent, err := gci.commitGraph.GetNodeByIndex(parentIndex)

+		if err != nil {

+			return nil, err

+		}

+

+		return &graphCommitNode{

+			hash:  hash,

+			index: parentIndex,

+			node:  parent,

+			gci:   gci,

+		}, nil

+	}

+

+	// Fallback to loading full commit object

+	commit, err := GetCommit(gci.s, hash)

+	if err != nil {

+		return nil, err

+	}

+

+	return &objectCommitNode{commit: commit}, nil

+}

+

+// Commit returns the full Commit object representing the commit graph node.

+func (gci *graphCommitNodeIndex) Commit(node CommitNode) (*Commit, error) {

+	if cgn, ok := node.(*graphCommitNode); ok {

+		return GetCommit(gci.s, cgn.ID())

+	}

+	co := node.(*objectCommitNode)

+	return co.commit, nil

+}

+

+// CommitTime returns the time when the commit was performed.

+//

+// CommitTime is present to fulfill the CommitNode interface.

+func (c *objectCommitNode) CommitTime() time.Time {

+	return c.commit.Committer.When

+}

+

+// ID returns the Commit object id referenced by the node.

+func (c *objectCommitNode) ID() plumbing.Hash {

+	return c.commit.ID()

+}

+

+// Tree returns the Tree referenced by the node.

+func (c *objectCommitNode) Tree() (*Tree, error) {

+	return c.commit.Tree()

+}

+

+func NewObjectCommitNodeIndex(s storer.EncodedObjectStorer) CommitNodeIndex {

+	return &objectCommitNodeIndex{s}

+}

+

+// NumParents returns the number of parents in a commit.

+func (oci *objectCommitNodeIndex) NumParents(node CommitNode) int {

+	co := node.(*objectCommitNode)

+	return co.commit.NumParents()

+}

+

+// ParentNodes return a CommitNodeIter for parents of specified node.

+func (oci *objectCommitNodeIndex) ParentNodes(node CommitNode) CommitNodeIter {

+	return newParentgraphCommitNodeIter(oci, node)

+}

+

+// ParentNode returns the ith parent of a commit.

+func (oci *objectCommitNodeIndex) ParentNode(node CommitNode, i int) (CommitNode, error) {

+	co := node.(*objectCommitNode)

+	parent, err := co.commit.Parent(i)

+	if err != nil {

+		return nil, err

+	}

+	return &objectCommitNode{commit: parent}, nil

+}

+

+// ParentHashes returns hashes of the parent commits for a specified node

+func (oci *objectCommitNodeIndex) ParentHashes(node CommitNode) []plumbing.Hash {

+	co := node.(*objectCommitNode)

+	return co.commit.ParentHashes

+}

+

+// NodeFromHash looks up a commit node by it's object hash

+func (oci *objectCommitNodeIndex) Get(hash plumbing.Hash) (CommitNode, error) {

+	commit, err := GetCommit(oci.s, hash)

+	if err != nil {

+		return nil, err

+	}

+

+	return &objectCommitNode{commit: commit}, nil

+}

+

+// Commit returns the full Commit object representing the commit graph node.

+func (oci *objectCommitNodeIndex) Commit(node CommitNode) (*Commit, error) {

+	co := node.(*objectCommitNode)

+	return co.commit, nil

+}

+

+// parentCommitNodeIter provides an iterator for parent commits from associated CommitNodeIndex.

+type parentCommitNodeIter struct {

+	gci  CommitNodeIndex

+	node CommitNode

+	i    int

+}

+

+func newParentgraphCommitNodeIter(gci CommitNodeIndex, node CommitNode) CommitNodeIter {

+	return &parentCommitNodeIter{gci, node, 0}

+}

+

+// Next moves the iterator to the next commit and returns a pointer to it. If

+// there are no more commits, it returns io.EOF.

+func (iter *parentCommitNodeIter) Next() (CommitNode, error) {

+	obj, err := iter.gci.ParentNode(iter.node, iter.i)

+	if err == ErrParentNotFound {

+		return nil, io.EOF

+	}

+	if err == nil {

+		iter.i++

+	}

+

+	return obj, err

+}

+

+// ForEach call the cb function for each commit contained on this iter until

+// an error appends or the end of the iter is reached. If ErrStop is sent

+// the iteration is stopped but no error is returned. The iterator is closed.

+func (iter *parentCommitNodeIter) ForEach(cb func(CommitNode) error) error {

+	for {

+		obj, err := iter.Next()

+		if err != nil {

+			if err == io.EOF {

+				return nil

+			}

+

+			return err

+		}

+

+		if err := cb(obj); err != nil {

+			if err == storer.ErrStop {

+				return nil

+			}

+

+			return err

+		}

+	}

+}

+

+func (iter *parentCommitNodeIter) Close() {

+}