LingXi-CRM/server/api/user.go

package api

import (
	"crm/models"
	"crm/response"
	"crm/service"
	"log"
	"regexp"
	"strconv"

	"github.com/gin-gonic/gin"
)

type UserApi struct {
	userService *service.UserService
}

func NewUserApi() *UserApi {
	userApi := UserApi{
		userService: &service.UserService{},
	}
	return &userApi
}

// 用户注册
func (u *UserApi) Register(context *gin.Context) {
	var param models.UserCreateParam
	if err := context.ShouldBind(&param); err != nil {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		log.Printf("[error]UserApi:Register:%s", err)
		return
	}
	if !verifyEmailFormat(param.Email) {
		response.Result(response.ErrCodeEmailFormatInvalid, nil, context)
		return
	}
	errCode := u.userService.Register(&param)
	response.Result(errCode, nil, context)
}

// 用户登录
func (u *UserApi) Login(context *gin.Context) {
	var param models.UserLoginParam
	if err := context.ShouldBind(&param); err != nil {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	if !verifyEmailFormat(param.Email) {
		response.Result(response.ErrCodeEmailFormatInvalid, nil, context)
		return
	}
	userInfo, errCode := u.userService.Login(&param)
	if userInfo == nil {
		response.Result(errCode, nil, context)
		return
	}
	response.Result(errCode, userInfo, context)
}

// 获取验证码
func (u *UserApi) GetVerifyCode(context *gin.Context) {
	email := context.Query("email")
	if email == "" {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	if !verifyEmailFormat(email) {
		response.Result(response.ErrCodeEmailFormatInvalid, nil, context)
		return
	}
	errCode := u.userService.GetVerifyCode(email)
	response.Result(errCode, nil, context)
}

// 忘记密码
func (u *UserApi) ForgotPass(context *gin.Context) {
	var param models.UserPassParam
	if verifyEmailFormat(param.Email) {
		response.Result(response.ErrCodeEmailFormatInvalid, nil, context)
		return
	}
	if err := context.ShouldBind(&param); err != nil {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	errCode := u.userService.ForgotPass(&param)
	response.Result(errCode, nil, context)
}

// 修改邮箱
func (u *UserApi) UpdateMail(context *gin.Context) {
	var param models.UserMailParam
	if verifyEmailFormat(param.Email) && verifyEmailFormat(param.NewEmail) {
		response.Result(response.ErrCodeEmailFormatInvalid, nil, context)
		return
	}
	if err := context.ShouldBind(&param); err != nil {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	errCode := u.userService.UpdateMail(&param)
	response.Result(errCode, nil, context)
}

// 退出登录
func (u *UserApi) Logout(context *gin.Context) {
	token := context.Request.Header.Get("token")
	if token == "" {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	errCode := u.userService.Logout(token)
	response.Result(errCode, nil, context)
}

// 注销账号
func (u *UserApi) Delete(context *gin.Context) {
	var param models.UserDeleteParam
	uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
	err := context.ShouldBind(&param)
	if uid <= 0 || err != nil {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	param.Id = int64(uid)
	errCode := u.userService.Delete(param)
	response.Result(errCode, nil, context)
}

// 获取用户信息
func (u *UserApi) GetInfo(context *gin.Context) {
	uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
	if uid <= 0 {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	userInfo, errCode := u.userService.GetInfo(int64(uid))
	response.Result(errCode, userInfo, context)
}

// 订阅个人版
func (u *UserApi) Buy(context *gin.Context) {
	uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
	if uid <= 0 {
		response.Result(response.ErrCodeParamInvalid, nil, context)
		return
	}
	versionInfo, errCode := u.userService.Buy(int64(uid))
	response.Result(errCode, versionInfo, context)
}

// 邮箱格式校验
func verifyEmailFormat(email string) bool {
	pattern := `\w+([-+.]\w+)*@\w+([-.]\w+)*\.\w+([-.]\w+)*`
	reg := regexp.MustCompile(pattern)
	return reg.MatchString(email)
}