107 lines
2.1 KiB
Go
107 lines
2.1 KiB
Go
package git
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/go-git/go-git/v5"
|
|
"github.com/go-git/go-git/v5/plumbing"
|
|
"github.com/go-git/go-git/v5/plumbing/object"
|
|
"github.com/go-git/go-git/v5/plumbing/storer"
|
|
)
|
|
|
|
const (
|
|
head = "HEAD"
|
|
|
|
defaultCommitCount = 10
|
|
)
|
|
|
|
// Repository is an abstraction for git-repository
|
|
type Repository interface {
|
|
Log(fromRev, toRev string) ([]Commit, error)
|
|
}
|
|
|
|
type repo struct {
|
|
r *git.Repository
|
|
}
|
|
|
|
type stopFn func(*object.Commit) error
|
|
|
|
// NewRepository return Repository from path
|
|
func NewRepository(path string) (Repository, error) {
|
|
r, err := git.PlainOpen(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &repo{
|
|
r: r,
|
|
}, nil
|
|
}
|
|
|
|
// Log return all commits between <from revision> and <to revision>
|
|
func (r *repo) Log(fromRev, toRev string) ([]Commit, error) {
|
|
if fromRev == "" {
|
|
fromRev = head
|
|
}
|
|
|
|
fromHash, err := r.r.ResolveRevision(plumbing.Revision(fromRev))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to resolve %s: %w", fromRev, err)
|
|
}
|
|
|
|
if toRev == "" {
|
|
return r.logWithStopFn(fromHash, nil, nil)
|
|
}
|
|
|
|
toHash, err := r.r.ResolveRevision(plumbing.Revision(toRev))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to resolve %s: %w", toRev, err)
|
|
}
|
|
|
|
return r.logWithStopFn(fromHash, nil, stopAtHash(toHash))
|
|
}
|
|
|
|
func (r *repo) logWithStopFn(fromHash *plumbing.Hash, beginStopFn, endStopFn stopFn) ([]Commit, error) {
|
|
cIter, err := r.r.Log(&git.LogOptions{
|
|
From: *fromHash,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to git log: %w", err)
|
|
}
|
|
|
|
commits := make([]Commit, 0, defaultCommitCount)
|
|
|
|
if err := cIter.ForEach(func(c *object.Commit) error {
|
|
if beginStopFn != nil {
|
|
if err := beginStopFn(c); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
commit := newCommit(c)
|
|
commits = append(commits, commit)
|
|
|
|
if endStopFn != nil {
|
|
if err := endStopFn(c); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}); err != nil {
|
|
return nil, fmt.Errorf("failed to iterate each git log: %w", err)
|
|
}
|
|
|
|
return commits, nil
|
|
}
|
|
|
|
func stopAtHash(hash *plumbing.Hash) stopFn {
|
|
return func(c *object.Commit) error {
|
|
if c.Hash == *hash {
|
|
return storer.ErrStop
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|