feat: keep track of new messages

crud.py

@@ -641,3 +641,17 @@ async def update_customer_profile(
         f"UPDATE nostrmarket.customers SET event_created_at = ?, meta = ? WHERE public_key = ?",
         (event_created_at, json.dumps(profile.dict()), public_key),
     )
+
+
+async def increment_customer_unread_messages(public_key: str):
+    await db.execute(
+        f"UPDATE nostrmarket.customers SET unread_messages = unread_messages + 1 WHERE public_key = ?",
+        (public_key,),
+    )
+
+
+async def update_customer_no_unread_messages(public_key: str):
+    await db.execute(
+        f"UPDATE nostrmarket.customers SET unread_messages = 0 WHERE public_key = ?",
+        (public_key,),
+    )

migrations.py

@@ -135,6 +135,7 @@ async def m001_initial(db):
             merchant_id TEXT NOT NULL,
             public_key TEXT NOT NULL,
             event_created_at INTEGER,
+            unread_messages INTEGER NOT NULL DEFAULT 1,
             meta TEXT NOT NULL DEFAULT '{}'
         );
         """

models.py

@@ -442,6 +442,7 @@ class Customer(BaseModel):
     public_key: str
     event_created_at: Optional[int]
     profile: Optional[CustomerProfile]
+    unread_messages: int = 0
 
     @classmethod
     def from_row(cls, row: Row) -> "Customer":

services.py

@@ -20,6 +20,7 @@ from .crud import (
     get_products_by_ids,
     get_stalls,
     get_wallet_for_product,
+    increment_customer_unread_messages,
     update_customer_profile,
     update_order_paid_status,
     update_product,
@@ -256,6 +257,8 @@ async def _handle_incoming_dms(
     customer = await get_customer(merchant.id, event.pubkey)
     if not customer:
         await _handle_new_customer(event, merchant)
+    else:
+        await increment_customer_unread_messages(event.pubkey)
 
     dm_reply = await _handle_dirrect_message(
         merchant.id,

static/components/direct-messages/direct-messages.html

@@ -9,7 +9,7 @@
     <q-card-section>
       <q-select
         v-model="activePublicKey"
-        :options="customers.map(c => ({label: `${c.profile.name || 'unknown'} ${c.profile.about || ''} (${c.public_key.slice(0, 16)}...${c.public_key.slice(c.public_key.length - 16)})`, value: c.public_key}))"
+        :options="customers.map(c => ({label: buildCustomerLabel(c), value: c.public_key}))"
         label="Select Customer"
         emit-value
         @input="selectActiveCustomer()"

static/components/direct-messages/direct-messages.js

@@ -20,6 +20,16 @@ async function directMessages(path) {
     },
     methods: {
       sendMessage: async function () {},
+      buildCustomerLabel: function (c) {
+        let label = `${c.profile.name || 'unknown'} ${c.profile.about || ''}`
+        if (c.unread_messages) {
+          label += `[new: ${c.unread_messages}]`
+        }
+        label += `  (${c.public_key.slice(0, 16)}...${c.public_key.slice(
+          c.public_key.length - 16
+        )}`
+        return label
+      },
       getDirectMessages: async function (pubkey) {
         if (!pubkey) {
           this.messages = []

views_api.py

@@ -44,6 +44,7 @@ from .crud import (
     get_stalls,
     get_zone,
     get_zones,
+    update_customer_no_unread_messages,
     update_merchant,
     update_order_shipped_status,
     update_product,
@@ -738,6 +739,7 @@ async def api_get_messages(
         assert merchant, f"Merchant cannot be found"
 
         messages = await get_direct_messages(merchant.id, public_key)
+        await update_customer_no_unread_messages(public_key)
         return messages
     except AssertionError as ex:
         raise HTTPException(