aimeat 1.25.0 → 1.26.0

package/dist/src/storage/providers/mongodb/index.js

@@ -507,6 +507,7 @@ export class PrismaStorage {
                 version: record.version,
                 flagCount: record.flagCount ?? 0,
                 allowedOrigins: record.allowedOrigins ?? [],
+                searchBlob: this.buildSearchBlob(record),
                 createdAt: new Date(record.createdAt),
                 updatedAt: new Date(record.updatedAt),
             },
@@ -519,6 +520,7 @@ export class PrismaStorage {
                 version: record.version,
                 flagCount: record.flagCount ?? 0,
                 allowedOrigins: record.allowedOrigins ?? [],
+                searchBlob: this.buildSearchBlob(record),
                 updatedAt: new Date(record.updatedAt),
             },
         });
@@ -542,6 +544,7 @@ export class PrismaStorage {
                 version: record.version,
                 flagCount: record.flagCount ?? 0,
                 allowedOrigins: record.allowedOrigins ?? [],
+                searchBlob: this.buildSearchBlob(record),
                 updatedAt: new Date(record.updatedAt),
             },
         });
@@ -670,6 +673,57 @@ export class PrismaStorage {
             return true;
         });
     }
+    /** Flatten a record into searchable text: key + string/number leaves of value + tags. Indexed
+     *  by the `searchBlob` `$text` index (created in init). Mirrors what SQLite FTS5 indexes. */
+    buildSearchBlob(record) {
+        const parts = [record.key, ...(record.tags ?? [])];
+        const collect = (v, depth) => {
+            if (depth > 6 || v == null)
+                return;
+            if (typeof v === 'string') {
+                if (v.trim())
+                    parts.push(v);
+            }
+            else if (typeof v === 'number' || typeof v === 'boolean')
+                parts.push(String(v));
+            else if (Array.isArray(v))
+                for (const x of v)
+                    collect(x, depth + 1);
+            else if (typeof v === 'object')
+                for (const x of Object.values(v))
+                    collect(x, depth + 1);
+        };
+        collect(record.value, 0);
+        return parts.join(' \n ');
+    }
+    async searchText(query, opts) {
+        this.ensureReady();
+        const tokens = query.toLowerCase().match(/[\p{L}\p{N}]+/gu);
+        if (!tokens || tokens.length === 0)
+            return [];
+        const where = { OR: tokens.map(tok => ({ searchBlob: { contains: tok, mode: 'insensitive' } })) };
+        if (opts?.ownerGaiis?.length)
+            where.ownerGaii = { in: opts.ownerGaiis };
+        if (opts?.visibility)
+            where.visibility = opts.visibility;
+        if (opts?.keyPrefix)
+            where.key = { startsWith: opts.keyPrefix };
+        const limit = opts?.limit ?? 50;
+        const rows = await this.prisma.memory.findMany({ where, take: limit * 4 });
+        // Rank: more distinct tokens matched first, then most-recently updated. (No bm25 here; the
+        // librarian's value is finding the right records, and an AI rerank can follow — design §4.)
+        return rows
+            .filter(r => !r.ttlHours || Date.now() <= new Date(r.createdAt).getTime() + r.ttlHours * 3600_000)
+            .filter(r => opts?.maxFlags === undefined || (r.flagCount ?? 0) <= opts.maxFlags)
+            .map(r => {
+            const blob = String(r.searchBlob ?? '').toLowerCase();
+            const score = tokens.reduce((n, t) => n + (blob.includes(t) ? 1 : 0), 0);
+            return { record: this.toMemoryRecord(r), score, updatedAt: r.updatedAt };
+        })
+            .sort((a, b) => (b.score - a.score) || (b.updatedAt.getTime() - a.updatedAt.getTime()))
+            .slice(0, limit)
+            .map(({ record, score }) => ({ record, score }));
+    }
     // ── Actions ─────────────────────────────────────────────────
     async createAction(action) {
         this.ensureReady();
@@ -4570,8 +4624,8 @@ export class PrismaStorage {
         this.ensureReady();
         await this.prisma.federationPeer.upsert({
             where: { nodeId: peer.nodeId },
-            create: { nodeId: peer.nodeId, url: peer.url, publicKey: peer.publicKey, status: peer.status, addedAt: new Date(peer.addedAt), lastSeen: new Date(peer.lastSeen), shareCatalogue: peer.shareCatalogue, replicateMemory: peer.replicateMemory, allowRouting: peer.allowRouting, peerMode: peer.peerMode || 'federation', allowFederatedAuth: peer.allowFederatedAuth ?? false, federationAuthScopes: peer.federationAuthScopes ?? [] },
-            update: { url: peer.url, publicKey: peer.publicKey, status: peer.status, lastSeen: new Date(peer.lastSeen), shareCatalogue: peer.shareCatalogue, replicateMemory: peer.replicateMemory, allowRouting: peer.allowRouting, peerMode: peer.peerMode || 'federation', allowFederatedAuth: peer.allowFederatedAuth ?? false, federationAuthScopes: peer.federationAuthScopes ?? [] },
+            create: { nodeId: peer.nodeId, url: peer.url, publicKey: peer.publicKey, status: peer.status, addedAt: new Date(peer.addedAt), lastSeen: new Date(peer.lastSeen), shareCatalogue: peer.shareCatalogue, replicateMemory: peer.replicateMemory, allowRouting: peer.allowRouting, peerMode: peer.peerMode || 'federation', allowFederatedAuth: peer.allowFederatedAuth ?? false, federationAuthScopes: peer.federationAuthScopes ?? [], tier: peer.tier ?? 'member', availability: peer.availability ?? null, expiresAt: peer.expiresAt ? new Date(peer.expiresAt) : null, heartbeatOk: peer.heartbeatOk ?? 0, heartbeatTotal: peer.heartbeatTotal ?? 0, availabilityWindow: peer.availabilityWindow ?? null, availabilityPct: peer.availabilityPct ?? null, softwareVersion: peer.softwareVersion ?? null, nodeCardHash: peer.nodeCardHash ?? null },
+            update: { url: peer.url, publicKey: peer.publicKey, status: peer.status, lastSeen: new Date(peer.lastSeen), shareCatalogue: peer.shareCatalogue, replicateMemory: peer.replicateMemory, allowRouting: peer.allowRouting, peerMode: peer.peerMode || 'federation', allowFederatedAuth: peer.allowFederatedAuth ?? false, federationAuthScopes: peer.federationAuthScopes ?? [], tier: peer.tier ?? 'member', availability: peer.availability ?? null, expiresAt: peer.expiresAt ? new Date(peer.expiresAt) : null, heartbeatOk: peer.heartbeatOk ?? 0, heartbeatTotal: peer.heartbeatTotal ?? 0, availabilityWindow: peer.availabilityWindow ?? null, availabilityPct: peer.availabilityPct ?? null, softwareVersion: peer.softwareVersion ?? null, nodeCardHash: peer.nodeCardHash ?? null },
         });
     }
     async listFederationPeers() {
@@ -4590,6 +4644,15 @@ export class PrismaStorage {
             peerMode: r.peerMode || 'federation',
             allowFederatedAuth: r.allowFederatedAuth ?? false,
             federationAuthScopes: r.federationAuthScopes ?? [],
+            tier: r.tier ?? 'member',
+            availability: r.availability ?? null,
+            expiresAt: r.expiresAt instanceof Date ? r.expiresAt.toISOString() : (r.expiresAt ?? null),
+            heartbeatOk: r.heartbeatOk ?? 0,
+            heartbeatTotal: r.heartbeatTotal ?? 0,
+            availabilityWindow: r.availabilityWindow ?? null,
+            availabilityPct: r.availabilityPct ?? null,
+            softwareVersion: r.softwareVersion ?? null,
+            nodeCardHash: r.nodeCardHash ?? null,
         }));
     }
     async deleteFederationPeer(nodeId) {
@@ -6996,6 +7059,69 @@ export class PrismaStorage {
             return false;
         }
     }
