opt: oauth2 callback type write to header

pull/242/head
zijiren233 10 months ago
parent 7f35a9594d
commit 83a03d0759

@ -148,6 +148,8 @@ func newAuthFunc(redirect string) stateHandler {
return func(ctx *gin.Context, pi provider.ProviderInterface, code string) {
log := ctx.MustGet("log").(*logrus.Entry)
ctx.Header("X-OAuth2-Type", CallbackTypeAuth)
ui, err := pi.GetUserInfo(ctx, code)
if err != nil {
log.Errorf("failed to get user info: %v", err)
@ -184,6 +186,7 @@ func newAuthFunc(redirect string) stateHandler {
if errors.Is(err, middlewares.ErrUserBanned) ||
errors.Is(err, middlewares.ErrUserPending) {
ctx.AbortWithStatusJSON(http.StatusOK, model.NewApiDataResp(gin.H{
"type": CallbackTypeAuth,
"message": err.Error(),
"role": user.Role,
}))

@ -69,6 +69,8 @@ func newBindFunc(userID, redirect string) stateHandler {
return func(ctx *gin.Context, pi provider.ProviderInterface, code string) {
log := ctx.MustGet("log").(*logrus.Entry)
ctx.Header("X-OAuth2-Type", CallbackTypeBind)
ui, err := pi.GetUserInfo(ctx, code)
if err != nil {
log.Errorf("failed to get user info: %v", err)

@ -1,6 +1,6 @@
package auth
type CallbackType string
type CallbackType = string
const (
CallbackTypeAuth CallbackType = "auth"

Loading…
Cancel
Save