mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-01-26 05:37:18 +02:00
move author name colouring code into its own file
This commit is contained in:
parent
b04b457246
commit
7561703e8d
103
pkg/gui/presentation/authors/authors.go
Normal file
103
pkg/gui/presentation/authors/authors.go
Normal file
@ -0,0 +1,103 @@
|
|||||||
|
package authors
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/md5"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gookit/color"
|
||||||
|
"github.com/jesseduffield/lazygit/pkg/gui/style"
|
||||||
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
||||||
|
"github.com/lucasb-eyer/go-colorful"
|
||||||
|
"github.com/mattn/go-runewidth"
|
||||||
|
)
|
||||||
|
|
||||||
|
var authorInitialCache = make(map[string]string)
|
||||||
|
var authorNameCache = make(map[string]string)
|
||||||
|
var authorStyleCache = make(map[string]style.TextStyle)
|
||||||
|
|
||||||
|
func ShortAuthor(authorName string) string {
|
||||||
|
if value, ok := authorInitialCache[authorName]; ok {
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
initials := getInitials(authorName)
|
||||||
|
if initials == "" {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
value := AuthorStyle(authorName).Sprint(initials)
|
||||||
|
authorInitialCache[authorName] = value
|
||||||
|
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func LongAuthor(authorName string) string {
|
||||||
|
if value, ok := authorNameCache[authorName]; ok {
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
paddedAuthorName := utils.WithPadding(authorName, 17)
|
||||||
|
truncatedName := utils.TruncateWithEllipsis(paddedAuthorName, 17)
|
||||||
|
value := AuthorStyle(authorName).Sprint(truncatedName)
|
||||||
|
authorNameCache[authorName] = value
|
||||||
|
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func AuthorStyle(authorName string) style.TextStyle {
|
||||||
|
if value, ok := authorStyleCache[authorName]; ok {
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
value := trueColorStyle(authorName)
|
||||||
|
|
||||||
|
authorStyleCache[authorName] = value
|
||||||
|
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func trueColorStyle(str string) style.TextStyle {
|
||||||
|
hash := md5.Sum([]byte(str))
|
||||||
|
c := colorful.Hsl(randFloat(hash[0:4])*360.0, 0.6+0.4*randFloat(hash[4:8]), 0.4+randFloat(hash[8:12])*0.2)
|
||||||
|
|
||||||
|
return style.New().SetFg(style.NewRGBColor(color.RGB(uint8(c.R*255), uint8(c.G*255), uint8(c.B*255))))
|
||||||
|
}
|
||||||
|
|
||||||
|
func randFloat(hash []byte) float64 {
|
||||||
|
return float64(randInt(hash, 100)) / 100
|
||||||
|
}
|
||||||
|
|
||||||
|
func randInt(hash []byte, max int) int {
|
||||||
|
sum := 0
|
||||||
|
for _, b := range hash {
|
||||||
|
sum = (sum + int(b)) % max
|
||||||
|
}
|
||||||
|
return sum
|
||||||
|
}
|
||||||
|
|
||||||
|
func getInitials(authorName string) string {
|
||||||
|
if authorName == "" {
|
||||||
|
return authorName
|
||||||
|
}
|
||||||
|
|
||||||
|
firstRune := getFirstRune(authorName)
|
||||||
|
if runewidth.RuneWidth(firstRune) > 1 {
|
||||||
|
return string(firstRune)
|
||||||
|
}
|
||||||
|
|
||||||
|
split := strings.Split(authorName, " ")
|
||||||
|
if len(split) == 1 {
|
||||||
|
return utils.LimitStr(authorName, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
return utils.LimitStr(split[0], 1) + utils.LimitStr(split[1], 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getFirstRune(str string) rune {
|
||||||
|
// just using the loop for the sake of getting the first rune
|
||||||
|
for _, r := range str {
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
// should never land here
|
||||||
|
return 0
|
||||||
|
}
|
@ -1,4 +1,4 @@
|
|||||||
package presentation
|
package authors
|
||||||
|
|
||||||
import "testing"
|
import "testing"
|
||||||
|
|
@ -1,17 +1,14 @@
|
|||||||
package presentation
|
package presentation
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/md5"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/gookit/color"
|
|
||||||
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
"github.com/jesseduffield/lazygit/pkg/commands/models"
|
||||||
|
"github.com/jesseduffield/lazygit/pkg/gui/presentation/authors"
|
||||||
"github.com/jesseduffield/lazygit/pkg/gui/style"
|
"github.com/jesseduffield/lazygit/pkg/gui/style"
|
||||||
"github.com/jesseduffield/lazygit/pkg/theme"
|
"github.com/jesseduffield/lazygit/pkg/theme"
|
||||||
"github.com/jesseduffield/lazygit/pkg/utils"
|
"github.com/jesseduffield/lazygit/pkg/utils"
|
||||||
"github.com/kyokomi/emoji/v2"
|
"github.com/kyokomi/emoji/v2"
|
||||||
colorful "github.com/lucasb-eyer/go-colorful"
|
|
||||||
"github.com/mattn/go-runewidth"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func GetCommitListDisplayStrings(commits []*models.Commit, fullDescription bool, cherryPickedCommitShaMap map[string]bool, diffName string, parseEmoji bool) [][]string {
|
func GetCommitListDisplayStrings(commits []*models.Commit, fullDescription bool, cherryPickedCommitShaMap map[string]bool, diffName string, parseEmoji bool) [][]string {
|
||||||
@ -72,7 +69,7 @@ func getFullDescriptionDisplayStringsForCommit(c *models.Commit, cherryPickedCom
|
|||||||
return []string{
|
return []string{
|
||||||
shaColor.Sprint(c.ShortSha()),
|
shaColor.Sprint(c.ShortSha()),
|
||||||
secondColumnString,
|
secondColumnString,
|
||||||
longAuthor(c.Author),
|
authors.LongAuthor(c.Author),
|
||||||
tagString + theme.DefaultTextColor.Sprint(name),
|
tagString + theme.DefaultTextColor.Sprint(name),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -116,84 +113,11 @@ func getDisplayStringsForCommit(c *models.Commit, cherryPickedCommitShaMap map[s
|
|||||||
|
|
||||||
return []string{
|
return []string{
|
||||||
shaColor.Sprint(c.ShortSha()),
|
shaColor.Sprint(c.ShortSha()),
|
||||||
shortAuthor(c.Author),
|
authors.ShortAuthor(c.Author),
|
||||||
actionString + tagString + theme.DefaultTextColor.Sprint(name),
|
actionString + tagString + theme.DefaultTextColor.Sprint(name),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var authorInitialCache = make(map[string]string)
|
|
||||||
var authorNameCache = make(map[string]string)
|
|
||||||
|
|
||||||
func shortAuthor(authorName string) string {
|
|
||||||
if value, ok := authorInitialCache[authorName]; ok {
|
|
||||||
return value
|
|
||||||
}
|
|
||||||
|
|
||||||
initials := getInitials(authorName)
|
|
||||||
if initials == "" {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
value := authorColor(authorName).Sprint(initials)
|
|
||||||
authorInitialCache[authorName] = value
|
|
||||||
|
|
||||||
return value
|
|
||||||
}
|
|
||||||
|
|
||||||
func longAuthor(authorName string) string {
|
|
||||||
if value, ok := authorNameCache[authorName]; ok {
|
|
||||||
return value
|
|
||||||
}
|
|
||||||
|
|
||||||
truncatedName := utils.TruncateWithEllipsis(authorName, 17)
|
|
||||||
value := authorColor(authorName).Sprint(truncatedName)
|
|
||||||
authorNameCache[authorName] = value
|
|
||||||
|
|
||||||
return value
|
|
||||||
}
|
|
||||||
|
|
||||||
func authorColor(authorName string) style.TextStyle {
|
|
||||||
hash := md5.Sum([]byte(authorName))
|
|
||||||
c := colorful.Hsl(randFloat(hash[0:4])*360.0, 0.6+0.4*randFloat(hash[4:8]), 0.4+randFloat(hash[8:12])*0.2)
|
|
||||||
|
|
||||||
return style.New().SetFg(style.NewRGBColor(color.RGB(uint8(c.R*255), uint8(c.G*255), uint8(c.B*255))))
|
|
||||||
}
|
|
||||||
|
|
||||||
func randFloat(hash []byte) float64 {
|
|
||||||
sum := 0
|
|
||||||
for _, b := range hash {
|
|
||||||
sum = (sum + int(b)) % 100
|
|
||||||
}
|
|
||||||
return float64(sum) / 100
|
|
||||||
}
|
|
||||||
|
|
||||||
func getInitials(authorName string) string {
|
|
||||||
if authorName == "" {
|
|
||||||
return authorName
|
|
||||||
}
|
|
||||||
|
|
||||||
firstRune := getFirstRune(authorName)
|
|
||||||
if runewidth.RuneWidth(firstRune) > 1 {
|
|
||||||
return string(firstRune)
|
|
||||||
}
|
|
||||||
|
|
||||||
split := strings.Split(authorName, " ")
|
|
||||||
if len(split) == 1 {
|
|
||||||
return utils.LimitStr(authorName, 2)
|
|
||||||
}
|
|
||||||
|
|
||||||
return utils.LimitStr(split[0], 1) + utils.LimitStr(split[1], 1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func getFirstRune(str string) rune {
|
|
||||||
// just using the loop for the sake of getting the first rune
|
|
||||||
for _, r := range str {
|
|
||||||
return r
|
|
||||||
}
|
|
||||||
// should never land here
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
func actionColorMap(str string) style.TextStyle {
|
func actionColorMap(str string) style.TextStyle {
|
||||||
switch str {
|
switch str {
|
||||||
case "pick":
|
case "pick":
|
||||||
|
Loading…
x
Reference in New Issue
Block a user