Add support for extensions to the HTTP protocol

strawberry/http/__init__.py

@@ -33,6 +33,7 @@ class GraphQLRequestData:
     query: Optional[str]
     variables: Optional[Dict[str, Any]]
     operation_name: Optional[str]
+    extensions: Optional[Dict[str, Any]]
 
 
 def parse_query_params(params: Dict[str, str]) -> Dict[str, Any]:
@@ -47,6 +48,7 @@ def parse_request_data(data: Mapping[str, Any]) -> GraphQLRequestData:
         query=data.get("query"),
         variables=data.get("variables"),
         operation_name=data.get("operationName"),
+        extensions=data.get("extensions"),
     )
 
 

strawberry/http/async_base_view.py

@@ -20,6 +20,7 @@
 from strawberry.schema.base import BaseSchema
 from strawberry.schema.exceptions import InvalidOperationTypeError
 from strawberry.types import ExecutionResult
+from strawberry.types.context_wrapper import ContextWrapper
 from strawberry.types.graphql import OperationType
 
 from .base import BaseView
@@ -102,11 +103,15 @@ async def execute_operation(
 
         assert self.schema
 
+        context_wrapper = ContextWrapper(
+            context=context, extensions=request_data.extensions
+        )
+
         return await self.schema.execute(
             request_data.query,
             root_value=root_value,
             variable_values=request_data.variables,
-            context_value=context,
+            context_value=context_wrapper,
             operation_name=request_data.operation_name,
             allowed_operation_types=allowed_operation_types,
         )
@@ -205,6 +210,7 @@ async def parse_http_body(
             query=data.get("query"),
             variables=data.get("variables"),
             operation_name=data.get("operationName"),
+            extensions=data.get("extensions"),
         )
 
     async def process_result(

strawberry/http/base.py

@@ -59,6 +59,12 @@ def parse_query_params(self, params: QueryParams) -> Dict[str, Any]:
             if variables:
                 params["variables"] = self.parse_json(variables)
 
+        if "extensions" in params:
+            extensions = params["extensions"]
+
+            if extensions:
+                params["extensions"] = self.parse_json(extensions)
+
         return params
 
     @property

strawberry/http/sync_base_view.py

@@ -21,6 +21,7 @@
 from strawberry.schema import BaseSchema
 from strawberry.schema.exceptions import InvalidOperationTypeError
 from strawberry.types import ExecutionResult
+from strawberry.types.context_wrapper import ContextWrapper
 from strawberry.types.graphql import OperationType
 
 from .base import BaseView
@@ -112,11 +113,15 @@ def execute_operation(
 
         assert self.schema
 
+        context_wrapper = ContextWrapper(
+            context=context, extensions=request_data.extensions
+        )
+
         return self.schema.execute_sync(
             request_data.query,
             root_value=root_value,
             variable_values=request_data.variables,
-            context_value=context,
+            context_value=context_wrapper,
             operation_name=request_data.operation_name,
             allowed_operation_types=allowed_operation_types,
         )
@@ -146,6 +151,7 @@ def parse_http_body(self, request: SyncHTTPRequestAdapter) -> GraphQLRequestData
             query=data.get("query"),
             variables=data.get("variables"),
             operation_name=data.get("operationName"),
+            extensions=data.get("extensions"),
         )
 
     def _handle_errors(

strawberry/types/context_wrapper.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+from typing import Any, Dict, Optional
+
+
+@dataclass
+class ContextWrapper:
+    context: Optional[Any]
+    extensions: Optional[Dict[str, Any]]

strawberry/types/info.py

@@ -16,6 +16,7 @@
 )
 from typing_extensions import TypeVar
 
+from .context_wrapper import ContextWrapper
 from .nodes import convert_selections
 
 if TYPE_CHECKING:
@@ -115,8 +116,19 @@
     @property
     def context(self) -> ContextType:
         """The context passed to the query execution."""
+        if isinstance(self._raw_info.context, ContextWrapper):
+            return self._raw_info.context.context
+
         return self._raw_info.context
 
+    @property
+    def input_extensions(self) -> Dict[str, Any]:
+        """The input extensions passed to the query execution."""
+        if isinstance(self._raw_info.context, ContextWrapper):
+            return self._raw_info.context.extensions
+
+        return {}
+
     @property
     def root_value(self) -> RootValueType:
         """The root value passed to the query execution."""

tests/http/clients/aiohttp.py

@@ -106,11 +106,16 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         async with TestClient(TestServer(self.app)) as client:
             body = self._build_body(
-                query=query, variables=variables, files=files, method=method
+                query=query,
+                variables=variables,
+                files=files,
+                method=method,
+                extensions=extensions,
             )
 
             if body and files:

tests/http/clients/asgi.py

@@ -97,10 +97,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         if method == "get":

tests/http/clients/base.py

@@ -55,6 +55,7 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response: ...
 
@@ -89,9 +90,15 @@ async def query(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
     ) -> Response:
         return await self._graphql_request(
-            method, query=query, headers=headers, variables=variables, files=files
+            method,
+            query=query,
+            headers=headers,
+            variables=variables,
+            files=files,
+            extensions=extensions,
         )
 
     def _get_headers(
@@ -117,6 +124,7 @@ def _build_body(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         method: Literal["get", "post"] = "post",
+        extensions: Optional[Dict[str, Any]] = None,
     ) -> Optional[Dict[str, object]]:
         if query is None:
             assert files is None
@@ -129,6 +137,9 @@ def _build_body(
         if variables:
             body["variables"] = variables
 
+        if extensions:
+            body["extensions"] = extensions
+
         if files:
             assert variables is not None
 
@@ -142,6 +153,9 @@ def _build_body(
         if method == "get" and variables:
             body["variables"] = json.dumps(variables)
 
+        if method == "get" and extensions:
+            body["extensions"] = json.dumps(extensions)
+
         return body
 
     @staticmethod

tests/http/clients/chalice.py

@@ -74,10 +74,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         data: Union[Dict[str, object], str, None] = None

tests/http/clients/channels.py

@@ -32,11 +32,16 @@
 
 
 def generate_get_path(
-    path, query: str, variables: Optional[Dict[str, Any]] = None
+    path,
+    query: str,
+    variables: Optional[Dict[str, Any]] = None,
+    extensions: Optional[Dict[str, Any]] = None,
 ) -> str:
     body: Dict[str, Any] = {"query": query}
     if variables is not None:
         body["variables"] = json_module.dumps(variables)
+    if extensions is not None:
+        body["extensions"] = json_module.dumps(extensions)
 
     parts = [f"{k}={v}" for k, v in body.items()]
     return f"{path}?{'&'.join(parts)}"
@@ -165,10 +170,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         headers = self._get_headers(method=method, headers=headers, files=files)
@@ -183,7 +193,7 @@ async def _graphql_request(
             endpoint_url = "/graphql"
         else:
             body = b""
-            endpoint_url = generate_get_path("/graphql", query, variables)
+            endpoint_url = generate_get_path("/graphql", query, variables, extensions)
 
         return await self.request(
             url=endpoint_url, method=method, body=body, headers=headers

tests/http/clients/django.py

@@ -101,13 +101,18 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         headers = self._get_headers(method=method, headers=headers, files=files)
         additional_arguments = {**kwargs, **headers}
 
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         data: Union[Dict[str, object], str, None] = None

tests/http/clients/fastapi.py

@@ -117,10 +117,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         if body:

tests/http/clients/flask.py

@@ -86,10 +86,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         data: Union[Dict[str, object], str, None] = None

tests/http/clients/litestar.py

@@ -98,10 +98,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         if body := self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         ):
             if method == "get":
                 kwargs["params"] = body

tests/http/clients/quart.py

@@ -79,10 +79,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         url = "/graphql"

tests/http/clients/sanic.py

@@ -76,10 +76,15 @@ async def _graphql_request(
         variables: Optional[Dict[str, object]] = None,
         files: Optional[Dict[str, BytesIO]] = None,
         headers: Optional[Dict[str, str]] = None,
+        extensions: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
     ) -> Response:
         body = self._build_body(
-            query=query, variables=variables, files=files, method=method
+            query=query,
+            variables=variables,
+            files=files,
+            method=method,
+            extensions=extensions,
         )
 
         if body: