mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-12 11:15:00 +02:00
82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
package presentation
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
|
"github.com/jesseduffield/lazygit/pkg/theme"
|
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
|
)
|
|
|
|
func GetBranchListDisplayStrings(branches []*models.Branch, fullDescription bool, diffName string) [][]string {
|
|
lines := make([][]string, len(branches))
|
|
|
|
for i := range branches {
|
|
diffed := branches[i].Name == diffName
|
|
lines[i] = getBranchDisplayStrings(branches[i], fullDescription, diffed)
|
|
}
|
|
|
|
return lines
|
|
}
|
|
|
|
// getBranchDisplayStrings returns the display string of branch
|
|
func getBranchDisplayStrings(b *models.Branch, fullDescription bool, diffed bool) []string {
|
|
displayName := b.Name
|
|
if b.DisplayName != "" {
|
|
displayName = b.DisplayName
|
|
}
|
|
|
|
nameColorAttr := GetBranchColor(b.Name)
|
|
if diffed {
|
|
nameColorAttr = theme.DiffTerminalColor
|
|
}
|
|
coloredName := utils.ColoredString(displayName, nameColorAttr)
|
|
if b.IsTrackingRemote() {
|
|
coloredName = fmt.Sprintf("%s %s", coloredName, ColoredBranchStatus(b))
|
|
}
|
|
|
|
recencyColor := color.FgCyan
|
|
if b.Recency == " *" {
|
|
recencyColor = color.FgGreen
|
|
}
|
|
|
|
if fullDescription {
|
|
return []string{utils.ColoredString(b.Recency, recencyColor), coloredName, utils.ColoredString(b.UpstreamName, color.FgYellow)}
|
|
}
|
|
|
|
return []string{utils.ColoredString(b.Recency, recencyColor), coloredName}
|
|
}
|
|
|
|
// GetBranchColor branch color
|
|
func GetBranchColor(name string) color.Attribute {
|
|
branchType := strings.Split(name, "/")[0]
|
|
|
|
switch branchType {
|
|
case "feature":
|
|
return color.FgGreen
|
|
case "bugfix":
|
|
return color.FgYellow
|
|
case "hotfix":
|
|
return color.FgRed
|
|
default:
|
|
return theme.DefaultTextColor
|
|
}
|
|
}
|
|
|
|
func ColoredBranchStatus(branch *models.Branch) string {
|
|
colour := color.FgYellow
|
|
if branch.MatchesUpstream() {
|
|
colour = color.FgGreen
|
|
} else if !branch.IsTrackingRemote() {
|
|
colour = color.FgRed
|
|
}
|
|
|
|
return utils.ColoredString(BranchStatus(branch), colour)
|
|
}
|
|
|
|
func BranchStatus(branch *models.Branch) string {
|
|
return fmt.Sprintf("↑%s↓%s", branch.Pushables, branch.Pullables)
|
|
}
|