mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-04-24 02:05:34 +03:00
95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
package git_commands
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
|
)
|
|
|
|
type StatusCommands struct {
|
|
*GitCommon
|
|
}
|
|
|
|
func NewStatusCommands(
|
|
gitCommon *GitCommon,
|
|
) *StatusCommands {
|
|
return &StatusCommands{
|
|
GitCommon: gitCommon,
|
|
}
|
|
}
|
|
|
|
func (self *StatusCommands) WorkingTreeState() models.WorkingTreeState {
|
|
result := models.WorkingTreeState{}
|
|
result.Rebasing, _ = self.IsInRebase()
|
|
result.Merging, _ = self.IsInMergeState()
|
|
result.CherryPicking, _ = self.IsInCherryPick()
|
|
result.Reverting, _ = self.IsInRevert()
|
|
return result
|
|
}
|
|
|
|
func (self *StatusCommands) IsBareRepo() bool {
|
|
return self.repoPaths.isBareRepo
|
|
}
|
|
|
|
func (self *StatusCommands) IsInRebase() (bool, error) {
|
|
exists, err := self.os.FileExists(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "rebase-merge"))
|
|
if err == nil && exists {
|
|
return true, nil
|
|
}
|
|
return self.os.FileExists(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "rebase-apply"))
|
|
}
|
|
|
|
// IsInMergeState states whether we are still mid-merge
|
|
func (self *StatusCommands) IsInMergeState() (bool, error) {
|
|
return self.os.FileExists(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "MERGE_HEAD"))
|
|
}
|
|
|
|
func (self *StatusCommands) IsInCherryPick() (bool, error) {
|
|
exists, err := self.os.FileExists(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "CHERRY_PICK_HEAD"))
|
|
if err != nil || !exists {
|
|
return exists, err
|
|
}
|
|
// Sometimes, CHERRY_PICK_HEAD is present during rebases even if no
|
|
// cherry-pick is in progress. I suppose this is because rebase used to be
|
|
// implemented as a series of cherry-picks, so this could be remnants of
|
|
// code that is shared between cherry-pick and rebase, or something. The way
|
|
// to tell if this is the case is to check for the presence of the
|
|
// stopped-sha file, which records the sha of the last pick that was
|
|
// executed before the rebase stopped, and seeing if the sha in that file is
|
|
// the same as the one in CHERRY_PICK_HEAD.
|
|
cherryPickHead, err := os.ReadFile(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "CHERRY_PICK_HEAD"))
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
stoppedSha, err := os.ReadFile(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "rebase-merge", "stopped-sha"))
|
|
if err != nil {
|
|
// If we get an error we assume the file doesn't exist
|
|
return true, nil
|
|
}
|
|
cherryPickHeadStr := strings.TrimSpace(string(cherryPickHead))
|
|
stoppedShaStr := strings.TrimSpace(string(stoppedSha))
|
|
// Need to use HasPrefix here because the cherry-pick HEAD is a full sha1,
|
|
// but stopped-sha is an abbreviated sha1
|
|
if strings.HasPrefix(cherryPickHeadStr, stoppedShaStr) {
|
|
return false, nil
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
func (self *StatusCommands) IsInRevert() (bool, error) {
|
|
return self.os.FileExists(filepath.Join(self.repoPaths.WorktreeGitDirPath(), "REVERT_HEAD"))
|
|
}
|
|
|
|
// Full ref (e.g. "refs/heads/mybranch") of the branch that is currently
|
|
// being rebased, or empty string when we're not in a rebase
|
|
func (self *StatusCommands) BranchBeingRebased() string {
|
|
for _, dir := range []string{"rebase-merge", "rebase-apply"} {
|
|
if bytesContent, err := os.ReadFile(filepath.Join(self.repoPaths.WorktreeGitDirPath(), dir, "head-name")); err == nil {
|
|
return strings.TrimSpace(string(bytesContent))
|
|
}
|
|
}
|
|
return ""
|
|
}
|