package system

import (
	"sundynix-go/global"
	"sundynix-go/model/commom/response"
	"sundynix-go/model/system"
	systemReq "sundynix-go/model/system/request"
	systemRes "sundynix-go/model/system/response"
	"sundynix-go/utils/jwt"

	"github.com/gin-gonic/gin"
	"github.com/mojocn/base64Captcha"
	"go.uber.org/zap"
)

var store = base64Captcha.DefaultMemStore

type AuthApi struct{}

// Login
// @Tags      登录相关
// @Summary   pc登录
// @accept    application/json
// @Produce   application/json
// @Param     data  body  systemReq.Login  true "用户名, 密码, 验证码,验证码id"
// @Success   200   {object}  response.Response{msg=string}  "登录成功"
// @Router    /api/auth/login [post]
func (a *AuthApi) Login(c *gin.Context) {
	var l systemReq.Login
	err := c.ShouldBindJSON(&l)
	if err != nil {
		response.FailWithMsg(err.Error(), c)
		return
	}
	if l.CaptchaId != "" && l.Captcha != "" && store.Verify(l.CaptchaId, l.Captcha, true) {
		u := &system.User{Account: l.Account, Password: l.Password}
		user, err := userService.Login(u)
		if err != nil {
			global.Logger.Error("登录失败! 用户名不存在或者密码错误!", zap.Error(err))
			response.FailWithMsg("用户名不存在或者密码错误", c)
			return
		}
		a.GetToken(c, *user)
		return
	}
	response.FailWithMsg("验证码错误", c)
}

// Logout
// @Tags      登录相关
// @Summary   pc登出
// @Security  ApiKeyAuth
// @Produce   application/json
// @Success   200   {object}  response.Response{msg=string}  "登出成功"
// @Router    /api/auth/logout [get]
func (a *AuthApi) Logout(c *gin.Context) {
	token := jwt.GetToken(c)
	userId := jwt.GetUserId(c)
	err := jwtService.PutBlacklist(userId, token)
	if err != nil {
		global.Logger.Error("登出失败!", zap.Error(err))
		response.FailWithMsg("登出失败", c)
		return
	}
	jwt.ClearToken(c)
	response.OkWithMsg("登出成功", c)

}

// Captcha
// @Tags      登录相关
// @Summary   获取验证码
// @Produce   application/json
// @Success   200   {object}  response.Response{data=systemRes.CaptchaRes}  "获取验证码"
// @Router    /api/auth/captcha [get]
func (u *AuthApi) Captcha(c *gin.Context) {
	var driver = base64Captcha.DriverString{
		Height:          80,
		Width:           240,
		NoiseCount:      2,
		ShowLineOptions: 4,
		Length:          4,
		Source:          "1234567890",
	}

	cp := base64Captcha.NewCaptcha(&driver, store)
	id, b64s, _, err := cp.Generate()
	if err != nil {
		global.Logger.Error("GenerateCaptcha err", zap.Error(err))
		response.FailWithMsg("GenerateCaptcha err", c)
		return
	}
	response.OkWithData(systemRes.CaptchaRes{
		CaptchaId: id,
		Captcha:   b64s,
	}, c)
}

func (a *AuthApi) GetToken(c *gin.Context, user system.User) {
	token, claims, err := jwt.GetLoginToken(&user)
	if err != nil {
		global.Logger.Error("GetToken err", zap.Error(err))
		response.FailWithMsg("GetToken err", c)
	}
	response.OkWithData(systemRes.LoginResponse{
		ExpiresAt: claims.RegisteredClaims.ExpiresAt.Unix() * 1000,
		Token:     token,
		User:      user,
	}, c)
}