Fix error handling in resource server

src/oidc_test/database.py

@@ -13,6 +13,7 @@ logger = logging.getLogger(__name__)
 class UserNotInDB(Exception):
     pass
 
+
 class Database:
     users: dict[str, User] = {}
     tokens: dict[str, OAuth2Token] = {}

src/oidc_test/main.py

@@ -56,7 +56,6 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(title="OIDC auth test", lifespan=lifespan)
 
-
 app.add_middleware(
     CORSMiddleware,
     allow_origins=settings.cors_origins,

src/oidc_test/resource_server.py

@@ -2,40 +2,59 @@ from datetime import datetime
 import logging
 
 from httpx import AsyncClient
-from fastapi import HTTPException, status
-from jwt import ExpiredSignatureError, InvalidKeyError, decode
 
 from .models import User
-from .auth_utils import oidc_providers_settings
-from .settings import settings
 
 logger = logging.getLogger(__name__)
 
 
-async def get_resource(id: str, user: User) -> dict:
+async def get_resource(resource_id: str, user: User) -> dict:
+    """
+    Resource processing: build an informative rely as a simple showcase
+    """
     pname = getattr(user.oidc_provider, "name", "?")
     resp = {
-        "hello": f"Hi {user.name} from an OAuth resource provider.",
+        "hello": f"Hi {user.name} from an OAuth resource provider",
-        "comment": f"I received a request for '{id}' with an access token signed by {pname}.",
+        "comment": f"I received a request for '{resource_id}' "
+        + f"with an access token signed by {pname}",
     }
-    scope = f"get:{id}"
+    # For the demo, resource resource_id matches a scope get:resource_id,
-    user_scopes = user.userinfo["scope"].split(" ")
+    # but this has to be refined for production
-    if scope in user_scopes:
+    required_scope = f"get:{resource_id}"
-        if id == "time":
+    # Check if the required scope is in the scopes allowed in userinfo
-            resp["time"] = datetime.now().strftime("%c")
+    if "required_scope" in user.userinfo:
-        elif id == "bs":
+        user_scopes = user.userinfo["required_scope"].split(" ")
-            async with AsyncClient() as client:
+        if required_scope in user_scopes:
-                bs = await client.get(
+            await process(user, required_scope, resp)
-                    "https://corporatebs-generator.sameerkumar.website/"
+        else:
+            ## For the showcase, giving a explanation.
+            ## Alternatively, raise HTTP_401_UNAUTHORIZED
+            resp["sorry"] = (
+                f"No scope {required_scope} in the access token "
+                + "but it is required for accessing this resource."
             )
+    else:
+        resp["sorry"] = "There is no scope in id token"
+    return resp
+
+
+async def process(user, resource_id, resp):
+    """
+    Too simple to be serious.
+    It's a good fit for a plugin architecture for production
+    """
+    assert user is not None
+    if resource_id == "time":
+        resp["time"] = datetime.now().strftime("%c")
+    elif resource_id == "bs":
+        async with AsyncClient() as client:
+            bs = await client.get("https://corporatebs-generator.sameerkumar.website/")
             resp["bs"] = bs.json().get("phrase", "Sorry, i am out of BS today.")
-        else:
-            resp["sorry"] = f"I don't known how to give '{id}' but i know corporate bs."
     else:
         resp["sorry"] = (
-            f"I don't serve the ressource {id} to you because there is no scope {scope} in the access token,"
+            f"I don't known how to give '{resource_id}' but i know corporate bs."
         )
-    return resp
+
 
 # assert user.oidc_provider is not None
 ### Get some info (TODO: refactor)