68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package utils
|
|
|
|
import (
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"unicode"
|
|
)
|
|
|
|
// GenerateSlug 生成URL友好的slug
|
|
func GenerateSlug(text string) string {
|
|
// 转换为小写
|
|
text = strings.ToLower(text)
|
|
|
|
// 移除重音字符
|
|
text = removeAccents(text)
|
|
|
|
// 替换空格和特殊字符为连字符
|
|
reg := regexp.MustCompile(`[^\p{L}\p{N}]+`)
|
|
text = reg.ReplaceAllString(text, "-")
|
|
|
|
// 移除首尾的连字符
|
|
text = strings.Trim(text, "-")
|
|
|
|
// 移除连续的连字符
|
|
reg = regexp.MustCompile(`-+`)
|
|
text = reg.ReplaceAllString(text, "-")
|
|
|
|
return text
|
|
}
|
|
|
|
// removeAccents 移除重音字符的转换函数
|
|
func removeAccents(text string) string {
|
|
var result strings.Builder
|
|
for _, r := range text {
|
|
if !unicode.Is(unicode.Mn, r) {
|
|
result.WriteRune(r)
|
|
}
|
|
}
|
|
return result.String()
|
|
}
|
|
|
|
// TruncateString 截断字符串到指定长度
|
|
func TruncateString(s string, length int) string {
|
|
if len(s) <= length {
|
|
return s
|
|
}
|
|
return s[:length]
|
|
}
|
|
|
|
// GenerateUniqueSlug 生成唯一的slug
|
|
func GenerateUniqueSlug(base string, existingCheck func(string) bool) string {
|
|
slug := GenerateSlug(base)
|
|
if !existingCheck(slug) {
|
|
return slug
|
|
}
|
|
|
|
// 如果存在重复，添加数字后缀
|
|
for i := 1; i <= 1000; i++ {
|
|
candidateSlug := slug + "-" + strconv.Itoa(i)
|
|
if !existingCheck(candidateSlug) {
|
|
return candidateSlug
|
|
}
|
|
}
|
|
|
|
// 如果还是重复，使用时间戳
|
|
return slug + "-" + GenerateRandomString(6)
|
|
} |