fix: Manually test the SDK and fix bugs

Author: KShivendu

setup.py

@@ -99,7 +99,7 @@
     ],
     keywords="",
     install_requires=[
-        "PyJWT>=2.0.0 ,<3.0.0",
+        "PyJWT>=2.6.0 ,<3.0.0",
         "httpx>=0.15.0 ,<0.24.0",
         "pycryptodome==3.10.*",
         "tldextract==3.1.0",

supertokens_python/recipe/session/access_token.py

@@ -82,7 +82,7 @@ def get_info_from_access_token(
                 )
                 payload = jwt.decode(  # type: ignore
                     jwt_info.raw_token_string,
-                    matching_key,
+                    matching_key.key,  # type: ignore
                     algorithms=["RS256"],
                     options={"verify_signature": True, "verify_exp": True},
                 )

supertokens_python/recipe/session/jwks.py

@@ -70,7 +70,7 @@ def get_latest_keys(self) -> List[PyJWK]:
 
         return all_keys
 
-    def get_matching_key_from_jwt(self, token: str) -> str:
+    def get_matching_key_from_jwt(self, token: str) -> PyJWK:
         header = decode_token(token, options={"verify_signature": False})["header"]
         kid: str = header["kid"]  # type: ignore
 
@@ -80,15 +80,15 @@ def get_matching_key_from_jwt(self, token: str) -> str:
         assert self.jwk_set is not None
 
         try:
-            return self.jwk_set[kid].key  # type: ignore
-        except KeyError:
+            return self.jwk_set[kid]  # type: ignore
+        except IndexError:
             if not self.is_cooling_down():
                 # One more attempt to fetch the latest keys
                 # and then try to find the key again.
                 self.reload()
                 try:
-                    return str(self.jwk_set[kid].key)  # type: ignore
-                except KeyError:
+                    return self.jwk_set[kid]  # type: ignore
+                except IndexError:
                     pass
 
         raise JWKSKeyNotFoundError("No key found for the given kid")

supertokens_python/recipe/session/recipe_implementation.py

@@ -59,9 +59,7 @@ def __init__(self, querier: Querier, config: SessionConfig, app_info: AppInfo):
         self.config = config
         self.app_info = app_info
 
-    @property
-    def JWK_clients(self) -> List[JWKClient]:
-        return [
+        self.JWK_clients = [
             JWKClient(
                 uri,
                 cooldown_duration=JWKRequestCooldownInMs,
@@ -238,6 +236,7 @@ async def get_session(
             self,
             self.config,
             access_token_str,
+            # FIXME: expiry time cannot be None
             build_front_token(response.session.userId, expiry_time, payload),
             None,  # refresh_token
             anti_csrf_token,

supertokens_python/recipe/session/session_functions.py

@@ -61,7 +61,11 @@ def __init__(
 
 class GetSessionAPIResponseSession:
     def __init__(
-        self, handle: str, userId: str, userDataInJWT: Dict[str, Any], expiryTime: int
+        self,
+        handle: str,
+        userId: str,
+        userDataInJWT: Dict[str, Any],
+        expiryTime: Optional[int] = None,
     ) -> None:
         self.handle = handle
         self.userId = userId
@@ -175,8 +179,8 @@ async def get_session(
 
         payload = parsed_access_token.payload
 
-        time_created = payload["timeCreated"]
-        expiry_time = payload["expiryTime"]
+        time_created = payload.get("timeCreated")
+        expiry_time = payload.get("expiryTime")
 
         if not isinstance(time_created, int) or not isinstance(expiry_time, int):
             raise e
@@ -276,8 +280,7 @@ async def get_session(
             GetSessionAPIResponseSession(
                 response["session"]["handle"],
                 response["session"]["userId"],
-                response["session"]["userData"],
-                response["session"]["expiresAt"],
+                response["session"]["userDataInJWT"],
             ),
             GetSessionAPIResponseAccessToken(
                 response["accessToken"]["token"],