feat: jwt middleware

middleware/auth.go

@@ -0,0 +1,29 @@
+package middleware
+
+import (
+	"github.com/gin-gonic/gin"
+	"nCovTrack-Backend/utils"
+	"net/http"
+)
+
+const UNAUTH_MSG = "unauthorized"
+
+func Auth() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		oldToken := c.Request.Header["Token"]
+		c.Writer.Header().Set("X-Token", "")
+		if len(oldToken) != 1 || oldToken[0] == "" {
+			utils.Err(c, http.StatusUnauthorized, http.StatusUnauthorized, UNAUTH_MSG)
+			c.Abort()
+			return
+		}
+		renewToken := utils.RenewToken(oldToken[0])
+		if renewToken == "" {
+			utils.Err(c, http.StatusUnauthorized, http.StatusUnauthorized, UNAUTH_MSG)
+			c.Abort()
+			return
+		}
+		c.Writer.Header().Set("X-Token", renewToken)
+		c.Next()
+	}
+}