diff options
Diffstat (limited to 'pkg/gui/presentation')
-rw-r--r-- | pkg/gui/presentation/commits.go | 27 | ||||
-rw-r--r-- | pkg/gui/presentation/commits_test.go | 71 |
2 files changed, 98 insertions, 0 deletions
diff --git a/pkg/gui/presentation/commits.go b/pkg/gui/presentation/commits.go index 53d3aaf0d..4affe6a57 100644 --- a/pkg/gui/presentation/commits.go +++ b/pkg/gui/presentation/commits.go @@ -39,6 +39,8 @@ type bisectBounds struct { func GetCommitListDisplayStrings( common *common.Common, commits []*models.Commit, + branches []*models.Branch, + currentBranchName string, fullDescription bool, cherryPickedCommitShaSet *set.Set[string], diffName string, @@ -99,6 +101,24 @@ func GetCommitListDisplayStrings( getGraphLine = func(idx int) string { return "" } } + // Determine the hashes of the local branches for which we want to show a + // branch marker in the commits list. We only want to do this for branches + // that are not the current branch, and not any of the main branches. The + // goal is to visualize stacks of local branches, so anything that doesn't + // contribute to a branch stack shouldn't show a marker. + branchHeadsToVisualize := set.NewFromSlice(lo.FilterMap(branches, + func(b *models.Branch, index int) (string, bool) { + return b.CommitHash, + // Don't consider branches that don't have a commit hash. As far + // as I can see, this happens for a detached head, so filter + // these out + b.CommitHash != "" && + // Don't show a marker for the current branch + b.Name != currentBranchName && + // Don't show a marker for main branches + !lo.Contains(common.UserConfig.Git.MainBranches, b.Name) + })) + lines := make([][]string, 0, len(filteredCommits)) var bisectStatus BisectStatus for i, commit := range filteredCommits { @@ -112,6 +132,7 @@ func GetCommitListDisplayStrings( lines = append(lines, displayCommit( common, commit, + branchHeadsToVisualize, cherryPickedCommitShaSet, diffName, timeFormat, @@ -260,6 +281,7 @@ func getBisectStatusText(bisectStatus BisectStatus, bisectInfo *git_commands.Bis func displayCommit( common *common.Common, commit *models.Commit, + branchHeadsToVisualize *set.Set[string], cherryPickedCommitShaSet *set.Set[string], diffName string, timeFormat string, @@ -290,6 +312,11 @@ func displayCommit( if len(commit.Tags) > 0 { tagString = theme.DiffTerminalColor.SetBold().Sprint(strings.Join(commit.Tags, " ")) + " " } + + if branchHeadsToVisualize.Includes(commit.Sha) && commit.Status != models.StatusMerged { + tagString = style.FgCyan.SetBold().Sprint( + lo.Ternary(icons.IsIconEnabled(), icons.BRANCH_ICON, "*") + " " + tagString) + } } name := commit.Name diff --git a/pkg/gui/presentation/commits_test.go b/pkg/gui/presentation/commits_test.go index 31c6c813d..b6684f80e 100644 --- a/pkg/gui/presentation/commits_test.go +++ b/pkg/gui/presentation/commits_test.go @@ -28,6 +28,8 @@ func TestGetCommitListDisplayStrings(t *testing.T) { scenarios := []struct { testName string commits []*models.Commit + branches []*models.Branch + currentBranchName string fullDescription bool cherryPickedCommitShaSet *set.Set[string] diffName string @@ -73,6 +75,73 @@ func TestGetCommitListDisplayStrings(t *testing.T) { `), }, { + testName: "commit with tags", + commits: []*models.Commit{ + {Name: "commit1", Sha: "sha1", Tags: []string{"tag1", "tag2"}}, + {Name: "commit2", Sha: "sha2"}, + }, + startIdx: 0, + length: 2, + showGraph: false, + bisectInfo: git_commands.NewNullBisectInfo(), + cherryPickedCommitShaSet: set.New[string](), + now: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC), + expected: formatExpected(` + sha1 tag1 tag2 commit1 + sha2 commit2 + `), + }, + { + testName: "show local branch head, except the current branch, main branches, or merged branches", + commits: []*models.Commit{ + {Name: "commit1", Sha: "sha1"}, + {Name: "commit2", Sha: "sha2"}, + {Name: "commit3", Sha: "sha3"}, + {Name: "commit4", Sha: "sha4", Status: models.StatusMerged}, + }, + branches: []*models.Branch{ + {Name: "current-branch", CommitHash: "sha1", Head: true}, + {Name: "other-branch", CommitHash: "sha2", Head: false}, + {Name: "master", CommitHash: "sha3", Head: false}, + {Name: "old-branch", CommitHash: "sha4", Head: false}, + }, + currentBranchName: "current-branch", + startIdx: 0, + length: 4, + showGraph: false, + bisectInfo: git_commands.NewNullBisectInfo(), + cherryPickedCommitShaSet: set.New[string](), + now: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC), + expected: formatExpected(` + sha1 commit1 + sha2 * commit2 + sha3 commit3 + sha4 commit4 + `), + }, + { + testName: "show local branch head and tag if both exist", + commits: []*models.Commit{ + {Name: "commit1", Sha: "sha1"}, + {Name: "commit2", Sha: "sha2", Tags: []string{"some-tag"}}, + {Name: "commit3", Sha: "sha3"}, + }, + branches: []*models.Branch{ + {Name: "some-branch", CommitHash: "sha2"}, + }, + startIdx: 0, + length: 3, + showGraph: false, + bisectInfo: git_commands.NewNullBisectInfo(), + cherryPickedCommitShaSet: set.New[string](), + now: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC), + expected: formatExpected(` + sha1 commit1 + sha2 * some-tag commit2 + sha3 commit3 + `), + }, + { testName: "showing graph", commits: []*models.Commit{ {Name: "commit1", Sha: "sha1", Parents: []string{"sha2", "sha3"}}, @@ -285,6 +354,8 @@ func TestGetCommitListDisplayStrings(t *testing.T) { result := GetCommitListDisplayStrings( common, s.commits, + s.branches, + s.currentBranchName, s.fullDescription, s.cherryPickedCommitShaSet, s.diffName, |