refactor: clean-up tasks.py

services.py

@@ -1,10 +1,14 @@
 import json
 from typing import Optional
+
+import httpx
 from loguru import logger
 
-from lnbits.core import create_invoice
+from lnbits.core import create_invoice, get_wallet, url_for
 
 from .crud import (
+    create_direct_message,
+    create_order,
     get_merchant_by_pubkey,
     get_merchant_for_user,
     get_order,
@@ -13,11 +17,14 @@ from .crud import (
     get_wallet_for_product,
     update_order_paid_status,
 )
+from .helpers import order_from_json
 from .models import (
+    Merchant,
     Nostrable,
     Order,
     OrderExtra,
     OrderStatusUpdate,
+    PartialDirectMessage,
     PartialOrder,
     PaymentOption,
     PaymentRequest,
@@ -26,7 +33,9 @@ from .nostr.event import NostrEvent
 from .nostr.nostr_client import publish_nostr_event
 
 
-async def create_order(user_id: str, data: PartialOrder) -> Optional[PaymentRequest]:
+async def create_new_order(
+    user_id: str, data: PartialOrder
+) -> Optional[PaymentRequest]:
     if await get_order(user_id, data.id):
         return None
     if data.event_id and await get_order_by_event_id(user_id, data.event_id):
@@ -102,4 +111,111 @@ async def handle_order_paid(order_id: str, merchant_pubkey: str):
         dm_event = merchant.build_dm_event(dm_content, order.pubkey)
         await publish_nostr_event(dm_event)
     except Exception as ex:
-        logger.warning(ex)
+        logger.warning(ex)
+
+
+async def process_nostr_message(msg: str):
+    try:
+        type, subscription_id, event = json.loads(msg)
+        subscription_name, public_key = subscription_id.split(":")
+        if type.upper() == "EVENT":
+            event = NostrEvent(**event)
+            if event.kind == 4:
+                await _handle_nip04_message(subscription_name, public_key, event)
+
+    except Exception as ex:
+        logger.warning(ex)
+
+
+async def _handle_nip04_message(
+    subscription_name: str, public_key: str, event: NostrEvent
+):
+    merchant = await get_merchant_by_pubkey(public_key)
+    assert merchant, f"Merchant not found for public key '{public_key}'"
+
+    clear_text_msg = merchant.decrypt_message(event.content, event.pubkey)
+    if subscription_name == "direct-messages-in":
+        await _handle_incoming_dms(event, merchant, clear_text_msg)
+    else:
+        await _handle_outgoing_dms(event, merchant, clear_text_msg)
+
+
+async def _handle_incoming_dms(
+    event: NostrEvent, merchant: Merchant, clear_text_msg: str
+):
+    dm_content = await _handle_dirrect_message(
+        merchant.id, event.pubkey, event.id, event.created_at, clear_text_msg
+    )
+    if dm_content:
+        dm_event = merchant.build_dm_event(dm_content, event.pubkey)
+        await publish_nostr_event(dm_event)
+
+
+async def _handle_outgoing_dms(
+    event: NostrEvent, merchant: Merchant, clear_text_msg: str
+):
+    sent_to = event.tag_values("p")
+    if len(sent_to) != 0:
+        dm = PartialDirectMessage(
+            event_id=event.id,
+            event_created_at=event.created_at,
+            message=clear_text_msg,  # exclude if json
+            public_key=sent_to[0],
+            incoming=True,
+        )
+        await create_direct_message(merchant.id, dm)
+
+
+async def _handle_dirrect_message(
+    merchant_id: str, from_pubkey: str, event_id: str, event_created_at: int, msg: str
+) -> Optional[str]:
+    order, text_msg = order_from_json(msg)
+    try:
+        if order:
+            order["pubkey"] = from_pubkey
+            order["event_id"] = event_id
+            order["event_created_at"] = event_created_at
+            return await _handle_new_order(PartialOrder(**order))
+        else:
+            print("### text_msg", text_msg)
+            dm = PartialDirectMessage(
+                event_id=event_id,
+                event_created_at=event_created_at,
+                message=text_msg,
+                public_key=from_pubkey,
+                incoming=True,
+            )
+            await create_direct_message(merchant_id, dm)
+            return None
+    except Exception as ex:
+        logger.warning(ex)
+        return None
+
+
+async def _handle_new_order(order: PartialOrder) -> Optional[str]:
+    ### todo: check that event_id not parsed already
+
+    order.validate_order()
+
+    first_product_id = order.items[0].product_id
+    wallet_id = await get_wallet_for_product(first_product_id)
+    assert wallet_id, f"Cannot find wallet id for product id: {first_product_id}"
+
+    wallet = await get_wallet(wallet_id)
+    assert wallet, f"Cannot find wallet for product id: {first_product_id}"
+
+    market_url = url_for(f"/nostrmarket/api/v1/order", external=True)
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            url=market_url,
+            headers={
+                "X-Api-Key": wallet.adminkey,
+            },
+            json=order.dict(),
+        )
+        resp.raise_for_status()
+        data = resp.json()
+        if data:
+            return json.dumps(data, separators=(",", ":"), ensure_ascii=False)
+
+    return None