[auth] Add custom header injection to JsonRpcBase (API key support)

xrpl/asyncio/clients/client.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import Optional
+from typing import Optional, Dict
 
 from typing_extensions import Final, Self
 
@@ -24,14 +24,24 @@ class Client(ABC):
     :meta private:
     """
 
-    def __init__(self: Self, url: str) -> None:
+    def __init__(
+        self: Self,
+        url: str,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> None:
         """
         Initializes a client.
 
         Arguments:
-            url: The url to which this client will connect
+            url: The URL to which this client will connect.
+            headers: Optional dictionary of default headers to include with each request.
+                These can be used to authenticate with private XRPL nodes or pass
+                custom metadata, such as:
+                - {"Authorization": "Bearer <token>"}
         """
         self.url = url
+        self.headers = headers or {}
         self.network_id: Optional[int] = None
         self.build_version: Optional[str] = None
 
@@ -66,7 +76,6 @@ async def get_network_id_and_build_version(client: Client) -> None:
     Raises:
         XRPLRequestFailureException: if the rippled API call fails.
     """
-    # the required values are already present, no need for further processing
     if client.network_id and client.build_version:
         return
 

xrpl/asyncio/clients/json_rpc_base.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from json import JSONDecodeError
+from typing import Optional, Dict
 
 from httpx import AsyncClient
 from typing_extensions import Self
@@ -21,16 +22,37 @@ class JsonRpcBase(Client):
     :meta private:
     """
 
+    def __init__(
+        self: Self,
+        url: str,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+    ) -> None:
+        """
+        Initializes a new JsonRpcBase client.
+
+        Arguments:
+            url: The URL of the XRPL node to connect to.
+            headers: Optional default headers for all requests (e.g. API key or Dhali payment-claim).
+        """
+        super().__init__(url)
+        self.headers = headers or {}
+
     async def _request_impl(
-        self: Self, request: Request, *, timeout: float = REQUEST_TIMEOUT
+        self: Self,
+        request: Request,
+        *,
+        timeout: float = REQUEST_TIMEOUT,
+        headers: Optional[Dict[str, str]] = None,
     ) -> Response:
         """
         Base ``_request_impl`` implementation for JSON RPC.
 
         Arguments:
             request: An object representing information about a rippled request.
             timeout: The duration within which we expect to hear a response from the
-            rippled validator.
+                rippled server.
+            headers: Optional additional headers to include for this request.
 
         Returns:
             The response from the server, as a Response object.
@@ -40,10 +62,18 @@ async def _request_impl(
 
         :meta private:
         """
+        # Merge global and per-request headers
+        merged_headers = {
+            "Content-Type": "application/json",
+            **self.headers,
+            **(headers or {}),
+        }
+
         async with AsyncClient(timeout=timeout) as http_client:
             response = await http_client.post(
                 self.url,
                 json=request_to_json_rpc(request),
+                headers=merged_headers,
             )
             try:
                 return json_to_response(response.json())
@@ -53,4 +83,4 @@ async def _request_impl(
                         "error": response.status_code,
                         "error_message": response.text,
                     }
-                )
+                ) from None