aimeat 1.16.2 → 1.18.0

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

@@ -195,6 +195,8 @@ export class MongoStorage {
                 languages: agent.languages ?? null,
                 mode: agent.mode ?? 'interactive',
                 maxConcurrentTasks: agent.maxConcurrentTasks ?? 1,
+                dailySpendLimit: agent.dailySpendLimit ?? null,
+                scheduleConstraintDefaults: agent.scheduleConstraintDefaults ?? null,
                 webhookUrl: agent.webhookUrl ?? null,
                 webhookSecret: agent.webhookSecret ?? null,
                 webhookEnabled: agent.webhookEnabled ?? false,
@@ -1432,7 +1434,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, languages: row.languages ?? 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, mode: (row.mode ?? 'interactive'), maxConcurrentTasks: row.maxConcurrentTasks ?? 1, 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, languages: row.languages ?? 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, mode: (row.mode ?? 'interactive'), maxConcurrentTasks: row.maxConcurrentTasks ?? 1, dailySpendLimit: row.dailySpendLimit ?? undefined, scheduleConstraintDefaults: row.scheduleConstraintDefaults ?? 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 };
@@ -3370,6 +3372,25 @@ export class MongoStorage {
         await this.prisma.notificationTemplate.deleteMany();
     }
     // ── Sessions (P3-7: Server-Side Session Tracking) ──────────
+    mapPrismaSession(s) {
+        const iso = (d) => d == null ? null : (d instanceof Date ? d.toISOString() : String(d));
+        return {
+            sessionId: s.sessionId,
+            gaii: s.gaii,
+            owner: s.owner,
+            issuedAt: iso(s.issuedAt),
+            expiresAt: iso(s.expiresAt),
+            revoked: s.revoked,
+            refreshTokenHash: s.refreshTokenHash ?? null,
+            prevTokenHash: s.prevTokenHash ?? null,
+            prevValidUntil: iso(s.prevValidUntil),
+            lastUsedAt: iso(s.lastUsedAt),
+            idleExpiresAt: iso(s.idleExpiresAt),
+            absoluteExpiresAt: iso(s.absoluteExpiresAt),
+            deviceLabel: s.deviceLabel ?? null,
+            userAgent: s.userAgent ?? null,
+        };
+    }
     async createSession(session) {
         this.ensureReady();
         await this.prisma.session.create({
@@ -3383,20 +3404,54 @@ export class MongoStorage {
             },
         });
     }
