✨ Add ApiKeysRpcApi subclient and integrate it into WebServerRpcClient

Author: pcrespov

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/webserver/v1/__init__.py

@@ -3,6 +3,7 @@
 Provides a class-based, modular RPC client for webserver services.
 """
 
+from .api_keys import ApiKeysRpcApi
 from .client import WebServerRpcClient
 from .functions import FunctionsRpcApi
 from .licenses import LicensesRpcApi
@@ -13,4 +14,5 @@
     "ProjectsRpcApi",
     "LicensesRpcApi",
     "FunctionsRpcApi",
+    "ApiKeysRpcApi",
 )

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/webserver/v1/api_keys.py

@@ -0,0 +1,58 @@
+"""API Keys RPC API subclient."""
+
+from models_library.basic_types import IDStr
+from models_library.products import ProductName
+from models_library.rpc.webserver.auth.api_keys import ApiKeyCreate, ApiKeyGet
+from models_library.users import UserID
+
+from ._base import BaseRpcApi
+
+
+class ApiKeysRpcApi(BaseRpcApi):
+    """RPC client for API key-related operations."""
+
+    async def create_api_key(
+        self,
+        *,
+        product_name: ProductName,
+        user_id: UserID,
+        api_key: ApiKeyCreate,
+    ) -> ApiKeyGet:
+        """Create an API key."""
+        return await self._request(
+            "create_api_key",
+            product_name=product_name,
+            user_id=user_id,
+            display_name=api_key.display_name,
+            expiration=api_key.expiration,
+        )
+
+    async def get_api_key(
+        self,
+        *,
+        product_name: ProductName,
+        user_id: UserID,
+        api_key_id: IDStr,
+    ) -> ApiKeyGet:
+        """Get an API key by ID."""
+        return await self._request(
+            "get_api_key",
+            product_name=product_name,
+            user_id=user_id,
+            api_key_id=api_key_id,
+        )
+
+    async def delete_api_key_by_key(
+        self,
+        *,
+        product_name: ProductName,
+        user_id: UserID,
+        api_key: str,
+    ) -> None:
+        """Delete an API key by key value."""
+        return await self._request(
+            "delete_api_key_by_key",
+            product_name=product_name,
+            user_id=user_id,
+            api_key=api_key,
+        )

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/webserver/v1/client.py

@@ -4,6 +4,7 @@
 from models_library.rpc.webserver import WEBSERVER_RPC_NAMESPACE
 from servicelib.rabbitmq import RabbitMQRPCClient
 
+from .api_keys import ApiKeysRpcApi
 from .functions import FunctionsRpcApi
 from .licenses import LicensesRpcApi
 from .projects import ProjectsRpcApi
@@ -24,6 +25,7 @@ def __init__(
         self.projects = ProjectsRpcApi(rpc_client, namespace)
         self.licenses = LicensesRpcApi(rpc_client, namespace)
         self.functions = FunctionsRpcApi(rpc_client, namespace)
+        self.api_keys = ApiKeysRpcApi(rpc_client, namespace)
 
     @property
     def namespace(self) -> RPCNamespace: