diff --git a/pkg/api/routes.go b/pkg/api/routes.go index fdb3d90..65d3810 100644 --- a/pkg/api/routes.go +++ b/pkg/api/routes.go @@ -2,6 +2,7 @@ package api import ( "wallet-api/pkg/controllers" + "wallet-api/pkg/middleware" "wallet-api/pkg/services" "wallet-api/pkg/utl/configs" @@ -12,14 +13,17 @@ import ( func Routes(s *gin.Engine, db *pg.DB) { ver := s.Group(configs.Prefix) - api := ver.Group("api") + api := ver.Group("api", middleware.Auth) register := ver.Group("register") login := ver.Group("login") + wallet := ver.Group("wallet", middleware.Auth) apiService := services.ApiService{Db: db} usersService := services.UsersService{Db: db} + walletService := services.WalletService{Db: db} controllers.NewApiController(&apiService, api) controllers.NewRegisterController(&usersService, register) controllers.NewLoginController(&usersService, login) + controllers.NewWalletsController(&walletService, wallet) } diff --git a/pkg/controllers/login.go b/pkg/controllers/login.go index 008fb4c..a6e6626 100644 --- a/pkg/controllers/login.go +++ b/pkg/controllers/login.go @@ -22,15 +22,15 @@ func NewLoginController(rs *services.UsersService, s *gin.RouterGroup) *LoginCon } func (rc *LoginController) Post(c *gin.Context) { - loginBody := new(models.LoginModel) - if err := c.ShouldBindJSON(&loginBody); err != nil { + body := new(models.LoginModel) + if err := c.ShouldBindJSON(&body); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } - returnedUser, returnException := rc.UsersService.Login(loginBody) + returnedUser, exceptionReturn := rc.UsersService.Login(body) - if returnException.Message != "" { - c.JSON(returnException.StatusCode, returnException) + if exceptionReturn.Message != "" { + c.JSON(exceptionReturn.StatusCode, exceptionReturn) } else { c.JSON(200, returnedUser) } diff --git a/pkg/controllers/register.go b/pkg/controllers/register.go index fbb6740..ccdff4e 100644 --- a/pkg/controllers/register.go +++ b/pkg/controllers/register.go @@ -4,7 +4,6 @@ import ( "net/http" "wallet-api/pkg/models" "wallet-api/pkg/services" - "wallet-api/pkg/utl/common" "github.com/gin-gonic/gin" ) @@ -23,23 +22,18 @@ func NewRegisterController(rs *services.UsersService, s *gin.RouterGroup) *Regis } func (rc *RegisterController) Post(c *gin.Context) { - registerBody := createUserModel() - if err := c.ShouldBindJSON(®isterBody); err != nil { + body := new(models.UserModel) + body.Init() + if err := c.ShouldBindJSON(&body); err != nil { c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) return } - returnedUser, returnException := rc.UsersService.Create(®isterBody) + returnedUser, exceptionReturn := rc.UsersService.Create(body) - if returnException.Message != "" { - c.JSON(returnException.StatusCode, returnException) + if exceptionReturn.Message != "" { + c.JSON(exceptionReturn.StatusCode, exceptionReturn) } else { c.JSON(200, returnedUser.Payload()) } } - -func createUserModel() models.UserModel { - commonModel := common.CreateDbModel() - userModel := models.UserModel{CommonModel: commonModel} - return userModel -} diff --git a/pkg/controllers/wallets.go b/pkg/controllers/wallets.go new file mode 100644 index 0000000..8af66f5 --- /dev/null +++ b/pkg/controllers/wallets.go @@ -0,0 +1,44 @@ +package controllers + +import ( + "wallet-api/pkg/models" + "wallet-api/pkg/services" + + "github.com/gin-gonic/gin" +) + +type WalletsController struct { + WalletService *services.WalletService +} + +func NewWalletsController(as *services.WalletService, s *gin.RouterGroup) *WalletsController { + wc := new(WalletsController) + wc.WalletService = as + + s.POST("", wc.New) + s.GET("", wc.Get) + + return wc +} + +func (wc *WalletsController) New(c *gin.Context) { + body := new(models.AuthModel) + + get := c.MustGet("auth") + body.Id = get.(*models.AuthModel).Id + + wm := wc.WalletService.New(body) + c.JSON(200, wm) +} + +func (wc *WalletsController) Get(c *gin.Context) { + body := new(models.AuthModel) + + embed, _ := c.GetQuery("embed") + auth := c.MustGet("auth") + body.Id = auth.(*models.AuthModel).Id + + wm := wc.WalletService.Get(body, embed) + + c.JSON(200, wm) +} diff --git a/pkg/middleware/auth.go b/pkg/middleware/auth.go index 6ff3722..54f1d58 100644 --- a/pkg/middleware/auth.go +++ b/pkg/middleware/auth.go @@ -1,7 +1,58 @@ package middleware -import "github.com/gin-gonic/gin" +import ( + "fmt" + "os" + "strings" + "wallet-api/pkg/models" + "wallet-api/pkg/utl/configs" + + "github.com/dgrijalva/jwt-go" + + "github.com/gin-gonic/gin" +) func Auth(c *gin.Context) { + exceptionReturn := new(models.ExceptionModel) + tokenString := ExtractToken(c) + secret := os.Getenv("ACCESS_SECRET") + if secret == "" { + secret = configs.Secret + } + token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) { + _, ok := token.Method.(*jwt.SigningMethodHMAC) + println(ok) + if !ok { + return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"]) + } + return []byte(secret), nil + }) + if err != nil { + exceptionReturn.ErrorCode = "401001" + exceptionReturn.StatusCode = 401 + exceptionReturn.Message = "Invalid token" + c.AbortWithStatusJSON(exceptionReturn.StatusCode, exceptionReturn) + } + claims, ok := token.Claims.(jwt.MapClaims) + if ok && token.Valid { + userId, _ := claims["id"].(string) + authModel := new(models.AuthModel) + authModel.Id = userId + + c.Set("auth", authModel) + } + c.Next() +} + +func ExtractToken(c *gin.Context) string { + bearerToken := c.GetHeader("Authorization") + tokenArr := strings.Split(bearerToken, " ") + if len(tokenArr) == 2 { + bearerCheck := strings.ToLower(tokenArr[0]) + if bearerCheck == "bearer" { + return tokenArr[1] + } + } + return "" } diff --git a/pkg/migrate/migrate.go b/pkg/migrate/migrate.go index 1252628..0322ef4 100644 --- a/pkg/migrate/migrate.go +++ b/pkg/migrate/migrate.go @@ -9,7 +9,10 @@ import ( func Start(conn *pg.DB) { apiMigration := migrations.ApiMigration{Db: conn} usersMigration := migrations.UsersMigration{Db: conn} + walletsMigration := migrations.WalletsMigration{Db: conn} apiMigration.Create() usersMigration.Create() + walletsMigration.Create() + walletsMigration.PopulateTypes() } diff --git a/pkg/migrate/migrations/wallets.go b/pkg/migrate/migrations/wallets.go new file mode 100644 index 0000000..3b27abd --- /dev/null +++ b/pkg/migrate/migrations/wallets.go @@ -0,0 +1,40 @@ +package migrations + +import ( + "fmt" + "log" + "wallet-api/pkg/models" + + "github.com/go-pg/pg/v10" + "github.com/go-pg/pg/v10/orm" +) + +type WalletsMigration struct { + Db *pg.DB +} + +func (am *WalletsMigration) Create() { + models := []interface{}{ + (*models.WalletTypeModel)(nil), + (*models.WalletModel)(nil), + } + + for _, model := range models { + err := am.Db.Model(model).CreateTable(&orm.CreateTableOptions{ + IfNotExists: false, + FKConstraints: true, + }) + if err != nil { + log.Printf("Error Creating Table: %s", err) + } else { + fmt.Println("Table created successfully") + } + } +} + +func (am *WalletsMigration) PopulateTypes() { + walletTypeModel := new(models.WalletTypeModel) + walletTypeModel.Init() + walletTypeModel.Name = "Test" + am.Db.Model(walletTypeModel).Insert() +} diff --git a/pkg/models/auth.go b/pkg/models/auth.go index 5d75a10..f432118 100644 --- a/pkg/models/auth.go +++ b/pkg/models/auth.go @@ -8,3 +8,7 @@ type LoginModel struct { Email string Password string } + +type AuthModel struct { + Id string +} diff --git a/pkg/models/db.go b/pkg/models/db.go index 95d6b15..418c505 100644 --- a/pkg/models/db.go +++ b/pkg/models/db.go @@ -1,9 +1,20 @@ package models -import "time" +import ( + "time" + + "github.com/google/uuid" +) type CommonModel struct { - Id string `json:"id" pg:"id"` + Id string `json:"id" pg:"id,pk"` DateCreated time.Time `json:"dateCreated" pg:"datecreated"` DateUpdated time.Time `json:"dateUpdated" pg:"dateupdated"` } + +func (cm *CommonModel) Init() { + date := time.Now() + cm.Id = uuid.NewString() + cm.DateCreated = date + cm.DateUpdated = date +} diff --git a/pkg/models/register.go b/pkg/models/register.go index ed39369..2d3d8ec 100644 --- a/pkg/models/register.go +++ b/pkg/models/register.go @@ -3,9 +3,9 @@ package models type UserModel struct { tableName struct{} `pg:"users,alias:users"` CommonModel - Username string `json:"username"` - Password string `json:"password"` - Email string `json:"email"` + Username string `json:"username" pg:"username"` + Password string `json:"password" pg:"password"` + Email string `json:"email" pg:"email"` } type UserReturnInfoModel struct { @@ -15,11 +15,11 @@ type UserReturnInfoModel struct { Email string `json:"email"` } -func (um *UserModel) Payload() UserReturnInfoModel { - payload := UserReturnInfoModel{ - CommonModel: um.CommonModel, - Username: um.Username, - Email: um.Email, - } +func (um *UserModel) Payload() *UserReturnInfoModel { + payload := new(UserReturnInfoModel) + payload.CommonModel = um.CommonModel + payload.Username = um.Username + payload.Email = um.Email + return payload } diff --git a/pkg/models/wallets.go b/pkg/models/wallets.go new file mode 100644 index 0000000..03221ea --- /dev/null +++ b/pkg/models/wallets.go @@ -0,0 +1,16 @@ +package models + +type WalletModel struct { + tableName struct{} `pg:"wallets,alias:wallets"` + CommonModel + WalletTypeID string `json:"walletTypeId" pg:"wallet_type_id"` + WalletType *WalletTypeModel `json:"walletType" pg:"rel:has-one,fk:wallet_type_id"` + UserID string `json:"userId" pg:"user_id"` + User *UserReturnInfoModel `json:"user" pg:"rel:has-one,fk:user_id"` +} + +type WalletTypeModel struct { + tableName struct{} `pg:"walletTypes,alias:walletTypes"` + CommonModel + Name string `json:"name"` +} diff --git a/pkg/services/users.go b/pkg/services/users.go index 80cddbe..3f7741b 100644 --- a/pkg/services/users.go +++ b/pkg/services/users.go @@ -5,6 +5,7 @@ import ( "time" "wallet-api/pkg/models" "wallet-api/pkg/utl/common" + "wallet-api/pkg/utl/configs" jwt "github.com/dgrijalva/jwt-go" "golang.org/x/crypto/bcrypt" @@ -16,16 +17,16 @@ type UsersService struct { Db *pg.DB } -func (us *UsersService) Create(registerBody *models.UserModel) (*models.UserModel, models.ExceptionModel) { - var checkModel models.UserModel - var exceptionReturn models.ExceptionModel +func (us *UsersService) Create(registerBody *models.UserModel) (*models.UserModel, *models.ExceptionModel) { + check := new(models.UserModel) + exceptionReturn := new(models.ExceptionModel) - us.Db.Model(&checkModel).Where("? = ?", pg.Ident("username"), registerBody.Username).WhereOr("? = ?", pg.Ident("email"), registerBody.Email).Select() - if checkModel.Username != "" || checkModel.Email != "" { + us.Db.Model(check).Where("? = ?", pg.Ident("username"), registerBody.Username).WhereOr("? = ?", pg.Ident("email"), registerBody.Email).Select() + if check.Username != "" || check.Email != "" { exceptionReturn.Message = "User already exists" exceptionReturn.ErrorCode = "400101" exceptionReturn.StatusCode = 400 - return &checkModel, exceptionReturn + return check, exceptionReturn } hashedPassword, err := bcrypt.GenerateFromPassword([]byte(registerBody.Password), bcrypt.DefaultCost) @@ -43,27 +44,27 @@ func (us *UsersService) Create(registerBody *models.UserModel) (*models.UserMode return registerBody, exceptionReturn } -func (us *UsersService) Login(loginBody *models.LoginModel) (models.TokenModel, models.ExceptionModel) { - var checkModel models.UserModel - var exceptionReturn models.ExceptionModel - var tokenPayload models.TokenModel +func (us *UsersService) Login(loginBody *models.LoginModel) (*models.TokenModel, *models.ExceptionModel) { + check := new(models.UserModel) + exceptionReturn := new(models.ExceptionModel) + tokenPayload := new(models.TokenModel) - us.Db.Model(&checkModel).Where("? = ?", pg.Ident("email"), loginBody.Email).Select() - if checkModel.Email == "" { + us.Db.Model(check).Where("? = ?", pg.Ident("email"), loginBody.Email).Select() + if check.Email == "" { exceptionReturn.Message = "Email not found" exceptionReturn.ErrorCode = "400103" exceptionReturn.StatusCode = 400 return tokenPayload, exceptionReturn } - if bcrypt.CompareHashAndPassword([]byte(checkModel.Password), []byte(loginBody.Password)) != nil { + if bcrypt.CompareHashAndPassword([]byte(check.Password), []byte(loginBody.Password)) != nil { exceptionReturn.Message = "Incorrect password" exceptionReturn.ErrorCode = "400104" exceptionReturn.StatusCode = 400 return tokenPayload, exceptionReturn } - token, err := CreateToken(checkModel) + token, err := CreateToken(check) common.CheckError(err) tokenPayload.Token = token @@ -71,15 +72,15 @@ func (us *UsersService) Login(loginBody *models.LoginModel) (models.TokenModel, return tokenPayload, exceptionReturn } -func CreateToken(user models.UserModel) (string, error) { +func CreateToken(user *models.UserModel) (string, error) { atClaims := jwt.MapClaims{} atClaims["authorized"] = true atClaims["id"] = user.Id - atClaims["exp"] = time.Now().Add(time.Minute * 15).Unix() + atClaims["exp"] = time.Now().Add(time.Minute).Unix() secret := os.Getenv("ACCESS_SECRET") if secret == "" { - secret = "Dond3sta" + secret = configs.Secret } at := jwt.NewWithClaims(jwt.SigningMethodHS256, atClaims) diff --git a/pkg/services/wallets.go b/pkg/services/wallets.go new file mode 100644 index 0000000..868811e --- /dev/null +++ b/pkg/services/wallets.go @@ -0,0 +1,40 @@ +package services + +import ( + "wallet-api/pkg/models" + "wallet-api/pkg/utl/common" + + "github.com/go-pg/pg/v10" +) + +type WalletService struct { + Db *pg.DB +} + +func (as *WalletService) New(am *models.AuthModel) *models.WalletModel { + walletType := as.GetType() + + walletModel := new(models.WalletModel) + walletModel.Init() + walletModel.UserID = am.Id + walletModel.WalletTypeID = walletType.Id + as.Db.Model(walletModel).Insert() + return walletModel +} + +func (as *WalletService) Get(am *models.AuthModel, embed string) *models.WalletModel { + wm := new(models.WalletModel) + + query := as.Db.Model(wm).Where("? = ?", pg.Ident("user_id"), am.Id) + common.GenerateEmbed(query, embed).Select() + + return wm +} + +func (as *WalletService) GetType() *models.WalletTypeModel { + wt := new(models.WalletTypeModel) + + as.Db.Model(wt).Select() + + return wt +} diff --git a/pkg/utl/common/common.go b/pkg/utl/common/common.go index 2acdeab..0e898b1 100644 --- a/pkg/utl/common/common.go +++ b/pkg/utl/common/common.go @@ -2,10 +2,6 @@ package common import ( "log" - "time" - "wallet-api/pkg/models" - - "github.com/google/uuid" ) func CheckError(err error) { @@ -13,13 +9,3 @@ func CheckError(err error) { log.Fatalf("Error occured. %v", err) } } - -func CreateDbModel() models.CommonModel { - date := time.Now() - dbModel := models.CommonModel{ - Id: uuid.NewString(), - DateCreated: date, - DateUpdated: date, - } - return dbModel -} diff --git a/pkg/utl/common/embeds.go b/pkg/utl/common/embeds.go new file mode 100644 index 0000000..3ed83fa --- /dev/null +++ b/pkg/utl/common/embeds.go @@ -0,0 +1,17 @@ +package common + +import ( + "strings" + + "github.com/go-pg/pg/v10/orm" +) + +func GenerateEmbed(qr *orm.Query, embed string) *orm.Query{ + if embed != "" { + rels := strings.Split(embed, ",") + for _, rel := range rels { + qr = qr.Relation(rel) + } + } + return qr +} diff --git a/pkg/utl/configs/configs.go b/pkg/utl/configs/configs.go index c89ecf1..14b38f9 100644 --- a/pkg/utl/configs/configs.go +++ b/pkg/utl/configs/configs.go @@ -3,4 +3,5 @@ package configs const ( Version = "0.0.1" Prefix = "v1" + Secret = "Donde4sta" )