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

package validator

import (
	"fmt"
	"net/url"
	"regexp"
	"strings"

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

// BusinessValidator 业务验证器
type BusinessValidator struct {
	validator *validator.Validate
}

// NewBusinessValidator 创建业务验证器
func NewBusinessValidator() *BusinessValidator {
	validate := validator.New()
	RegisterCustomValidators(validate)
	
	return &BusinessValidator{
		validator: validate,
	}
}

// ValidateStruct 验证结构体
func (bv *BusinessValidator) ValidateStruct(data interface{}) error {
	return bv.validator.Struct(data)
}

// ValidateField 验证单个字段
func (bv *BusinessValidator) ValidateField(field interface{}, tag string) error {
	return bv.validator.Var(field, tag)
}

// 以下是具体的业务验证方法，可以在业务逻辑中直接调用

// ValidatePhone 验证手机号
func (bv *BusinessValidator) ValidatePhone(phone string) error {
	if phone == "" {
		return fmt.Errorf("手机号不能为空")
	}
	matched, _ := regexp.MatchString(`^1[3-9]\d{9}$`, phone)
	if !matched {
		return fmt.Errorf("手机号格式不正确")
	}
	return nil
}

// ValidatePassword 验证密码强度
func (bv *BusinessValidator) ValidatePassword(password string) error {
	if password == "" {
		return fmt.Errorf("密码不能为空")
	}
	if len(password) < 8 {
		return fmt.Errorf("密码长度不能少于8位")
	}
	hasUpper := regexp.MustCompile(`[A-Z]`).MatchString(password)
	hasLower := regexp.MustCompile(`[a-z]`).MatchString(password)
	hasDigit := regexp.MustCompile(`\d`).MatchString(password)
	
	if !hasUpper {
		return fmt.Errorf("密码必须包含大写字母")
	}
	if !hasLower {
		return fmt.Errorf("密码必须包含小写字母")
	}
	if !hasDigit {
		return fmt.Errorf("密码必须包含数字")
	}
	return nil
}

// ValidateUsername 验证用户名
func (bv *BusinessValidator) ValidateUsername(username string) error {
	if username == "" {
		return fmt.Errorf("用户名不能为空")
	}
	matched, _ := regexp.MatchString(`^[a-zA-Z][a-zA-Z0-9_]{2,19}$`, username)
	if !matched {
		return fmt.Errorf("用户名格式不正确，只能包含字母、数字、下划线，且必须以字母开头，长度3-20位")
	}
	return nil
}

// ValidateSocialCreditCode 验证统一社会信用代码
func (bv *BusinessValidator) ValidateSocialCreditCode(code string) error {
	if code == "" {
		return fmt.Errorf("统一社会信用代码不能为空")
	}
	matched, _ := regexp.MatchString(`^[0-9A-HJ-NPQRTUWXY]{2}\d{6}[0-9A-HJ-NPQRTUWXY]{10}$`, code)
	if !matched {
		return fmt.Errorf("统一社会信用代码格式不正确，必须是18位统一社会信用代码")
	}
	return nil
}

// ValidateIDCard 验证身份证号
func (bv *BusinessValidator) ValidateIDCard(idCard string) error {
	if idCard == "" {
		return fmt.Errorf("身份证号不能为空")
	}
	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)
	if !matched {
		return fmt.Errorf("身份证号格式不正确，必须是18位身份证号")
	}
	return nil
}

// ValidateUUID 验证UUID
func (bv *BusinessValidator) ValidateUUID(uuid string) error {
	if uuid == "" {
		return fmt.Errorf("UUID不能为空")
	}
	matched, _ := regexp.MatchString(`^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`, uuid)
	if !matched {
		return fmt.Errorf("UUID格式不正确")
	}
	return nil
}

// ValidateURL 验证URL
func (bv *BusinessValidator) ValidateURL(urlStr string) error {
	if urlStr == "" {
		return fmt.Errorf("URL不能为空")
	}
	_, err := url.ParseRequestURI(urlStr)
	if err != nil {
		return fmt.Errorf("URL格式不正确: %v", err)
	}
	return nil
}

// ValidateProductCode 验证产品代码
func (bv *BusinessValidator) ValidateProductCode(code string) error {
	if code == "" {
		return fmt.Errorf("产品代码不能为空")
	}
	matched, _ := regexp.MatchString(`^[a-zA-Z0-9_-]{3,50}$`, code)
	if !matched {
		return fmt.Errorf("产品代码格式不正确，只能包含字母、数字、下划线、连字符，长度3-50位")
	}
	return nil
}

// ValidateEmail 验证邮箱
func (bv *BusinessValidator) ValidateEmail(email string) error {
	if email == "" {
		return fmt.Errorf("邮箱不能为空")
	}
	matched, _ := regexp.MatchString(`^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$`, email)
	if !matched {
		return fmt.Errorf("邮箱格式不正确")
	}
	return nil
}

// ValidateSortOrder 验证排序方向
func (bv *BusinessValidator) ValidateSortOrder(sortOrder string) error {
	if sortOrder == "" {
		return nil // 允许为空
	}
	if sortOrder != "asc" && sortOrder != "desc" {
		return fmt.Errorf("排序方向必须是 asc 或 desc")
	}
	return nil
}

// ValidatePrice 验证价格
func (bv *BusinessValidator) ValidatePrice(price float64) error {
	if price < 0 {
		return fmt.Errorf("价格不能为负数")
	}
	return nil
}

// ValidateStringLength 验证字符串长度
func (bv *BusinessValidator) ValidateStringLength(str string, fieldName string, min, max int) error {
	length := len(strings.TrimSpace(str))
	if min > 0 && length < min {
		return fmt.Errorf("%s长度不能少于%d位", fieldName, min)
	}
	if max > 0 && length > max {
		return fmt.Errorf("%s长度不能超过%d位", fieldName, max)
	}
	return nil
}

// ValidateRequired 验证必填字段
func (bv *BusinessValidator) ValidateRequired(value interface{}, fieldName string) error {
	if value == nil {
		return fmt.Errorf("%s不能为空", fieldName)
	}
	
	switch v := value.(type) {
	case string:
		if strings.TrimSpace(v) == "" {
			return fmt.Errorf("%s不能为空", fieldName)
		}
	case *string:
		if v == nil || strings.TrimSpace(*v) == "" {
			return fmt.Errorf("%s不能为空", fieldName)
		}
	}
	
	return nil
}

// ValidateRange 验证数值范围
func (bv *BusinessValidator) ValidateRange(value float64, fieldName string, min, max float64) error {
	if value < min {
		return fmt.Errorf("%s不能小于%v", fieldName, min)
	}
	if value > max {
		return fmt.Errorf("%s不能大于%v", fieldName, max)
	}
	return nil
}

// ValidateSliceNotEmpty 验证切片不为空
func (bv *BusinessValidator) ValidateSliceNotEmpty(slice interface{}, fieldName string) error {
	if slice == nil {
		return fmt.Errorf("%s不能为空", fieldName)
	}
	
	switch v := slice.(type) {
	case []string:
		if len(v) == 0 {
			return fmt.Errorf("%s不能为空", fieldName)
		}
	case []int:
		if len(v) == 0 {
			return fmt.Errorf("%s不能为空", fieldName)
		}
	}
	
	return nil
}