84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package middleware
|
|
|
|
import (
|
|
"john/handler"
|
|
"john/internal/storage/dao/postgresql"
|
|
"john/internal/storage/model"
|
|
"john/internal/storage/validators"
|
|
"john/internal/user"
|
|
"john/internal/utils"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Validator struct {
|
|
userService *user.Service
|
|
*handler.Context
|
|
}
|
|
|
|
func NewValidator(ctx *handler.Context, db *postgresql.DatabasePostgreSQL) *Validator {
|
|
return &Validator{userService: user.NewService(user.NewDatabase(db)), Context: ctx}
|
|
}
|
|
func (v Validator) HasValidUserId(gc *gin.Context) {
|
|
userId := gc.Param("userId")
|
|
err := v.Validator.VarCtx(gc, userId, "uuid4")
|
|
if err != nil {
|
|
utils.JSONError(gc.Writer, validators.NewDataValidationAPIError(err))
|
|
return
|
|
}
|
|
}
|
|
|
|
func (v Validator) UserExists(gc *gin.Context) {
|
|
userId := gc.Param("userId")
|
|
user, err := v.userService.GetUserById(userId)
|
|
if e, ok := err.(*model.APIError); ok {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).WithField("type", e.Type).Error("error GetUser: get user error")
|
|
utils.JSONErrorWithMessage(gc.Writer, *e, e.Description)
|
|
} else if err != nil {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).Error("error while get user")
|
|
utils.JSONError(gc.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
|
|
if user == nil {
|
|
utils.JSONErrorWithMessage(gc.Writer, model.ErrNotFound, "User not found")
|
|
return
|
|
}
|
|
}
|
|
|
|
func (v Validator) UserdIdMatchOAuthToken(gc *gin.Context) {
|
|
userId := gc.Param("userId")
|
|
usrParam, err := v.userService.GetUserById(userId)
|
|
if e, ok := err.(*model.APIError); ok {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).WithField("type", e.Type).Error("error GetUser: get user error")
|
|
utils.JSONErrorWithMessage(gc.Writer, *e, e.Description)
|
|
} else if err != nil {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).Error("error while get user")
|
|
utils.JSONError(gc.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
googleUserId, exists := gc.Get("googleUserId")
|
|
if exists == false {
|
|
utils.GetLoggerFromCtx(gc).Error("error while getting google user id")
|
|
utils.JSONError(gc.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
usr, err := v.userService.GetUserFromGoogleID(googleUserId.(string))
|
|
if e, ok := err.(*model.APIError); ok {
|
|
utils.GetLogger().Info(err)
|
|
utils.GetLoggerFromCtx(gc).WithError(err).WithField("type", e.Type).Error("error GetUserFromGoogleID: get user from google user id")
|
|
utils.JSONErrorWithMessage(gc.Writer, *e, e.Description)
|
|
return
|
|
} else if err != nil {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).Error("error while get user from google user id")
|
|
utils.JSONError(gc.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
|
|
if usr == nil || usr.ID != usrParam.ID {
|
|
utils.GetLoggerFromCtx(gc).WithError(err).Error("User in path doesn't match authenticated user")
|
|
utils.JSONError(gc.Writer, model.ErrBadRequestFormat)
|
|
return
|
|
}
|
|
}
|