Refactor and implement queueing messages

core/lib/deferred.py

@@ -0,0 +1,54 @@
+# Deferred processing library
+from core.util import logs
+from pydantic import BaseModel
+from typing import Annotated
+from uuid import UUID
+from pydantic import ValidationError
+from core.models import QueuedMessage, Message
+from core.clients import signal
+from core.lib.prompts.functions import delete_messages
+from asgiref.sync import sync_to_async
+
+
+log = logs.get_logger("deferred")
+
+
+class DeferredRequest(BaseModel):
+    type: str
+    method: str
+    user_id: int
+    message_id: Annotated[str, UUID]
+
+async def process_deferred(data: dict):
+    try:
+        validated_data = DeferredRequest(**data)
+        log.info(f"Validated Data: {validated_data}")
+        # Process the validated data
+    except ValidationError as e:
+        log.info(f"Validation Error: {e}")
+        return
+    
+    method = validated_data.method
+    user_id = validated_data.user_id
+    message_id = validated_data.message_id
+
+    try:
+        message = await sync_to_async(QueuedMessage.objects.get)(
+            user_id=user_id,
+            id=message_id,
+        )
+        log.info(f"Got {message}")
+    except QueuedMessage.DoesNotExist:
+        log.info(f"Didn't get message from {message_id}")
+        return
+
+    if message.session.identifier.service == "signal":
+        log.info(f"Is sisngla")
+        if method == "accept_message":
+            await signal.send_message(message)
+        else:
+            log.warning(f"Method not yet supported: {method}")
+            return
+    else:
+        log.warning(f"Protocol not supported: {message.session.identifier.service}")
+        return