aimeat 1.4.8 → 1.6.1

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

@@ -122,6 +122,7 @@ export class MongoStorage {
                 morselBalance: agent.morselBalance,
                 allowedOrigins: agent.allowedOrigins ?? [],
                 defaultScopes: agent.defaultScopes ?? ['*'],
+                federate: agent.federate ?? false,
                 createdAt: new Date(agent.createdAt),
                 lastSeen: new Date(agent.lastSeen),
             },
@@ -531,6 +532,7 @@ export class MongoStorage {
                 tags: action.tags,
                 webhookUrl: action.webhookUrl ?? null,
                 semantic: action.semantic ?? null,
+                federate: action.federate ?? false,
                 createdAt: new Date(action.createdAt),
                 updatedAt: new Date(action.updatedAt),
             },
@@ -731,6 +733,7 @@ export class MongoStorage {
                 visibility: board.visibility,
                 ownerGaii: board.ownerGaii,
                 allowedGaiis: board.allowedGaiis,
+                federate: board.federate ?? false,
                 createdAt: new Date(board.createdAt),
             },
         });
@@ -751,10 +754,13 @@ export class MongoStorage {
         const rows = await this.prisma.board.findMany({ where });
         return rows.map((r) => this.toBoardRecord(r));
     }
-    async updateBoardVisibility(id, visibility) {
+    async updateBoardVisibility(id, visibility, federate) {
         this.ensureReady();
         try {
-            const row = await this.prisma.board.update({ where: { boardId: id }, data: { visibility } });
+            const data = { visibility };
+            if (federate !== undefined)
+                data.federate = federate;
+            const row = await this.prisma.board.update({ where: { boardId: id }, data });
             return this.toBoardRecord(row);
         }
         catch {
@@ -1143,6 +1149,7 @@ export class MongoStorage {
                 size: file.size,
                 data: file.data,
                 tags: file.tags || [],
+                federate: file.federate ?? false,
                 createdAt: new Date(file.createdAt),
             },
         });
@@ -1155,16 +1162,16 @@ export class MongoStorage {
         });
         if (!row)
             return null;
-        return { key: row.key, ownerGaii: row.ownerGaii, visibility: row.visibility, mimeType: row.mimeType, size: row.size, data: Buffer.from(row.data), tags: row.tags || [], createdAt: row.createdAt.toISOString() };
+        return { key: row.key, ownerGaii: row.ownerGaii, visibility: row.visibility, mimeType: row.mimeType, size: row.size, data: Buffer.from(row.data), tags: row.tags || [], federate: row.federate ?? false, createdAt: row.createdAt.toISOString() };
     }
     async listStorageFiles(ownerGaii) {
         this.ensureReady();
         const rows = await this.prisma.storageFile.findMany({
             where: { ownerGaii },
-            select: { key: true, ownerGaii: true, visibility: true, mimeType: true, size: true, tags: true, createdAt: true },
+            select: { key: true, ownerGaii: true, visibility: true, mimeType: true, size: true, tags: true, federate: true, createdAt: true },
         });
         return rows.map((r) => ({
-            key: r.key, ownerGaii: r.ownerGaii, visibility: r.visibility, mimeType: r.mimeType, size: r.size, data: Buffer.alloc(0), tags: r.tags || [], createdAt: r.createdAt.toISOString(),
+            key: r.key, ownerGaii: r.ownerGaii, visibility: r.visibility, mimeType: r.mimeType, size: r.size, data: Buffer.alloc(0), tags: r.tags || [], federate: r.federate ?? false, createdAt: r.createdAt.toISOString(),
         }));
     }
     async deleteStorageFile(ownerGaii, key) {
@@ -1244,6 +1251,16 @@ export class MongoStorage {
             return null;
         }
     }