+    async appendMessageDeliveryLog(log) {
+        this.ensureReady();
+        await this.prisma.messageDeliveryLog.create({
+            data: {
+                id: log.id, messageId: log.messageId, origin: log.origin, targetNodeId: log.targetNodeId,
+                status: log.status, httpStatus: log.httpStatus ?? null, errorMessage: log.errorMessage ?? null,
+                latencyMs: log.latencyMs, createdAt: new Date(log.createdAt),
+            },
+        });
+    }
+    toMessageDeliveryLog(row) {
+        const rec = {
+            id: row.id, messageId: row.messageId, origin: row.origin, targetNodeId: row.targetNodeId,
+            status: row.status, latencyMs: row.latencyMs ?? 0,
+            createdAt: row.createdAt instanceof Date ? row.createdAt.toISOString() : row.createdAt,
+        };
+        if (row.httpStatus != null)
+            rec.httpStatus = row.httpStatus;
+        if (row.errorMessage)
+            rec.errorMessage = row.errorMessage;
+        return rec;
+    }
+    async listMessageDeliveryLogs(limit = 100) {
+        this.ensureReady();
+        const rows = await this.prisma.messageDeliveryLog.findMany({ orderBy: { createdAt: 'desc' }, take: limit });
+        return rows.map((r) => this.toMessageDeliveryLog(r));
+    }
+    async getMessageDeliveryStats() {
+        this.ensureReady();
+        const since = new Date(Date.now() - 24 * 3600_000);
+        const [all, recent] = await Promise.all([
+            this.prisma.messageDeliveryLog.groupBy({ by: ['status'], _count: { _all: true } }),
+            this.prisma.messageDeliveryLog.groupBy({ by: ['status'], where: { createdAt: { gte: since } }, _count: { _all: true } }),
+        ]);
+        const byStatus = {};
+        const byStatus24h = {};
+        let total = 0, total24h = 0;
+        for (const r of all) {
+            byStatus[r.status] = r._count._all;
+            total += r._count._all;
+        }
+        for (const r of recent) {
+            byStatus24h[r.status] = r._count._all;
+            total24h += r._count._all;
+        }
+        const nodes = await this.prisma.messageDeliveryLog.groupBy({ by: ['targetNodeId'], _count: { _all: true } });
+        const topTargetNodes = await Promise.all(nodes.sort((a, b) => b._count._all - a._count._all).slice(0, 10).map(async (n) => ({
+            nodeId: n.targetNodeId,
+            total: n._count._all,
+            failed: await this.prisma.messageDeliveryLog.count({ where: { targetNodeId: n.targetNodeId, status: { in: ['failed', 'undeliverable'] } } }),
+        })));
+        return { total, total24h, byStatus, byStatus24h, topTargetNodes };
+    }
+    async pruneMessageDeliveryLogs(keep = 10000) {
+        this.ensureReady();
+        const cutoff = await this.prisma.messageDeliveryLog.findMany({
+            orderBy: { createdAt: 'desc' }, skip: keep, take: 1, select: { createdAt: true },
+        });
+        if (cutoff.length === 0)
+            return 0;
+        const res = await this.prisma.messageDeliveryLog.deleteMany({ where: { createdAt: { lt: cutoff[0].createdAt } } });
+        return res.count;
+    }
     async getContact(ownerGhii, contactId) {
         this.ensureReady();
         const row = await this.prisma.contactConsent.findUnique({ where: { id: this.contactDocId(ownerGhii, contactId) } });