+    async createOwnerSession(session) {
+        this.ensureReady();
+        await this.prisma.session.create({
+            data: {
+                sessionId: session.sessionId,
+                gaii: session.gaii,
+                owner: session.owner,
+                issuedAt: new Date(session.issuedAt),
+                // expiresAt mirrors the idle window so listActiveSessions reflects refresh-token life.
+                expiresAt: new Date(session.idleExpiresAt),
+                revoked: false,
+                refreshTokenHash: session.refreshTokenHash,
+                idleExpiresAt: new Date(session.idleExpiresAt),
+                absoluteExpiresAt: new Date(session.absoluteExpiresAt),
+                lastUsedAt: new Date(session.lastUsedAt),
+                deviceLabel: session.deviceLabel ?? null,
+                userAgent: session.userAgent ?? null,
+            },
+        });
+    }
     async listActiveSessions(owner) {
         this.ensureReady();
         const sessions = await this.prisma.session.findMany({
             where: { owner, revoked: false },
             orderBy: { issuedAt: 'desc' },
         });
-        return sessions.map((s) => ({
-            sessionId: s.sessionId,
-            gaii: s.gaii,
-            owner: s.owner,
-            issuedAt: s.issuedAt instanceof Date ? s.issuedAt.toISOString() : String(s.issuedAt),
-            expiresAt: s.expiresAt instanceof Date ? s.expiresAt.toISOString() : String(s.expiresAt),
-            revoked: s.revoked,
-        }));
+        return sessions.map((s) => this.mapPrismaSession(s));
+    }
+    async getSessionByRefreshHash(tokenHash) {
+        this.ensureReady();
+        const s = await this.prisma.session.findFirst({
+            where: { OR: [{ refreshTokenHash: tokenHash }, { prevTokenHash: tokenHash }] },
+        });
+        return s ? this.mapPrismaSession(s) : null;
+    }
+    async rotateSessionRefresh(sessionId, update) {
+        this.ensureReady();
+        await this.prisma.session.update({
+            where: { sessionId },
+            data: {
+                refreshTokenHash: update.refreshTokenHash,
+                prevTokenHash: update.prevTokenHash,
+                prevValidUntil: update.prevValidUntil ? new Date(update.prevValidUntil) : null,
+                idleExpiresAt: new Date(update.idleExpiresAt),
+                expiresAt: new Date(update.expiresAt),
+                lastUsedAt: new Date(update.lastUsedAt),
+            },
+        });
     }
     async revokeSession(sessionId) {
         this.ensureReady();
@@ -3421,6 +3476,80 @@ export class MongoStorage {
             return false;
         return session.revoked;
     }
+    async pruneExpiredSessions(nowIso) {
+        this.ensureReady();
+        const now = new Date(nowIso);
+        // Remove fully-dead rows; keep revoked-but-unexpired rows so isSessionRevoked
+        // still rejects their short-lived access tokens.
+        const result = await this.prisma.session.deleteMany({
+            where: { OR: [{ expiresAt: { lt: now } }, { absoluteExpiresAt: { lt: now } }] },
+        });
+        return result.count;
+    }
+    // ── Personal Access Tokens ────────────────────────────────
+    mapPat(p) {
+        const iso = (d) => d == null ? null : (d instanceof Date ? d.toISOString() : String(d));
+        return {
+            id: p.id,
+            tokenHash: p.tokenHash,
+            label: p.label,
+            owner: p.owner,
+            scopes: p.scopes ?? [],
+            grantOwner: p.grantOwner,
+            grantOperator: p.grantOperator,
+            readOwnerData: p.readOwnerData,
+            gaii: p.gaii,
+            createdAt: iso(p.createdAt),
+            expiresAt: iso(p.expiresAt),
+            lastUsedAt: iso(p.lastUsedAt),
+            revoked: p.revoked,
+        };
+    }
+    async createPat(pat) {
+        this.ensureReady();
+        await this.prisma.personalAccessToken.create({
+            data: {
+                id: pat.id,
+                tokenHash: pat.tokenHash,
+                label: pat.label,
+                owner: pat.owner,
+                scopes: pat.scopes ?? [],
+                grantOwner: pat.grantOwner,
+                grantOperator: pat.grantOperator,
+                readOwnerData: pat.readOwnerData,
+                gaii: pat.gaii,
+                createdAt: new Date(pat.createdAt),
+                expiresAt: pat.expiresAt ? new Date(pat.expiresAt) : null,
+                lastUsedAt: pat.lastUsedAt ? new Date(pat.lastUsedAt) : null,
+                revoked: false,
+            },
+        });
+    }
+    async getPatByHash(tokenHash) {
+        this.ensureReady();
+        const p = await this.prisma.personalAccessToken.findFirst({ where: { tokenHash, revoked: false } });
+        return p ? this.mapPat(p) : null;
+    }
+    async listPats(owner) {
+        this.ensureReady();
+        const pats = await this.prisma.personalAccessToken.findMany({
+            where: { owner, revoked: false },
+            orderBy: { createdAt: 'desc' },
+        });
+        return pats.map((p) => this.mapPat(p));
+    }
+    async revokePat(id, owner) {
+        this.ensureReady();
+        const existing = await this.prisma.personalAccessToken.findFirst({ where: { id, owner, revoked: false } });
+        if (!existing)
+            return false;
+        await this.prisma.personalAccessToken.update({ where: { id }, data: { revoked: true } });
+        return true;
+    }
+    async touchPat(id, usedAtIso) {
+        this.ensureReady();
+        await this.prisma.personalAccessToken.update({ where: { id }, data: { lastUsedAt: new Date(usedAtIso) } });
+    }
     // ── Token Revocation ──────────────────────────────────────
     async revokeToken(tokenHash, expiresAt) {
         this.ensureReady();
@@ -3843,6 +3972,16 @@ export class MongoStorage {
             createdBy: row.createdBy,
             createdAt: row.createdAt instanceof Date ? row.createdAt.toISOString() : row.createdAt,
             updatedAt: row.updatedAt instanceof Date ? row.updatedAt.toISOString() : row.updatedAt,
+            ownerScope: row.ownerScope ?? undefined,
+            agentName: row.agentName ?? undefined,
+            agentGaii: row.agentGaii ?? undefined,
+            createdByAgent: row.createdByAgent ?? undefined,
+            displayName: row.displayName ?? undefined,
+            description: row.description ?? undefined,
+            purpose: row.purpose ?? undefined,
+            timezone: row.timezone ?? undefined,
+            constraints: row.constraints ?? undefined,
+            runCount: row.runCount ?? undefined,
         };
     }
     async createScheduledJob(record) {
@@ -3867,6 +4006,16 @@ export class MongoStorage {
                 createdBy: record.createdBy,
                 createdAt: new Date(record.createdAt),
                 updatedAt: new Date(record.updatedAt),
+                ownerScope: record.ownerScope,
+                agentName: record.agentName,
+                agentGaii: record.agentGaii,
+                createdByAgent: record.createdByAgent ?? false,
+                displayName: record.displayName,
+                description: record.description,
+                purpose: record.purpose,
+                timezone: record.timezone,
+                constraints: record.constraints,
+                runCount: record.runCount ?? 0,
             },
         });
         return record;
