mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2024-12-28 20:39:40 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(git): branch status as setter

Browse source
This commit is contained in:
Jan De Dobbeleer 2021-12-08 13:36:44 +01:00 committed by Jan De Dobbeleer
parent 04f841d1b3
commit 930fc2bfcc
2 changed files with 22 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/segment_git.go
View file

@ -182,7 +182,7 @@ func (g *git) string() string {
if displayStatus || statusColorsEnabled {
g.setGitStatus()
g.setGitHEADContext()
g.BranchStatus = g.getBranchStatus()
g.setBranchStatus()
} else {
g.Working = &GitStatus{}
g.Staging = &GitStatus{}
@ -224,7 +224,8 @@ func (g *git) init(props properties, env environmentInfo) {
g.env = env
}
func (g *git) getBranchStatus() string {
func (g *git) setBranchStatus() {
getBranchStatus := func() string {
if g.Ahead > 0 && g.Behind > 0 {
return fmt.Sprintf(" %s%d %s%d", g.props.getString(BranchAheadIcon, "\u2191"), g.Ahead, g.props.getString(BranchBehindIcon, "\u2193"), g.Behind)
}
@ -241,6 +242,8 @@ func (g *git) getBranchStatus() string {
return fmt.Sprintf(" %s", g.props.getString(BranchGoneIcon, "\u2262"))
}
return ""
}
g.BranchStatus = getBranchStatus()
}
func (g *git) getUpstreamIcon() string {

3
src/segment_git_test.go
View file

@ -486,7 +486,8 @@ func TestGetBranchStatus(t *testing.T) {
Behind: tc.Behind,
Upstream: tc.Upstream,
}
assert.Equal(t, tc.Expected, g.getBranchStatus(), tc.Case)
g.setBranchStatus()
assert.Equal(t, tc.Expected, g.BranchStatus, tc.Case)
}
}