116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
|
package avatar
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/base64"
|
|||
|
|
"fmt"
|
|||
|
|
"unicode/utf8"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 预定义一组好看的颜色
|
|||
|
|
var colors = []string{
|
|||
|
|
"#FF6B6B", "#4ECDC4", "#45B7D1", "#96CEB4",
|
|||
|
|
"#FFEAA7", "#DDA0DD", "#98D8C8", "#F7DC6F",
|
|||
|
|
"#BB8FCE", "#85C1E9", "#F8B500", "#00CED1",
|
|||
|
|
"#E74C3C", "#3498DB", "#2ECC71", "#9B59B6",
|
|||
|
|
"#1ABC9C", "#F39C12", "#E67E22", "#16A085",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SVG模板
|
|||
|
|
const svgTemplate = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 100 100" width="%d" height="%d">
|
|||
|
|
<rect width="100" height="100" fill="%s"/>
|
|||
|
|
<text x="50" y="50" font-family="Arial, sans-serif" font-size="40" font-weight="bold" fill="#ffffff" text-anchor="middle" dominant-baseline="central">%s</text>
|
|||
|
|
</svg>`
|
|||
|
|
|
|||
|
|
// GenerateSVGAvatar 根据用户名生成SVG头像
|
|||
|
|
// username: 用户名
|
|||
|
|
// size: 头像尺寸（像素）
|
|||
|
|
func GenerateSVGAvatar(username string, size int) string {
|
|||
|
|
initials := getInitials(username)
|
|||
|
|
color := stringToColor(username)
|
|||
|
|
return fmt.Sprintf(svgTemplate, size, size, color, initials)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateAvatarDataURI 生成Data URI格式的头像
|
|||
|
|
// 可以直接在HTML img标签或CSS background-image中使用
|
|||
|
|
func GenerateAvatarDataURI(username string, size int) string {
|
|||
|
|
svg := GenerateSVGAvatar(username, size)
|
|||
|
|
encoded := base64.StdEncoding.EncodeToString([]byte(svg))
|
|||
|
|
return fmt.Sprintf("data:image/svg+xml;base64,%s", encoded)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getInitials 获取用户名首字母
|
|||
|
|
// 中文取第一个字，英文取首字母（最多2个）
|
|||
|
|
func getInitials(username string) string {
|
|||
|
|
if username == "" {
|
|||
|
|
return "?"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查是否是中文字符
|
|||
|
|
firstRune, _ := utf8.DecodeRuneInString(username)
|
|||
|
|
if isChinese(firstRune) {
|
|||
|
|
// 中文直接返回第一个字符
|
|||
|
|
return string(firstRune)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 英文处理：取前两个单词的首字母
|
|||
|
|
// 例如: "John Doe" -> "JD", "john" -> "J"
|
|||
|
|
result := []rune{}
|
|||
|
|
for i, r := range username {
|
|||
|
|
if i == 0 {
|
|||
|
|
result = append(result, toUpper(r))
|
|||
|
|
} else if r == ' ' || r == '_' || r == '-' {
|
|||
|
|
// 找到下一个字符作为第二个首字母
|
|||
|
|
nextIdx := i + 1
|
|||
|
|
if nextIdx < len(username) {
|
|||
|
|
nextRune, _ := utf8.DecodeRuneInString(username[nextIdx:])
|
|||
|
|
if nextRune != utf8.RuneError && nextRune != ' ' {
|
|||
|
|
result = append(result, toUpper(nextRune))
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(result) == 0 {
|
|||
|
|
return "?"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 最多返回2个字符
|
|||
|
|
if len(result) > 2 {
|
|||
|
|
result = result[:2]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return string(result)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// isChinese 判断是否是中文字符
|
|||
|
|
func isChinese(r rune) bool {
|
|||
|
|
return r >= 0x4E00 && r <= 0x9FFF
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// toUpper 将字母转换为大写
|
|||
|
|
func toUpper(r rune) rune {
|
|||
|
|
if r >= 'a' && r <= 'z' {
|
|||
|
|
return r - 32
|
|||
|
|
}
|
|||
|
|
return r
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// stringToColor 根据字符串生成颜色
|
|||
|
|
// 使用简单的哈希算法确保同一用户名每次生成的颜色一致
|
|||
|
|
func stringToColor(s string) string {
|
|||
|
|
if s == "" {
|
|||
|
|
return colors[0]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
hash := 0
|
|||
|
|
for _, r := range s {
|
|||
|
|
hash = (hash*31 + int(r)) % len(colors)
|
|||
|
|
}
|
|||
|
|
if hash < 0 {
|
|||
|
|
hash = -hash
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return colors[hash%len(colors)]
|
|||
|
|
}
|