hongze_yb_en_api/controller/auth.go

package controller

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/go-playground/validator/v10"
	"hongze/hongze_yb_en_api/controller/resp"
	"hongze/hongze_yb_en_api/global"
	"hongze/hongze_yb_en_api/models/english_report_email"
	"hongze/hongze_yb_en_api/models/msg_code"
	"hongze/hongze_yb_en_api/models/session"
	"hongze/hongze_yb_en_api/services"
	"hongze/hongze_yb_en_api/utils"
	"time"
)

type AuthController struct {
}

func (a *AuthController) Login(c *gin.Context) {
	req := new(services.LoginReq)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}
	if req.Account == "" {
		resp.Fail("邮箱或手机号错误", c)
		return
	}
	sysUser, err := english_report_email.CheckUserPwd(req.Type, req.Account, req.Password)
	if err != nil {
		resp.FailData("Login failed. Please check your entries and try again.", "Err:"+err.Error(), c)
		return
	}
	if sysUser == nil {
		resp.FailData("Login failed. Please check your entries and try again.", "Err:"+err.Error(), c)
		return
	}
	if sysUser.Enable == 0 {
		resp.FailData("您的账号已被禁用，如需登录，请联系管理员", "Err:"+err.Error(), c)
		return
	}
	if sysUser.Status == 3 {
		resp.FailData("Your trial has ended</br>Enjoyed your experience with us?</br>Contact us at stephanie@hzinsights.com to extend your trial.", "Err:"+err.Error(), c)
		return
	}
	account := utils.MD5(req.Account)
	token, err := utils.GenToken(account)
	sysSession := new(session.EnglishYbSession)
	sysSession.UserId = int(sysUser.Id)
	//现在要求永不过期
	sysSession.ExpireTime = time.Now().AddDate(99, 0, 0)
	sysSession.CreatedTime = time.Now()
	sysSession.LastUpdatedTime = time.Now()
	sysSession.AccessToken = token
	err = sysSession.AddSession()
	fmt.Println("id:", sysSession.SessionId)
	if err != nil {
		resp.FailData("新增session信息失败", "Err:"+err.Error(), c)
		return
	}

	resp.OkData("登陆成功", sysSession, c)

}

func (a *AuthController) Register(c *gin.Context) {
	req := new(services.RegisterReq)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}

	item, err := msg_code.GetMsgCode(req.Email, req.SmsCode)
	if err != nil {
		if err == utils.ErrNoRow {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		} else {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		}
	}
	if item == nil {
		resp.Fail("验证码错误,请重新输入", c)
		return
	}

	emailItem, err := english_report_email.CheckUser(req.Email)
	if err != nil && err != utils.ErrNoRow{
		resp.Fail("验证码错误,请重新输入", c)
		return
	}
	if emailItem != nil{
		if emailItem.Status == 1{
			//已经是正式用户，更新密码即可
			emailItem.Password = req.Password
			emailItem.ModifyTime = time.Now()

			err = emailItem.Update([]string{"Password","ModifyTime"})
			if err != nil {
				resp.FailMsg("修改密码失败", "修改密码失败，Err:"+err.Error(), c)
				return
			}
			resp.Ok("注册成功", c)
		} else if emailItem.Status == 2{
			resp.Fail("There is already a user account associated with this email address. Please log in instead.", c)
			return
		} else if emailItem.Status == 3 {
			resp.Fail("Your trial has ended</br>Enjoyed your experience with us?</br>Contact us at stephanie@hzinsights.com to extend your trial.", c)
			return
		}
	}

	//状态为临时
	user := english_report_email.Email{
		Name:        req.Name,
		CompanyName: req.CompanyName,
		Email:       req.Email,
		Password:    req.Password,
		Enable:      1,
		Status:      2,
	}
	user.Set()

	err = user.Add()
	if err != nil {
		resp.FailData("新增用户信息失败", "Err:"+err.Error(), c)
		return
	}

	resp.Ok("注册成功", c)
}

// @Title 修改密码
// @Description 修改密码
// @Param	request	body models.ModifyPwdReq true "type json string"
// @Success 200 {object} models.LoginResp
// @router /modifyPwd [post]
func (a *AuthController) ModifyPwd(c *gin.Context) {
	req := new(services.ModifyPwdReq)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}
	userinfo := services.GetInfoByClaims(c)

	if req.NewPwd == "" {
		resp.Fail("新密码不能为空", c)
		return
	}

	if req.OldPwd == "" {
		resp.Fail("旧密码不能为空", c)
		return
	}

	if req.OldPwd != userinfo.Password {
		resp.Fail("旧密码错误，请重新输入", c)
		return
	}

	emailitem := english_report_email.Email{
		Id:       userinfo.Id,
		Password: req.NewPwd,
	}
	emailitem.ModifyTime = time.Now()

	err = emailitem.Update([]string{"Password"})
	if err != nil {
		resp.FailMsg("修改密码失败", "修改密码失败，Err:"+err.Error(), c)
		return
	}

	resp.Ok("修改成功", c)
}