@@ -3885,6 +4034,10 @@ export class MongoStorage {
             where.extensionName = filter.extensionName;
         if (filter?.enabled !== undefined)
             where.enabled = filter.enabled;
+        if (filter?.ownerScope !== undefined)
+            where.ownerScope = filter.ownerScope;
+        if (filter?.agentGaii !== undefined)
+            where.agentGaii = filter.agentGaii;
         const rows = await this.prisma.scheduledJob.findMany({ where });
         return rows.map((r) => this.toScheduledJobRecord(r));
     }
@@ -3922,6 +4075,26 @@ export class MongoStorage {
             data.lastRunDurationMs = updates.lastRunDurationMs;
         if (updates.nextRunAt !== undefined)
             data.nextRunAt = updates.nextRunAt ? new Date(updates.nextRunAt) : null;
+        if (updates.ownerScope !== undefined)
+            data.ownerScope = updates.ownerScope;
+        if (updates.agentName !== undefined)
+            data.agentName = updates.agentName;
+        if (updates.agentGaii !== undefined)
+            data.agentGaii = updates.agentGaii;
+        if (updates.createdByAgent !== undefined)
+            data.createdByAgent = updates.createdByAgent;
+        if (updates.displayName !== undefined)
+            data.displayName = updates.displayName;
+        if (updates.description !== undefined)
+            data.description = updates.description;
+        if (updates.purpose !== undefined)
+            data.purpose = updates.purpose;
+        if (updates.timezone !== undefined)
+            data.timezone = updates.timezone;
+        if (updates.constraints !== undefined)
+            data.constraints = updates.constraints;
+        if (updates.runCount !== undefined)
+            data.runCount = updates.runCount;
         const row = await this.prisma.scheduledJob.update({ where: { id }, data });
         return this.toScheduledJobRecord(row);
     }
@@ -3949,6 +4122,7 @@ export class MongoStorage {
                 durationMs: entry.durationMs,
                 memoryReads: entry.memoryReads,
                 memoryWrites: entry.memoryWrites,
+                taskId: entry.taskId,
                 createdAt: new Date(entry.createdAt),
             },
         });
@@ -4007,6 +4181,7 @@ export class MongoStorage {
             durationMs: row.durationMs,
             memoryReads: Array.isArray(row.memoryReads) ? row.memoryReads : JSON.parse(row.memoryReads || '[]'),
             memoryWrites: Array.isArray(row.memoryWrites) ? row.memoryWrites : JSON.parse(row.memoryWrites || '[]'),
+            taskId: row.taskId ?? undefined,
             createdAt: row.createdAt instanceof Date ? row.createdAt.toISOString() : row.createdAt,
         };
     }
@@ -5455,6 +5630,7 @@ export class MongoStorage {
             completedAt: row.completedAt ? (row.completedAt instanceof Date ? row.completedAt.toISOString() : row.completedAt) : undefined,
             deliverableKey: row.deliverableKey ?? undefined,
             rating: row.rating ?? undefined,
+            triage: (row.triage ?? undefined),
         };
     }
     toTaskEventRecord(row) {
@@ -5490,6 +5666,7 @@ export class MongoStorage {
                 completedAt: record.completedAt ? new Date(record.completedAt) : null,
                 deliverableKey: record.deliverableKey ?? null,
                 rating: record.rating ?? null,
+                triage: record.triage ?? null,
             },
         });
         return record;
@@ -5569,6 +5746,7 @@ export class MongoStorage {
                     completedAt: merged.completedAt ? new Date(merged.completedAt) : null,
                     deliverableKey: merged.deliverableKey ?? null,
                     rating: merged.rating ?? null,
+                    triage: merged.triage ?? null,
                 },
             });
             return this.toTaskRecord(row);