lan
4d8f2ec997
Set up project files and add .gitignore to exclude local build/runtime artifacts. Made-with: Cursor
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)]
|
||
}
|