aimeat 1.7.0 → 1.9.0

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

@@ -186,6 +186,18 @@ export class MongoStorage {
                 technicalCapabilities: agent.technicalCapabilities ?? null,
                 domainCapabilities: agent.domainCapabilities ?? null,
                 activityStats: agent.activityStats ?? null,
+                modulesLoaded: agent.modulesLoaded ?? null,
+                agentLimitations: agent.agentLimitations ?? null,
+                webhookUrl: agent.webhookUrl ?? null,
+                webhookSecret: agent.webhookSecret ?? null,
+                webhookEnabled: agent.webhookEnabled ?? false,
+                webhookLastSuccess: agent.webhookLastSuccess ? new Date(agent.webhookLastSuccess) : null,
+                webhookLastFailure: agent.webhookLastFailure ? new Date(agent.webhookLastFailure) : null,
+                webhookFailCount: agent.webhookFailCount ?? 0,
+                platform: agent.platform ?? null,
+                platformVersion: agent.platformVersion ?? null,
+                platformDetectedBy: agent.platformDetectedBy ?? null,
+                tags: agent.tags ?? [],
                 createdAt: new Date(agent.createdAt),
                 lastSeen: new Date(agent.lastSeen),
             },
@@ -197,6 +209,11 @@ export class MongoStorage {
         const row = await this.prisma.agent.findUnique({ where: { gaii } });
         return row ? this.toAgentRecord(row) : null;
     }