// GetSmsCode 获取短信验证码接口
// @Tags 用户模块
// @Summary  获取短信验证码
// @Description 获取短信验证码接口
// @Security ApiKeyAuth
// @securityDefinitions.basic BasicAuth
// @Param Mobile	query string true "手机号"
// @Param AreaNum	query string true "手机国际区号（中国大陆：86）"
// @Accept  json
// @Product json
// @Success 200 {string} string 获取验证码成功
// @Failure 400 {string} string 手机号不能为空,请输入手机号
// @Router /smsCode [get]
func (a *AuthController) GetSmsCode(c *gin.Context) {
	mobile := c.DefaultQuery("Mobile", "")
	areaNum := c.DefaultQuery("AreaNum", "")
	err, errMsg := services.SendSmsCode(mobile, areaNum)
	if err != nil {
		if errMsg != "" {
			errMsg = "获取验证码失败"
		}
		resp.Fail(errMsg, c)
		return
	}

	resp.Ok("获取验证码成功", c)
}

// GetEmailCode 获取邮箱验证码接口
// @Tags 用户模块
// @Summary  获取邮箱验证码
// @Description 获取邮箱验证码
// @Security ApiKeyAuth
// @securityDefinitions.basic BasicAuth
// @Param email	query string true "电子邮箱账号"
// @Accept  json
// @Product json
// @Success 200 {string} string 获取验证码成功
// @Failure 400 {string} string 请输入邮箱地址
// @Router /emailCode [get]
func (a *AuthController) GetEmailCode(c *gin.Context) {
	name := c.DefaultQuery("Name", "")
	if name == "" {
		resp.Fail("请输入客户名称", c)
		return
	}
	email := c.DefaultQuery("Email", "")
	if email == "" {
		resp.Fail("请输入邮箱地址", c)
		return
	}
	if !utils.ValidateEmailFormatat(email) {
		resp.Fail("邮箱格式错误，请重新输入", c)
		return
	}

	err, errMsg := services.SendEmailCode(name, email)
	if err != nil {
		if errMsg != "" {
			errMsg = "获取验证码失败"
		}
		resp.Fail(errMsg, c)
		return
	}

	resp.Ok("获取验证码成功", c)
}

func (a *AuthController) BindMobile(c *gin.Context) {
	req := new(services.BindMobileReq)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}
	userinfo := services.GetInfoByClaims(c)

	if req.Mobile == "" {
		resp.Fail("手机号不能为空", c)
		return
	}

	if req.SmsCode == "" {
		resp.Fail("验证码不能为空", c)
		return
	}

	item, err := msg_code.GetMsgCode(req.Mobile, req.SmsCode)
	if err != nil {
		if err == utils.ErrNoRow {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		} else {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		}
	}
	if item == nil {
		resp.Fail("验证码错误,请重新输入", c)
		return
	}

	user := english_report_email.Email{
		Id:     userinfo.Id,
		Mobile: req.Mobile,
	}
	user.ModifyTime = time.Now()

	err = user.Update([]string{"Mobile"})
	if err != nil {
		resp.FailMsg("绑定手机号失败", "修改手机号失败，Err:"+err.Error(), c)
		return
	}

	resp.Ok("绑定成功", c)
}

func (a *AuthController) ForgetPwd(c *gin.Context) {
	req := new(services.ForgetPwdReq)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}
	userinfo := services.GetInfoByClaims(c)

	if req.Account == "" {
		resp.Fail("手机号不能为空", c)
		return
	}

	if req.SmsCode == "" {
		resp.Fail("验证码不能为空", c)
		return
	}

	if req.Password == "" {
		resp.Fail("新密码不能为空", c)
		return
	}

	item, err := msg_code.GetMsgCode(req.Account, req.SmsCode)
	if err != nil {
		if err == utils.ErrNoRow {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		} else {
			resp.Fail("校验验证码失败，Err:"+err.Error(), c)
			return
		}
	}
	if item == nil {
		resp.Fail("验证码错误,请重新输入", c)
		return
	}

	emailitem := english_report_email.Email{
		Id:       userinfo.Id,
		Password: req.Password,
	}
	emailitem.ModifyTime = time.Now()

	err = emailitem.Update([]string{"Password"})
	if err != nil {
		resp.FailMsg("修改密码失败", "修改密码失败，Err:"+err.Error(), c)
		return
	}

	resp.Ok("修改成功", c)
}

func (a *AuthController) ModifyMobile(c *gin.Context) {
	req := new(services.ModifyMobile)
	err := c.ShouldBind(&req)
	if err != nil {
		errs, ok := err.(validator.ValidationErrors)
		if !ok {
			resp.FailData("参数解析失败", "Err:"+err.Error(), c)
			return
		}
		resp.FailData("参数解析失败", errs.Translate(global.Trans), c)
		return
	}
	userinfo := services.GetInfoByClaims(c)

	if req.NewMobile == "" {
		resp.Fail("新密码不能为空", c)
		return
	}

	if req.OldMobile == "" {
		resp.Fail("旧密码不能为空", c)
		return
	}

	if req.OldMobile != userinfo.Mobile {
		resp.Fail("旧密码错误，请重新输入", c)
		return
	}

	if req.SmsCode == "" {
		resp.Fail("验证码不能为空", c)
		return
	}

	emailitem := english_report_email.Email{
		Id:     userinfo.Id,
		Mobile: req.NewMobile,
	}
	emailitem.ModifyTime = time.Now()

	err = emailitem.Update([]string{"Mobile"})
	if err != nil {
		resp.FailMsg("修改手机号失败", "修改手机号失败，Err:"+err.Error(), c)
		return
	}

	resp.Ok("修改成功", c)
}