creates notifier

Author: pcrespov

services/payments/src/simcore_service_payments/services/socketio.py

@@ -1,12 +1,17 @@
 import logging
+from dataclasses import dataclass
 
 import socketio
 from fastapi import FastAPI
 from fastapi.encoders import jsonable_encoder
 from models_library.api_schemas_payments.socketio import (
     SOCKET_IO_PAYMENT_COMPLETED_EVENT,
+    SOCKET_IO_PAYMENT_METHOD_ACKED_EVENT,
+)
+from models_library.api_schemas_webserver.wallets import (
+    PaymentMethodTransaction,
+    PaymentTransaction,
 )
-from models_library.api_schemas_webserver.wallets import PaymentTransaction
 from models_library.users import GroupID
 from servicelib.socketio_utils import cleanup_socketio_async_pubsub_manager
 from settings_library.rabbit import RabbitSettings
@@ -16,6 +21,37 @@
 _logger = logging.getLogger(__name__)
 
 
+@dataclass
+class Notifier:
+    _sio_manager: socketio.AsyncAioPikaManager
+
+    async def notify_payment_completed(
+        self,
+        user_primary_group_id: GroupID,
+        payment: PaymentTransaction,
+    ):
+        # NOTE: We assume that the user has been added to all
+        # rooms associated to his groups
+        assert payment.completed_at is not None  # nosec
+
+        return await self._sio_manager.emit(
+            SOCKET_IO_PAYMENT_COMPLETED_EVENT,
+            data=jsonable_encoder(payment, by_alias=True),
+            room=f"{user_primary_group_id}",
+        )
+
+    async def notify_payment_method_acked(
+        self,
+        user_primary_group_id: GroupID,
+        payment_method: PaymentMethodTransaction,
+    ):
+        return await self._sio_manager.emit(
+            SOCKET_IO_PAYMENT_METHOD_ACKED_EVENT,
+            data=jsonable_encoder(payment_method, by_alias=True),
+            room=f"{user_primary_group_id}",
+        )
+
+
 def setup_socketio(app: FastAPI):
     settings: RabbitSettings = get_rabbitmq_settings(app)
 
@@ -31,6 +67,9 @@ async def _on_startup() -> None:
             url=settings.dsn, logger=_logger, write_only=True
         )
 
+        # NOTE: this might be moved somewhere else when notifier incorporates emails etc
+        app.state.notifier = Notifier(_sio_manager=app.state.external_socketio)
+
     async def _on_shutdown() -> None:
         if app.state.external_socketio:
             await cleanup_socketio_async_pubsub_manager(
@@ -47,14 +86,9 @@ async def notify_payment_completed(
     user_primary_group_id: GroupID,
     payment: PaymentTransaction,
 ):
-    # NOTE: We assume that the user has been added to all
-    # rooms associated to his groups
-    assert payment.completed_at is not None  # nosec
 
-    external_socketio: socketio.AsyncAioPikaManager = app.state.external_socketio
+    notifier: Notifier = app.state.notifier
 
-    return await external_socketio.emit(
-        SOCKET_IO_PAYMENT_COMPLETED_EVENT,
-        data=jsonable_encoder(payment, by_alias=True),
-        room=f"{user_primary_group_id}",
+    return await notifier.notify_payment_completed(
+        user_primary_group_id=user_primary_group_id, payment=payment
     )