119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package service_auth
|
|
|
|
import (
|
|
"context"
|
|
"nixcn-cms/data"
|
|
"nixcn-cms/internal/authcode"
|
|
"nixcn-cms/internal/authtoken"
|
|
"nixcn-cms/internal/exception"
|
|
"nixcn-cms/service/shared"
|
|
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type TokenData struct {
|
|
Code string `json:"code"`
|
|
}
|
|
|
|
type TokenPayload struct {
|
|
Context context.Context
|
|
Data *TokenData
|
|
}
|
|
|
|
type TokenResult struct {
|
|
Common shared.CommonResult
|
|
Data *TokenResponse
|
|
}
|
|
|
|
type TokenResponse struct {
|
|
AccessToken string `json:"access_token"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
}
|
|
|
|
func (self *AuthServiceImpl) Token(payload *TokenPayload) (result *TokenResult) {
|
|
authCode, ok := authcode.VerifyAuthCode(payload.Context, payload.Data.Code)
|
|
if !ok {
|
|
exception := new(exception.Builder).
|
|
SetStatus(exception.StatusUser).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeSpecific).
|
|
SetOriginal(exception.AuthTokenInvalidToken).
|
|
Throw(payload.Context)
|
|
|
|
result = &TokenResult{
|
|
Common: shared.CommonResult{
|
|
HttpCode: 403,
|
|
Exception: exception,
|
|
},
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
userData := new(data.User)
|
|
user, err := userData.GetByEmail(payload.Context, &authCode.Email)
|
|
if err != nil {
|
|
exception := new(exception.Builder).
|
|
SetStatus(exception.StatusServer).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeCommon).
|
|
SetOriginal(exception.CommonErrorInternal).
|
|
SetError(err).
|
|
Throw(payload.Context)
|
|
|
|
result = &TokenResult{
|
|
Common: shared.CommonResult{
|
|
HttpCode: 500,
|
|
Exception: exception,
|
|
},
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
JwtTool := authtoken.Token{
|
|
Application: viper.GetString("server.application"),
|
|
}
|
|
accessToken, refreshToken, err := JwtTool.IssueTokens(payload.Context, authCode.ClientId, user.UserId)
|
|
if err != nil {
|
|
exception := new(exception.Builder).
|
|
SetStatus(exception.StatusServer).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeSpecific).
|
|
SetOriginal(exception.AuthTokenGenFailed).
|
|
SetError(err).
|
|
Throw(payload.Context)
|
|
|
|
result = &TokenResult{
|
|
Common: shared.CommonResult{
|
|
HttpCode: 500,
|
|
Exception: exception,
|
|
},
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
result = &TokenResult{
|
|
Common: shared.CommonResult{
|
|
HttpCode: 200,
|
|
Exception: new(exception.Builder).
|
|
SetStatus(exception.StatusSuccess).
|
|
SetService(exception.ServiceAuth).
|
|
SetEndpoint(exception.EndpointAuthServiceToken).
|
|
SetType(exception.TypeCommon).
|
|
SetOriginal(exception.CommonSuccess).
|
|
Throw(payload.Context),
|
|
},
|
|
Data: &TokenResponse{
|
|
AccessToken: accessToken,
|
|
RefreshToken: refreshToken,
|
|
},
|
|
}
|
|
|
|
return
|
|
}
|