133 lines
3.8 KiB
Go
Executable File
133 lines
3.8 KiB
Go
Executable File
package user
|
|
|
|
import (
|
|
"budget/handler"
|
|
"budget/internal/storage/dao/postgresql"
|
|
"budget/internal/storage/model"
|
|
"budget/internal/utils"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"google.golang.org/api/oauth2/v2"
|
|
)
|
|
|
|
type Context struct {
|
|
service *Service
|
|
db *Database
|
|
*handler.Context
|
|
}
|
|
|
|
func NewHandler(ctx *handler.Context, db *postgresql.DatabasePostgreSQL) *Context {
|
|
database := NewDatabase(db)
|
|
service := NewService(database)
|
|
return &Context{service: service, db: database, Context: ctx}
|
|
}
|
|
|
|
func (uc *Context) GetAllUsers(c *gin.Context) {
|
|
users, err := uc.db.GetAllUsers()
|
|
if err != nil {
|
|
utils.GetLoggerFromCtx(c).WithError(err).Error("error while getting users")
|
|
utils.JSONErrorWithMessage(c.Writer, model.ErrInternalServer, "Error while getting users")
|
|
return
|
|
}
|
|
utils.JSON(c.Writer, http.StatusOK, users)
|
|
}
|
|
|
|
func (hc *Context) ConnectUser(c *gin.Context) {
|
|
authorizationHeader := c.GetHeader("Authorization")
|
|
authorizationHeaderSplitted := strings.Split(authorizationHeader, " ")
|
|
if len(authorizationHeaderSplitted) != 2 {
|
|
utils.JSONError(c.Writer, model.ErrBadRequestFormat)
|
|
return
|
|
}
|
|
|
|
oauth2Service, err := oauth2.New(&http.Client{})
|
|
if oauth2Service == nil {
|
|
utils.GetLoggerFromCtx(c).WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
tokenInfoCall := oauth2Service.Tokeninfo()
|
|
tokenInfoCall.IdToken(authorizationHeaderSplitted[1])
|
|
tokenInfo, err := tokenInfoCall.Do()
|
|
if err != nil {
|
|
utils.GetLoggerFromCtx(c).WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrBadRequestFormat)
|
|
return
|
|
}
|
|
user, err := hc.service.GetUserFromGoogleID(tokenInfo.UserId)
|
|
if err != nil {
|
|
if castedError, ok := err.(*model.APIError); ok {
|
|
if castedError.Type == model.ErrNotFound.Type {
|
|
user, err := hc.service.CreateUserFromGoogleToken(tokenInfo.UserId, tokenInfo.Email)
|
|
if err != nil {
|
|
utils.GetLoggerFromCtx(c).WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
utils.JSON(c.Writer, 200, user)
|
|
return
|
|
}
|
|
utils.JSONError(c.Writer, *castedError)
|
|
return
|
|
}
|
|
utils.GetLoggerFromCtx(c).WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
utils.JSON(c.Writer, 200, user)
|
|
}
|
|
|
|
func (hc *Context) CreateUser(c *gin.Context) {
|
|
authorizationHeader := c.GetHeader("Authorization")
|
|
authorizationHeaderSplitted := strings.Split(authorizationHeader, " ")
|
|
if len(authorizationHeaderSplitted) != 2 {
|
|
utils.JSONError(c.Writer, model.ErrBadRequestFormat)
|
|
return
|
|
}
|
|
|
|
oauth2Service, err := oauth2.New(&http.Client{})
|
|
if oauth2Service == nil {
|
|
utils.GetLogger().WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
tokenInfoCall := oauth2Service.Tokeninfo()
|
|
tokenInfoCall.IdToken(authorizationHeaderSplitted[1])
|
|
tokenInfo, err := tokenInfoCall.Do()
|
|
if err != nil {
|
|
utils.GetLogger().WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrBadRequestFormat)
|
|
return
|
|
}
|
|
user, err := hc.service.GetUserFromGoogleID(tokenInfo.UserId)
|
|
if err != nil {
|
|
utils.GetLogger().WithError(err).Error(err)
|
|
if castedError, ok := err.(*model.APIError); ok {
|
|
if castedError.Type == model.ErrNotFound.Type {
|
|
user, err := hc.service.CreateUserFromGoogleToken(tokenInfo.UserId, tokenInfo.Email)
|
|
if err != nil {
|
|
utils.GetLogger().WithError(err).Error(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
utils.JSON(c.Writer, http.StatusCreated, user)
|
|
return
|
|
}
|
|
utils.JSONError(c.Writer, *castedError)
|
|
return
|
|
}
|
|
utils.GetLogger().Info(err)
|
|
utils.JSONError(c.Writer, model.ErrInternalServer)
|
|
return
|
|
}
|
|
utils.JSON(c.Writer, http.StatusOK, user)
|
|
}
|
|
|
|
func (hc *Context) GetUser(c *gin.Context) {
|
|
userID := c.Param("userId")
|
|
user, _ := hc.service.GetUserById(userID)
|
|
utils.JSON(c.Writer, http.StatusOK, user)
|
|
}
|