+    async getAgentByName(name, nodeId) {
+        this.ensureReady();
+        const row = await this.prisma.agent.findFirst({ where: { name } });
+        return row ? this.toAgentRecord(row) : null;
+    }
     async getAgentsByOwner(owner) {
         this.ensureReady();
         const rows = await this.prisma.agent.findMany({ where: { owner } });
@@ -208,7 +225,8 @@ export class MongoStorage {
             const row = await this.prisma.agent.update({ where: { gaii }, data: updates });
             return this.toAgentRecord(row);
         }
-        catch {
+        catch (err) {
+            logger.warn('updateAgent failed for %s: %s', gaii, err.message);
             return null;
         }
     }
@@ -294,6 +312,19 @@ export class MongoStorage {
         catch { /* not found */ }
         // Agent activity
         await this.prisma.agentActivity.deleteMany({ where: { agentGaii: gaii } });
+        // Agent messages
+        await this.prisma.agentMessage.deleteMany({ where: { agentGaii: gaii } });
+        // Telemetry events
+        await this.prisma.telemetryEvent.deleteMany({ where: { agentGaii: gaii } });
+        // Webhook delivery logs
+        await this.prisma.webhookDeliveryLog.deleteMany({ where: { agentGaii: gaii } });
+        // Onboarding record
+        try {
+            await this.prisma.agentOnboarding.delete({ where: { agentGaii: gaii } });
+        }
+        catch { /* not found */ }
+        // Sharing groups
+        await this.prisma.sharingGroup.deleteMany({ where: { ownerGaii: gaii } });
     }
     async listAgents() {
         this.ensureReady();
@@ -1380,7 +1411,7 @@ 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, federate: row.federate ?? false, technicalCapabilities: row.technicalCapabilities ?? undefined, domainCapabilities: row.domainCapabilities ?? undefined, activityStats: row.activityStats ?? 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, technicalCapabilities: row.technicalCapabilities ?? undefined, domainCapabilities: row.domainCapabilities ?? undefined, activityStats: row.activityStats ?? undefined, modulesLoaded: row.modulesLoaded ?? undefined, agentLimitations: row.agentLimitations ?? undefined, webhookUrl: row.webhookUrl ?? undefined, webhookSecret: row.webhookSecret ?? undefined, webhookEnabled: row.webhookEnabled ?? false, webhookLastSuccess: row.webhookLastSuccess ? row.webhookLastSuccess.toISOString() : undefined, webhookLastFailure: row.webhookLastFailure ? row.webhookLastFailure.toISOString() : undefined, webhookFailCount: row.webhookFailCount ?? 0, platform: row.platform ?? undefined, platformVersion: row.platformVersion ?? undefined, platformDetectedBy: row.platformDetectedBy ?? undefined, tags: row.tags?.length ? row.tags : undefined, createdAt: row.createdAt.toISOString(), lastSeen: row.lastSeen.toISOString() };
     }
     toMemoryRecord(row) {
         return { key: row.key, ownerGaii: row.ownerGaii, value: row.value, visibility: row.visibility, groupId: row.groupId ?? undefined, 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 };
@@ -5690,7 +5721,7 @@ export class MongoStorage {
     }
     // ── Agent Directives ──
     toDirectivesRecord(row) {
-        return {
+        const record = {
             agentGaii: row.agentGaii,
             purpose: row.purpose,
             rules: row.rules,
@@ -5698,6 +5729,9 @@ export class MongoStorage {
             resources: row.resources,
             updatedAt: row.updatedAt instanceof Date ? row.updatedAt.toISOString() : row.updatedAt,
         };
+        if (row.budgetLimits)
+            record.budgetLimits = row.budgetLimits;
+        return record;
     }
     toOwnerDefaultsRecord(row) {
         const record = {
@@ -5727,12 +5761,14 @@ export class MongoStorage {
                 rules: record.rules,
                 memoryAreas: record.memoryAreas,
                 resources: record.resources,
+                budgetLimits: record.budgetLimits ?? null,
             },
             update: {
                 purpose: record.purpose,
                 rules: record.rules,
                 memoryAreas: record.memoryAreas,
                 resources: record.resources,
+                budgetLimits: record.budgetLimits ?? null,
             },
         });
         return this.toDirectivesRecord(row);
@@ -5947,6 +5983,221 @@ export class MongoStorage {
         results.sort((a, b) => b.updatedAt.localeCompare(a.updatedAt));
         return results;
     }
+    // ══════════════════════════════════════════════════════════
+    // ── Agent Onboarding ──
+    // ══════════════════════════════════════════════════════════
+    toOnboardingRecord(row) {
+        return {
+            agentGaii: row.agentGaii,
+            status: row.status,
+            startedAt: row.startedAt instanceof Date ? row.startedAt.toISOString() : row.startedAt,
+            completedAt: row.completedAt instanceof Date ? row.completedAt.toISOString() : row.completedAt ?? undefined,
+            steps: row.steps,
+            readinessScore: row.readinessScore ?? undefined,
+            readinessLevel: row.readinessLevel ?? undefined,
+            detectedPlatform: row.detectedPlatform ?? undefined,
+            installedRuntime: row.installedRuntime ?? undefined,
+            onboardingBaseline: row.onboardingBaseline ?? undefined,
+            operationalHealth: row.operationalHealth ?? undefined,
+            healthComponents: row.healthComponents ?? undefined,
+            healthRecalculatedAt: row.healthRecalculatedAt instanceof Date ? row.healthRecalculatedAt.toISOString() : row.healthRecalculatedAt ?? undefined,
+            readinessOverride: row.readinessOverride ?? undefined,
+        };
+    }
+    async createOnboarding(record) {
+        this.ensureReady();
+        const row = await this.prisma.agentOnboarding.create({
+            data: {
+                agentGaii: record.agentGaii,
+                status: record.status,
+                startedAt: new Date(record.startedAt),
+                completedAt: record.completedAt ? new Date(record.completedAt) : null,
+                steps: record.steps,
+                readinessScore: record.readinessScore ?? null,
+                readinessLevel: record.readinessLevel ?? null,
+                detectedPlatform: record.detectedPlatform ?? null,
+                installedRuntime: record.installedRuntime ?? null,
+                onboardingBaseline: record.onboardingBaseline ?? null,
+                operationalHealth: record.operationalHealth ?? null,
+                healthComponents: record.healthComponents ?? null,
+                healthRecalculatedAt: record.healthRecalculatedAt ? new Date(record.healthRecalculatedAt) : null,
+                readinessOverride: record.readinessOverride ?? null,
+            },
+        });
+        return this.toOnboardingRecord(row);
+    }
+    async getOnboarding(agentGaii) {
+        this.ensureReady();
+        try {
+            const row = await this.prisma.agentOnboarding.findUnique({ where: { agentGaii } });
+            return row ? this.toOnboardingRecord(row) : null;
+        }
+        catch {
+            return null;
+        }
+    }
+    async updateOnboarding(agentGaii, updates) {
+        this.ensureReady();
+        try {
+            const data = {};
+            if (updates.status !== undefined)
+                data.status = updates.status;
+            if (updates.startedAt !== undefined)
+                data.startedAt = new Date(updates.startedAt);
+            if (updates.completedAt !== undefined)
+                data.completedAt = updates.completedAt ? new Date(updates.completedAt) : null;
+            if (updates.steps !== undefined)
+                data.steps = updates.steps;
+            if (updates.readinessScore !== undefined)
+                data.readinessScore = updates.readinessScore ?? null;
+            if (updates.readinessLevel !== undefined)
+                data.readinessLevel = updates.readinessLevel ?? null;
+            if (updates.detectedPlatform !== undefined)
+                data.detectedPlatform = updates.detectedPlatform ?? null;
+            if (updates.installedRuntime !== undefined)
+                data.installedRuntime = updates.installedRuntime ?? null;
+            if (updates.onboardingBaseline !== undefined)
+                data.onboardingBaseline = updates.onboardingBaseline ?? null;
+            if (updates.operationalHealth !== undefined)
+                data.operationalHealth = updates.operationalHealth ?? null;
+            if (updates.healthComponents !== undefined)
+                data.healthComponents = updates.healthComponents ?? null;
+            if (updates.healthRecalculatedAt !== undefined)
+                data.healthRecalculatedAt = updates.healthRecalculatedAt ? new Date(updates.healthRecalculatedAt) : null;
+            if (updates.readinessOverride !== undefined)
+                data.readinessOverride = updates.readinessOverride ?? null;
+            const row = await this.prisma.agentOnboarding.update({ where: { agentGaii }, data });
+            return this.toOnboardingRecord(row);
+        }
+        catch {
+            return null;
+        }
+    }
+    async deleteOnboarding(agentGaii) {
+        this.ensureReady();
+        try {
+            await this.prisma.agentOnboarding.delete({ where: { agentGaii } });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    async listOnboardingByOwner(owner) {
+        this.ensureReady();
+        const rows = await this.prisma.agentOnboarding.findMany({
+            where: { agentGaii: { contains: `#${owner}@` } },
+            orderBy: { startedAt: 'desc' },
+        });
+        return rows.map((r) => this.toOnboardingRecord(r));
+    }
+    async listOnboardingByStatus(status) {
+        this.ensureReady();
+        const rows = await this.prisma.agentOnboarding.findMany({
+            where: { status },
+            orderBy: { startedAt: 'desc' },
+        });
+        return rows.map((r) => this.toOnboardingRecord(r));
+    }
+    // ── Agent Telemetry ──
+    toTelemetryEvent(row) {
+        return {
+            id: row.id,
+            agentGaii: row.agentGaii,
+            type: row.type,
+            data: row.data,
+            sessionId: row.sessionId ?? undefined,
+            taskId: row.taskId ?? undefined,
+            createdAt: row.createdAt.toISOString(),
+        };
+    }
+    async appendTelemetry(event) {
+        this.ensureReady();
+        await this.prisma.telemetryEvent.create({
+            data: {
+                id: event.id,
+                agentGaii: event.agentGaii,
+                type: event.type,
+                data: event.data,
+                sessionId: event.sessionId ?? null,
+                taskId: event.taskId ?? null,
+                createdAt: new Date(event.createdAt),
+            },
+        });
+    }
+    async listTelemetry(agentGaii, opts) {
+        this.ensureReady();
+        const where = { agentGaii };
+        if (opts.type)
+            where.type = opts.type;
+        if (opts.since)
+            where.createdAt = { gt: new Date(opts.since) };
+        const rows = await this.prisma.telemetryEvent.findMany({
+            where,
+            orderBy: { createdAt: 'desc' },
+            take: opts.limit ?? 50,
+        });
+        return rows.map((r) => this.toTelemetryEvent(r));
+    }
+    // ── Webhook Delivery Log ──
+    toDeliveryLog(row) {
+        return {
+            id: row.id,
+            agentGaii: row.agentGaii,
+            event: row.event,
+            payload: row.payload,
+            status: row.status,
+            httpStatus: row.httpStatus ?? undefined,
+            errorMessage: row.errorMessage ?? undefined,
+            attemptCount: row.attemptCount,
+            latencyMs: row.latencyMs,
+            createdAt: row.createdAt.toISOString(),
+        };
+    }
+    async appendDeliveryLog(log) {
+        this.ensureReady();
+        await this.prisma.webhookDeliveryLog.create({
+            data: {
+                id: log.id,
+                agentGaii: log.agentGaii,
+                event: log.event,
+                payload: log.payload,
+                status: log.status,
+                httpStatus: log.httpStatus ?? null,
+                errorMessage: log.errorMessage ?? null,
+                attemptCount: log.attemptCount,
+                latencyMs: log.latencyMs,
+                createdAt: new Date(log.createdAt),
+            },
+        });
+    }
+    async listDeliveryLog(agentGaii, limit) {
+        this.ensureReady();
+        const rows = await this.prisma.webhookDeliveryLog.findMany({
+            where: { agentGaii },
+            orderBy: { createdAt: 'desc' },
+            take: limit ?? 50,
+        });
+        return rows.map((r) => this.toDeliveryLog(r));
+    }
+    async pruneDeliveryLog(agentGaii, keepCount) {
+        this.ensureReady();
+        // Find the cutoff: the keepCount-th newest record's createdAt
+        const cutoffRows = await this.prisma.webhookDeliveryLog.findMany({
+            where: { agentGaii },
+            orderBy: { createdAt: 'desc' },
+            skip: keepCount,
+            take: 1,
+            select: { createdAt: true },
+        });
+        if (cutoffRows.length === 0)
+            return 0;
+        const cutoff = cutoffRows[0].createdAt;
+        const result = await this.prisma.webhookDeliveryLog.deleteMany({
+            where: { agentGaii, createdAt: { lte: cutoff } },
+        });
+        return result.count;
+    }
 }
 /**
  * Wildcard pattern matching: supports '*' (one segment) and '**' (multiple segments)