feat: add validate token in routes

dlmoraes · Oct 13, 2023 · 433ae58 · 433ae58
1 parent 8829120
commit 433ae58
Show file tree

Hide file tree

Showing 2 changed files with 62 additions and 0 deletions.
diff --git a/api/account.go b/api/account.go
@@ -6,6 +6,7 @@ import (
 	"time"
 
 	db "github.com/dlmoraes/gofinance-backend/db/sqlc"
+	"github.com/dlmoraes/gofinance-backend/util"
 	"github.com/gin-gonic/gin"
 )
 
@@ -20,6 +21,11 @@ type createAccountRequest struct {
 }
 
 func (server *Server) createAccount(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req createAccountRequest
 	err := ctx.ShouldBindJSON(&req)
 	if err != nil {
@@ -65,6 +71,11 @@ type getAccountRequest struct {
 }
 
 func (server *Server) getAccount(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getAccountRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {
@@ -90,6 +101,11 @@ type deleteAccountRequest struct {
 }
 
 func (server *Server) deleteAccount(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getAccountRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {
@@ -118,6 +134,11 @@ type updateAccountRequest struct {
 }
 
 func (server *Server) updateAccount(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req updateAccountRequest
 	err := ctx.ShouldBindJSON(&req)
 	if err != nil {
@@ -150,6 +171,11 @@ type getAccountsRequest struct {
 }
 
 func (server *Server) getAccounts(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getAccountsRequest
 	err := ctx.ShouldBindJSON(&req)
 
@@ -182,6 +208,11 @@ type getAccountsGraphRequest struct {
 }
 
 func (server *Server) getAccountGraph(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getAccountsGraphRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {
@@ -208,6 +239,11 @@ type getAccountsReportsRequest struct {
 }
 
 func (server *Server) getAccountsReports(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getAccountsReportsRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {

diff --git a/api/category.go b/api/category.go
@@ -5,6 +5,7 @@ import (
 	"net/http"
 
 	db "github.com/dlmoraes/gofinance-backend/db/sqlc"
+	"github.com/dlmoraes/gofinance-backend/util"
 	"github.com/gin-gonic/gin"
 )
 
@@ -16,6 +17,11 @@ type createCategoryRequest struct {
 }
 
 func (server *Server) createCategory(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req createCategoryRequest
 	err := ctx.ShouldBindJSON(&req)
 	if err != nil {
@@ -43,6 +49,11 @@ type getCategoryRequest struct {
 }
 
 func (server *Server) getCategory(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getCategoryRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {
@@ -68,6 +79,11 @@ type deleteCategoryRequest struct {
 }
 
 func (server *Server) deleteCategory(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getCategoryRequest
 	err := ctx.ShouldBindUri(&req)
 	if err != nil {
@@ -95,6 +111,11 @@ type updateCategoryRequest struct {
 }
 
 func (server *Server) updateCategory(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req updateCategoryRequest
 	err := ctx.ShouldBindJSON(&req)
 	if err != nil {
@@ -124,6 +145,11 @@ type getCategoriesRequest struct {
 }
 
 func (server *Server) getCategories(ctx *gin.Context) {
+	errOnValidateToken := util.GetTokenInHeaderAndVerify(ctx)
+	if errOnValidateToken != nil {
+		return
+	}
+
 	var req getCategoriesRequest
 	err := ctx.ShouldBindJSON(&req)
 	if err != nil {