Adding new class to WebSocket Authorizer

Author: leandrodamascena

docs/utilities/data_classes.md

@@ -131,12 +131,18 @@ It is used for [API Gateway Rest API Lambda Authorizer payload](https://docs.aws
 
 Use **`APIGatewayAuthorizerRequestEvent`** for type `REQUEST` and **`APIGatewayAuthorizerTokenEvent`** for type `TOKEN`.
 
-=== "app.py"
+=== "Rest APIs"
 
-    ```python hl_lines="2-4 8"
+    ```python hl_lines="2-4 8 18"
     --8<-- "examples/event_sources/src/apigw_authorizer_request.py"
     ```
 
+=== "WebSocket APIs"
+
+    ```python hl_lines="2-4 8 18"
+    --8<-- "examples/event_sources/src/apigw_authorizer_request_websocket.py"
+    ```
+
 === "API Gateway Authorizer Request Example Event"
 
     ```json hl_lines="3 11"

examples/event_sources/src/apigw_authorizer_request_websocket.py

@@ -0,0 +1,29 @@
+from aws_lambda_powertools.utilities.data_classes import event_source
+from aws_lambda_powertools.utilities.data_classes.api_gateway_authorizer_event import (
+    APIGatewayAuthorizerRequestEvent,
+    APIGatewayAuthorizerResponseWebSocket,
+)
+
+
+@event_source(data_class=APIGatewayAuthorizerRequestEvent)
+def lambda_handler(event: APIGatewayAuthorizerRequestEvent, context):
+    # Simple auth check (replace with your actual auth logic)
+    is_authorized = event.headers.get("HeaderAuth1") == "headerValue1"
+
+    if not is_authorized:
+        return {"principalId": "", "policyDocument": {"Version": "2012-10-17", "Statement": []}}
+
+    arn = event.parsed_arn
+
+    policy = APIGatewayAuthorizerResponseWebSocket(
+        principal_id="user",
+        context={"user": "example"},
+        region=arn.region,
+        aws_account_id=arn.aws_account_id,
+        api_id=arn.api_id,
+        stage=arn.stage,
+    )
+
+    policy.allow_all_routes()
+
+    return policy.asdict()