mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-10 11:10:18 +02:00
e33fe37a99
We've been sometimes using lo and sometimes using my slices package, and we need to pick one for consistency. Lo is more extensive and better maintained so we're going with that. My slices package was a superset of go's own slices package so in some places I've just used the official one (the methods were just wrappers anyway). I've also moved the remaining methods into the utils package.
83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package git_commands
|
|
|
|
import (
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
|
"github.com/jesseduffield/lazygit/pkg/commands/oscommands"
|
|
"github.com/jesseduffield/lazygit/pkg/common"
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
type StashLoader struct {
|
|
*common.Common
|
|
cmd oscommands.ICmdObjBuilder
|
|
}
|
|
|
|
func NewStashLoader(
|
|
common *common.Common,
|
|
cmd oscommands.ICmdObjBuilder,
|
|
) *StashLoader {
|
|
return &StashLoader{
|
|
Common: common,
|
|
cmd: cmd,
|
|
}
|
|
}
|
|
|
|
func (self *StashLoader) GetStashEntries(filterPath string) []*models.StashEntry {
|
|
if filterPath == "" {
|
|
return self.getUnfilteredStashEntries()
|
|
}
|
|
|
|
cmdArgs := NewGitCmd("stash").Arg("list", "--name-only").ToArgv()
|
|
rawString, err := self.cmd.New(cmdArgs).DontLog().RunWithOutput()
|
|
if err != nil {
|
|
return self.getUnfilteredStashEntries()
|
|
}
|
|
stashEntries := []*models.StashEntry{}
|
|
var currentStashEntry *models.StashEntry
|
|
lines := utils.SplitLines(rawString)
|
|
isAStash := func(line string) bool { return strings.HasPrefix(line, "stash@{") }
|
|
re := regexp.MustCompile(`stash@\{(\d+)\}`)
|
|
|
|
outer:
|
|
for i := 0; i < len(lines); i++ {
|
|
if !isAStash(lines[i]) {
|
|
continue
|
|
}
|
|
match := re.FindStringSubmatch(lines[i])
|
|
idx, err := strconv.Atoi(match[1])
|
|
if err != nil {
|
|
return self.getUnfilteredStashEntries()
|
|
}
|
|
currentStashEntry = self.stashEntryFromLine(lines[i], idx)
|
|
for i+1 < len(lines) && !isAStash(lines[i+1]) {
|
|
i++
|
|
if lines[i] == filterPath {
|
|
stashEntries = append(stashEntries, currentStashEntry)
|
|
continue outer
|
|
}
|
|
}
|
|
}
|
|
return stashEntries
|
|
}
|
|
|
|
func (self *StashLoader) getUnfilteredStashEntries() []*models.StashEntry {
|
|
cmdArgs := NewGitCmd("stash").Arg("list", "-z", "--pretty=%gs").ToArgv()
|
|
|
|
rawString, _ := self.cmd.New(cmdArgs).DontLog().RunWithOutput()
|
|
return lo.Map(utils.SplitNul(rawString), func(line string, index int) *models.StashEntry {
|
|
return self.stashEntryFromLine(line, index)
|
|
})
|
|
}
|
|
|
|
func (c *StashLoader) stashEntryFromLine(line string, index int) *models.StashEntry {
|
|
return &models.StashEntry{
|
|
Name: line,
|
|
Index: index,
|
|
}
|
|
}
|