pkg/commands/git_commands/stash_loader.go - master - githubmirror/lazygit

pkg/commands/git_commands/stash_loader.go
103 строки2 KB
Stefan Haller
Add hash field to models.StashEntry
25 авг 2025, 20:23
25 авг 2025, 20:23ee81a8e
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", "--pretty=%gd:%H|%ct|%gs").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++ {
		match := re.FindStringSubmatch(lines[i])
		if match == nil {
			continue
		}
		idx, err := strconv.Atoi(match[1])
		if err != nil {
			return self.getUnfilteredStashEntries()
		}
		currentStashEntry = stashEntryFromLine(match[2], idx)
		for i+1 < len(lines) && !isAStash(lines[i+1]) {
			i++
			if strings.HasPrefix(lines[i], filterPath) {
				stashEntries = append(stashEntries, currentStashEntry)
				continue outer
			}
		}
	}
	return stashEntries
}

func (self *StashLoader) getUnfilteredStashEntries() []*models.StashEntry {
	cmdArgs := NewGitCmd("stash").Arg("list", "-z", "--pretty=%H|%ct|%gs").ToArgv()

	rawString, _ := self.cmd.New(cmdArgs).DontLog().RunWithOutput()
	return lo.Map(utils.SplitNul(rawString), func(line string, index int) *models.StashEntry {
		return stashEntryFromLine(line, index)
	})
}

func stashEntryFromLine(line string, index int) *models.StashEntry {
	model := &models.StashEntry{
		Name:  line,
		Index: index,
	}

	hash, line, ok := strings.Cut(line, "|")
	if !ok {
		return model
	}
	model.Hash = hash

	tstr, msg, ok := strings.Cut(line, "|")
	if !ok {
		return model
	}

	t, err := strconv.ParseInt(tstr, 10, 64)
	if err != nil {
		return model
	}

	model.Name = msg
	model.Recency = utils.UnixToTimeAgo(t)

	return model
}