@agentunion/fastaun 0.5.31 → 0.5.33

package/dist/client/delivery.d.ts

@@ -174,6 +174,9 @@ export declare class MessageDeliveryEngine {
     pendingPullUpper(ns: string): number;
     hasAnyPendingPull(): boolean;
     onPullGateIdle(ns: string): void;
+    queueMessageClassPull(params: Record<string, any>): void;
+    restoreMessageClassPulls(): Promise<void>;
+    private pullMessageClassPage;
     private recordBackgroundPullNeed;
     observeBackgroundPullResponse(ns: string, afterSeq: number, result: unknown, limit?: number): void;
     private finishBackgroundPull;

package/dist/client/delivery.js

@@ -165,8 +165,11 @@ export class MessageDeliveryEngine {
         this.groupPullVersions.clear();
     }
     groupPullNamespace(namespace) {
-        const groupId = namespace.startsWith('group_event:') ? namespace.slice('group_event:'.length)
+        let groupId = namespace.startsWith('group_event:') ? namespace.slice('group_event:'.length)
             : namespace.startsWith('group:') ? namespace.slice('group:'.length) : '';
+        if (namespace.startsWith('message_class/')) {
+            groupId = /^message_class\/[^/]+\/(?:group|volatile_group)\/(.+)$/.exec(namespace)?.[1] ?? '';
+        }
         return groupId ? `group:${normalizeGroupAid(groupId)}` : '';
     }
     automaticGroupPullStopped(namespace) {
@@ -1684,12 +1687,64 @@ export class MessageDeliveryEngine {
         }
         this.schedulePendingGroupPullIfNeeded(ns, 'pull-gate-idle');
     }
+    queueMessageClassPull(params) {
+        const client = this.runtime.client;
+        if (client._sessionOptions?.background_sync === false || !client._aid || client._closing)
+            return;
+        const group = Boolean(params.group_id ?? params.group_aid);
+        if (!client._controlMessages().supports(params.message_class, group))
+            return;
+        this.ensurePeerContext();
+        const ns = `message_class/${params.message_class}/${group ? params.volatile ? 'volatile_group' : 'group' : 'peer'}/${group ? params.group_id ?? params.group_aid : params.sender_aid}`;
+        if (this.automaticGroupPullStopped(ns))
+            return;
+        this.recordBackgroundPullNeed(ns, 0, true, true);
+        this.backgroundPullNeeds.get(ns).messageClassRequest = { ...params };
+        this.requestBackgroundPullRecovery();
+    }
+    async restoreMessageClassPulls() {
+        const client = this.runtime.client;
+        if (!client._aid || client._sessionOptions?.background_sync === false)
+            return;
+        const saved = client._tokenStore.loadInstanceState?.(client._aid, client._deviceId, client._slotId)?.message_control_v1 ?? {};
+        for (const key of Object.keys(saved)) {
+            const [messageClass, kind, scope] = JSON.parse(key);
+            this.queueMessageClassPull({ message_class: messageClass,
+                ...(kind === 'peer' ? { sender_aid: scope } : { group_id: scope }), ...(kind === 'volatile_group' ? { volatile: true } : {}) });
+        }
+    }
+    async pullMessageClassPage(ns, need, generation) {
+        const client = this.runtime.client;
+        const request = need.messageClassRequest;
+        const trigger = need.trigger;
+        try {
+            const result = await this.runAutomaticPull(ns, () => client._controlMessages().call(request.group_id || request.group_aid ? 'group.v2.pull' : 'message.v2.pull', request, true));
+            if (generation !== this.captureInlineAckGeneration() || trigger !== need.trigger)
+                return;
+            if (result._pubsub_backpressure === true) {
+                need.retryAt = Infinity;
+                return;
+            }
+            if (result.has_more === true || result.volatile_has_more === true) {
+                if (result.head_seq > 0)
+                    need.messageClassRequest = { ...request, head_seq: result.head_seq };
+            }
+            else
+                this.backgroundPullNeeds.delete(ns);
+        }
+        catch (error) {
+            if (generation === this.captureInlineAckGeneration()) {
+                need.retryAt = this.automaticPullFailures.get(ns)?.retryAt ?? Date.now() + BACKGROUND_PULL_RETRY_DELAY_MS;
+            }
+            throw error;
+        }
+    }
     recordBackgroundPullNeed(ns, throughSeq = 0, probe = false, restart = false) {
         if (this.automaticGroupPullStopped(ns))
             return;
         const client = this.runtime.client;
         if (client._sessionOptions?.background_sync === false || !client._aid
-            || !(ns === `p2p:${client._aid}` || ns.startsWith('group:') || ns.startsWith('group_event:')))
+            || !(ns === `p2p:${client._aid}` || ns.startsWith('group:') || ns.startsWith('group_event:') || ns.startsWith('message_class/')))
             return;
         const need = this.backgroundPullNeeds.get(ns) ?? {
             throughSeq: 0, hasMore: false, probe, lastAfter: -1, retryAt: 0, empty: false, trigger: {},
@@ -1784,6 +1839,13 @@ export class MessageDeliveryEngine {
                 }
                 if (ns.startsWith('p2p:') && !this.hasP2pApplicationSubscription())
                     continue;
+                if (need.messageClassRequest && need.messageClassRequest.message_class !== 'sdk_control') {
+                    const events = need.messageClassRequest.group_id || need.messageClassRequest.group_aid
+                        ? ['group.message_created', 'group.message_batch'] : ['message.received', 'message.batch'];
+                    const handlers = client._dispatcher._snapshotApplicationHandlers?.(events);
+                    if (handlers && ![...handlers.values()].some((selected) => selected.length > 0))
+                        continue;
+                }
                 if (this.backgroundPullTasks.has(ns) || this.realtimeSyncing?.has(ns) || this.onlineUnreadHintTasks?.has(ns))
                     continue;
                 const eventScope = ns.startsWith('group_event:');
@@ -1793,7 +1855,7 @@ export class MessageDeliveryEngine {
                 if (filling || (!eventScope && client._rpcPipeline?.hasPullActivity?.(ns, false))
                     || (eventScope && ['pulling', 'processing', 'pulled'].includes(client._rpcPipeline?.getPullLifecycle?.(ns))))
                     continue;
-                if (this.syncState(ns).ack >= need.throughSeq && !need.hasMore && !need.probe) {
+                if (!need.messageClassRequest && this.syncState(ns).ack >= need.throughSeq && !need.hasMore && !need.probe) {
                     this.backgroundPullNeeds.delete(ns);
                     continue;
                 }
@@ -1805,6 +1867,8 @@ export class MessageDeliveryEngine {
                 const task = Promise.resolve().then(() => {
                     if (generation !== this.captureInlineAckGeneration())
                         return;
+                    if (need.messageClassRequest)
+                        return this.pullMessageClassPage(ns, need, generation);
                     return eventScope ? this.fillGroupEventGap(ns.slice('group_event:'.length))
                         : ns.startsWith('group:') ? this.fillGroupGap(ns.slice(6)) : this.fillP2pGap();
                 }).finally(() => {