From 9531cc14fb69f1e5f1b285f1ba72d10f5facc439 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=20Jurmanovi=C4=87?= <fjurma12@outlook.com>
Date: Tue, 25 May 2021 22:45:27 +0200
Subject: [PATCH] secret code from body instead header

---
 pkg/middleware/secretCode.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/pkg/middleware/secretCode.go b/pkg/middleware/secretCode.go
index fcf8f9f..304fb39 100644
--- a/pkg/middleware/secretCode.go
+++ b/pkg/middleware/secretCode.go
@@ -1,6 +1,7 @@
 package middleware
 
 import (
+	"net/http"
 	"os"
 	"wallet-api/pkg/models"
 	"wallet-api/pkg/utl/configs"
@@ -15,7 +16,6 @@ func SecretCode(c *gin.Context) {
 	if secret == "" {
 		secret = configs.SecretCode
 	}
-	print(secret, secretCode)
 	if secret != secretCode {
 		exceptionReturn.ErrorCode = "401101"
 		exceptionReturn.StatusCode = 401
@@ -26,6 +26,14 @@ func SecretCode(c *gin.Context) {
 }
 
 func ExtractCode(c *gin.Context) string {
-	secret := c.GetHeader("SECRET_CODE")
-	return secret
+	secret := new(SecretCodeModel)
+	if err := c.ShouldBindJSON(&secret); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return ""
+	}
+	return secret.SecretCode
+}
+
+type SecretCodeModel struct {
+	SecretCode string `json:"secretCode"`
 }