From 1b317b1b9b763bd406ef9fb6607d78755bdf3203 Mon Sep 17 00:00:00 2001
From: Vlad Stan <stan.v.vlad@gmail.com>
Date: Mon, 6 Mar 2023 18:01:05 +0200
Subject: [PATCH] refactor: extract simpler methods

---
 models.py | 13 +--------
 tasks.py  | 85 +++++++++++++++++++++++++++++--------------------------
 2 files changed, 46 insertions(+), 52 deletions(-)

diff --git a/models.py b/models.py
index cd93d38..711a3f4 100644
--- a/models.py
+++ b/models.py
@@ -256,6 +256,7 @@ class OrderContact(BaseModel):
     nostr: Optional[str]
     phone: Optional[str]
     email: Optional[str]
+    address: Optional[str]
 
 
 class PartialOrder(BaseModel):
@@ -307,15 +308,3 @@ class PaymentRequest(BaseModel):
     id: str
     message: Optional[str]
     payment_options: List[PaymentOption]
-
-    def to_nostr_event(self, author_pubkey: str, to_pubkey: str) -> NostrEvent:
-        event = NostrEvent(
-            pubkey=author_pubkey,
-            created_at=round(time.time()),
-            kind=4,
-            tags=[["p", to_pubkey]],
-            content=json.dumps(self.dict(), separators=(",", ":"), ensure_ascii=False),
-        )
-        event.id = event.event_id
-
-        return event
diff --git a/tasks.py b/tasks.py
index b5e9619..9a78344 100644
--- a/tasks.py
+++ b/tasks.py
@@ -89,62 +89,67 @@ async def handle_message(msg: str):
         if type.upper() == "EVENT":
             event = NostrEvent(**event)
             if event.kind == 4:
-                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)
-                dm_resp = await handle_dirrect_message(
-                    event.pubkey, event.id, clear_text_msg
-                )
-                if dm_resp:
-                    dm_event = merchant.build_dm_event(dm_resp, event.pubkey)
-                    await publish_nostr_event(dm_event)
+                await handle_nip04_message(public_key, event)
 
     except Exception as ex:
         logger.warning(ex)
 
 
+async def handle_nip04_message(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)
+    dm_resp = await handle_dirrect_message(event.pubkey, event.id, clear_text_msg)
+    if dm_resp:
+        dm_event = merchant.build_dm_event(dm_resp, event.pubkey)
+        await publish_nostr_event(dm_event)
+
+
 async def handle_dirrect_message(
     from_pubkey: str, event_id: str, msg: str
 ) -> Optional[str]:
     order, text_msg = order_from_json(msg)
     try:
         if order:
-            ### check that event_id not parsed already
             order["pubkey"] = from_pubkey
             order["event_id"] = event_id
-            partial_order = PartialOrder(**order)
-            partial_order.validate_order()
-            assert len(partial_order.items) != 0, "Order has no items. Order: " + msg
-
-            first_product_id = partial_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,
-                )
-                resp.raise_for_status()
-                data = resp.json()
-                return (
-                    json.dumps(data, separators=(",", ":"), ensure_ascii=False)
-                    if data
-                    else None
-                )
+            return await handle_new_order(PartialOrder(**order))
         else:
             print("### text_msg", text_msg)
             return None
     except Exception as ex:
         logger.warning(ex)
         return None
+
+
+async def handle_new_order(order: PartialOrder):
+    ### check that event_id not parsed already
+
+    order.validate_order()
+    assert (
+        len(order.items) != 0
+    ), f"Order has no items. Order:  '{order.id}' ({order.event_id})"
+
+    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