@unicitylabs/openclaw-unicity 0.5.2 → 0.5.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@unicitylabs/openclaw-unicity",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "description": "Unicity wallet identity and encrypted DMs for OpenClaw agents — powered by Sphere SDK",
   "type": "module",
   "main": "src/index.ts",

package/src/channel.ts

@@ -248,23 +248,24 @@ export const unicityChannelPlugin = {
       const seenDmIds = new Set<string>();
       const DM_SEEN_MAX = 1000;
 
+      // Collect all known representations of "self" so we can detect echoed-back
+      // messages.  The SDK's built-in self-check compares transport pubkey against
+      // chainPubkey, but those may differ in format (33-byte compressed vs 32-byte
+      // x-only Nostr key), letting self-messages slip through.
+      const selfPubkeys = new Set<string>();
+      if (sphere.identity?.chainPubkey) selfPubkeys.add(sphere.identity.chainPubkey);
+      const myNostrPubkey = sphere.groupChat?.getMyPublicKey?.() ?? null;
+      if (myNostrPubkey) selfPubkeys.add(myNostrPubkey);
 
-      const unsub = sphere.communications.onDirectMessage((msg) => {
-        // Deduplicate: skip already-processed messages (relays may deliver dupes)
-        if (msg.id && seenDmIds.has(msg.id)) return;
-        if (msg.id) {
-          seenDmIds.add(msg.id);
-          if (seenDmIds.size > DM_SEEN_MAX) {
-            const first = seenDmIds.values().next().value!;
-            seenDmIds.delete(first);
-          }
-        }
+      // Per-sender queue: while a reply is being generated for a sender, hold
+      // the latest incoming message so it gets processed after the current reply
+      // finishes.  Prevents duplicate replies while not dropping follow-ups.
+      type DmMsg = { id: string; senderPubkey: string; senderNametag?: string; recipientPubkey: string; content: string; timestamp: number; isRead: boolean };
+      const sendersInFlight = new Set<string>();
+      const pendingPerSender = new Map<string, DmMsg>();
 
-        // Skip historical messages replayed on connect
-        if (msg.timestamp && msg.timestamp < dmStartTime) {
-          ctx.log?.debug(`[${ctx.account.accountId}] Skipping historical DM (ts=${msg.timestamp} < start=${dmStartTime})`);
-          return;
-        }
+      function dispatchDm(msg: DmMsg): void {
+        sendersInFlight.add(msg.senderPubkey);
 
         // Immediately signal that we're composing a reply
         sphere.communications.sendComposingIndicator(msg.senderPubkey)
@@ -343,7 +344,47 @@ export const unicityChannelPlugin = {
           })
           .catch((err: unknown) => {
             ctx.log?.error(`[${ctx.account.accountId}] Reply dispatch error: ${err}`);
+          })
+          .finally(() => {
+            sendersInFlight.delete(msg.senderPubkey);
+            // If a follow-up message arrived while we were replying, process it now
+            const pending = pendingPerSender.get(msg.senderPubkey);
+            if (pending) {
+              pendingPerSender.delete(msg.senderPubkey);
+              ctx.log?.info(`[${ctx.account.accountId}] Processing queued DM from ${peerId}`);
+              dispatchDm(pending);
+            }
           });
+      }
+
+      const unsub = sphere.communications.onDirectMessage((msg) => {
+        // Skip messages from self (own DMs echoed back by the relay)
+        if (selfPubkeys.has(msg.senderPubkey)) return;
+
+        // Deduplicate: skip already-processed messages (relays may deliver dupes)
+        if (msg.id && seenDmIds.has(msg.id)) return;
+        if (msg.id) {
+          seenDmIds.add(msg.id);
+          if (seenDmIds.size > DM_SEEN_MAX) {
+            const first = seenDmIds.values().next().value!;
+            seenDmIds.delete(first);
+          }
+        }
+
+        // Skip historical messages replayed on connect
+        if (msg.timestamp && msg.timestamp < dmStartTime) {
+          ctx.log?.debug(`[${ctx.account.accountId}] Skipping historical DM (ts=${msg.timestamp} < start=${dmStartTime})`);
+          return;
+        }
+
+        // Per-sender lock: queue if we're already generating a reply for this sender
+        if (sendersInFlight.has(msg.senderPubkey)) {
+          ctx.log?.debug(`[${ctx.account.accountId}] Queuing DM from ${msg.senderPubkey.slice(0, 16)}… — reply in flight`);
+          pendingPerSender.set(msg.senderPubkey, msg);
+          return;
+        }
+
+        dispatchDm(msg);
       });
 
       ctx.log?.info(`[${ctx.account.accountId}] Unicity DM listener active`);
@@ -480,11 +521,6 @@ export const unicityChannelPlugin = {
         replyToId?: string;
       };
 
-      // Nostr pubkey for self-message detection and reply-to-self detection.
-      // Group messages use the 32-byte x-only Nostr pubkey (event.pubkey),
-      // NOT the 33-byte compressed chainPubkey.
-      const myNostrPubkey = sphere.groupChat?.getMyPublicKey?.() ?? null;
-
       // Detect if a group message is a reply to one of the agent's own messages.
       // Used to set WasMentioned so the mention gate treats replies-to-self as
       // implicit mentions (same pattern Discord uses for thread replies).