95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package auth
|
|
|
|
import (
|
|
"nixcn-cms/data"
|
|
"nixcn-cms/internal/exception"
|
|
"nixcn-cms/pkgs/authcode"
|
|
"nixcn-cms/pkgs/authtoken"
|
|
"nixcn-cms/utils"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type TokenRequest struct {
|
|
Code string `json:"code"`
|
|
}
|
|
|
|
func Token(c *gin.Context) {
|
|
var req TokenRequest
|
|
|
|
err := c.ShouldBindJSON(&req)
|
|
if err != nil {
|
|
errorCode := new(exception.Builder).
|
|
SetStatus(exception.StatusUser).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeCommon).
|
|
SetOriginal(exception.CommonErrorInvalidInput).
|
|
SetError(err).
|
|
Build(c)
|
|
utils.HttpResponse(c, 400, errorCode)
|
|
return
|
|
}
|
|
|
|
authCode, ok := authcode.VerifyAuthCode(c, req.Code)
|
|
if !ok {
|
|
errorCode := new(exception.Builder).
|
|
SetStatus(exception.StatusUser).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeSpecific).
|
|
SetOriginal(exception.AuthTokenInvalidToken).
|
|
Build(c)
|
|
utils.HttpResponse(c, 403, errorCode)
|
|
return
|
|
}
|
|
|
|
userData := new(data.User)
|
|
user, err := userData.GetByEmail(c, authCode.Email)
|
|
if err != nil {
|
|
errorCode := new(exception.Builder).
|
|
SetStatus(exception.StatusServer).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeCommon).
|
|
SetOriginal(exception.CommonErrorInternal).
|
|
SetError(err).
|
|
Build(c)
|
|
utils.HttpResponse(c, 500, errorCode)
|
|
return
|
|
}
|
|
|
|
// Generate jwt
|
|
JwtTool := authtoken.Token{
|
|
Application: viper.GetString("server.application"),
|
|
}
|
|
accessToken, refreshToken, err := JwtTool.IssueTokens(c, authCode.ClientId, user.UserId)
|
|
if err != nil {
|
|
errorCode := new(exception.Builder).
|
|
SetStatus(exception.StatusServer).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeSpecific).
|
|
SetOriginal(exception.AuthTokenGenFailed).
|
|
SetError(err).
|
|
Build(c)
|
|
utils.HttpResponse(c, 500, errorCode)
|
|
return
|
|
}
|
|
|
|
tokenResp := struct {
|
|
AccessToken string `json:"access_token"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
}{accessToken, refreshToken}
|
|
|
|
errorCode := new(exception.Builder).
|
|
SetStatus(exception.StatusServer).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeCommon).
|
|
SetOriginal(exception.CommonSuccess).
|
|
Build(c)
|
|
utils.HttpResponse(c, 200, errorCode, tokenResp)
|
|
}
|