feat: [AUTH-5729] Scaffold idp for codegen (#284)

Author: max-stytch

stytch/b2b/api/idp.py

@@ -1,3 +1,9 @@
+# !!!
+# WARNING: This file is autogenerated
+# Only modify code within MANUAL() sections
+# or your changes may be overwritten later!
+# !!!
+
 from __future__ import annotations
 
 from typing import Any, Dict, Optional
@@ -25,26 +31,14 @@ def __init__(
         self.api_base = api_base
         self.sync_client = sync_client
         self.async_client = async_client
+        self.policy_cache = policy_cache
         self.jwks_client = jwks_client
         self.project_id = project_id
-        self.non_custom_claim_keys = [
-            "aud",
-            "exp",
-            "iat",
-            "iss",
-            "jti",
-            "nbf",
-            "sub",
-            "active",
-            "client_id",
-            "request_id",
-            "scope",
-            "status_code",
-            "token_type",
-            "https://stytch.com/organization",
-        ]
-        self.policy_cache = policy_cache
 
+    # MANUAL(IDP)(SERVICE_METHOD)
+    # ADDIMPORT: from stytch.b2b.models.sessions import AuthorizationCheck
+    # ADDIMPORT: from stytch.shared import jwt_helpers, rbac_local
+    # ADDIMPORT: from stytch.consumer.models.idp import IDPTokenClaims, IDPTokenResponse
     def introspect_token_network(
         self,
         token: str,
@@ -80,7 +74,7 @@ def introspect_token_network(
         if not jwtResponse.active:
             return None
         custom_claims = {
-            k: v for k, v in res.json.items() if k not in self.non_custom_claim_keys
+            k: v for k, v in res.json.items() if k not in self._non_custom_claim_keys()
         }
         organization_claim = res.json["https://stytch.com/organization"]
         organization_id = organization_claim["organization_id"]
@@ -142,7 +136,7 @@ async def introspect_token_network_async(
         if not jwtResponse.active:
             return None
         custom_claims = {
-            k: v for k, v in res.json.items() if k not in self.non_custom_claim_keys
+            k: v for k, v in res.json.items() if k not in self._non_custom_claim_keys()
         }
         organization_claim = res.json["https://stytch.com/organization"]
         organization_id = organization_claim["organization_id"]
@@ -222,3 +216,22 @@ def introspect_access_token_local(
             token_type="access_token",
             organization_claim=org_claim,
         )
+
+    def _non_custom_claim_keys(self):
+        return [
+            "aud",
+            "exp",
+            "iat",
+            "iss",
+            "jti",
+            "nbf",
+            "sub",
+            "active",
+            "client_id",
+            "request_id",
+            "scope",
+            "status_code",
+            "token_type",
+        ]
+
+    # ENDMANUAL(IDP)

stytch/consumer/api/idp.py

@@ -1,3 +1,9 @@
+# !!!
+# WARNING: This file is autogenerated
+# Only modify code within MANUAL() sections
+# or your changes may be overwritten later!
+# !!!
+
 from __future__ import annotations
 
 from typing import Any, Dict, Optional
@@ -25,25 +31,14 @@ def __init__(
         self.api_base = api_base
         self.sync_client = sync_client
         self.async_client = async_client
+        self.policy_cache = policy_cache
         self.jwks_client = jwks_client
         self.project_id = project_id
-        self.non_custom_claim_keys = [
-            "aud",
-            "exp",
-            "iat",
-            "iss",
-            "jti",
-            "nbf",
-            "sub",
-            "active",
-            "client_id",
-            "request_id",
-            "scope",
-            "status_code",
-            "token_type",
-        ]
-        self.policy_cache = policy_cache
 
+    # MANUAL(IDP)(SERVICE_METHOD)
+    # ADDIMPORT: from stytch.shared import jwt_helpers, rbac_local
+    # ADDIMPORT: from stytch.consumer.models.idp import IDPTokenClaims, IDPTokenResponse
+    # ADDIMPORT: from stytch.consumer.models.sessions import AuthorizationCheck
     def introspect_token_network(
         self,
         token: str,
@@ -78,7 +73,7 @@ def introspect_token_network(
         res = self.sync_client.post_form(url, data, headers)
         jwtResponse = IDPTokenResponse.from_json(res.response.status_code, res.json)
         custom_claims = {
-            k: v for k, v in res.json.items() if k not in self.non_custom_claim_keys
+            k: v for k, v in res.json.items() if k not in self._non_custom_claim_keys()
         }
         if not jwtResponse.active:
             return None
@@ -135,7 +130,7 @@ async def introspect_token_network_async(
         res = await self.async_client.post_form(url, data, headers)
         jwtResponse = IDPTokenResponse.from_json(res.response.status, res.json)
         custom_claims = {
-            k: v for k, v in res.json.items() if k not in self.non_custom_claim_keys
+            k: v for k, v in res.json.items() if k not in self._non_custom_claim_keys()
         }
         if not jwtResponse.active:
             return None
@@ -207,3 +202,22 @@ def introspect_access_token_local(
             not_before=generic_claims.reserved_claims["nbf"],
             token_type="access_token",
         )
+
+    def _non_custom_claim_keys(self):
+        return [
+            "aud",
+            "exp",
+            "iat",
+            "iss",
+            "jti",
+            "nbf",
+            "sub",
+            "active",
+            "client_id",
+            "request_id",
+            "scope",
+            "status_code",
+            "token_type",
+        ]
+
+    # ENDMANUAL(IDP)

stytch/consumer/models/idp.py

@@ -1,10 +1,28 @@
+# !!!
+# WARNING: This file is autogenerated
+# Only modify code within MANUAL() sections
+# or your changes may be overwritten later!
+# !!!
+
+from __future__ import annotations
+
 from typing import Any, Dict, List, Optional
 
 import pydantic
 
 from stytch.core.response_base import ResponseBase
 
 
+class ScopeResult(pydantic.BaseModel):
+    scope: str
+    description: str
+    is_grantable: bool
+
+
+# MANUAL(IDP)(TYPES)
+# ADDIMPORT: from stytch.core.response_base import ResponseBase
+
+
 class IDPTokenResponse(ResponseBase):
     """Response type for `IDP.introspect_token_network`.
     Fields:
@@ -54,3 +72,6 @@ class IDPTokenClaims(pydantic.BaseModel):
     not_before: Optional[int]
     token_type: Optional[str]
     organization_claim: Optional[Dict[str, Any]] = None
+
+
+# ENDMANUAL(IDP)