tyapi-server/internal/shared/validator/custom_validators.go

package validator

import (
	"net/url"
	"regexp"

	"github.com/go-playground/validator/v10"
)

// RegisterCustomValidators 注册所有自定义验证器
func RegisterCustomValidators(validate *validator.Validate) {
	// 手机号验证器
	validate.RegisterValidation("phone", validatePhone)
	
	// 用户名验证器（字母开头，允许字母数字下划线，3-20位）
	validate.RegisterValidation("username", validateUsername)
	
	// 强密码验证器（至少8位，包含大小写字母和数字）
	validate.RegisterValidation("strong_password", validateStrongPassword)
	
	// 统一社会信用代码验证器
	validate.RegisterValidation("social_credit_code", validateSocialCreditCode)
	
	// 身份证号验证器
	validate.RegisterValidation("id_card", validateIDCard)
	
	// 价格验证器（非负数）
	validate.RegisterValidation("price", validatePrice)
	
	// 排序方向验证器
	validate.RegisterValidation("sort_order", validateSortOrder)
	
	// 产品代码验证器（字母数字下划线连字符，3-50位）
	validate.RegisterValidation("product_code", validateProductCode)
	
	// UUID验证器
	validate.RegisterValidation("uuid", validateUUID)
	
	// URL验证器
	validate.RegisterValidation("url", validateURL)
}

// validatePhone 手机号验证
func validatePhone(fl validator.FieldLevel) bool {
	phone := fl.Field().String()
	matched, _ := regexp.MatchString(`^1[3-9]\d{9}$`, phone)
	return matched
}

// validateUsername 用户名验证
func validateUsername(fl validator.FieldLevel) bool {
	username := fl.Field().String()
	matched, _ := regexp.MatchString(`^[a-zA-Z][a-zA-Z0-9_]{2,19}$`, username)
	return matched
}

// validateStrongPassword 强密码验证
func validateStrongPassword(fl validator.FieldLevel) bool {
	password := fl.Field().String()
	if len(password) < 8 {
		return false
	}
	hasUpper := regexp.MustCompile(`[A-Z]`).MatchString(password)
	hasLower := regexp.MustCompile(`[a-z]`).MatchString(password)
	hasDigit := regexp.MustCompile(`\d`).MatchString(password)
	return hasUpper && hasLower && hasDigit
}

// validateSocialCreditCode 统一社会信用代码验证
func validateSocialCreditCode(fl validator.FieldLevel) bool {
	code := fl.Field().String()
	matched, _ := regexp.MatchString(`^[0-9A-HJ-NPQRTUWXY]{2}\d{6}[0-9A-HJ-NPQRTUWXY]{10}$`, code)
	return matched
}

// validateIDCard 身份证号验证
func validateIDCard(fl validator.FieldLevel) bool {
	idCard := fl.Field().String()
	matched, _ := regexp.MatchString(`^[1-9]\d{5}(18|19|20)\d{2}((0[1-9])|(1[0-2]))(([0-2][1-9])|10|20|30|31)\d{3}[\dXx]$`, idCard)
	return matched
}

// validatePrice 价格验证
func validatePrice(fl validator.FieldLevel) bool {
	price := fl.Field().Float()
	return price >= 0
}

// validateSortOrder 排序方向验证
func validateSortOrder(fl validator.FieldLevel) bool {
	sortOrder := fl.Field().String()
	return sortOrder == "" || sortOrder == "asc" || sortOrder == "desc"
}

// validateProductCode 产品代码验证
func validateProductCode(fl validator.FieldLevel) bool {
	code := fl.Field().String()
	matched, _ := regexp.MatchString(`^[a-zA-Z0-9_-]{3,50}$`, code)
	return matched
}

// validateUUID UUID验证
func validateUUID(fl validator.FieldLevel) bool {
	uuid := fl.Field().String()
	matched, _ := regexp.MatchString(`^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`, uuid)
	return matched
}

// validateURL URL验证
func validateURL(fl validator.FieldLevel) bool {
	urlStr := fl.Field().String()
	_, err := url.ParseRequestURI(urlStr)
	return err == nil
}