added additional parameters for token generation

app/utils/auth_utils.py

@@ -35,20 +35,18 @@ def validate_mobile(mobile: str) -> bool:
     """
     return mobile.isdigit() and len(mobile) == 10
 
-def generate_tokens(identifier: str) -> dict:
-    """
-    Generate access and refresh tokens.
-
-    Args:
-        identifier (str): Email or mobile number.
-
-    Returns:
-        dict: Access and refresh tokens.
-    """
+def generate_tokens(identifier: str, merchant_id: str = None, role: str = None) -> dict:
     access_token_expiry = datetime.utcnow() + timedelta(minutes=30)
     refresh_token_expiry = datetime.utcnow() + timedelta(days=1)
 
-    access_token = jwt.encode({"sub": identifier, "exp": access_token_expiry}, SECRET_KEY, algorithm=ALGORITHM)
+    payload = {"sub": identifier, "exp": access_token_expiry}
+    
+    if merchant_id:
+        payload["merchant_id"] = merchant_id
+    if role:
+        payload["role"] = role
+
+    access_token = jwt.encode(payload, SECRET_KEY, algorithm=ALGORITHM)
     refresh_token = jwt.encode({"sub": identifier, "exp": refresh_token_expiry}, SECRET_KEY, algorithm=ALGORITHM)
 
     return {"access_token": access_token, "refresh_token": refresh_token, "token_type": "bearer"}