6
0
mirror of https://github.com/grdl/git-get.git synced 2026-02-04 17:24:49 +00:00

Add a basic branch iterator for getting branches state

This commit is contained in:
Grzegorz Dlugoszewski
2020-05-18 17:35:32 +02:00
parent 13f69bb4b5
commit 4ce4381c57
3 changed files with 59 additions and 5 deletions

36
branch.go Normal file
View File

@@ -0,0 +1,36 @@
package main
import (
"fmt"
git "github.com/libgit2/git2go/v30"
"github.com/pkg/errors"
)
type BranchStatus struct {
Name string
IsRemote bool
HasUpstream bool
NeedsPull bool
NeedsPush bool
}
func Branches(repo *git.Repository) ([]*git.Branch, error) {
it, err := repo.NewBranchIterator(git.BranchAll)
if err != nil {
return nil, errors.Wrap(err, "Failed creating branch iterator")
}
it.ForEach(func(branch *git.Branch, branchType git.BranchType) error {
fmt.Print(branch.IsRemote())
upstream, err := branch.Upstream()
if err != nil {
fmt.Println(err.Error())
} else {
fmt.Println(upstream.Name())
}
return nil
})
return nil, nil
}

18
branch_test.go Normal file
View File

@@ -0,0 +1,18 @@
package main
import (
"fmt"
"testing"
git "github.com/libgit2/git2go/v30"
)
func TestBranches(t *testing.T) {
repo, err := git.OpenRepository("/home/grdl/workspace/gitlab.com/grdl/git-get")
checkFatal(t, err)
branches, err := Branches(repo)
checkFatal(t, err)
fmt.Println(len(branches))
}

View File

@@ -17,12 +17,12 @@ const (
func GetStatus(path string) ([]RepoStatus, error) {
repo, err := git.OpenRepository(path)
if err != nil {
return nil, errors.Wrap(err, "failed opening repository")
return nil, errors.Wrap(err, "Failed opening repository")
}
entries, err := statusEntries(repo)
if err != nil {
return nil, errors.Wrap(err, "failed opening repository")
return nil, errors.Wrap(err, "Failed opening repository")
}
statusSet := make(map[RepoStatus]bool)
@@ -59,19 +59,19 @@ func statusEntries(repo *git.Repository) ([]git.StatusEntry, error) {
status, err := repo.StatusList(opts)
if err != nil {
return nil, errors.Wrap(err, "failed getting repository status")
return nil, errors.Wrap(err, "Failed getting repository status")
}
entryCount, err := status.EntryCount()
if err != nil {
return nil, errors.Wrap(err, "failed getting repository status count")
return nil, errors.Wrap(err, "Failed getting repository status count")
}
var entries []git.StatusEntry
for i := 0; i < entryCount; i++ {
entry, err := status.ByIndex(i)
if err != nil {
return nil, errors.Wrap(err, "failed getting repository status entry")
return nil, errors.Wrap(err, "Failed getting repository status entry")
}
entries = append(entries, entry)