Merge branch 'develop' into main

cmd/api/main.go

@@ -20,7 +20,7 @@ func main() {
 	r := gin.New()
 	r.Use(middleware.CORSMiddleware())
 
-	conn := db.CreateConnection(dbUrl, ctx)
+	conn := db.CreateConnection(ctx, dbUrl)
 	api.Init(r, conn)
 
 	server.Start(r)

cmd/migrate/main.go

@@ -15,6 +15,6 @@ func main() {
 	dbUrl := os.Getenv("DATABASE_URL")
 	ctx := context.Background()
 
-	conn := db.CreateConnection(dbUrl, ctx)
+	conn := db.CreateConnection(ctx, dbUrl)
 	migrate.Start(conn, "")
 }

pkg/api/routes.go

@@ -21,6 +21,11 @@ func Routes(s *gin.Engine, db *pg.DB) {
 	transactionType := ver.Group("transaction-type", middleware.Auth)
 	subscription := ver.Group("subscription", middleware.Auth)
 	subscriptionType := ver.Group("subscription-type", middleware.Auth)
+	
+	s.NoRoute(func(c *gin.Context) {
+		c.JSON(404, gin.H{"code": "PAGE_NOT_FOUND", "message": "Page not found"})
+	})
+	
 
 	apiService := services.ApiService{Db: db}
 	usersService := services.UsersService{Db: db}

pkg/controllers/api.go

@@ -11,6 +11,7 @@ type ApiController struct {
 	ApiService *services.ApiService
 }
 
+// Initializes ApiController.
 func NewApiController(as *services.ApiService, s *gin.RouterGroup) *ApiController {
 	ac := new(ApiController)
 	ac.ApiService = as
@@ -21,11 +22,15 @@ func NewApiController(as *services.ApiService, s *gin.RouterGroup) *ApiControlle
 	return ac
 }
 
+// ROUTE (GET /api).
 func (ac *ApiController) getFirst(c *gin.Context) {
 	apiModel := ac.ApiService.GetFirst(c)
 	c.JSON(200, apiModel)
 }
 
+// ROUTE (POST /api/migrate).
+//
+// Requires "SECRET_CODE", "VERSION" (optional) from body.
 func (ac *ApiController) postMigrate(c *gin.Context) {
 	migrateModel := c.MustGet("migrate")
 	version := migrateModel.(middleware.SecretCodeModel).Version

pkg/controllers/auth.go

@@ -13,6 +13,7 @@ type AuthController struct {
 	UsersService *services.UsersService
 }
 
+// Initializes AuthController.
 func NewAuthController(rs *services.UsersService, s *gin.RouterGroup) *AuthController {
 	rc := new(AuthController)
 	rc.UsersService = rs
@@ -25,6 +26,7 @@ func NewAuthController(rs *services.UsersService, s *gin.RouterGroup) *AuthContr
 	return rc
 }
 
+// ROUTE (POST /auth/login).
 func (rc *AuthController) PostLogin(c *gin.Context) {
 	body := new(models.Login)
 	if err := c.ShouldBind(&body); err != nil {
@@ -40,6 +42,7 @@ func (rc *AuthController) PostLogin(c *gin.Context) {
 	}
 }
 
+// ROUTE (POST /auth/register).
 func (rc *AuthController) PostRegister(c *gin.Context) {
 	body := new(models.User)
 	body.Init()
@@ -56,6 +59,8 @@ func (rc *AuthController) PostRegister(c *gin.Context) {
 		c.JSON(200, returnedUser.Payload())
 	}
 }
+
+// ROUTE (DELETE /auth/deactivate).
 func (rc *AuthController) Delete(c *gin.Context) {
 	auth := new(models.Auth)
 	authGet := c.MustGet("auth")
@@ -70,6 +75,7 @@ func (rc *AuthController) Delete(c *gin.Context) {
 	}
 }
 
+// ROUTE (GET /auth/check-token).
 func (rc *AuthController) CheckToken(c *gin.Context) {
 	token, _ := c.GetQuery("token")
 	re := new(models.CheckToken)

pkg/controllers/controllers.go

@@ -10,6 +10,7 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+// Gets query parameters and populates FilteredResponse model.
 func FilteredResponse(c *gin.Context) *models.FilteredResponse {
 	filtered := new(models.FilteredResponse)
 	page, _ := c.GetQuery("page")

pkg/controllers/subscriptionTypes.go

@@ -12,6 +12,7 @@ type SubscriptionTypeController struct {
 	SubscriptionTypeService *services.SubscriptionTypeService
 }
 
+// Initializes SubscriptionTypeController.
 func NewSubscriptionTypeController(as *services.SubscriptionTypeService, s *gin.RouterGroup) *SubscriptionTypeController {
 	wc := new(SubscriptionTypeController)
 	wc.SubscriptionTypeService = as
@@ -22,6 +23,7 @@ func NewSubscriptionTypeController(as *services.SubscriptionTypeService, s *gin.
 	return wc
 }
 
+// ROUTE (POST /subscription-types)
 func (wc *SubscriptionTypeController) New(c *gin.Context) {
 	body := new(models.NewSubscriptionTypeBody)
 	if err := c.ShouldBind(body); err != nil {
@@ -33,6 +35,7 @@ func (wc *SubscriptionTypeController) New(c *gin.Context) {
 	c.JSON(200, wm)
 }
 
+// ROUTE (GET /subscription-types)
 func (wc *SubscriptionTypeController) GetAll(c *gin.Context) {
 	embed, _ := c.GetQuery("embed")
 

pkg/controllers/subscriptions.go

@@ -12,16 +12,25 @@ type SubscriptionController struct {
 	SubscriptionService *services.SubscriptionService
 }
 
+// Initializes SubscriptionController.
 func NewSubscriptionController(as *services.SubscriptionService, s *gin.RouterGroup) *SubscriptionController {
 	wc := new(SubscriptionController)
 	wc.SubscriptionService = as
 
 	s.POST("", wc.New)
+	s.PUT("/:id", wc.Edit)
+	s.GET("/:id", wc.Get)
 	s.GET("", wc.GetAll)
 
+	se := s.Group("/end")
+	{
+		se.PUT("/:id", wc.End)
+	}
+
 	return wc
 }
 
+// ROUTE (POST /subscription)
 func (wc *SubscriptionController) New(c *gin.Context) {
 	body := new(models.NewSubscriptionBody)
 	if err := c.ShouldBind(body); err != nil {
@@ -33,6 +42,59 @@ func (wc *SubscriptionController) New(c *gin.Context) {
 	c.JSON(200, wm)
 }
 
+// ROUTE (PUT /subscription/:id)
+func (wc *SubscriptionController) Edit(c *gin.Context) {
+	body := new(models.SubscriptionEdit)
+	if err := c.ShouldBind(body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	id := c.Param("id")
+
+	wm := wc.SubscriptionService.Edit(c, body, id)
+	c.JSON(200, wm)
+}
+
+// ROUTE (GET /subscription/:id)
+func (wc *SubscriptionController) Get(c *gin.Context) {
+	body := new(models.Auth)
+	params := new(models.Params)
+
+	auth := c.MustGet("auth")
+	body.Id = auth.(*models.Auth).Id
+
+	id := c.Param("id")
+
+	embed, _ := c.GetQuery("embed")
+	params.Embed = embed
+
+	fr := wc.SubscriptionService.Get(c, body, id, params)
+
+	c.JSON(200, fr)
+}
+
+// ROUTE (PUT /subscription/end/:id)
+func (wc *SubscriptionController) End(c *gin.Context) {
+	body := new(models.Auth)
+
+	auth := c.MustGet("auth")
+	body.Id = auth.(*models.Auth).Id
+
+	end := new(models.SubscriptionEnd)
+	if err := c.ShouldBind(end); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	id := c.Param("id")
+
+	fr := wc.SubscriptionService.End(c, id)
+
+	c.JSON(200, fr)
+}
+
+// ROUTE (GET /subscription)
 func (wc *SubscriptionController) GetAll(c *gin.Context) {
 	body := new(models.Auth)
 	auth := c.MustGet("auth")

pkg/controllers/transactionTypes.go

@@ -12,6 +12,7 @@ type TransactionTypeController struct {
 	TransactionTypeService *services.TransactionTypeService
 }
 
+// Initializes TransactionTypeController.
 func NewTransactionTypeController(as *services.TransactionTypeService, s *gin.RouterGroup) *TransactionTypeController {
 	wc := new(TransactionTypeController)
 	wc.TransactionTypeService = as
@@ -22,6 +23,7 @@ func NewTransactionTypeController(as *services.TransactionTypeService, s *gin.Ro
 	return wc
 }
 
+// ROUTE (POST /transaction-types)
 func (wc *TransactionTypeController) New(c *gin.Context) {
 	body := new(models.NewTransactionTypeBody)
 	if err := c.ShouldBind(body); err != nil {
@@ -33,6 +35,7 @@ func (wc *TransactionTypeController) New(c *gin.Context) {
 	c.JSON(200, wm)
 }
 
+// ROUTE (GET /transaction-types)
 func (wc *TransactionTypeController) GetAll(c *gin.Context) {
 	embed, _ := c.GetQuery("embed")
 

pkg/controllers/transactions.go

@@ -12,16 +12,20 @@ type TransactionController struct {
 	TransactionService *services.TransactionService
 }
 
+// Initializes TransactionController.
 func NewTransactionController(as *services.TransactionService, s *gin.RouterGroup) *TransactionController {
 	wc := new(TransactionController)
 	wc.TransactionService = as
 
 	s.POST("", wc.New)
 	s.GET("", wc.GetAll)
+	s.PUT("/:id", wc.Edit)
+	s.GET("/:id", wc.Get)
 
 	return wc
 }
 
+// ROUTE (POST /transactions)
 func (wc *TransactionController) New(c *gin.Context) {
 	body := new(models.NewTransactionBody)
 	if err := c.ShouldBind(body); err != nil {
@@ -33,6 +37,7 @@ func (wc *TransactionController) New(c *gin.Context) {
 	c.JSON(200, wm)
 }
 
+// ROUTE (GET /transactions)
 func (wc *TransactionController) GetAll(c *gin.Context) {
 	body := new(models.Auth)
 	auth := c.MustGet("auth")
@@ -45,3 +50,35 @@ func (wc *TransactionController) GetAll(c *gin.Context) {
 
 	c.JSON(200, fr)
 }
+
+// ROUTE (PUT /transactions/:id)
+func (wc *TransactionController) Edit(c *gin.Context) {
+	body := new(models.TransactionEdit)
+	if err := c.ShouldBind(body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	id := c.Param("id")
+
+	wm := wc.TransactionService.Edit(c, body, id)
+	c.JSON(200, wm)
+}
+
+// ROUTE (GET /transactions/:id)
+func (wc *TransactionController) Get(c *gin.Context) {
+	body := new(models.Auth)
+	params := new(models.Params)
+
+	auth := c.MustGet("auth")
+	body.Id = auth.(*models.Auth).Id
+
+	id := c.Param("id")
+
+	embed, _ := c.GetQuery("embed")
+	params.Embed = embed
+
+	fr := wc.TransactionService.Get(c, body, id, params)
+
+	c.JSON(200, fr)
+}

pkg/controllers/wallets-header.go

@@ -11,6 +11,7 @@ type WalletsHeaderController struct {
 	WalletService *services.WalletService
 }
 
+// Initializes WalletsHeaderController.
 func NewWalletsHeaderController(as *services.WalletService, s *gin.RouterGroup) *WalletsHeaderController {
 	wc := new(WalletsHeaderController)
 	wc.WalletService = as
@@ -20,6 +21,7 @@ func NewWalletsHeaderController(as *services.WalletService, s *gin.RouterGroup)
 	return wc
 }
 
+// ROUTE (GET /wallet/wallet-header)
 func (wc *WalletsHeaderController) Get(c *gin.Context) {
 	body := new(models.Auth)
 

pkg/controllers/wallets.go

@@ -12,16 +12,20 @@ type WalletsController struct {
 	WalletService *services.WalletService
 }
 
+// Initializes WalletsController.
 func NewWalletsController(as *services.WalletService, s *gin.RouterGroup) *WalletsController {
 	wc := new(WalletsController)
 	wc.WalletService = as
 
 	s.POST("", wc.New)
 	s.GET("", wc.GetAll)
+	s.PUT("/:id", wc.Edit)
+	s.GET("/:id", wc.Get)
 
 	return wc
 }
 
+// ROUTE (POST /wallet)
 func (wc *WalletsController) New(c *gin.Context) {
 	body := new(models.NewWalletBody)
 
@@ -37,18 +41,7 @@ func (wc *WalletsController) New(c *gin.Context) {
 	c.JSON(200, wm)
 }
 
-func (wc *WalletsController) Get(c *gin.Context) {
-	body := new(models.Auth)
-
-	embed, _ := c.GetQuery("embed")
-	auth := c.MustGet("auth")
-	body.Id = auth.(*models.Auth).Id
-
-	wm := wc.WalletService.Get(c, body, embed)
-
-	c.JSON(200, wm)
-}
-
+// ROUTE (GET /wallet)
 func (wc *WalletsController) GetAll(c *gin.Context) {
 	body := new(models.Auth)
 	auth := c.MustGet("auth")
@@ -59,5 +52,32 @@ func (wc *WalletsController) GetAll(c *gin.Context) {
 	wc.WalletService.GetAll(c, body, fr)
 
 	c.JSON(200, fr)
-
+}
+
+// ROUTE (PUT /wallet/:id)
+func (wc *WalletsController) Edit(c *gin.Context) {
+	body := new(models.WalletEdit)
+	if err := c.ShouldBind(body); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	id := c.Param("id")
+
+	wm := wc.WalletService.Edit(c, body, id)
+	c.JSON(200, wm)
+}
+
+// ROUTE (GET /wallet/:id)
+func (wc *WalletsController) Get(c *gin.Context) {
+	params := new(models.Params)
+
+	id := c.Param("id")
+
+	embed, _ := c.GetQuery("embed")
+	params.Embed = embed
+
+	fr := wc.WalletService.Get(c, id, params)
+
+	c.JSON(200, fr)
 }

pkg/middleware/auth.go

@@ -12,6 +12,9 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+// Auth Middleware.
+//
+// Checks if token from header is valid and extracts the id.
 func Auth(c *gin.Context) {
 	exceptionReturn := new(models.Exception)
 	tokenString := ExtractToken(c)
@@ -34,6 +37,7 @@ func Auth(c *gin.Context) {
 	c.Next()
 }
 
+// Extracts token from header
 func ExtractToken(c *gin.Context) string {
 	bearerToken := c.GetHeader("Authorization")
 	tokenArr := strings.Split(bearerToken, " ")
@@ -46,6 +50,7 @@ func ExtractToken(c *gin.Context) string {
 	return ""
 }
 
+// Checks if token is valid
 func CheckToken(tokenString string) (*jwt.Token, error) {
 	secret := os.Getenv("ACCESS_SECRET")
 	if secret == "" {

pkg/middleware/cors.go

@@ -2,6 +2,9 @@ package middleware
 
 import "github.com/gin-gonic/gin"
 
+// CORS Middleware.
+//
+// Add needed headers to make cors functioning.
 func CORSMiddleware() gin.HandlerFunc {
 	return func(c *gin.Context) {
 		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")

pkg/middleware/secretCode.go

@@ -9,6 +9,9 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
+// Secret Code Middleware.
+//
+// Checks if secret code from body is valid.
 func SecretCode(c *gin.Context) {
 	exceptionReturn := new(models.Exception)
 	secretCode := ExtractCode(c)
@@ -26,6 +29,7 @@ func SecretCode(c *gin.Context) {
 	c.Next()
 }
 
+// Extracts the secret code from body
 func ExtractCode(c *gin.Context) SecretCodeModel {
 	secret := new(SecretCodeModel)
 	if err := c.ShouldBindJSON(&secret); err != nil {
@@ -37,5 +41,5 @@ func ExtractCode(c *gin.Context) SecretCodeModel {
 
 type SecretCodeModel struct {
 	SecretCode string `json:"secretCode"`
-	Version string `json:"version"`
+	Version    string `json:"version"`
 }

pkg/migrate/1_create_table_api.go

@@ -2,13 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
+// Creates api table if it does not exist.
 func CreateTableApi(db pg.DB) error {
 
 	models := []interface{}{
@@ -20,10 +22,10 @@ func CreateTableApi(db pg.DB) error {
 			IfNotExists: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"api\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"api\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/2_create_table_users.go

@@ -2,14 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
-
+// Creates api users if it does not exist.
 func CreateTableUsers(db pg.DB) error {
 	models := []interface{}{
 		(*models.User)(nil),
@@ -20,10 +21,10 @@ func CreateTableUsers(db pg.DB) error {
 			IfNotExists: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"users\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"users\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/3_create_table_wallets.go

@@ -2,13 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
+// Creates wallets table if it does not exist.
 func CreateTableWallets(db pg.DB) error {
 	models := []interface{}{
 		(*models.Wallet)(nil),
@@ -20,10 +22,10 @@ func CreateTableWallets(db pg.DB) error {
 			FKConstraints: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"wallets\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"wallets\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/4_create_table_transaction_types.go

@@ -2,13 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
+// Creates transactionTypes table if it does not exist.
 func CreateTableTransactionTypes(db pg.DB) error {
 	models := []interface{}{
 		(*models.TransactionType)(nil),
@@ -20,10 +22,10 @@ func CreateTableTransactionTypes(db pg.DB) error {
 			FKConstraints: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"api\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"api\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/5_create_table_transactions.go

@@ -2,14 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
-
+// Creates api transactions if it does not exist.
 func CreateTableTransactions(db pg.DB) error {
 	models := []interface{}{
 		(*models.Transaction)(nil),
@@ -21,10 +22,10 @@ func CreateTableTransactions(db pg.DB) error {
 			FKConstraints: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"transactions\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"transactions\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/6_create_table_subscription_types.go

@@ -2,13 +2,15 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
 	"log"
 	"wallet-api/pkg/models"
 
+	"github.com/go-pg/pg/v10"
+
 	"github.com/go-pg/pg/v10/orm"
 )
 
+// Creates subscriptionTypes table if it does not exist.
 func CreateTableSubscriptionTypes(db pg.DB) error {
 	models := []interface{}{
 		(*models.SubscriptionType)(nil),
@@ -20,10 +22,10 @@ func CreateTableSubscriptionTypes(db pg.DB) error {
 			FKConstraints: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"subscriptionTypes\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"subscriptionTypes\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/7_create_table_subscriptions.go

@@ -2,12 +2,14 @@ package migrate
 
 import (
 	"fmt"
-	"github.com/go-pg/pg/v10"
-	"github.com/go-pg/pg/v10/orm"
 	"log"
 	"wallet-api/pkg/models"
+
+	"github.com/go-pg/pg/v10"
+	"github.com/go-pg/pg/v10/orm"
 )
 
+// Creates subscriptions table if it does not exist.
 func CreateTableSubscriptions(db pg.DB) error {
 	models := []interface{}{
 		(*models.Subscription)(nil),
@@ -19,10 +21,10 @@ func CreateTableSubscriptions(db pg.DB) error {
 			FKConstraints: true,
 		})
 		if err != nil {
-			log.Printf("Error Creating Table: %s", err)
+			log.Printf("Error creating table \"subscriptions\": %s", err)
 			return err
 		} else {
-			fmt.Println("Table created successfully")
+			fmt.Println("Table \"subscriptions\" created successfully")
 		}
 	}
 	return nil

pkg/migrate/8_populate_subscription_types.go

@@ -1,10 +1,14 @@
 package migrate
 
 import (
-	"github.com/go-pg/pg/v10"
+	"fmt"
+	"log"
 	"wallet-api/pkg/models"
+
+	"github.com/go-pg/pg/v10"
 )
 
+// Populates subscriptionTypes table if it does not exist.
 func PopulateSubscriptionTypes(db pg.DB) error {
 	daily := new(models.SubscriptionType)
 	weekly := new(models.SubscriptionType)
@@ -28,12 +32,35 @@ func PopulateSubscriptionTypes(db pg.DB) error {
 	yearly.Type = "yearly"
 
 	_, err := db.Model(daily).Where("? = ?", pg.Ident("type"), daily.Type).SelectOrInsert()
-
+	if err != nil {
+		log.Printf("Error inserting row into \"subscriptionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"subscriptionTypes\" table.")
+	}
 	_, err = db.Model(weekly).Where("? = ?", pg.Ident("type"), weekly.Type).SelectOrInsert()
+	if err != nil {
+		log.Printf("Error inserting row into \"subscriptionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"subscriptionTypes\" table.")
+	}
 
 	_, err = db.Model(monthly).Where("? = ?", pg.Ident("type"), monthly.Type).SelectOrInsert()
+	if err != nil {
+		log.Printf("Error inserting row into \"subscriptionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"subscriptionTypes\" table.")
+	}
 
 	_, err = db.Model(yearly).Where("? = ?", pg.Ident("type"), yearly.Type).SelectOrInsert()
+	if err != nil {
+		log.Printf("Error inserting row into \"subscriptionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"subscriptionTypes\" table.")
+	}
 
 	return err
 }

pkg/migrate/9_populate_transaction_types.go

@@ -1,10 +1,14 @@
 package migrate
 
 import (
-	"github.com/go-pg/pg/v10"
+	"fmt"
+	"log"
 	"wallet-api/pkg/models"
+
+	"github.com/go-pg/pg/v10"
 )
 
+// Populates transactionTypes table if it does not exist.
 func PopulateTransactionTypes(db pg.DB) error {
 	gain := new(models.TransactionType)
 	expense := new(models.TransactionType)
@@ -18,8 +22,20 @@ func PopulateTransactionTypes(db pg.DB) error {
 	expense.Type = "expense"
 
 	_, err := db.Model(gain).Where("? = ?", pg.Ident("type"), gain.Type).SelectOrInsert()
+	if err != nil {
+		log.Printf("Error inserting row into \"transactionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"transactionTypes\" table.")
+	}
 
 	_, err = db.Model(expense).Where("? = ?", pg.Ident("type"), expense.Type).SelectOrInsert()
+	if err != nil {
+		log.Printf("Error inserting row into \"transactionTypes\" table: %s", err)
+		return err
+	} else {
+		fmt.Println("Row inserted successfully into \"transactionTypes\" table.")
+	}
 
 	return err
 }

pkg/migrate/migrate.go

@@ -4,6 +4,7 @@ import (
 	"github.com/go-pg/pg/v10"
 )
 
+// Starts database migration.
 func Start(conn *pg.DB, version string) {
 	migration001 := Migration{
 		Version: "001",
@@ -43,7 +44,6 @@ func Start(conn *pg.DB, version string) {
 }
 
 type Migration struct {
-	Version string
+	Version    string
 	Migrations []interface{}
 }
-

pkg/models/auth.go

@@ -5,8 +5,9 @@ type Token struct {
 }
 
 type Login struct {
-	Email    string `form:"email"`
-	Password string `form:"password"`
+	Email    string `json:"email" form:"email"`
+	Password string `json:"password" form:"password"`
+	RememberMe bool `json:"rememberMe" form:"rememberMe"`
 }
 
 type Auth struct {

pkg/models/models.go

@@ -4,11 +4,7 @@ import "github.com/gin-gonic/gin"
 
 type FilteredResponse struct {
 	Items        interface{} `json:"items"`
-	SortBy       string      `json:"sortBy"`
-	Embed        string      `json:"embed"`
-	Page         int         `json:"page"`
-	Rpp          int         `json:"rpp"`
-	TotalRecords int         `json:"totalRecords"`
+	Params
 }
 
 type ResponseFunc func(*gin.Context) *[]interface{}
@@ -16,3 +12,11 @@ type ResponseFunc func(*gin.Context) *[]interface{}
 type MessageResponse struct {
 	Message string `json:"message"`
 }
+
+type Params struct {
+	SortBy       string      `json:"sortBy"`
+	Embed        string      `json:"embed"`
+	Page         int         `json:"page"`
+	Rpp          int         `json:"rpp"`
+	TotalRecords int         `json:"totalRecords"`
+}

pkg/models/subscriptions.go

@@ -23,6 +23,16 @@ type Subscription struct {
 	Amount              float32           `json:"amount", pg:"amount,default:0"`
 }
 
+type SubscriptionEdit struct {
+	tableName   struct{}    `pg:"subscriptions,alias:subscriptions"`
+	Id          string      `json:"id" form:"id"`
+	Description string      `json:"description" form:"description"`
+	EndDate     time.Time   `json:"endDate" form:"endDate" `
+	HasEnd      bool        `json:"hasEnd" form:"hasEnd"`
+	WalletID    string      `json:"walletId" form:"walletId"`
+	Amount      json.Number `json:"amount" form:"amount"`
+}
+
 type NewSubscriptionBody struct {
 	WalletID           string      `json:"walletId" form:"walletId"`
 	TransactionTypeID  string      `json:"transactionTypeId" form:"transactionTypeId"`
@@ -35,6 +45,10 @@ type NewSubscriptionBody struct {
 	Amount             json.Number `json:"amount" form:"amount"`
 }
 
+type SubscriptionEnd struct {
+	Id string `json:"id" form:"id"`
+}
+
 func (cm *Subscription) ToTrans() *Transaction {
 	trans := new(Transaction)
 	trans.Init()
@@ -60,7 +74,7 @@ func (cm *Subscription) HasNew() bool {
 			}
 			return false
 		case "weekly":
-			lastDate := time.Now().AddDate(0, 0, -(7*cm.CustomRange))
+			lastDate := time.Now().AddDate(0, 0, -(7 * cm.CustomRange))
 			if cm.LastTransactionDate.Before(lastDate) {
 				return true
 			}
@@ -80,4 +94,4 @@ func (cm *Subscription) HasNew() bool {
 		}
 	}
 	return true
-}
+}

pkg/models/transactions.go

@@ -26,3 +26,12 @@ type NewTransactionBody struct {
 	Description       string      `json:"description" form:"description"`
 	Amount            json.Number `json:"amount" form:"amount"`
 }
+
+type TransactionEdit struct {
+	Id                string      `json:"id" form:"id"`
+	WalletID          string      `json:"walletId" form:"walletId"`
+	TransactionTypeID string      `json:"transactionTypeId" form:"transactionTypeId"`
+	TransactionDate   time.Time   `json:"transactionDate" form:"transactionDate"`
+	Description       string      `json:"description" form:"description"`
+	Amount            json.Number `json:"amount" form:"amount"`
+}

pkg/models/wallets.go

@@ -28,3 +28,8 @@ type WalletTransactions struct {
 	LastMonth      float32
 	NextMonth      float32
 }
+
+type WalletEdit struct {
+	Id   string `json:"id" form:"id"`
+	Name string `json:"name" form:"name"`
+}

pkg/services/api.go

@@ -12,6 +12,7 @@ type ApiService struct {
 	Db *pg.DB
 }
 
+// Gets first row from API table.
 func (as *ApiService) GetFirst(ctx context.Context) models.ApiModel {
 	db := as.Db.WithContext(ctx)
 
@@ -20,6 +21,9 @@ func (as *ApiService) GetFirst(ctx context.Context) models.ApiModel {
 	return apiModel
 }
 
+// Starts database migration.
+//
+// Takes migration version.
 func (as *ApiService) PostMigrate(ctx context.Context, version string) (*models.MessageResponse, *models.Exception) {
 	db := as.Db.WithContext(ctx)
 

pkg/services/services.go

@@ -7,6 +7,7 @@ import (
 	"github.com/go-pg/pg/v10"
 )
 
+// Adds filters to query and executes it.
 func FilteredResponse(qry *pg.Query, mdl interface{}, filtered *models.FilteredResponse) {
 	if filtered.Page == 0 {
 		filtered.Page = 1

pkg/services/subscriptionTypes.go

@@ -12,6 +12,7 @@ type SubscriptionTypeService struct {
 	Db *pg.DB
 }
 
+// Inserts new row to subscription type table.
 func (as *SubscriptionTypeService) New(ctx context.Context, body *models.NewSubscriptionTypeBody) *models.SubscriptionType {
 	db := as.Db.WithContext(ctx)
 
@@ -26,6 +27,7 @@ func (as *SubscriptionTypeService) New(ctx context.Context, body *models.NewSubs
 	return tm
 }
 
+// Gets all rows from subscription type table.
 func (as *SubscriptionTypeService) GetAll(ctx context.Context, embed string) *[]models.SubscriptionType {
 	db := as.Db.WithContext(ctx)
 

pkg/services/subscriptions.go

@@ -5,6 +5,7 @@ import (
 	"math"
 	"time"
 	"wallet-api/pkg/models"
+	"wallet-api/pkg/utl/common"
 
 	"github.com/go-pg/pg/v10"
 )
@@ -13,6 +14,7 @@ type SubscriptionService struct {
 	Db *pg.DB
 }
 
+// Inserts new row to subscription table.
 func (as *SubscriptionService) New(ctx context.Context, body *models.NewSubscriptionBody) *models.Subscription {
 	db := as.Db.WithContext(ctx)
 
@@ -47,6 +49,29 @@ func (as *SubscriptionService) New(ctx context.Context, body *models.NewSubscrip
 	return tm
 }
 
+// Gets row from subscription table by id.
+func (as *SubscriptionService) Get(ctx context.Context, am *models.Auth, id string, params *models.Params) *models.Subscription {
+	db := as.Db.WithContext(ctx)
+
+	wm := new(models.Subscription)
+	wm.Id = id
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	qry := tx.Model(wm)
+	common.GenerateEmbed(qry, params.Embed).WherePK().Select()
+
+	if (*wm).HasNew() {
+		as.SubToTrans(wm, tx)
+	}
+
+	tx.Commit()
+
+	return wm
+}
+
+// Gets filtered rows from subscription table.
 func (as *SubscriptionService) GetAll(ctx context.Context, am *models.Auth, walletId string, filtered *models.FilteredResponse) {
 	db := as.Db.WithContext(ctx)
 
@@ -69,6 +94,52 @@ func (as *SubscriptionService) GetAll(ctx context.Context, am *models.Auth, wall
 	tx.Commit()
 }
 
+// Updates row from subscription table by id.
+func (as *SubscriptionService) Edit(ctx context.Context, body *models.SubscriptionEdit, id string) *models.Subscription {
+	db := as.Db.WithContext(ctx)
+
+	amount, _ := body.Amount.Float64()
+
+	tm := new(models.Subscription)
+	tm.Id = id
+	tm.EndDate = body.EndDate
+	tm.HasEnd = body.HasEnd
+	tm.Description = body.Description
+	tm.WalletID = body.WalletID
+	tm.Amount = float32(math.Round(amount*100) / 100)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tx.Model(tm).WherePK().UpdateNotZero()
+
+	tx.Commit()
+
+	return tm
+}
+
+// Updates row in subscription table by id.
+//
+// Ends subscription with current date.
+func (as *SubscriptionService) End(ctx context.Context, id string) *models.Subscription {
+	db := as.Db.WithContext(ctx)
+
+	tm := new(models.Subscription)
+	tm.Id = id
+	tm.EndDate = time.Now()
+	tm.HasEnd = true
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tx.Model(tm).WherePK().UpdateNotZero()
+
+	tx.Commit()
+
+	return tm
+}
+
+// Generates and Inserts new Transaction rows from the subscription model.
 func (as *SubscriptionService) SubToTrans(subModel *models.Subscription, tx *pg.Tx) {
 
 	now := time.Now()

pkg/services/transactionTypes.go

@@ -12,6 +12,7 @@ type TransactionTypeService struct {
 	Db *pg.DB
 }
 
+// Inserts new row to transaction type table.
 func (as *TransactionTypeService) New(ctx context.Context, body *models.NewTransactionTypeBody) *models.TransactionType {
 	db := as.Db.WithContext(ctx)
 
@@ -26,6 +27,7 @@ func (as *TransactionTypeService) New(ctx context.Context, body *models.NewTrans
 	return tm
 }
 
+// Gets all rows from transaction type table.
 func (as *TransactionTypeService) GetAll(ctx context.Context, embed string) *[]models.TransactionType {
 	db := as.Db.WithContext(ctx)
 

pkg/services/transactions.go

@@ -5,6 +5,7 @@ import (
 	"math"
 	"time"
 	"wallet-api/pkg/models"
+	"wallet-api/pkg/utl/common"
 
 	"github.com/go-pg/pg/v10"
 )
@@ -14,11 +15,15 @@ type TransactionService struct {
 	Ss *SubscriptionService
 }
 
+// Inserts new row to transaction table.
 func (as *TransactionService) New(ctx context.Context, body *models.NewTransactionBody) *models.Transaction {
 	db := as.Db.WithContext(ctx)
 
 	tm := new(models.Transaction)
 
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
 	amount, _ := body.Amount.Float64()
 
 	tm.Init()
@@ -32,11 +37,13 @@ func (as *TransactionService) New(ctx context.Context, body *models.NewTransacti
 		tm.TransactionDate = time.Now()
 	}
 
-	db.Model(tm).Insert()
+	tx.Model(tm).Insert()
+	tx.Commit()
 
 	return tm
 }
 
+// Gets filtered rows from transaction table.
 func (as *TransactionService) GetAll(ctx context.Context, am *models.Auth, walletId string, filtered *models.FilteredResponse) {
 	db := as.Db.WithContext(ctx)
 
@@ -67,3 +74,45 @@ func (as *TransactionService) GetAll(ctx context.Context, am *models.Auth, walle
 
 	tx.Commit()
 }
+
+// Updates row in transaction table by id.
+func (as *TransactionService) Edit(ctx context.Context, body *models.TransactionEdit, id string) *models.Transaction {
+	db := as.Db.WithContext(ctx)
+
+	amount, _ := body.Amount.Float64()
+
+	tm := new(models.Transaction)
+	tm.Id = id
+	tm.Description = body.Description
+	tm.WalletID = body.WalletID
+	tm.TransactionTypeID = body.TransactionTypeID
+	tm.TransactionDate = body.TransactionDate
+	tm.Amount = float32(math.Round(amount*100) / 100)
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tx.Model(tm).WherePK().UpdateNotZero()
+
+	tx.Commit()
+
+	return tm
+}
+
+// Gets row from transaction table by id.
+func (as *TransactionService) Get(ctx context.Context, am *models.Auth, id string, params *models.Params) *models.Transaction {
+	db := as.Db.WithContext(ctx)
+
+	wm := new(models.Transaction)
+	wm.Id = id
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	qry := tx.Model(wm)
+	common.GenerateEmbed(qry, params.Embed).WherePK().Select()
+
+	tx.Commit()
+
+	return wm
+}

pkg/services/users.go

@@ -18,6 +18,7 @@ type UsersService struct {
 	Db *pg.DB
 }
 
+// Inserts new row to users table.
 func (us *UsersService) Create(ctx context.Context, registerBody *models.User) (*models.User, *models.Exception) {
 	db := us.Db.WithContext(ctx)
 
@@ -52,6 +53,7 @@ func (us *UsersService) Create(ctx context.Context, registerBody *models.User) (
 	return registerBody, exceptionReturn
 }
 
+// Gets row from users table by email and valid password.
 func (us *UsersService) Login(ctx context.Context, loginBody *models.Login) (*models.Token, *models.Exception) {
 	db := us.Db.WithContext(ctx)
 
@@ -81,7 +83,7 @@ func (us *UsersService) Login(ctx context.Context, loginBody *models.Login) (*mo
 		return tokenPayload, exceptionReturn
 	}
 
-	token, err := CreateToken(check)
+	token, err := CreateToken(check, loginBody.RememberMe)
 	common.CheckError(err)
 
 	tokenPayload.Token = token
@@ -89,6 +91,9 @@ func (us *UsersService) Login(ctx context.Context, loginBody *models.Login) (*mo
 	return tokenPayload, exceptionReturn
 }
 
+// Updates row in users table.
+//
+// IsActive column is set to false
 func (us *UsersService) Deactivate(ctx context.Context, auth *models.Auth) (*models.MessageResponse, *models.Exception) {
 	db := us.Db.WithContext(ctx)
 
@@ -124,11 +129,18 @@ func (us *UsersService) Deactivate(ctx context.Context, auth *models.Auth) (*mod
 	return mm, me
 }
 
-func CreateToken(user *models.User) (string, error) {
+// Generates new jwt token.
+//
+// It encodes the user id. Based on rememberMe it is valid through 48hours or 2hours.
+func CreateToken(user *models.User, rememberMe bool) (string, error) {
 	atClaims := jwt.MapClaims{}
 	atClaims["authorized"] = true
 	atClaims["id"] = user.Id
-	atClaims["exp"] = time.Now().Add(time.Hour * 2).Unix()
+	if rememberMe {
+		atClaims["exp"] = time.Now().Add(time.Hour * 48).Unix()
+	} else {
+		atClaims["exp"] = time.Now().Add(time.Hour * 2).Unix()
+	}
 
 	secret := os.Getenv("ACCESS_SECRET")
 	if secret == "" {

pkg/services/wallets.go

@@ -14,6 +14,7 @@ type WalletService struct {
 	Ss *SubscriptionService
 }
 
+// Inserts row to wallets table.
 func (as *WalletService) New(ctx context.Context, am *models.NewWalletBody) *models.Wallet {
 	db := as.Db.WithContext(ctx)
 
@@ -25,17 +26,43 @@ func (as *WalletService) New(ctx context.Context, am *models.NewWalletBody) *mod
 	return walletModel
 }
 
-func (as *WalletService) Get(ctx context.Context, am *models.Auth, embed string) *models.Wallet {
+// Updates row in wallets table by id.
+func (as *WalletService) Edit(ctx context.Context, body *models.WalletEdit, id string) *models.Wallet {
+	db := as.Db.WithContext(ctx)
+
+	tm := new(models.Wallet)
+	tm.Id = id
+	tm.Name = body.Name
+
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	tx.Model(tm).WherePK().UpdateNotZero()
+
+	tx.Commit()
+
+	return tm
+}
+
+// Gets row in wallets table by id.
+func (as *WalletService) Get(ctx context.Context, id string, params *models.Params) *models.Wallet {
 	db := as.Db.WithContext(ctx)
 
 	wm := new(models.Wallet)
+	wm.Id = id
 
-	query := db.Model(wm).Where("? = ?", pg.Ident("user_id"), am.Id)
-	common.GenerateEmbed(query, embed).Select()
+	tx, _ := db.Begin()
+	defer tx.Rollback()
+
+	qry := tx.Model(wm)
+	common.GenerateEmbed(qry, params.Embed).WherePK().Select()
+
+	tx.Commit()
 
 	return wm
 }
 
+// Gets filtered rows from wallets table.
 func (as *WalletService) GetAll(ctx context.Context, am *models.Auth, filtered *models.FilteredResponse) {
 	db := as.Db.WithContext(ctx)
 	wm := new([]models.Wallet)
@@ -44,6 +71,9 @@ func (as *WalletService) GetAll(ctx context.Context, am *models.Auth, filtered *
 	FilteredResponse(query, wm, filtered)
 }
 
+// Gets row from wallets table.
+//
+// Calculates previous month, current and next month totals.
 func (as *WalletService) GetHeader(ctx context.Context, am *models.Auth, walletId string) *models.WalletHeader {
 	db := as.Db.WithContext(ctx)
 
@@ -148,10 +178,12 @@ func (as *WalletService) GetHeader(ctx context.Context, am *models.Auth, walletI
 	wm.Currency = "USD"
 	wm.WalletId = walletId
 
-
 	return wm
 }
 
+// Appends Transaction to the belonging walletId
+//
+// If missing, it creates the item list.
 func addWhere(s *[]models.WalletTransactions, walletId string, e models.Transaction) {
 	var exists bool
 	for a := range *s {

pkg/utl/configs/configs.go

@@ -1,7 +1,7 @@
 package configs
 
 const (
-	Version    = "0.0.73"
+	Version    = "0.0.81"
 	Prefix     = "v1"
 	Secret     = "Donde4sta"
 	SecretCode = "brasno"

pkg/utl/db/db.go

@@ -8,7 +8,7 @@ import (
 	"github.com/go-pg/pg/v10"
 )
 
-func CreateConnection(dbUrl string, ctx context.Context) *pg.DB {
+func CreateConnection(ctx context.Context, dbUrl string) *pg.DB {
 	opt, err := pg.ParseURL(dbUrl)
 	common.CheckError(err)
 	conn := pg.Connect(opt)