Expose Generation property on CommitNode

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

Author: filipnavara

plumbing/object/commitnode.go

@@ -25,6 +25,9 @@ type CommitNode interface {
 	ParentNode(i int) (CommitNode, error)
 	// ParentHashes returns hashes of the parent commits for a specified node
 	ParentHashes() []plumbing.Hash
+	// Generation returns the generation of the commit for reachability analysis.
+	// Objects with newer generation are not reachable from objects of older generation.
+	Generation() uint64
 	// Commit returns the full commit object from the node
 	Commit() (*Commit, error)
 }

plumbing/object/commitnode_graph.go

@@ -110,6 +110,13 @@ func (c *graphCommitNode) ParentHashes() []plumbing.Hash {
 	return c.commitData.ParentHashes
 }
 
+func (c *graphCommitNode) Generation() uint64 {
+	// If the commit-graph file was generated with older Git version that
+	// set the generation to zero for every commit the generation assumption
+	// is still valid. It is just less useful.
+	return uint64(c.commitData.Generation)
+}
+
 func (c *graphCommitNode) Commit() (*Commit, error) {
 	return GetCommit(c.gci.s, c.hash)
 }

plumbing/object/commitnode_object.go

@@ -1,6 +1,7 @@
 package object
 
 import (
+	"math"
 	"time"
 
 	"gopkg.in/src-d/go-git.v4/plumbing"
@@ -76,6 +77,13 @@ func (c *objectCommitNode) ParentHashes() []plumbing.Hash {
 	return c.commit.ParentHashes
 }
 
+func (c *objectCommitNode) Generation() uint64 {
+	// Commit nodes representing objects outside of the commit graph can never
+	// be reached by objects from the commit-graph thus we return the highest
+	// possible value.
+	return math.MaxUint64
+}
+
 func (c *objectCommitNode) Commit() (*Commit, error) {
 	return c.commit, nil
 }