+    async deletePeeringRequest(id) {
+        this.ensureReady();
+        try {
+            await this.prisma.peeringRequest.delete({ where: { requestId: id } });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
     // ── Chunked Uploads (kept in-memory — transient) ───────────
     async createChunkedUpload(record) {
         this.chunkedUploads.set(record.uploadId, record);
@@ -1283,19 +1300,19 @@ export class MongoStorage {
         return { name: row.name, displayName: row.displayName ?? undefined, publicKey: row.publicKey, roles: row.roles, createdAt: row.createdAt.toISOString() };
     }
     toAgentRecord(row) {
-        return { name: row.name, owner: row.owner, gaii: row.gaii, displayName: row.displayName ?? undefined, description: row.description ?? undefined, capabilities: row.capabilities, publicKey: row.publicKey, trustScore: row.trustScore, morselBalance: row.morselBalance, defaultScopes: row.defaultScopes?.length ? row.defaultScopes : undefined, allowedOrigins: row.allowedOrigins?.length ? row.allowedOrigins : undefined, createdAt: row.createdAt.toISOString(), lastSeen: row.lastSeen.toISOString() };
+        return { name: row.name, owner: row.owner, gaii: row.gaii, displayName: row.displayName ?? undefined, description: row.description ?? undefined, capabilities: row.capabilities, publicKey: row.publicKey, trustScore: row.trustScore, morselBalance: row.morselBalance, defaultScopes: row.defaultScopes?.length ? row.defaultScopes : undefined, allowedOrigins: row.allowedOrigins?.length ? row.allowedOrigins : undefined, federate: row.federate ?? false, createdAt: row.createdAt.toISOString(), lastSeen: row.lastSeen.toISOString() };
     }
     toMemoryRecord(row) {
         return { key: row.key, ownerGaii: row.ownerGaii, value: row.value, visibility: row.visibility, tags: row.tags, ttlHours: row.ttlHours, version: row.version, createdAt: row.createdAt.toISOString(), updatedAt: row.updatedAt.toISOString(), flagCount: row.flagCount ?? undefined, allowedOrigins: row.allowedOrigins?.length ? row.allowedOrigins : undefined };
     }
     toActionRecord(row) {
-        return { id: row.actionId, providerGaii: row.providerGaii, displayName: row.displayName, description: row.description, category: row.category ?? undefined, inputSchema: row.inputSchema, outputSchema: row.outputSchema, pricing: { baseMorsels: row.pricingBaseMorsels, perUnit: row.pricingPerUnit }, estimatedTimeSeconds: row.estimatedTimeSeconds ?? undefined, maxInputSizeBytes: row.maxInputSizeBytes ?? undefined, tags: row.tags, webhookUrl: row.webhookUrl ?? undefined, semantic: row.semantic ?? undefined, createdAt: row.createdAt.toISOString(), updatedAt: row.updatedAt.toISOString() };
+        return { id: row.actionId, providerGaii: row.providerGaii, displayName: row.displayName, description: row.description, category: row.category ?? undefined, inputSchema: row.inputSchema, outputSchema: row.outputSchema, pricing: { baseMorsels: row.pricingBaseMorsels, perUnit: row.pricingPerUnit }, estimatedTimeSeconds: row.estimatedTimeSeconds ?? undefined, maxInputSizeBytes: row.maxInputSizeBytes ?? undefined, tags: row.tags, webhookUrl: row.webhookUrl ?? undefined, semantic: row.semantic ?? undefined, federate: row.federate ?? false, createdAt: row.createdAt.toISOString(), updatedAt: row.updatedAt.toISOString() };
     }
     toWorkRecord(row) {
         return { trackingCode: row.trackingCode, status: row.status, actionId: row.actionId, providerGaii: row.providerGaii, requesterGaii: row.requesterGaii, input: row.input, output: row.output, cost: { basePrice: row.costBasePrice, networkFee: row.costNetworkFee, total: row.costTotal, inEscrow: row.costInEscrow }, ttlExpiresAt: row.ttlExpiresAt.toISOString(), callbackUrl: row.callbackUrl ?? undefined, rating: row.ratingScore != null ? { score: row.ratingScore, comment: row.ratingComment ?? undefined } : undefined, createdAt: row.createdAt.toISOString(), updatedAt: row.updatedAt.toISOString() };
     }
     toBoardRecord(row) {
-        return { id: row.boardId, name: row.name, description: row.description ?? undefined, visibility: row.visibility, ownerGaii: row.ownerGaii, allowedGaiis: row.allowedGaiis, createdAt: row.createdAt.toISOString() };
+        return { id: row.boardId, name: row.name, description: row.description ?? undefined, visibility: row.visibility, ownerGaii: row.ownerGaii, allowedGaiis: row.allowedGaiis, federate: row.federate ?? false, createdAt: row.createdAt.toISOString() };
     }
     toPostRecord(row) {
         return { id: row.postId, boardId: row.boardId, authorGaii: row.authorGaii, title: row.title, body: row.body, category: row.category ?? undefined, tags: row.tags, ttlExpiresAt: row.ttlExpiresAt?.toISOString(), reactions: row.reactions, replyTo: row.replyTo ?? undefined, createdAt: row.createdAt.toISOString() };
@@ -1828,6 +1845,14 @@ export class MongoStorage {
         }
         return results;
     }
+    async expireStaleConsents(before) {
+        this.ensureReady();
+        const result = await this.prisma.consent.updateMany({
+            where: { status: 'active', expires: { not: null, lt: new Date(before) } },
+            data: { status: 'expired' },
+        });
+        return result.count;
+    }
     // Consent Audit
     async addConsentAuditEntry(entry) {
         this.ensureReady();
@@ -2128,6 +2153,11 @@ export class MongoStorage {
         const result = await this.prisma.match.deleteMany({ where: { expiresAt: { lt: new Date() }, status: { not: 'accepted' } } });
         return result.count;
     }
+    async deleteMatchesByProfile(profile) {
+        this.ensureReady();
+        const result = await this.prisma.match.deleteMany({ where: { OR: [{ profileA: profile }, { profileB: profile }] } });
+        return result.count;
+    }
     async listAllMatches(limit = 10000) {
         this.ensureReady();
         const rows = await this.prisma.match.findMany({ take: Math.min(limit, 10000) });
@@ -3603,6 +3633,11 @@ export class MongoStorage {
         }
         return broken;
     }
+    async deleteLinksByContributor(gaii) {
+        this.ensureReady();
+        const result = await this.prisma.knowledgeLink.deleteMany({ where: { linkedBy: gaii } });
+        return result.count;
+    }
     // ══════════════════════════════════════════════════════════
     // ── Knowledge: Operator Reviews (Prisma-persisted) ──
     // ══════════════════════════════════════════════════════════
@@ -3648,6 +3683,11 @@ export class MongoStorage {
         });
         return rows.map((r) => this.toOperatorReviewRecord(r));
     }
+    async deleteReviewsByOperator(gaii) {
+        this.ensureReady();
+        const result = await this.prisma.knowledgeReview.deleteMany({ where: { operatorGaii: gaii } });
+        return result.count;
+    }
     // ══════════════════════════════════════════════════════════
     // ── Scheduler: Scheduled Jobs ──
     // ══════════════════════════════════════════════════════════
@@ -3915,6 +3955,11 @@ export class MongoStorage {
             return false;
         }
     }
+    async deleteExtensionInstancesByOwner(ownerIdentity) {
+        this.ensureReady();
+        const result = await this.prisma.extensionInstance.deleteMany({ where: { createdBy: ownerIdentity } });
+        return result.count;
+    }
     // ══════════════════════════════════════════════════════════
     // ── Federation Peers (persisted active peer connections) ──
     // ══════════════════════════════════════════════════════════
@@ -3922,8 +3967,8 @@ export class MongoStorage {
         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) },
