2021-12-30 08:36:21 +02:00
|
|
|
package loaders
|
2019-11-16 07:00:47 +02:00
|
|
|
|
|
|
|
import (
|
2019-11-17 01:04:06 +02:00
|
|
|
"fmt"
|
|
|
|
"regexp"
|
2019-11-17 01:23:06 +02:00
|
|
|
"strings"
|
2020-09-29 10:41:06 +02:00
|
|
|
|
2022-03-19 06:36:46 +02:00
|
|
|
"github.com/jesseduffield/generics/slices"
|
2021-12-30 08:36:21 +02:00
|
|
|
gogit "github.com/jesseduffield/go-git/v5"
|
2020-09-29 12:28:39 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
2021-12-30 08:36:21 +02:00
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands/oscommands"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/common"
|
2019-11-16 07:00:47 +02:00
|
|
|
)
|
|
|
|
|
2021-12-30 08:36:21 +02:00
|
|
|
type RemoteLoader struct {
|
|
|
|
*common.Common
|
|
|
|
cmd oscommands.ICmdObjBuilder
|
|
|
|
getGoGitRemotes func() ([]*gogit.Remote, error)
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewRemoteLoader(
|
|
|
|
common *common.Common,
|
|
|
|
cmd oscommands.ICmdObjBuilder,
|
|
|
|
getGoGitRemotes func() ([]*gogit.Remote, error),
|
|
|
|
) *RemoteLoader {
|
|
|
|
return &RemoteLoader{
|
|
|
|
Common: common,
|
|
|
|
cmd: cmd,
|
|
|
|
getGoGitRemotes: getGoGitRemotes,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *RemoteLoader) GetRemotes() ([]*models.Remote, error) {
|
2022-01-05 02:57:32 +02:00
|
|
|
remoteBranchesStr, err := self.cmd.New("git branch -r").DontLog().RunWithOutput()
|
2019-11-16 07:00:47 +02:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2021-12-30 08:36:21 +02:00
|
|
|
goGitRemotes, err := self.getGoGitRemotes()
|
2019-11-16 07:00:47 +02:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// first step is to get our remotes from go-git
|
2022-03-19 10:51:48 +02:00
|
|
|
remotes := slices.Map(goGitRemotes, func(goGitRemote *gogit.Remote) *models.Remote {
|
2019-11-17 03:07:36 +02:00
|
|
|
remoteName := goGitRemote.Config().Name
|
2019-11-16 07:00:47 +02:00
|
|
|
|
2021-12-23 09:35:29 +02:00
|
|
|
re := regexp.MustCompile(fmt.Sprintf(`(?m)^\s*%s\/([\S]+)`, remoteName))
|
2019-11-17 01:04:06 +02:00
|
|
|
matches := re.FindAllStringSubmatch(remoteBranchesStr, -1)
|
2022-03-19 10:51:48 +02:00
|
|
|
branches := slices.Map(matches, func(match []string) *models.RemoteBranch {
|
2022-03-19 06:36:46 +02:00
|
|
|
return &models.RemoteBranch{
|
2019-11-17 03:07:36 +02:00
|
|
|
Name: match[1],
|
|
|
|
RemoteName: remoteName,
|
2019-11-17 01:04:06 +02:00
|
|
|
}
|
2022-03-19 06:36:46 +02:00
|
|
|
})
|
2019-11-16 07:00:47 +02:00
|
|
|
|
2022-03-19 06:36:46 +02:00
|
|
|
return &models.Remote{
|
2019-11-16 07:00:47 +02:00
|
|
|
Name: goGitRemote.Config().Name,
|
|
|
|
Urls: goGitRemote.Config().URLs,
|
|
|
|
Branches: branches,
|
|
|
|
}
|
2022-03-19 06:36:46 +02:00
|
|
|
})
|
2019-11-16 07:00:47 +02:00
|
|
|
|
2019-11-17 01:23:06 +02:00
|
|
|
// now lets sort our remotes by name alphabetically
|
2022-03-19 06:36:46 +02:00
|
|
|
slices.SortFunc(remotes, func(a, b *models.Remote) bool {
|
2019-11-17 01:23:06 +02:00
|
|
|
// we want origin at the top because we'll be most likely to want it
|
2022-03-19 06:36:46 +02:00
|
|
|
if a.Name == "origin" {
|
2019-11-17 01:23:06 +02:00
|
|
|
return true
|
|
|
|
}
|
2022-03-19 06:36:46 +02:00
|
|
|
if b.Name == "origin" {
|
2019-11-17 01:23:06 +02:00
|
|
|
return false
|
|
|
|
}
|
2022-03-19 06:36:46 +02:00
|
|
|
return strings.ToLower(a.Name) < strings.ToLower(b.Name)
|
2019-11-17 01:23:06 +02:00
|
|
|
})
|
|
|
|
|
2019-11-16 07:00:47 +02:00
|
|
|
return remotes, nil
|
|
|
|
}
|