feat: Add role_code to auth context

trianggianggara · Jul 12, 2024 · 88d101a · 88d101a
1 parent b44dd46
commit 88d101a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/internal/delivery/middleware/auth.go b/internal/delivery/middleware/auth.go
@@ -65,10 +65,19 @@ func Auth(next echo.HandlerFunc) echo.HandlerFunc {
 			email = ""
 		}
 
+		var roleCode string
+		destructRoleCode := token.Claims.(jwt.MapClaims)["role_code"]
+		if destructRoleCode != nil {
+			roleCode = destructRoleCode.(string)
+		} else {
+			roleCode = ""
+		}
+
 		authCtx := &baseModel.AuthContext{
 			ID:    id,
 			Name:  name,
 			Email: email,
+			RoleCode: roleCode,
 		}
 		ctx := ctxval.SetAuthValue(c.Request().Context(), authCtx)
 		newRequest := c.Request().WithContext(ctx)

diff --git a/internal/model/base/context.go b/internal/model/base/context.go
@@ -6,9 +6,10 @@ import (
 )
 
 type AuthContext struct {
-	ID    uuid.UUID
-	Name  string
-	Email string
+	ID       uuid.UUID
+	Name     string
+	Email    string
+	RoleCode string
 }
 
 type TrxContext struct {