-            update: { url: peer.url, publicKey: peer.publicKey, status: peer.status, lastSeen: new Date(peer.lastSeen) },
+            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 ?? [] },
         });
     }
     async listFederationPeers() {
@@ -3936,6 +3981,12 @@ export class MongoStorage {
             status: r.status,
             addedAt: r.addedAt instanceof Date ? r.addedAt.toISOString() : r.addedAt,
             lastSeen: r.lastSeen instanceof Date ? r.lastSeen.toISOString() : r.lastSeen,
+            shareCatalogue: r.shareCatalogue ?? true,
+            replicateMemory: r.replicateMemory ?? true,
+            allowRouting: r.allowRouting ?? true,
+            peerMode: r.peerMode || 'federation',
+            allowFederatedAuth: r.allowFederatedAuth ?? false,
+            federationAuthScopes: r.federationAuthScopes ?? [],
         }));
     }
     async deleteFederationPeer(nodeId) {
@@ -4083,6 +4134,11 @@ export class MongoStorage {
         });
         return result.count;
     }
+    async deleteDeviceAuthByOwner(ownerName) {
+        this.ensureReady();
+        const result = await this.prisma.deviceAuth.deleteMany({ where: { ownerName } });
+        return result.count;
+    }
     toDeviceAuthRecord(row) {
         return {
             deviceCode: row.deviceCode,
@@ -5196,6 +5252,50 @@ export class MongoStorage {
         const trust = { ...cap.trust, vouchCount: Math.max(0, cap.trust.vouchCount - 1) };
         await this.prisma.capability.update({ where: { id }, data: { trust: trust } });
     }
+    // ── Stats Persistence ──
+    async flushStats(counters) {
+        const ops = Object.entries(counters).map(([key, value]) => this.prisma.statsCounter.upsert({
+            where: { id: key },
+            create: { id: key, value },
+            update: { value },
+        }));
+        await Promise.all(ops);
+    }
+    async loadStats() {
+        const rows = await this.prisma.statsCounter.findMany();
+        const result = {};
+        for (const row of rows) {
+            result[row.id] = row.value;
+        }
+        return result;
+    }
+    async flushDailyHistory(history) {
+        const cutoff = new Date();
+        cutoff.setDate(cutoff.getDate() - 90);
+        const cutoffStr = cutoff.toISOString().split('T')[0];
+        const ops = [];
+        for (const [date, counters] of Object.entries(history)) {
+            for (const [key, value] of Object.entries(counters)) {
+                ops.push(this.prisma.statsDailyHistory.upsert({
+                    where: { date_key: { date, key } },
+                    create: { date, key, value },
+                    update: { value },
+                }));
+            }
+        }
+        ops.push(this.prisma.statsDailyHistory.deleteMany({ where: { date: { lt: cutoffStr } } }));
+        await Promise.all(ops);
+    }
+    async loadDailyHistory() {
+        const rows = await this.prisma.statsDailyHistory.findMany({ orderBy: { date: 'asc' } });
+        const result = {};
+        for (const row of rows) {
+            if (!result[row.date])
+                result[row.date] = {};
+            result[row.date][row.key] = row.value;
+        }
+        return result;
+    }
 }
 /**
  * Wildcard pattern matching: supports '*' (one segment) and '**' (multiple segments)