128 lines
3.1 KiB
Go
128 lines
3.1 KiB
Go
package utils
|
||
|
||
import (
|
||
"regexp"
|
||
"strings"
|
||
"unicode"
|
||
)
|
||
|
||
// IsValidEmail 验证邮箱格式
|
||
func IsValidEmail(email string) bool {
|
||
emailRegex := regexp.MustCompile(`^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$`)
|
||
return emailRegex.MatchString(email)
|
||
}
|
||
|
||
// IsValidUsername 验证用户名格式
|
||
func IsValidUsername(username string) bool {
|
||
// 用户名长度3-20，只能包含字母、数字、下划线
|
||
if len(username) < 3 || len(username) > 20 {
|
||
return false
|
||
}
|
||
|
||
usernameRegex := regexp.MustCompile(`^[a-zA-Z0-9_]+$`)
|
||
return usernameRegex.MatchString(username)
|
||
}
|
||
|
||
// IsValidPassword 验证密码强度
|
||
func IsValidPassword(password string) bool {
|
||
// 密码长度至少6位
|
||
if len(password) < 6 {
|
||
return false
|
||
}
|
||
|
||
// 检查是否包含字母和数字
|
||
hasLetter := false
|
||
hasNumber := false
|
||
|
||
for _, char := range password {
|
||
if unicode.IsLetter(char) {
|
||
hasLetter = true
|
||
}
|
||
if unicode.IsNumber(char) {
|
||
hasNumber = true
|
||
}
|
||
}
|
||
|
||
return hasLetter && hasNumber
|
||
}
|
||
|
||
// IsValidSlug 验证slug格式
|
||
func IsValidSlug(slug string) bool {
|
||
// slug只能包含小写字母、数字和连字符
|
||
if len(slug) == 0 || len(slug) > 100 {
|
||
return false
|
||
}
|
||
|
||
slugRegex := regexp.MustCompile(`^[a-z0-9-]+$`)
|
||
return slugRegex.MatchString(slug) && !strings.HasPrefix(slug, "-") && !strings.HasSuffix(slug, "-")
|
||
}
|
||
|
||
// IsValidHexColor 验证十六进制颜色代码
|
||
func IsValidHexColor(color string) bool {
|
||
colorRegex := regexp.MustCompile(`^#[a-fA-F0-9]{6}$`)
|
||
return colorRegex.MatchString(color)
|
||
}
|
||
|
||
// IsValidURL 验证URL格式
|
||
func IsValidURL(url string) bool {
|
||
urlRegex := regexp.MustCompile(`^https?://[^\s/$.?#].[^\s]*$`)
|
||
return urlRegex.MatchString(url)
|
||
}
|
||
|
||
// SanitizeString 清理字符串，移除HTML标签和特殊字符
|
||
func SanitizeString(input string) string {
|
||
// 移除HTML标签
|
||
htmlRegex := regexp.MustCompile(`<[^>]*>`)
|
||
cleaned := htmlRegex.ReplaceAllString(input, "")
|
||
|
||
// 移除多余的空白字符
|
||
whitespaceRegex := regexp.MustCompile(`\s+`)
|
||
cleaned = whitespaceRegex.ReplaceAllString(cleaned, " ")
|
||
|
||
return strings.TrimSpace(cleaned)
|
||
}
|
||
|
||
// ValidateImageFormat 验证图片格式
|
||
func ValidateImageFormat(filename string) bool {
|
||
allowedExtensions := []string{".jpg", ".jpeg", ".png", ".gif", ".webp"}
|
||
lowerFilename := strings.ToLower(filename)
|
||
|
||
for _, ext := range allowedExtensions {
|
||
if strings.HasSuffix(lowerFilename, ext) {
|
||
return true
|
||
}
|
||
}
|
||
|
||
return false
|
||
}
|
||
|
||
// ValidateFileSize 验证文件大小（字节）
|
||
func ValidateFileSize(size int64, maxSizeMB int64) bool {
|
||
maxSizeBytes := maxSizeMB * 1024 * 1024
|
||
return size <= maxSizeBytes && size > 0
|
||
}
|
||
|
||
// NormalizeString 标准化字符串（去空格、转小写）
|
||
func NormalizeString(s string) string {
|
||
return strings.ToLower(strings.TrimSpace(s))
|
||
}
|
||
|
||
// ContainsOnlyASCII 检查字符串是否只包含ASCII字符
|
||
func ContainsOnlyASCII(s string) bool {
|
||
for _, char := range s {
|
||
if char > 127 {
|
||
return false
|
||
}
|
||
}
|
||
return true
|
||
}
|
||
|
||
// Contains 检查切片是否包含指定元素
|
||
func Contains(slice []string, item string) bool {
|
||
for _, s := range slice {
|
||
if s == item {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
} |