legit/git/git.go

123 lines
2.4 KiB
Go
Raw Normal View History

2022-12-11 05:52:47 +00:00
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"
)
2022-12-11 15:46:42 +00:00
type GitRepo struct {
r *git.Repository
h plumbing.Hash
2022-12-11 05:52:47 +00:00
}
2022-12-11 15:46:42 +00:00
func Open(path string, ref string) (*GitRepo, error) {
var err error
g := GitRepo{}
g.r, err = git.PlainOpen(path)
2022-12-11 05:52:47 +00:00
if err != nil {
2022-12-11 15:46:42 +00:00
return nil, fmt.Errorf("opening %s: %w", path, err)
2022-12-11 05:52:47 +00:00
}
2022-12-11 15:46:42 +00:00
if ref == "" {
head, err := g.r.Head()
if err != nil {
return nil, fmt.Errorf("getting head of %s: %w", path, err)
}
g.h = head.Hash()
2022-12-11 05:52:47 +00:00
} else {
2022-12-11 15:46:42 +00:00
hash, err := g.r.ResolveRevision(plumbing.Revision(ref))
2022-12-11 05:52:47 +00:00
if err != nil {
2022-12-11 15:46:42 +00:00
return nil, fmt.Errorf("resolving rev %s for %s: %w", ref, path, err)
2022-12-11 05:52:47 +00:00
}
2022-12-11 15:46:42 +00:00
g.h = *hash
}
return &g, nil
}
2022-12-11 05:52:47 +00:00
2022-12-11 15:46:42 +00:00
func (g *GitRepo) Commits() ([]*object.Commit, error) {
ci, err := g.r.Log(&git.LogOptions{From: g.h})
if err != nil {
return nil, fmt.Errorf("commits from ref: %w", err)
2022-12-11 05:52:47 +00:00
}
2022-12-11 15:46:42 +00:00
commits := []*object.Commit{}
ci.ForEach(func(c *object.Commit) error {
commits = append(commits, c)
return nil
})
return commits, nil
2022-12-11 05:52:47 +00:00
}
2022-12-12 16:03:09 +00:00
func (g *GitRepo) LastCommit() (*object.Commit, error) {
c, err := g.r.CommitObject(g.h)
if err != nil {
return nil, fmt.Errorf("last commit: %w", err)
}
return c, nil
}
2022-12-11 15:46:42 +00:00
func (g *GitRepo) FileContent(path string) (string, error) {
c, err := g.r.CommitObject(g.h)
2022-12-11 08:48:25 +00:00
if err != nil {
return "", fmt.Errorf("commit object: %w", err)
}
tree, err := c.Tree()
if err != nil {
return "", fmt.Errorf("file tree: %w", err)
}
file, err := tree.File(path)
if err != nil {
return "", err
}
return file.Contents()
}
2022-12-12 16:28:47 +00:00
func (g *GitRepo) Tags() ([]*object.Tag, error) {
ti, err := g.r.TagObjects()
if err != nil {
return nil, fmt.Errorf("tag objects: %w", err)
}
tags := []*object.Tag{}
_ = ti.ForEach(func(t *object.Tag) error {
tags = append(tags, t)
return nil
})
return tags, nil
}
func (g *GitRepo) Branches() ([]*plumbing.Reference, error) {
bi, err := g.r.Branches()
if err != nil {
return nil, fmt.Errorf("branchs: %w", err)
}
branches := []*plumbing.Reference{}
_ = bi.ForEach(func(ref *plumbing.Reference) error {
branches = append(branches, ref)
return nil
})
return branches, nil
}
2022-12-13 04:28:40 +00:00
func (g *GitRepo) FindMainBranch(branches []string) (string, error) {
for _, b := range branches {
_, err := g.r.ResolveRevision(plumbing.Revision(b))
if err == nil {
return b, nil
}
}
return "", fmt.Errorf("unable to find main branch")
}