add team and role id

middleware/auth/auth.go

@@ -10,6 +10,8 @@ import (
 
 const (
 	CtxRequestHeaderUserId        = "user_id"
+	CtxRequestHeaderTeamId        = "team_id"
+	CtxRequestHeaderRoleId        = "role_id"
 	ctxRequestHeaderAuthorization = "Authorization"
 	ctxRequestCookieAuthorization = "ak"
 	ctxRequestTokenExpired        = "expired"
@@ -58,6 +60,20 @@ func Auth(authKey string, session Session) gin.HandlerFunc {
 				ctx.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"msg": "auth failed, mapClaims[CtxRequestHeaderUserId].(float64) error"})
 				return
 			}
+
+			if tid, ok := mapClaims[CtxRequestHeaderTeamId].(float64); ok {
+				ctx.Set(CtxRequestHeaderTeamId, int64(tid))
+			} else {
+				ctx.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"msg": "auth failed, mapClaims[CtxRequestHeaderTeamId].(float64) error"})
+				return
+			}
+
+			if rid, ok := mapClaims[CtxRequestHeaderRoleId].(float64); ok {
+				ctx.Set(CtxRequestHeaderRoleId, int64(rid))
+			} else {
+				ctx.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"msg": "auth failed, mapClaims[CtxRequestHeaderRoleId].(float64) error"})
+				return
+			}
 		} else {
 			ctx.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"msg": "auth failed, token.Claims.(jwt.MapClaims) error"})
 			return

middleware/auth/optional_auth.go

@@ -38,6 +38,14 @@ func OptionalAuth(authKey string) gin.HandlerFunc {
 				if uid, ok := mapClaims[CtxRequestHeaderUserId].(float64); ok {
 					ctx.Set(CtxRequestHeaderUserId, int64(uid))
 				}
+
+				if tid, ok := mapClaims[CtxRequestHeaderTeamId].(float64); ok {
+					ctx.Set(CtxRequestHeaderTeamId, int64(tid))
+				}
+
+				if rid, ok := mapClaims[CtxRequestHeaderRoleId].(float64); ok {
+					ctx.Set(CtxRequestHeaderRoleId, int64(rid))
+				}
 			}
 		}
 	}