aimeat 1.4.7 → 1.5.0

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() };
@@ -3916,62 +3933,106 @@ export class MongoStorage {
         }
     }
     // ══════════════════════════════════════════════════════════
-    // ── Replication Queue (B.1) — in-memory (transient queue)
+    // ── Federation Peers (persisted active peer connections) ──
+    // ══════════════════════════════════════════════════════════
+    async saveFederationPeer(peer) {
+        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' },
+            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' },
+        });
+    }
+    async listFederationPeers() {
+        this.ensureReady();
+        const rows = await this.prisma.federationPeer.findMany();
+        return rows.map((r) => ({
+            nodeId: r.nodeId,
+            url: r.url,
+            publicKey: r.publicKey,
+            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',
+        }));
+    }
+    async deleteFederationPeer(nodeId) {
+        this.ensureReady();
+        try {
+            await this.prisma.federationPeer.delete({ where: { nodeId } });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    // ══════════════════════════════════════════════════════════
+    // ── Replication Queue (B.1) — persisted to MongoDB
     // ══════════════════════════════════════════════════════════
-    replicationQueue = new Map();
     async enqueueReplication(entry) {
+        this.ensureReady();
         const id = randomUUID();
-        const full = {
-            ...entry,
-            id,
-            attempts: 0,
-            lastAttemptAt: null,
-            status: 'pending',
-        };
-        this.replicationQueue.set(id, full);
+        await this.prisma.replicationQueue.create({
+            data: {
+                id,
+                type: entry.type,
+                targetPeers: entry.targetPeers,
+                payload: entry.payload != null ? JSON.stringify(entry.payload) : null,
+                createdAt: new Date(entry.createdAt),
+                attempts: 0,
+                lastAttemptAt: null,
+                status: 'pending',
+            },
+        });
         return id;
     }
     async dequeueReplication(peerId, limit) {
-        const results = [];
-        // Iterate in insertion order (Map preserves insertion order)
-        for (const entry of this.replicationQueue.values()) {
-            if (entry.status === 'pending' && entry.targetPeers.includes(peerId)) {
-                results.push(entry);
-                if (results.length >= limit)
-                    break;
-            }
-        }
-        return results.sort((a, b) => a.createdAt.localeCompare(b.createdAt));
+        this.ensureReady();
+        const rows = await this.prisma.replicationQueue.findMany({
+            where: { status: 'pending', targetPeers: { has: peerId } },
+            orderBy: { createdAt: 'asc' },
+            take: limit,
+        });
+        return rows.map((r) => ({
+            id: r.id,
+            type: r.type,
+            targetPeers: r.targetPeers,
+            payload: r.payload ? JSON.parse(r.payload) : null,
+            createdAt: r.createdAt instanceof Date ? r.createdAt.toISOString() : r.createdAt,
+            attempts: r.attempts,
+            lastAttemptAt: r.lastAttemptAt instanceof Date ? r.lastAttemptAt.toISOString() : r.lastAttemptAt,
+            status: r.status,
+        }));
     }
     async markReplicationSent(ids) {
-        for (const id of ids) {
-            const entry = this.replicationQueue.get(id);
-            if (entry)
-                entry.status = 'sent';
-        }
+        this.ensureReady();
+        await this.prisma.replicationQueue.updateMany({
+            where: { id: { in: ids } },
+            data: { status: 'sent' },
+        });
     }
     async markReplicationFailed(ids) {
+        this.ensureReady();
         for (const id of ids) {
-            const entry = this.replicationQueue.get(id);
-            if (entry) {
-                entry.status = 'failed';
-                entry.attempts++;
-                entry.lastAttemptAt = new Date().toISOString();
-            }
+            await this.prisma.replicationQueue.update({
+                where: { id },
+                data: { status: 'failed', attempts: { increment: 1 }, lastAttemptAt: new Date() },
+            }).catch(() => { });
         }
     }
     async pruneReplicationQueue(maxAge) {
-        let pruned = 0;
-        for (const [id, entry] of this.replicationQueue) {
-            if (new Date(entry.createdAt) < maxAge || entry.status === 'sent') {
-                this.replicationQueue.delete(id);
-                pruned++;
-            }
-        }
-        return pruned;
+        this.ensureReady();
+        const result = await this.prisma.replicationQueue.deleteMany({
+            where: { OR: [{ createdAt: { lt: maxAge } }, { status: 'sent' }] },
+        });
+        return result.count;
     }
     async replicationQueueSize() {
-        return this.replicationQueue.size;
+        this.ensureReady();
+        return this.prisma.replicationQueue.count();
     }
     // ── Device Authorization (RFC 8628) ──
     async createDeviceAuth(req) {