fixup! Visualize local branch heads in commits panel

Author: stefanhaller

pkg/gui/presentation/commits.go

@@ -106,14 +106,14 @@ func GetCommitListDisplayStrings(
 	// 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 := lo.FilterMap(branches,
+	branchHeadsToVisualize := lo.SliceToMap(lo.FilterMap(branches,
 		func(b *models.Branch, index int) (string, bool) {
 			return 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)
-		})
+		}), func(s string) (string, struct{}) { return s, struct{}{} })
 
 	lines := make([][]string, 0, len(filteredCommits))
 	var bisectStatus BisectStatus
@@ -277,7 +277,7 @@ func getBisectStatusText(bisectStatus BisectStatus, bisectInfo *git_commands.Bis
 func displayCommit(
 	common *common.Common,
 	commit *models.Commit,
-	branchHeadsToVisualize []string,
+	branchHeadsToVisualize map[string]struct{},
 	cherryPickedCommitShaSet *set.Set[string],
 	diffName string,
 	timeFormat string,
@@ -307,7 +307,7 @@ func displayCommit(
 	} else {
 		if len(commit.Tags) > 0 {
 			tagString = theme.DiffTerminalColor.SetBold().Sprint(strings.Join(commit.Tags, " ")) + " "
-		} else if lo.Contains(branchHeadsToVisualize, commit.Sha) && commit.Status != models.StatusMerged {
+		} else if _, found := branchHeadsToVisualize[commit.Sha]; found && commit.Status != models.StatusMerged {
 			tagString = style.FgCyan.SetBold().Sprint(
 				lo.Ternary(icons.IsIconEnabled(), icons.BRANCH_ICON, "*") + " ")
 		}