mirrors/lazygit
1
0
Fork 0
mirror of https://github.com/jesseduffield/lazygit.git synced 2025-05-12 21:05:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

get branches with git for-each-ref

Browse source
This commit is contained in:
Jesse Duffield 2019-11-17 10:04:06 +11:00
parent 6bd0979b4a
commit b7f2d0366b
2 changed files with 12 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

51
pkg/commands/loading_remotes.go
View file

@ -1,46 +1,17 @@
package commands
import (
"io/ioutil"
"os"
"path/filepath"
"strings"
"fmt"
"regexp"
)
func (c *GitCommand) GetBranchesFromDir(dirPath string) ([]*Branch, error) {
branches := []*Branch{}
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
if err != nil {
// it's possible that go-git is referencing a remote we don't have locally
// in which case we'll just swallow this error
c.Log.Warn(err)
return nil
}
if info.IsDir() {
return nil
}
// it's a file: we need to get the path and work out the branch name from that
fileContents, err := ioutil.ReadFile(path)
if err != nil {
return err
}
branches = append(branches, &Branch{
Name: strings.TrimPrefix(path, dirPath)[1:], // stripping prefix slash
Hash: strings.TrimSpace(string(fileContents)),
})
return nil
})
func (c *GitCommand) GetRemotes() ([]*Remote, error) {
// get remote branches
remoteBranchesStr, err := c.OSCommand.RunCommandWithOutput("git for-each-ref --format='%(refname:strip=2)' refs/remotes")
if err != nil {
return nil, err
}
return branches, nil
}
func (c *GitCommand) GetRemotes() ([]*Remote, error) {
goGitRemotes, err := c.Repo.Remotes()
if err != nil {
return nil, err
@ -51,11 +22,13 @@ func (c *GitCommand) GetRemotes() ([]*Remote, error) {
for i, goGitRemote := range goGitRemotes {
name := goGitRemote.Config().Name
// can't seem to easily get the branches of the remotes from go-git so we'll
// traverse the directory recursively
branches, err := c.GetBranchesFromDir(filepath.Join(".git", "refs", "remotes", name))
if err != nil {
return nil, err
re := regexp.MustCompile(fmt.Sprintf("%s\\/(.*)", name))
matches := re.FindAllStringSubmatch(remoteBranchesStr, -1)
branches := make([]*Branch, len(matches))
for j, match := range matches {
branches[j] = &Branch{
Name: match[1],
}
}
remotes[i] = &Remote{