@cat-factory/node-server 0.80.5 → 0.81.0

package/dist/repositories/drizzle.js

@@ -2,7 +2,7 @@ import { LLM_WARNING_FINISH_REASONS } from '@cat-factory/kernel';
 import { agentRunKindSchema, decodeInitiativeRow, isWebSearchProvider, } from '@cat-factory/contracts';
 import { decodeEnum, tryDecodeRows, blockInsertValues, blockPatchToColumns, parseIssueIntakeColumn, serializeIssueIntakeColumn, rowToBlock, rowToExecution, executionToDetail, rowToPipeline, rowToSandboxExperiment, rowToSandboxFixture, rowToSandboxGrade, rowToSandboxPromptVersion, rowToSandboxRun, rowToWorkspace, } from '@cat-factory/server';
 import { and, asc, count, desc, eq, gte, inArray, isNull, lt, ne, notInArray, or, sql, } from 'drizzle-orm';
-import { accountInvitations, passwordResetTokens, accountSettings, localSettings, accounts, agentContextSnapshots, agentSearchQueries, agentRuns, blocks, consensusSessions, incidentEnrichmentConnections, observabilityConnections, packageRegistryConnections, emailConnections, llmCallMetrics, provisioningLog, sharedStacks, memberships, mergeThresholdPresets, releaseHealthConfigs, pipelineScheduleRuns, pipelineSchedules, pipelines, requirementReviews, docInterviewSessions, kaizenGradings, kaizenVerifiedCombos, clarityReviews, binaryArtifacts, brainstormSessions, initiatives, sandboxPromptVersions, sandboxFixtures, sandboxExperiments, sandboxRuns, sandboxGrades, services, tokenUsage, trackerSettings, modelPresets, userIdentities, users, workspaceFragmentDefaults, workspaceServices, workspaceSettings, workspaces, } from '../db/schema.js';
+import { accountInvitations, passwordResetTokens, accountSettings, localSettings, accounts, agentContextSnapshots, agentSearchQueries, agentRuns, blocks, consensusSessions, incidentEnrichmentConnections, observabilityConnections, packageRegistryConnections, emailConnections, llmCallMetrics, provisioningLog, sharedStacks, memberships, mergeThresholdPresets, releaseHealthConfigs, pipelineScheduleRuns, pipelineSchedules, pipelines, requirementReviews, docInterviewSessions, kaizenGradings, kaizenVerifiedCombos, clarityReviews, binaryArtifacts, brainstormSessions, initiatives, sandboxPromptVersions, sandboxFixtures, sandboxExperiments, sandboxRuns, sandboxGrades, services, tokenUsage, trackerSettings, modelPresets, userIdentities, users, userSettings, workspaceFragmentDefaults, workspaceServices, workspaceSettings, workspaces, } from '../db/schema.js';
 // Drizzle/Postgres implementations of the core kernel repository ports. The
 // row<->domain mapping is the SAME shared mapping the Cloudflare D1 repos use
 // (@cat-factory/server), so behaviour matches across stores; this layer only owns
@@ -532,6 +532,7 @@ function rowToAccount(row) {
         ...(row.default_cloud_provider
             ? { defaultCloudProvider: row.default_cloud_provider }
             : {}),
+        ...(row.spend_monthly_limit != null ? { spendMonthlyLimit: row.spend_monthly_limit } : {}),
     };
 }
 class DrizzleAccountRepository {
@@ -567,18 +568,23 @@ class DrizzleAccountRepository {
             owner_user_id: account.ownerUserId,
             created_at: account.createdAt,
             default_cloud_provider: account.defaultCloudProvider ?? null,
+            spend_monthly_limit: account.spendMonthlyLimit ?? null,
         });
     }
     async rename(id, name) {
         await this.db.update(accounts).set({ name }).where(eq(accounts.id, id));
     }
     async updateSettings(id, patch) {
-        if (!('defaultCloudProvider' in patch))
+        const set = {};
+        if ('defaultCloudProvider' in patch) {
+            set.default_cloud_provider = patch.defaultCloudProvider ?? null;
+        }
+        if ('spendMonthlyLimit' in patch) {
+            set.spend_monthly_limit = patch.spendMonthlyLimit ?? null;
+        }
+        if (Object.keys(set).length === 0)
             return;
-        await this.db
-            .update(accounts)
-            .set({ default_cloud_provider: patch.defaultCloudProvider ?? null })
-            .where(eq(accounts.id, id));
+        await this.db.update(accounts).set(set).where(eq(accounts.id, id));
     }
     async findPersonalByUser(userId) {
         const [row] = await this.db
@@ -930,6 +936,8 @@ class DrizzleTokenUsageRepository {
         await this.db.insert(tokenUsage).values({
             id: usage.id,
             workspace_id: usage.workspaceId,
+            account_id: usage.accountId,
+            user_id: usage.userId,
             execution_id: usage.executionId,
             agent_kind: usage.agentKind,
             provider: usage.provider,
@@ -974,6 +982,36 @@ class DrizzleTokenUsageRepository {
             costEstimate: row?.cost ?? 0,
         };
     }
+    async totalsSinceForAccount(accountId, epochMs) {
+        const [row] = await this.db
+            .select({
+            input: sql `coalesce(sum(${tokenUsage.input_tokens}), 0)::bigint`,
+            output: sql `coalesce(sum(${tokenUsage.output_tokens}), 0)::bigint`,
+            cost: sql `coalesce(sum(${tokenUsage.cost_estimate}), 0)::float8`,
+        })
+            .from(tokenUsage)
+            .where(and(eq(tokenUsage.account_id, accountId), gte(tokenUsage.created_at, epochMs)));
+        return {
+            inputTokens: Number(row?.input ?? 0),
+            outputTokens: Number(row?.output ?? 0),
+            costEstimate: row?.cost ?? 0,
+        };
+    }
+    async totalsSinceForUser(userId, epochMs) {
+        const [row] = await this.db
+            .select({
+            input: sql `coalesce(sum(${tokenUsage.input_tokens}), 0)::bigint`,
+            output: sql `coalesce(sum(${tokenUsage.output_tokens}), 0)::bigint`,
+            cost: sql `coalesce(sum(${tokenUsage.cost_estimate}), 0)::float8`,
+        })
+            .from(tokenUsage)
+            .where(and(eq(tokenUsage.user_id, userId), gte(tokenUsage.created_at, epochMs)));
+        return {
+            inputTokens: Number(row?.input ?? 0),
+            outputTokens: Number(row?.output ?? 0),
+            costEstimate: row?.cost ?? 0,
+        };
+    }
     async deleteOlderThan(epochMs) {
         const deleted = await this.db
             .delete(tokenUsage)
@@ -982,6 +1020,29 @@ class DrizzleTokenUsageRepository {
         return deleted.length;
     }
 }
+class DrizzleUserSettingsRepository {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    async get(userId) {
+        const [row] = await this.db.select().from(userSettings).where(eq(userSettings.user_id, userId));
+        return row ? { spendMonthlyLimit: row.spend_monthly_limit } : null;
+    }
+    async upsert(userId, settings) {
+        await this.db
+            .insert(userSettings)
+            .values({
+            user_id: userId,
+            spend_monthly_limit: settings.spendMonthlyLimit ?? null,
+            updated_at: Date.now(),
+        })
+            .onConflictDoUpdate({
+            target: userSettings.user_id,
+            set: { spend_monthly_limit: settings.spendMonthlyLimit ?? null, updated_at: Date.now() },
+        });
+    }
+}
 function rowToLlmMetric(row) {
     return {
         id: row.id,
@@ -3770,6 +3831,7 @@ export function createDrizzleRepositories(db, clock) {
         mergePresetRepository: new DrizzleMergePresetRepository(db),
         sharedStackRepository: new DrizzleSharedStackRepository(db),
         workspaceSettingsRepository: new DrizzleWorkspaceSettingsRepository(db),
+        userSettingsRepository: new DrizzleUserSettingsRepository(db),
         observabilityConnectionRepository: new DrizzleObservabilityConnectionRepository(db),
         packageRegistryConnectionRepository: new DrizzlePackageRegistryConnectionRepository(db),
         incidentEnrichmentConnectionRepository: new DrizzleIncidentEnrichmentConnectionRepository(db),