36 lines
953 B
Python
36 lines
953 B
Python
from aws_lambda_powertools.event_handler.api_gateway import (
|
|
APIGatewayHttpResolver,
|
|
Response,
|
|
)
|
|
from aws_lambda_powertools.event_handler.middlewares import (
|
|
BaseMiddlewareHandler,
|
|
NextMiddleware,
|
|
)
|
|
from pydantic import UUID4, BaseModel, EmailStr, Field
|
|
|
|
|
|
class User(BaseModel):
|
|
id: str | UUID4 = Field(alias='sub')
|
|
name: str
|
|
email: EmailStr
|
|
email_verified: bool
|
|
|
|
|
|
class AuthenticationMiddleware(BaseMiddlewareHandler):
|
|
"""This middleware extracts user authentication details from
|
|
the jwt_claim authorizer context and makes them available
|
|
in the application context.
|
|
"""
|
|
|
|
def handler(
|
|
self,
|
|
app: APIGatewayHttpResolver,
|
|
next_middleware: NextMiddleware,
|
|
) -> Response:
|
|
jwt_claim = app.current_event.request_context.authorizer.jwt_claim
|
|
|
|
if jwt_claim:
|
|
app.append_context(user=User.model_validate(jwt_claim))
|
|
|
|
return next_middleware(app)
|