@mastra/factory 0.0.0-aggregate-trace-20261015182851

package/dist/storage/domains/credentials/base.js

@@ -0,0 +1,294 @@
+import { createPlaintextFactorySecretEncryption } from "../../../secret-encryption.js";
+import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
+//#region src/storage/domains/credentials/base.ts
+/** Mirror of `AuthStorage.getApiKey()`'s expiry check. */
+function isOAuthCredentialExpired(credential, now = Date.now()) {
+	return now >= credential.expires;
+}
+const MODEL_CREDENTIALS_SCHEMA = {
+	name: "model_provider_credentials",
+	columns: {
+		id: { type: "uuid-pk" },
+		org_id: { type: "text" },
+		user_id: {
+			type: "text",
+			nullable: true
+		},
+		provider: { type: "text" },
+		type: { type: "text" },
+		data: { type: "json" },
+		created_at: { type: "timestamp" },
+		updated_at: { type: "timestamp" }
+	},
+	uniqueIndexes: [{
+		name: "model_provider_credentials_user_unique",
+		columns: [
+			"org_id",
+			"user_id",
+			"provider"
+		],
+		whereNotNull: "user_id"
+	}, {
+		name: "model_provider_credentials_org_unique",
+		columns: ["org_id", "provider"],
+		whereNull: "user_id"
+	}]
+};
+const OAUTH_LOGIN_SESSIONS_SCHEMA = {
+	name: "oauth_login_sessions",
+	columns: {
+		session_id: {
+			type: "text",
+			primaryKey: true
+		},
+		org_id: { type: "text" },
+		user_id: { type: "text" },
+		provider: { type: "text" },
+		kind: { type: "text" },
+		credential_scope: {
+			type: "text",
+			nullable: true
+		},
+		pending: { type: "json" },
+		expires_at: { type: "timestamp" },
+		next_poll_at: {
+			type: "timestamp",
+			nullable: true
+		},
+		created_at: { type: "timestamp" }
+	}
+};
+function toSessionRow(db) {
+	return {
+		sessionId: db.session_id,
+		orgId: db.org_id,
+		userId: db.user_id,
+		provider: db.provider,
+		kind: db.kind,
+		...db.credential_scope ? { credentialScope: db.credential_scope } : {},
+		pending: db.pending,
+		expiresAt: db.expires_at,
+		nextPollAt: db.next_poll_at,
+		createdAt: db.created_at
+	};
+}
+/** Filter selecting exactly the tenant's row for a provider (`null` = org row). */
+function tenantWhere(tenant, provider) {
+	return {
+		org_id: tenant.orgId,
+		provider,
+		user_id: tenant.userId ?? null
+	};
+}
+/**
+* Model-credentials storage, written once against the generic
+* `FactoryStorageOps` surface. `refreshOAuth()` and `claimLoginSession()`
+* ride `updateAtomic` so concurrent replicas serialize instead of
+* invalidating each other's rotating tokens / double-claiming a flow.
+*/
+var ModelCredentialsStorage = class extends FactoryStorageDomain {
+	encryption;
+	constructor(encryption = createPlaintextFactorySecretEncryption()) {
+		super("model-credentials");
+		this.encryption = encryption;
+	}
+	async init() {
+		await this.ensureCollections([MODEL_CREDENTIALS_SCHEMA, OAUTH_LOGIN_SESSIONS_SCHEMA]);
+		const rows = await this.ops.findMany("model_provider_credentials", {});
+		await Promise.all(rows.map((row) => this.#migrateCredential(row.id)));
+	}
+	async dangerouslyClearAll() {
+		await this.ops.deleteMany("oauth_login_sessions", {});
+		await this.ops.deleteMany("model_provider_credentials", {});
+	}
+	get #db() {
+		return this.ops;
+	}
+	async #decryptCredential(value) {
+		return (await this.encryption.decrypt(value)).value;
+	}
+	async #migrateCredential(id) {
+		await this.#db.updateAtomic("model_provider_credentials", { id }, async (row) => {
+			const decrypted = await this.encryption.decrypt(row.data);
+			if (!decrypted.needsReencryption) return null;
+			return { data: await this.encryption.encrypt(decrypted.value) };
+		});
+	}
+	/** Read the tenant's credential for a provider at exactly that scope. */
+	async getCredential(tenant, provider) {
+		const row = await this.#db.findOne("model_provider_credentials", tenantWhere(tenant, provider));
+		return row ? this.#decryptCredential(row.data) : void 0;
+	}
+	/** Upsert the tenant's credential (`created_at` is preserved on update). */
+	async setCredential(tenant, provider, credential) {
+		const where = tenantWhere(tenant, provider);
+		const encrypted = await this.encryption.encrypt(credential);
+		const update = async () => this.#db.updateMany("model_provider_credentials", where, {
+			type: credential.type,
+			data: encrypted,
+			updated_at: /* @__PURE__ */ new Date()
+		});
+		let lastError;
+		for (let attempt = 0; attempt < 2; attempt++) {
+			if (await update() > 0) return;
+			const now = /* @__PURE__ */ new Date();
+			try {
+				await this.#db.insertOne("model_provider_credentials", {
+					org_id: tenant.orgId,
+					user_id: tenant.userId ?? null,
+					provider,
+					type: credential.type,
+					data: encrypted,
+					created_at: now,
+					updated_at: now
+				});
+				return;
+			} catch (error) {
+				if (!(error instanceof UniqueViolationError)) throw error;
+				lastError = error;
+			}
+		}
+		throw lastError;
+	}
+	/** Delete the tenant's credential at exactly that scope. True when a row was removed. */
+	async removeCredential(tenant, provider) {
+		return await this.#db.deleteMany("model_provider_credentials", tenantWhere(tenant, provider)) > 0;
+	}
+	/** All credentials visible to a user: their own rows plus the org's shared rows. */
+	async listCredentials(orgId, userId) {
+		const [userRows, orgRows] = await Promise.all([this.#db.findMany("model_provider_credentials", {
+			org_id: orgId,
+			user_id: userId
+		}), this.#db.findMany("model_provider_credentials", {
+			org_id: orgId,
+			user_id: null
+		})]);
+		return Promise.all([...userRows.map(async (row) => ({
+			provider: row.provider,
+			scope: "user",
+			credential: await this.#decryptCredential(row.data),
+			updatedAt: row.updated_at
+		})), ...orgRows.map(async (row) => ({
+			provider: row.provider,
+			scope: "org",
+			credential: await this.#decryptCredential(row.data),
+			updatedAt: row.updated_at
+		}))]);
+	}
+	/** Resolve a provider credential from exactly one tenancy scope. */
+	async resolveCredential(orgId, userId, provider, scope) {
+		const readUser = async () => {
+			const row = await this.#db.findOne("model_provider_credentials", {
+				org_id: orgId,
+				provider,
+				user_id: userId
+			});
+			return row ? {
+				provider: row.provider,
+				scope: "user",
+				credential: await this.#decryptCredential(row.data)
+			} : void 0;
+		};
+		const readOrg = async () => {
+			const row = await this.#db.findOne("model_provider_credentials", {
+				org_id: orgId,
+				provider,
+				user_id: null
+			});
+			return row ? {
+				provider: row.provider,
+				scope: "org",
+				credential: await this.#decryptCredential(row.data)
+			} : void 0;
+		};
+		return scope === "org" ? readOrg() : readUser();
+	}
+	/**
+	* Refresh the tenant's OAuth credential under the backend's atomic
+	* read-modify-write so concurrent replicas don't invalidate each other's
+	* rotating refresh tokens. The expiry is re-checked under the lock —
+	* another replica may have refreshed already, in which case `refreshFn` is
+	* skipped and the fresh credential is returned. Returns `undefined` when no
+	* OAuth row exists for the tenant.
+	*/
+	async refreshOAuth(tenant, provider, refreshFn) {
+		let result;
+		await this.#db.updateAtomic("model_provider_credentials", tenantWhere(tenant, provider), async (row) => {
+			const decrypted = await this.encryption.decrypt(row.data);
+			if (decrypted.value.type !== "oauth") return null;
+			const current = decrypted.value;
+			if (!isOAuthCredentialExpired(current)) {
+				result = current;
+				return decrypted.needsReencryption ? { data: await this.encryption.encrypt(current) } : null;
+			}
+			const next = await refreshFn(current);
+			result = next;
+			return {
+				data: await this.encryption.encrypt(next),
+				updated_at: /* @__PURE__ */ new Date()
+			};
+		});
+		return result;
+	}
+	/** Persist a pending login flow started by a web route. */
+	async createLoginSession(input) {
+		return toSessionRow(await this.#db.insertOne("oauth_login_sessions", {
+			session_id: input.sessionId,
+			org_id: input.orgId,
+			user_id: input.userId,
+			provider: input.provider,
+			kind: input.kind,
+			credential_scope: input.credentialScope ?? null,
+			pending: input.pending,
+			expires_at: input.expiresAt,
+			next_poll_at: input.nextPollAt ?? null,
+			created_at: /* @__PURE__ */ new Date()
+		}));
+	}
+	/**
+	* Read a pending login flow. An expired session is deleted on read (TTL
+	* cleanup) and reported as absent.
+	*/
+	async getLoginSession(sessionId) {
+		const row = await this.#db.findOne("oauth_login_sessions", { session_id: sessionId });
+		if (!row) return void 0;
+		if (row.expires_at.getTime() <= Date.now()) {
+			await this.#db.deleteMany("oauth_login_sessions", { session_id: sessionId });
+			return;
+		}
+		return toSessionRow(row);
+	}
+	/**
+	* Atomically claim a due login session for one upstream completion/poll attempt.
+	* Returns undefined when the session is missing, expired, owned by another
+	* tenant/provider, or already claimed by a concurrent request.
+	*/
+	async claimLoginSession(sessionId, owner) {
+		const now = Date.now();
+		let claimed = false;
+		const updated = await this.#db.updateAtomic("oauth_login_sessions", { session_id: sessionId }, (row) => {
+			if (row.org_id !== owner.orgId || row.user_id !== owner.userId || row.provider !== owner.provider || row.kind !== owner.kind) return null;
+			if (row.expires_at.getTime() <= now) return null;
+			if (row.next_poll_at && row.next_poll_at.getTime() > now) return null;
+			claimed = true;
+			return { next_poll_at: row.expires_at };
+		});
+		return claimed && updated ? toSessionRow(updated) : void 0;
+	}
+	/** Update a pending flow's serialized state and/or next allowed poll time. */
+	async touchLoginSession(sessionId, updates) {
+		const set = {};
+		if (updates.pending !== void 0) set.pending = updates.pending;
+		if (updates.nextPollAt !== void 0) set.next_poll_at = updates.nextPollAt;
+		if (Object.keys(set).length === 0) return;
+		await this.#db.updateMany("oauth_login_sessions", { session_id: sessionId }, set);
+	}
+	/** Remove a pending login flow (completed, cancelled, or failed). */
+	async deleteLoginSession(sessionId) {
+		await this.#db.deleteMany("oauth_login_sessions", { session_id: sessionId });
+	}
+};
+//#endregion
+export { MODEL_CREDENTIALS_SCHEMA, ModelCredentialsStorage, OAUTH_LOGIN_SESSIONS_SCHEMA, isOAuthCredentialExpired };
+
+//# sourceMappingURL=base.js.map

package/dist/storage/domains/credentials/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","names":["#migrateCredential","#db","#decryptCredential"],"sources":["../../../../src/storage/domains/credentials/base.ts"],"sourcesContent":["/**\n * Model provider credentials domain — tenant-scoped OAuth tokens and API keys\n * for model providers (Anthropic/Claude Max, OpenAI Codex, GitHub Copilot,\n * xAI, plus plain API-key providers).\n *\n * Tenancy is two-level:\n * - **user-scoped** rows (`userId` present): personal plan OAuth tokens and\n *   personal API keys. OAuth plan tokens are ONLY ever user-scoped — they are\n *   personal subscriptions, and sharing one would bill/rate-limit a single\n *   account for the whole org.\n * - **org-scoped** rows (`userId` absent): shared API keys set by an org\n *   admin, inherited by every member.\n *\n * Model calls resolve exactly one scope: user or org. Deployed tenant stores\n * never fall back to the other scope or implicitly borrow server environment keys.\n *\n * Secrets posture: credentials are stored server-side only and never returned\n * to the client (same as `linear_connections.access_token`). Encryption at\n * rest is out of scope, matching the existing posture.\n *\n * The domain also owns `oauth_login_sessions`: pending web login flows\n * (paste-code PKCE verifiers, device-code poll state) persisted server-side so\n * any replica can complete or poll a flow started on another.\n */\n\nimport type { AuthCredential, OAuthCredential } from '@mastra/code-sdk/auth/types';\nimport { FactoryStorageDomain, UniqueViolationError } from '@mastra/core/storage';\nimport type { CollectionSchema, CollectionWhere, FactoryStorageOps } from '@mastra/core/storage';\nimport { createPlaintextFactorySecretEncryption } from '../../../secret-encryption.js';\nimport type { FactorySecretEncryption } from '../../../secret-encryption.js';\n\n/** Owning tenant of a credential row. `userId` absent = org-scoped row. */\nexport interface CredentialTenant {\n  orgId: string;\n  userId?: string;\n}\n\n/** Which tenancy level a credential row lives at. */\nexport type CredentialScope = 'user' | 'org';\n\n/** One stored credential, annotated with its scope. */\nexport interface CredentialRecord {\n  provider: string;\n  scope: CredentialScope;\n  credential: AuthCredential;\n  updatedAt: Date;\n}\n\n/** Result of per-caller resolution: the winning credential and its scope. */\nexport interface ResolvedCredential {\n  provider: string;\n  scope: CredentialScope;\n  credential: AuthCredential;\n}\n\n/** Kind of pending web login flow a session row tracks. */\nexport type LoginSessionKind = 'paste-code' | 'device-code';\n\n/** Scope the completed credential should be written to. */\nexport type LoginCredentialScope = 'user' | 'org';\n\n/** One pending OAuth login flow, persisted so any replica can continue it. */\nexport interface LoginSessionRow {\n  sessionId: string;\n  orgId: string;\n  userId: string;\n  provider: string;\n  kind: LoginSessionKind;\n  /** Where the completed credential lands; absent rows default to `'user'`. */\n  credentialScope?: LoginCredentialScope;\n  /** Serialized flow state (PKCE verifier, device-code pending state, ...). */\n  pending: Record<string, unknown>;\n  expiresAt: Date;\n  /** Earliest time the next upstream poll is allowed (device-code flows). */\n  nextPollAt: Date | null;\n  createdAt: Date;\n}\n\nexport interface CreateLoginSessionInput {\n  sessionId: string;\n  orgId: string;\n  userId: string;\n  provider: string;\n  kind: LoginSessionKind;\n  credentialScope?: LoginCredentialScope;\n  pending: Record<string, unknown>;\n  expiresAt: Date;\n  nextPollAt?: Date | null;\n}\n\n/** Mirror of `AuthStorage.getApiKey()`'s expiry check. */\nexport function isOAuthCredentialExpired(credential: OAuthCredential, now = Date.now()): boolean {\n  return now >= credential.expires;\n}\n\nexport const MODEL_CREDENTIALS_SCHEMA: CollectionSchema = {\n  name: 'model_provider_credentials',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    user_id: { type: 'text', nullable: true },\n    provider: { type: 'text' },\n    type: { type: 'text' },\n    data: { type: 'json' },\n    created_at: { type: 'timestamp' },\n    updated_at: { type: 'timestamp' },\n  },\n  uniqueIndexes: [\n    // Scope is encoded in `user_id` nullability (NULL = org-scoped shared\n    // row), enforced by two partial unique indexes.\n    {\n      name: 'model_provider_credentials_user_unique',\n      columns: ['org_id', 'user_id', 'provider'],\n      whereNotNull: 'user_id',\n    },\n    { name: 'model_provider_credentials_org_unique', columns: ['org_id', 'provider'], whereNull: 'user_id' },\n  ],\n};\n\nexport const OAUTH_LOGIN_SESSIONS_SCHEMA: CollectionSchema = {\n  name: 'oauth_login_sessions',\n  columns: {\n    session_id: { type: 'text', primaryKey: true },\n    org_id: { type: 'text' },\n    user_id: { type: 'text' },\n    provider: { type: 'text' },\n    kind: { type: 'text' },\n    credential_scope: { type: 'text', nullable: true },\n    pending: { type: 'json' },\n    expires_at: { type: 'timestamp' },\n    next_poll_at: { type: 'timestamp', nullable: true },\n    created_at: { type: 'timestamp' },\n  },\n};\n\n/** Column shape of one `model_provider_credentials` row as returned by ops. */\ninterface CredentialDbRow extends Record<string, unknown> {\n  id: string;\n  provider: string;\n  user_id: string | null;\n  data: unknown;\n  updated_at: Date;\n}\n\n/** Column shape of one `oauth_login_sessions` row as returned by ops. */\ninterface LoginSessionDbRow extends Record<string, unknown> {\n  session_id: string;\n  org_id: string;\n  user_id: string;\n  provider: string;\n  kind: LoginSessionKind;\n  credential_scope: LoginCredentialScope | null;\n  pending: Record<string, unknown>;\n  expires_at: Date;\n  next_poll_at: Date | null;\n  created_at: Date;\n}\n\nfunction toSessionRow(db: LoginSessionDbRow): LoginSessionRow {\n  return {\n    sessionId: db.session_id,\n    orgId: db.org_id,\n    userId: db.user_id,\n    provider: db.provider,\n    kind: db.kind,\n    ...(db.credential_scope ? { credentialScope: db.credential_scope } : {}),\n    pending: db.pending,\n    expiresAt: db.expires_at,\n    nextPollAt: db.next_poll_at,\n    createdAt: db.created_at,\n  };\n}\n\n/** Filter selecting exactly the tenant's row for a provider (`null` = org row). */\nfunction tenantWhere(tenant: CredentialTenant, provider: string): CollectionWhere {\n  return { org_id: tenant.orgId, provider, user_id: tenant.userId ?? null };\n}\n\n/**\n * Model-credentials storage, written once against the generic\n * `FactoryStorageOps` surface. `refreshOAuth()` and `claimLoginSession()`\n * ride `updateAtomic` so concurrent replicas serialize instead of\n * invalidating each other's rotating tokens / double-claiming a flow.\n */\nexport class ModelCredentialsStorage extends FactoryStorageDomain {\n  constructor(private readonly encryption: FactorySecretEncryption = createPlaintextFactorySecretEncryption()) {\n    super('model-credentials');\n  }\n\n  async init(): Promise<void> {\n    await this.ensureCollections([MODEL_CREDENTIALS_SCHEMA, OAUTH_LOGIN_SESSIONS_SCHEMA]);\n    const rows = await this.ops.findMany<CredentialDbRow>('model_provider_credentials', {});\n    await Promise.all(rows.map(row => this.#migrateCredential(row.id)));\n  }\n\n  async dangerouslyClearAll(): Promise<void> {\n    await this.ops.deleteMany('oauth_login_sessions', {});\n    await this.ops.deleteMany('model_provider_credentials', {});\n  }\n\n  get #db(): FactoryStorageOps {\n    return this.ops;\n  }\n\n  async #decryptCredential(value: unknown): Promise<AuthCredential> {\n    return (await this.encryption.decrypt<AuthCredential>(value)).value;\n  }\n\n  async #migrateCredential(id: string): Promise<void> {\n    await this.#db.updateAtomic<CredentialDbRow>('model_provider_credentials', { id }, async row => {\n      const decrypted = await this.encryption.decrypt<AuthCredential>(row.data);\n      if (!decrypted.needsReencryption) return null;\n      return { data: await this.encryption.encrypt(decrypted.value) };\n    });\n  }\n\n  /** Read the tenant's credential for a provider at exactly that scope. */\n  async getCredential(tenant: CredentialTenant, provider: string): Promise<AuthCredential | undefined> {\n    const row = await this.#db.findOne<CredentialDbRow>('model_provider_credentials', tenantWhere(tenant, provider));\n    return row ? this.#decryptCredential(row.data) : undefined;\n  }\n\n  /** Upsert the tenant's credential (`created_at` is preserved on update). */\n  async setCredential(tenant: CredentialTenant, provider: string, credential: AuthCredential): Promise<void> {\n    const where = tenantWhere(tenant, provider);\n    const encrypted = await this.encryption.encrypt(credential);\n    const update = async () =>\n      this.#db.updateMany('model_provider_credentials', where, {\n        type: credential.type,\n        data: encrypted,\n        updated_at: new Date(),\n      });\n\n    let lastError: unknown;\n    for (let attempt = 0; attempt < 2; attempt++) {\n      if ((await update()) > 0) return;\n\n      const now = new Date();\n      try {\n        await this.#db.insertOne<CredentialDbRow>('model_provider_credentials', {\n          org_id: tenant.orgId,\n          user_id: tenant.userId ?? null,\n          provider,\n          type: credential.type,\n          data: encrypted,\n          created_at: now,\n          updated_at: now,\n        });\n        return;\n      } catch (error) {\n        if (!(error instanceof UniqueViolationError)) throw error;\n        lastError = error;\n      }\n    }\n    throw lastError;\n  }\n\n  /** Delete the tenant's credential at exactly that scope. True when a row was removed. */\n  async removeCredential(tenant: CredentialTenant, provider: string): Promise<boolean> {\n    const deleted = await this.#db.deleteMany('model_provider_credentials', tenantWhere(tenant, provider));\n    return deleted > 0;\n  }\n\n  /** All credentials visible to a user: their own rows plus the org's shared rows. */\n  async listCredentials(orgId: string, userId: string): Promise<CredentialRecord[]> {\n    // SQL `IN` never matches NULL, so the org rows need their own query.\n    const [userRows, orgRows] = await Promise.all([\n      this.#db.findMany<CredentialDbRow>('model_provider_credentials', { org_id: orgId, user_id: userId }),\n      this.#db.findMany<CredentialDbRow>('model_provider_credentials', { org_id: orgId, user_id: null }),\n    ]);\n    return Promise.all([\n      ...userRows.map(async row => ({\n        provider: row.provider,\n        scope: 'user' as const,\n        credential: await this.#decryptCredential(row.data),\n        updatedAt: row.updated_at,\n      })),\n      ...orgRows.map(async row => ({\n        provider: row.provider,\n        scope: 'org' as const,\n        credential: await this.#decryptCredential(row.data),\n        updatedAt: row.updated_at,\n      })),\n    ]);\n  }\n\n  /** Resolve a provider credential from exactly one tenancy scope. */\n  async resolveCredential(\n    orgId: string,\n    userId: string,\n    provider: string,\n    scope: CredentialScope,\n  ): Promise<ResolvedCredential | undefined> {\n    const readUser = async (): Promise<ResolvedCredential | undefined> => {\n      const row = await this.#db.findOne<CredentialDbRow>('model_provider_credentials', {\n        org_id: orgId,\n        provider,\n        user_id: userId,\n      });\n      return row\n        ? { provider: row.provider, scope: 'user', credential: await this.#decryptCredential(row.data) }\n        : undefined;\n    };\n    const readOrg = async (): Promise<ResolvedCredential | undefined> => {\n      const row = await this.#db.findOne<CredentialDbRow>('model_provider_credentials', {\n        org_id: orgId,\n        provider,\n        user_id: null,\n      });\n      return row\n        ? { provider: row.provider, scope: 'org', credential: await this.#decryptCredential(row.data) }\n        : undefined;\n    };\n    return scope === 'org' ? readOrg() : readUser();\n  }\n\n  /**\n   * Refresh the tenant's OAuth credential under the backend's atomic\n   * read-modify-write so concurrent replicas don't invalidate each other's\n   * rotating refresh tokens. The expiry is re-checked under the lock —\n   * another replica may have refreshed already, in which case `refreshFn` is\n   * skipped and the fresh credential is returned. Returns `undefined` when no\n   * OAuth row exists for the tenant.\n   */\n  async refreshOAuth(\n    tenant: CredentialTenant,\n    provider: string,\n    refreshFn: (current: OAuthCredential) => Promise<OAuthCredential>,\n  ): Promise<OAuthCredential | undefined> {\n    let result: OAuthCredential | undefined;\n    await this.#db.updateAtomic<CredentialDbRow>(\n      'model_provider_credentials',\n      tenantWhere(tenant, provider),\n      async row => {\n        const decrypted = await this.encryption.decrypt<AuthCredential>(row.data);\n        if (decrypted.value.type !== 'oauth') return null;\n        const current = decrypted.value;\n        // Re-check under the lock: another replica may have refreshed while we waited.\n        if (!isOAuthCredentialExpired(current)) {\n          result = current;\n          return decrypted.needsReencryption ? { data: await this.encryption.encrypt(current) } : null;\n        }\n        const next = await refreshFn(current);\n        result = next;\n        return { data: await this.encryption.encrypt(next), updated_at: new Date() };\n      },\n    );\n    return result;\n  }\n\n  /** Persist a pending login flow started by a web route. */\n  async createLoginSession(input: CreateLoginSessionInput): Promise<LoginSessionRow> {\n    const inserted = await this.#db.insertOne<LoginSessionDbRow>('oauth_login_sessions', {\n      session_id: input.sessionId,\n      org_id: input.orgId,\n      user_id: input.userId,\n      provider: input.provider,\n      kind: input.kind,\n      credential_scope: input.credentialScope ?? null,\n      pending: input.pending,\n      expires_at: input.expiresAt,\n      next_poll_at: input.nextPollAt ?? null,\n      created_at: new Date(),\n    });\n    return toSessionRow(inserted);\n  }\n\n  /**\n   * Read a pending login flow. An expired session is deleted on read (TTL\n   * cleanup) and reported as absent.\n   */\n  async getLoginSession(sessionId: string): Promise<LoginSessionRow | undefined> {\n    const row = await this.#db.findOne<LoginSessionDbRow>('oauth_login_sessions', { session_id: sessionId });\n    if (!row) return undefined;\n    if (row.expires_at.getTime() <= Date.now()) {\n      await this.#db.deleteMany('oauth_login_sessions', { session_id: sessionId });\n      return undefined;\n    }\n    return toSessionRow(row);\n  }\n\n  /**\n   * Atomically claim a due login session for one upstream completion/poll attempt.\n   * Returns undefined when the session is missing, expired, owned by another\n   * tenant/provider, or already claimed by a concurrent request.\n   */\n  async claimLoginSession(\n    sessionId: string,\n    owner: Pick<LoginSessionRow, 'orgId' | 'userId' | 'provider' | 'kind'>,\n  ): Promise<LoginSessionRow | undefined> {\n    const now = Date.now();\n    let claimed = false;\n    const updated = await this.#db.updateAtomic<LoginSessionDbRow>(\n      'oauth_login_sessions',\n      { session_id: sessionId },\n      row => {\n        if (\n          row.org_id !== owner.orgId ||\n          row.user_id !== owner.userId ||\n          row.provider !== owner.provider ||\n          row.kind !== owner.kind\n        ) {\n          return null;\n        }\n        if (row.expires_at.getTime() <= now) return null;\n        if (row.next_poll_at && row.next_poll_at.getTime() > now) return null;\n        claimed = true;\n        // Park the next poll at expiry so a concurrent claim loses until the\n        // claimer either touches the session forward or deletes it.\n        return { next_poll_at: row.expires_at };\n      },\n    );\n    return claimed && updated ? toSessionRow(updated) : undefined;\n  }\n\n  /** Update a pending flow's serialized state and/or next allowed poll time. */\n  async touchLoginSession(\n    sessionId: string,\n    updates: { pending?: Record<string, unknown>; nextPollAt?: Date | null },\n  ): Promise<void> {\n    const set: Record<string, unknown> = {};\n    if (updates.pending !== undefined) set.pending = updates.pending;\n    if (updates.nextPollAt !== undefined) set.next_poll_at = updates.nextPollAt;\n    if (Object.keys(set).length === 0) return;\n    await this.#db.updateMany('oauth_login_sessions', { session_id: sessionId }, set);\n  }\n\n  /** Remove a pending login flow (completed, cancelled, or failed). */\n  async deleteLoginSession(sessionId: string): Promise<void> {\n    await this.#db.deleteMany('oauth_login_sessions', { session_id: sessionId });\n  }\n}\n"],"mappings":";;;;AA2FA,SAAgB,yBAAyB,YAA6B,MAAM,KAAK,IAAI,GAAY;CAC/F,OAAO,OAAO,WAAW;AAC3B;AAEA,MAAa,2BAA6C;CACxD,MAAM;CACN,SAAS;EACP,IAAI,EAAE,MAAM,UAAU;EACtB,QAAQ,EAAE,MAAM,OAAO;EACvB,SAAS;GAAE,MAAM;GAAQ,UAAU;EAAK;EACxC,UAAU,EAAE,MAAM,OAAO;EACzB,MAAM,EAAE,MAAM,OAAO;EACrB,MAAM,EAAE,MAAM,OAAO;EACrB,YAAY,EAAE,MAAM,YAAY;EAChC,YAAY,EAAE,MAAM,YAAY;CAClC;CACA,eAAe,CAGb;EACE,MAAM;EACN,SAAS;GAAC;GAAU;GAAW;EAAU;EACzC,cAAc;CAChB,GACA;EAAE,MAAM;EAAyC,SAAS,CAAC,UAAU,UAAU;EAAG,WAAW;CAAU,CACzG;AACF;AAEA,MAAa,8BAAgD;CAC3D,MAAM;CACN,SAAS;EACP,YAAY;GAAE,MAAM;GAAQ,YAAY;EAAK;EAC7C,QAAQ,EAAE,MAAM,OAAO;EACvB,SAAS,EAAE,MAAM,OAAO;EACxB,UAAU,EAAE,MAAM,OAAO;EACzB,MAAM,EAAE,MAAM,OAAO;EACrB,kBAAkB;GAAE,MAAM;GAAQ,UAAU;EAAK;EACjD,SAAS,EAAE,MAAM,OAAO;EACxB,YAAY,EAAE,MAAM,YAAY;EAChC,cAAc;GAAE,MAAM;GAAa,UAAU;EAAK;EAClD,YAAY,EAAE,MAAM,YAAY;CAClC;AACF;AAyBA,SAAS,aAAa,IAAwC;CAC5D,OAAO;EACL,WAAW,GAAG;EACd,OAAO,GAAG;EACV,QAAQ,GAAG;EACX,UAAU,GAAG;EACb,MAAM,GAAG;EACT,GAAI,GAAG,mBAAmB,EAAE,iBAAiB,GAAG,iBAAiB,IAAI,CAAC;EACtE,SAAS,GAAG;EACZ,WAAW,GAAG;EACd,YAAY,GAAG;EACf,WAAW,GAAG;CAChB;AACF;;AAGA,SAAS,YAAY,QAA0B,UAAmC;CAChF,OAAO;EAAE,QAAQ,OAAO;EAAO;EAAU,SAAS,OAAO,UAAU;CAAK;AAC1E;;;;;;;AAQA,IAAa,0BAAb,cAA6C,qBAAqB;CACnC;CAA7B,YAAY,aAAuD,uCAAuC,GAAG;EAC3G,MAAM,mBAAmB;EADE,KAAA,aAAA;CAE7B;CAEA,MAAM,OAAsB;EAC1B,MAAM,KAAK,kBAAkB,CAAC,0BAA0B,2BAA2B,CAAC;EACpF,MAAM,OAAO,MAAM,KAAK,IAAI,SAA0B,8BAA8B,CAAC,CAAC;EACtF,MAAM,QAAQ,IAAI,KAAK,KAAI,QAAO,KAAKA,mBAAmB,IAAI,EAAE,CAAC,CAAC;CACpE;CAEA,MAAM,sBAAqC;EACzC,MAAM,KAAK,IAAI,WAAW,wBAAwB,CAAC,CAAC;EACpD,MAAM,KAAK,IAAI,WAAW,8BAA8B,CAAC,CAAC;CAC5D;CAEA,IAAIC,MAAyB;EAC3B,OAAO,KAAK;CACd;CAEA,MAAMC,mBAAmB,OAAyC;EAChE,QAAQ,MAAM,KAAK,WAAW,QAAwB,KAAK,EAAA,CAAG;CAChE;CAEA,MAAMF,mBAAmB,IAA2B;EAClD,MAAM,KAAKC,IAAI,aAA8B,8BAA8B,EAAE,GAAG,GAAG,OAAM,QAAO;GAC9F,MAAM,YAAY,MAAM,KAAK,WAAW,QAAwB,IAAI,IAAI;GACxE,IAAI,CAAC,UAAU,mBAAmB,OAAO;GACzC,OAAO,EAAE,MAAM,MAAM,KAAK,WAAW,QAAQ,UAAU,KAAK,EAAE;EAChE,CAAC;CACH;;CAGA,MAAM,cAAc,QAA0B,UAAuD;EACnG,MAAM,MAAM,MAAM,KAAKA,IAAI,QAAyB,8BAA8B,YAAY,QAAQ,QAAQ,CAAC;EAC/G,OAAO,MAAM,KAAKC,mBAAmB,IAAI,IAAI,IAAI,KAAA;CACnD;;CAGA,MAAM,cAAc,QAA0B,UAAkB,YAA2C;EACzG,MAAM,QAAQ,YAAY,QAAQ,QAAQ;EAC1C,MAAM,YAAY,MAAM,KAAK,WAAW,QAAQ,UAAU;EAC1D,MAAM,SAAS,YACb,KAAKD,IAAI,WAAW,8BAA8B,OAAO;GACvD,MAAM,WAAW;GACjB,MAAM;GACN,4BAAY,IAAI,KAAK;EACvB,CAAC;EAEH,IAAI;EACJ,KAAK,IAAI,UAAU,GAAG,UAAU,GAAG,WAAW;GAC5C,IAAK,MAAM,OAAO,IAAK,GAAG;GAE1B,MAAM,sBAAM,IAAI,KAAK;GACrB,IAAI;IACF,MAAM,KAAKA,IAAI,UAA2B,8BAA8B;KACtE,QAAQ,OAAO;KACf,SAAS,OAAO,UAAU;KAC1B;KACA,MAAM,WAAW;KACjB,MAAM;KACN,YAAY;KACZ,YAAY;IACd,CAAC;IACD;GACF,SAAS,OAAO;IACd,IAAI,EAAE,iBAAiB,uBAAuB,MAAM;IACpD,YAAY;GACd;EACF;EACA,MAAM;CACR;;CAGA,MAAM,iBAAiB,QAA0B,UAAoC;EAEnF,OAAO,MADe,KAAKA,IAAI,WAAW,8BAA8B,YAAY,QAAQ,QAAQ,CAAC,IACpF;CACnB;;CAGA,MAAM,gBAAgB,OAAe,QAA6C;EAEhF,MAAM,CAAC,UAAU,WAAW,MAAM,QAAQ,IAAI,CAC5C,KAAKA,IAAI,SAA0B,8BAA8B;GAAE,QAAQ;GAAO,SAAS;EAAO,CAAC,GACnG,KAAKA,IAAI,SAA0B,8BAA8B;GAAE,QAAQ;GAAO,SAAS;EAAK,CAAC,CACnG,CAAC;EACD,OAAO,QAAQ,IAAI,CACjB,GAAG,SAAS,IAAI,OAAM,SAAQ;GAC5B,UAAU,IAAI;GACd,OAAO;GACP,YAAY,MAAM,KAAKC,mBAAmB,IAAI,IAAI;GAClD,WAAW,IAAI;EACjB,EAAE,GACF,GAAG,QAAQ,IAAI,OAAM,SAAQ;GAC3B,UAAU,IAAI;GACd,OAAO;GACP,YAAY,MAAM,KAAKA,mBAAmB,IAAI,IAAI;GAClD,WAAW,IAAI;EACjB,EAAE,CACJ,CAAC;CACH;;CAGA,MAAM,kBACJ,OACA,QACA,UACA,OACyC;EACzC,MAAM,WAAW,YAAqD;GACpE,MAAM,MAAM,MAAM,KAAKD,IAAI,QAAyB,8BAA8B;IAChF,QAAQ;IACR;IACA,SAAS;GACX,CAAC;GACD,OAAO,MACH;IAAE,UAAU,IAAI;IAAU,OAAO;IAAQ,YAAY,MAAM,KAAKC,mBAAmB,IAAI,IAAI;GAAE,IAC7F,KAAA;EACN;EACA,MAAM,UAAU,YAAqD;GACnE,MAAM,MAAM,MAAM,KAAKD,IAAI,QAAyB,8BAA8B;IAChF,QAAQ;IACR;IACA,SAAS;GACX,CAAC;GACD,OAAO,MACH;IAAE,UAAU,IAAI;IAAU,OAAO;IAAO,YAAY,MAAM,KAAKC,mBAAmB,IAAI,IAAI;GAAE,IAC5F,KAAA;EACN;EACA,OAAO,UAAU,QAAQ,QAAQ,IAAI,SAAS;CAChD;;;;;;;;;CAUA,MAAM,aACJ,QACA,UACA,WACsC;EACtC,IAAI;EACJ,MAAM,KAAKD,IAAI,aACb,8BACA,YAAY,QAAQ,QAAQ,GAC5B,OAAM,QAAO;GACX,MAAM,YAAY,MAAM,KAAK,WAAW,QAAwB,IAAI,IAAI;GACxE,IAAI,UAAU,MAAM,SAAS,SAAS,OAAO;GAC7C,MAAM,UAAU,UAAU;GAE1B,IAAI,CAAC,yBAAyB,OAAO,GAAG;IACtC,SAAS;IACT,OAAO,UAAU,oBAAoB,EAAE,MAAM,MAAM,KAAK,WAAW,QAAQ,OAAO,EAAE,IAAI;GAC1F;GACA,MAAM,OAAO,MAAM,UAAU,OAAO;GACpC,SAAS;GACT,OAAO;IAAE,MAAM,MAAM,KAAK,WAAW,QAAQ,IAAI;IAAG,4BAAY,IAAI,KAAK;GAAE;EAC7E,CACF;EACA,OAAO;CACT;;CAGA,MAAM,mBAAmB,OAA0D;EAajF,OAAO,aAAa,MAZG,KAAKA,IAAI,UAA6B,wBAAwB;GACnF,YAAY,MAAM;GAClB,QAAQ,MAAM;GACd,SAAS,MAAM;GACf,UAAU,MAAM;GAChB,MAAM,MAAM;GACZ,kBAAkB,MAAM,mBAAmB;GAC3C,SAAS,MAAM;GACf,YAAY,MAAM;GAClB,cAAc,MAAM,cAAc;GAClC,4BAAY,IAAI,KAAK;EACvB,CAAC,CAC2B;CAC9B;;;;;CAMA,MAAM,gBAAgB,WAAyD;EAC7E,MAAM,MAAM,MAAM,KAAKA,IAAI,QAA2B,wBAAwB,EAAE,YAAY,UAAU,CAAC;EACvG,IAAI,CAAC,KAAK,OAAO,KAAA;EACjB,IAAI,IAAI,WAAW,QAAQ,KAAK,KAAK,IAAI,GAAG;GAC1C,MAAM,KAAKA,IAAI,WAAW,wBAAwB,EAAE,YAAY,UAAU,CAAC;GAC3E;EACF;EACA,OAAO,aAAa,GAAG;CACzB;;;;;;CAOA,MAAM,kBACJ,WACA,OACsC;EACtC,MAAM,MAAM,KAAK,IAAI;EACrB,IAAI,UAAU;EACd,MAAM,UAAU,MAAM,KAAKA,IAAI,aAC7B,wBACA,EAAE,YAAY,UAAU,IACxB,QAAO;GACL,IACE,IAAI,WAAW,MAAM,SACrB,IAAI,YAAY,MAAM,UACtB,IAAI,aAAa,MAAM,YACvB,IAAI,SAAS,MAAM,MAEnB,OAAO;GAET,IAAI,IAAI,WAAW,QAAQ,KAAK,KAAK,OAAO;GAC5C,IAAI,IAAI,gBAAgB,IAAI,aAAa,QAAQ,IAAI,KAAK,OAAO;GACjE,UAAU;GAGV,OAAO,EAAE,cAAc,IAAI,WAAW;EACxC,CACF;EACA,OAAO,WAAW,UAAU,aAAa,OAAO,IAAI,KAAA;CACtD;;CAGA,MAAM,kBACJ,WACA,SACe;EACf,MAAM,MAA+B,CAAC;EACtC,IAAI,QAAQ,YAAY,KAAA,GAAW,IAAI,UAAU,QAAQ;EACzD,IAAI,QAAQ,eAAe,KAAA,GAAW,IAAI,eAAe,QAAQ;EACjE,IAAI,OAAO,KAAK,GAAG,CAAC,CAAC,WAAW,GAAG;EACnC,MAAM,KAAKA,IAAI,WAAW,wBAAwB,EAAE,YAAY,UAAU,GAAG,GAAG;CAClF;;CAGA,MAAM,mBAAmB,WAAkC;EACzD,MAAM,KAAKA,IAAI,WAAW,wBAAwB,EAAE,YAAY,UAAU,CAAC;CAC7E;AACF"}

package/dist/storage/domains/custom-providers/base.d.ts

@@ -0,0 +1,59 @@
+import { FactoryStorageDomain } from '@mastra/core/storage';
+import type { CollectionSchema } from '@mastra/core/storage';
+import type { FactorySecretEncryption } from '../../../secret-encryption.js';
+/**
+ * A user-defined OpenAI-compatible provider. The DB-backed counterpart of the
+ * local `settings.json` `customProviders` entries: org-scoped rows keyed by the
+ * derived provider id (name slug), API key stored alongside so a deployed
+ * server never reads the shared settings file.
+ */
+export interface CustomProviderRecord {
+    id: string;
+    orgId: string;
+    createdBy: string;
+    /** Stable slug derived from the name (`getCustomProviderId`). */
+    providerId: string;
+    name: string;
+    url: string;
+    apiKey: string | null;
+    models: string[];
+    createdAt: Date;
+    updatedAt: Date;
+}
+export interface UpsertCustomProviderInput {
+    providerId: string;
+    name: string;
+    url: string;
+    apiKey?: string;
+    models: string[];
+}
+export declare const CUSTOM_PROVIDERS_SCHEMA: CollectionSchema;
+export declare class CustomProvidersStorage extends FactoryStorageDomain {
+    #private;
+    private readonly encryption;
+    constructor(encryption?: FactorySecretEncryption);
+    init(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
+    /**
+     * Create or wholesale-replace a provider by `(orgId, providerId)` — mirrors
+     * the settings.json upsert semantics (no key retention: an absent `apiKey`
+     * clears the stored key). `previousProviderId` renames the provider: the
+     * common path is a single atomic in-place update of the old row; renaming
+     * onto an id that already exists overwrites the target first and removes the
+     * old row last, so no interleaving or failure can lose the provider.
+     */
+    upsert({ orgId, userId, input, previousProviderId, }: {
+        orgId: string;
+        userId: string;
+        input: UpsertCustomProviderInput;
+        previousProviderId?: string;
+    }): Promise<CustomProviderRecord>;
+    list({ orgId }: {
+        orgId: string;
+    }): Promise<CustomProviderRecord[]>;
+    delete({ orgId, providerId }: {
+        orgId: string;
+        providerId: string;
+    }): Promise<boolean>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/storage/domains/custom-providers/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/custom-providers/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAwB,MAAM,sBAAsB,CAAC;AAClF,OAAO,KAAK,EAAE,gBAAgB,EAAqB,MAAM,sBAAsB,CAAC;AAEhF,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AAE7E;;;;;GAKG;AACH,MAAM,WAAW,oBAAoB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,iEAAiE;IACjE,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,yBAAyB;IACxC,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,eAAO,MAAM,uBAAuB,EAAE,gBAerC,CAAC;AAeF,qBAAa,sBAAuB,SAAQ,oBAAoB;;IAClD,OAAO,CAAC,QAAQ,CAAC,UAAU;gBAAV,UAAU,GAAE,uBAAkE;IAIrG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAMrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAgD1C;;;;;;;OAOG;IACG,MAAM,CAAC,EACX,KAAK,EACL,MAAM,EACN,KAAK,EACL,kBAAkB,GACnB,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,yBAAyB,CAAC;QACjC,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAqG3B,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,oBAAoB,EAAE,CAAC;IASnE,MAAM,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,OAAO,CAAC;CAG7F"}

package/dist/storage/domains/custom-providers/base.js

@@ -0,0 +1,176 @@
+import { createPlaintextFactorySecretEncryption } from "../../../secret-encryption.js";
+import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
+//#region src/storage/domains/custom-providers/base.ts
+const CUSTOM_PROVIDERS_SCHEMA = {
+	name: "custom_providers",
+	columns: {
+		id: { type: "uuid-pk" },
+		org_id: { type: "text" },
+		created_by: { type: "text" },
+		provider_id: { type: "text" },
+		name: { type: "text" },
+		url: { type: "text" },
+		api_key: {
+			type: "text",
+			nullable: true
+		},
+		models: { type: "json" },
+		created_at: { type: "timestamp" },
+		updated_at: { type: "timestamp" }
+	},
+	uniqueIndexes: [{
+		name: "custom_providers_org_provider_key",
+		columns: ["org_id", "provider_id"]
+	}]
+};
+var CustomProvidersStorage = class extends FactoryStorageDomain {
+	encryption;
+	constructor(encryption = createPlaintextFactorySecretEncryption()) {
+		super("custom-providers");
+		this.encryption = encryption;
+	}
+	async init() {
+		await this.ensureCollections([CUSTOM_PROVIDERS_SCHEMA]);
+		const rows = await this.ops.findMany("custom_providers", {});
+		await Promise.all(rows.filter((row) => row.api_key !== null).map((row) => this.#migrateApiKey(row.id)));
+	}
+	async dangerouslyClearAll() {
+		await this.ops.deleteMany("custom_providers", {});
+	}
+	get #db() {
+		return this.ops;
+	}
+	async #toRecord(row) {
+		const apiKey = row.api_key === null ? null : (await this.#decryptApiKey(row.api_key)).value;
+		return {
+			id: row.id,
+			orgId: row.org_id,
+			createdBy: row.created_by,
+			providerId: row.provider_id,
+			name: row.name,
+			url: row.url,
+			apiKey,
+			models: Array.isArray(row.models) ? row.models : [],
+			createdAt: row.created_at,
+			updatedAt: row.updated_at
+		};
+	}
+	/**
+	* API keys are always strings. A legacy raw key that happens to be valid JSON
+	* (e.g. `12345`) decodes to a non-string, so keep the stored text instead.
+	*/
+	async #decryptApiKey(stored) {
+		const decrypted = await this.encryption.decrypt(stored);
+		return {
+			value: typeof decrypted.value === "string" ? decrypted.value : stored,
+			needsReencryption: decrypted.needsReencryption
+		};
+	}
+	async #encryptApiKey(apiKey) {
+		return apiKey === void 0 ? null : this.encryption.encrypt(apiKey);
+	}
+	async #migrateApiKey(id) {
+		await this.#db.updateAtomic("custom_providers", { id }, async (row) => {
+			if (row.api_key === null) return null;
+			const decrypted = await this.#decryptApiKey(row.api_key);
+			return decrypted.needsReencryption ? { api_key: await this.encryption.encrypt(decrypted.value) } : null;
+		});
+	}
+	/**
+	* Create or wholesale-replace a provider by `(orgId, providerId)` — mirrors
+	* the settings.json upsert semantics (no key retention: an absent `apiKey`
+	* clears the stored key). `previousProviderId` renames the provider: the
+	* common path is a single atomic in-place update of the old row; renaming
+	* onto an id that already exists overwrites the target first and removes the
+	* old row last, so no interleaving or failure can lose the provider.
+	*/
+	async upsert({ orgId, userId, input, previousProviderId }) {
+		const now = /* @__PURE__ */ new Date();
+		const encryptedApiKey = await this.#encryptApiKey(input.apiKey);
+		const renameFrom = previousProviderId && previousProviderId !== input.providerId ? previousProviderId : void 0;
+		if (renameFrom) {
+			if (!await this.#db.findOne("custom_providers", {
+				org_id: orgId,
+				provider_id: input.providerId
+			})) {
+				const renamed = await this.#db.updateAtomic("custom_providers", {
+					org_id: orgId,
+					provider_id: renameFrom
+				}, () => ({
+					provider_id: input.providerId,
+					name: input.name,
+					url: input.url,
+					api_key: encryptedApiKey,
+					models: input.models,
+					updated_at: now
+				}));
+				if (renamed) return this.#toRecord(renamed);
+			}
+		}
+		const record = await this.#write({
+			orgId,
+			userId,
+			input,
+			encryptedApiKey,
+			now
+		});
+		if (renameFrom) await this.#db.deleteMany("custom_providers", {
+			org_id: orgId,
+			provider_id: renameFrom
+		});
+		return record;
+	}
+	/**
+	* Update-first, then insert-and-catch-unique-violation (see `queue_health`):
+	* concurrent creates of the same provider both succeed (last write wins on
+	* the single row) instead of one failing on the unique index, and a row
+	* deleted mid-flight is recreated rather than reported as a stale success.
+	*/
+	async #write({ orgId, userId, input, encryptedApiKey, now }) {
+		const updateExisting = () => this.#db.updateAtomic("custom_providers", {
+			org_id: orgId,
+			provider_id: input.providerId
+		}, () => ({
+			name: input.name,
+			url: input.url,
+			api_key: encryptedApiKey,
+			models: input.models,
+			updated_at: now
+		}));
+		const updated = await updateExisting();
+		if (updated) return this.#toRecord(updated);
+		try {
+			const row = await this.#db.insertOne("custom_providers", {
+				org_id: orgId,
+				created_by: userId,
+				provider_id: input.providerId,
+				name: input.name,
+				url: input.url,
+				api_key: encryptedApiKey,
+				models: input.models,
+				created_at: now,
+				updated_at: now
+			});
+			return this.#toRecord(row);
+		} catch (error) {
+			if (!(error instanceof UniqueViolationError)) throw error;
+			const row = await updateExisting();
+			if (!row) throw error;
+			return this.#toRecord(row);
+		}
+	}
+	async list({ orgId }) {
+		const rows = await this.#db.findMany("custom_providers", { org_id: orgId }, { orderBy: [["name", "asc"]] });
+		return Promise.all(rows.map((row) => this.#toRecord(row)));
+	}
+	async delete({ orgId, providerId }) {
+		return await this.#db.deleteMany("custom_providers", {
+			org_id: orgId,
+			provider_id: providerId
+		}) > 0;
+	}
+};
+//#endregion
+export { CUSTOM_PROVIDERS_SCHEMA, CustomProvidersStorage };
+
+//# sourceMappingURL=base.js.map

package/dist/storage/domains/custom-providers/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","names":["#migrateApiKey","#db","#toRecord","#decryptApiKey","#encryptApiKey","#write"],"sources":["../../../../src/storage/domains/custom-providers/base.ts"],"sourcesContent":["import { FactoryStorageDomain, UniqueViolationError } from '@mastra/core/storage';\nimport type { CollectionSchema, FactoryStorageOps } from '@mastra/core/storage';\nimport { createPlaintextFactorySecretEncryption } from '../../../secret-encryption.js';\nimport type { FactorySecretEncryption } from '../../../secret-encryption.js';\n\n/**\n * A user-defined OpenAI-compatible provider. The DB-backed counterpart of the\n * local `settings.json` `customProviders` entries: org-scoped rows keyed by the\n * derived provider id (name slug), API key stored alongside so a deployed\n * server never reads the shared settings file.\n */\nexport interface CustomProviderRecord {\n  id: string;\n  orgId: string;\n  createdBy: string;\n  /** Stable slug derived from the name (`getCustomProviderId`). */\n  providerId: string;\n  name: string;\n  url: string;\n  apiKey: string | null;\n  models: string[];\n  createdAt: Date;\n  updatedAt: Date;\n}\n\nexport interface UpsertCustomProviderInput {\n  providerId: string;\n  name: string;\n  url: string;\n  apiKey?: string;\n  models: string[];\n}\n\nexport const CUSTOM_PROVIDERS_SCHEMA: CollectionSchema = {\n  name: 'custom_providers',\n  columns: {\n    id: { type: 'uuid-pk' },\n    org_id: { type: 'text' },\n    created_by: { type: 'text' },\n    provider_id: { type: 'text' },\n    name: { type: 'text' },\n    url: { type: 'text' },\n    api_key: { type: 'text', nullable: true },\n    models: { type: 'json' },\n    created_at: { type: 'timestamp' },\n    updated_at: { type: 'timestamp' },\n  },\n  uniqueIndexes: [{ name: 'custom_providers_org_provider_key', columns: ['org_id', 'provider_id'] }],\n};\n\ninterface CustomProviderDbRow extends Record<string, unknown> {\n  id: string;\n  org_id: string;\n  created_by: string;\n  provider_id: string;\n  name: string;\n  url: string;\n  api_key: string | null;\n  models: string[];\n  created_at: Date;\n  updated_at: Date;\n}\n\nexport class CustomProvidersStorage extends FactoryStorageDomain {\n  constructor(private readonly encryption: FactorySecretEncryption = createPlaintextFactorySecretEncryption()) {\n    super('custom-providers');\n  }\n\n  async init(): Promise<void> {\n    await this.ensureCollections([CUSTOM_PROVIDERS_SCHEMA]);\n    const rows = await this.ops.findMany<CustomProviderDbRow>('custom_providers', {});\n    await Promise.all(rows.filter(row => row.api_key !== null).map(row => this.#migrateApiKey(row.id)));\n  }\n\n  async dangerouslyClearAll(): Promise<void> {\n    await this.ops.deleteMany('custom_providers', {});\n  }\n\n  get #db(): FactoryStorageOps {\n    return this.ops;\n  }\n\n  async #toRecord(row: CustomProviderDbRow): Promise<CustomProviderRecord> {\n    const apiKey = row.api_key === null ? null : (await this.#decryptApiKey(row.api_key)).value;\n    return {\n      id: row.id,\n      orgId: row.org_id,\n      createdBy: row.created_by,\n      providerId: row.provider_id,\n      name: row.name,\n      url: row.url,\n      apiKey,\n      models: Array.isArray(row.models) ? row.models : [],\n      createdAt: row.created_at,\n      updatedAt: row.updated_at,\n    };\n  }\n\n  /**\n   * API keys are always strings. A legacy raw key that happens to be valid JSON\n   * (e.g. `12345`) decodes to a non-string, so keep the stored text instead.\n   */\n  async #decryptApiKey(stored: string): Promise<{ value: string; needsReencryption: boolean }> {\n    const decrypted = await this.encryption.decrypt<unknown>(stored);\n    return {\n      value: typeof decrypted.value === 'string' ? decrypted.value : stored,\n      needsReencryption: decrypted.needsReencryption,\n    };\n  }\n\n  async #encryptApiKey(apiKey: string | undefined): Promise<string | null> {\n    return apiKey === undefined ? null : this.encryption.encrypt(apiKey);\n  }\n\n  async #migrateApiKey(id: string): Promise<void> {\n    await this.#db.updateAtomic<CustomProviderDbRow>('custom_providers', { id }, async row => {\n      if (row.api_key === null) return null;\n      const decrypted = await this.#decryptApiKey(row.api_key);\n      return decrypted.needsReencryption ? { api_key: await this.encryption.encrypt(decrypted.value) } : null;\n    });\n  }\n\n  /**\n   * Create or wholesale-replace a provider by `(orgId, providerId)` — mirrors\n   * the settings.json upsert semantics (no key retention: an absent `apiKey`\n   * clears the stored key). `previousProviderId` renames the provider: the\n   * common path is a single atomic in-place update of the old row; renaming\n   * onto an id that already exists overwrites the target first and removes the\n   * old row last, so no interleaving or failure can lose the provider.\n   */\n  async upsert({\n    orgId,\n    userId,\n    input,\n    previousProviderId,\n  }: {\n    orgId: string;\n    userId: string;\n    input: UpsertCustomProviderInput;\n    previousProviderId?: string;\n  }): Promise<CustomProviderRecord> {\n    const now = new Date();\n    const encryptedApiKey = await this.#encryptApiKey(input.apiKey);\n    const renameFrom = previousProviderId && previousProviderId !== input.providerId ? previousProviderId : undefined;\n\n    if (renameFrom) {\n      const target = await this.#db.findOne<CustomProviderDbRow>('custom_providers', {\n        org_id: orgId,\n        provider_id: input.providerId,\n      });\n      if (!target) {\n        // In-place rename: one atomic write with no delete/insert gap, so a\n        // failure can never drop the provider (and `created_at`/`created_by`\n        // are preserved). If a concurrent create grabs the new id between the\n        // lookup and this update, the unique index rejects the update and the\n        // old row stays intact — the request fails cleanly and a retry\n        // converges on the rename-onto-existing path below.\n        const renamed = await this.#db.updateAtomic<CustomProviderDbRow>(\n          'custom_providers',\n          { org_id: orgId, provider_id: renameFrom },\n          () => ({\n            provider_id: input.providerId,\n            name: input.name,\n            url: input.url,\n            api_key: encryptedApiKey,\n            models: input.models,\n            updated_at: now,\n          }),\n        );\n        if (renamed) return this.#toRecord(renamed);\n        // Old row already gone — fall through to a plain create of the new id.\n      }\n      // Rename onto an existing id: overwrite the target below, then delete\n      // the old row last — a failure in the gap leaves a redundant duplicate\n      // (recoverable by re-submitting or deleting) instead of losing data.\n    }\n\n    const record = await this.#write({ orgId, userId, input, encryptedApiKey, now });\n    if (renameFrom) {\n      await this.#db.deleteMany('custom_providers', { org_id: orgId, provider_id: renameFrom });\n    }\n    return record;\n  }\n\n  /**\n   * Update-first, then insert-and-catch-unique-violation (see `queue_health`):\n   * concurrent creates of the same provider both succeed (last write wins on\n   * the single row) instead of one failing on the unique index, and a row\n   * deleted mid-flight is recreated rather than reported as a stale success.\n   */\n  async #write({\n    orgId,\n    userId,\n    input,\n    encryptedApiKey,\n    now,\n  }: {\n    orgId: string;\n    userId: string;\n    input: UpsertCustomProviderInput;\n    encryptedApiKey: string | null;\n    now: Date;\n  }): Promise<CustomProviderRecord> {\n    const updateExisting = () =>\n      this.#db.updateAtomic<CustomProviderDbRow>(\n        'custom_providers',\n        { org_id: orgId, provider_id: input.providerId },\n        () => ({\n          name: input.name,\n          url: input.url,\n          api_key: encryptedApiKey,\n          models: input.models,\n          updated_at: now,\n        }),\n      );\n\n    const updated = await updateExisting();\n    if (updated) return this.#toRecord(updated);\n\n    try {\n      const row = await this.#db.insertOne<CustomProviderDbRow>('custom_providers', {\n        org_id: orgId,\n        created_by: userId,\n        provider_id: input.providerId,\n        name: input.name,\n        url: input.url,\n        api_key: encryptedApiKey,\n        models: input.models,\n        created_at: now,\n        updated_at: now,\n      });\n      return this.#toRecord(row);\n    } catch (error) {\n      if (!(error instanceof UniqueViolationError)) throw error;\n      // Lost the insert race — apply this write to the winning row.\n      const row = await updateExisting();\n      if (!row) throw error;\n      return this.#toRecord(row);\n    }\n  }\n\n  async list({ orgId }: { orgId: string }): Promise<CustomProviderRecord[]> {\n    const rows = await this.#db.findMany<CustomProviderDbRow>(\n      'custom_providers',\n      { org_id: orgId },\n      { orderBy: [['name', 'asc']] },\n    );\n    return Promise.all(rows.map(row => this.#toRecord(row)));\n  }\n\n  async delete({ orgId, providerId }: { orgId: string; providerId: string }): Promise<boolean> {\n    return (await this.#db.deleteMany('custom_providers', { org_id: orgId, provider_id: providerId })) > 0;\n  }\n}\n"],"mappings":";;;AAiCA,MAAa,0BAA4C;CACvD,MAAM;CACN,SAAS;EACP,IAAI,EAAE,MAAM,UAAU;EACtB,QAAQ,EAAE,MAAM,OAAO;EACvB,YAAY,EAAE,MAAM,OAAO;EAC3B,aAAa,EAAE,MAAM,OAAO;EAC5B,MAAM,EAAE,MAAM,OAAO;EACrB,KAAK,EAAE,MAAM,OAAO;EACpB,SAAS;GAAE,MAAM;GAAQ,UAAU;EAAK;EACxC,QAAQ,EAAE,MAAM,OAAO;EACvB,YAAY,EAAE,MAAM,YAAY;EAChC,YAAY,EAAE,MAAM,YAAY;CAClC;CACA,eAAe,CAAC;EAAE,MAAM;EAAqC,SAAS,CAAC,UAAU,aAAa;CAAE,CAAC;AACnG;AAeA,IAAa,yBAAb,cAA4C,qBAAqB;CAClC;CAA7B,YAAY,aAAuD,uCAAuC,GAAG;EAC3G,MAAM,kBAAkB;EADG,KAAA,aAAA;CAE7B;CAEA,MAAM,OAAsB;EAC1B,MAAM,KAAK,kBAAkB,CAAC,uBAAuB,CAAC;EACtD,MAAM,OAAO,MAAM,KAAK,IAAI,SAA8B,oBAAoB,CAAC,CAAC;EAChF,MAAM,QAAQ,IAAI,KAAK,QAAO,QAAO,IAAI,YAAY,IAAI,CAAC,CAAC,KAAI,QAAO,KAAKA,eAAe,IAAI,EAAE,CAAC,CAAC;CACpG;CAEA,MAAM,sBAAqC;EACzC,MAAM,KAAK,IAAI,WAAW,oBAAoB,CAAC,CAAC;CAClD;CAEA,IAAIC,MAAyB;EAC3B,OAAO,KAAK;CACd;CAEA,MAAMC,UAAU,KAAyD;EACvE,MAAM,SAAS,IAAI,YAAY,OAAO,QAAQ,MAAM,KAAKC,eAAe,IAAI,OAAO,EAAA,CAAG;EACtF,OAAO;GACL,IAAI,IAAI;GACR,OAAO,IAAI;GACX,WAAW,IAAI;GACf,YAAY,IAAI;GAChB,MAAM,IAAI;GACV,KAAK,IAAI;GACT;GACA,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,IAAI,SAAS,CAAC;GAClD,WAAW,IAAI;GACf,WAAW,IAAI;EACjB;CACF;;;;;CAMA,MAAMA,eAAe,QAAwE;EAC3F,MAAM,YAAY,MAAM,KAAK,WAAW,QAAiB,MAAM;EAC/D,OAAO;GACL,OAAO,OAAO,UAAU,UAAU,WAAW,UAAU,QAAQ;GAC/D,mBAAmB,UAAU;EAC/B;CACF;CAEA,MAAMC,eAAe,QAAoD;EACvE,OAAO,WAAW,KAAA,IAAY,OAAO,KAAK,WAAW,QAAQ,MAAM;CACrE;CAEA,MAAMJ,eAAe,IAA2B;EAC9C,MAAM,KAAKC,IAAI,aAAkC,oBAAoB,EAAE,GAAG,GAAG,OAAM,QAAO;GACxF,IAAI,IAAI,YAAY,MAAM,OAAO;GACjC,MAAM,YAAY,MAAM,KAAKE,eAAe,IAAI,OAAO;GACvD,OAAO,UAAU,oBAAoB,EAAE,SAAS,MAAM,KAAK,WAAW,QAAQ,UAAU,KAAK,EAAE,IAAI;EACrG,CAAC;CACH;;;;;;;;;CAUA,MAAM,OAAO,EACX,OACA,QACA,OACA,sBAMgC;EAChC,MAAM,sBAAM,IAAI,KAAK;EACrB,MAAM,kBAAkB,MAAM,KAAKC,eAAe,MAAM,MAAM;EAC9D,MAAM,aAAa,sBAAsB,uBAAuB,MAAM,aAAa,qBAAqB,KAAA;EAExG,IAAI,YAKE;OAAA,CAAC,MAJgB,KAAKH,IAAI,QAA6B,oBAAoB;IAC7E,QAAQ;IACR,aAAa,MAAM;GACrB,CAAC,GACY;IAOX,MAAM,UAAU,MAAM,KAAKA,IAAI,aAC7B,oBACA;KAAE,QAAQ;KAAO,aAAa;IAAW,UAClC;KACL,aAAa,MAAM;KACnB,MAAM,MAAM;KACZ,KAAK,MAAM;KACX,SAAS;KACT,QAAQ,MAAM;KACd,YAAY;IACd,EACF;IACA,IAAI,SAAS,OAAO,KAAKC,UAAU,OAAO;GAE5C;;EAMF,MAAM,SAAS,MAAM,KAAKG,OAAO;GAAE;GAAO;GAAQ;GAAO;GAAiB;EAAI,CAAC;EAC/E,IAAI,YACF,MAAM,KAAKJ,IAAI,WAAW,oBAAoB;GAAE,QAAQ;GAAO,aAAa;EAAW,CAAC;EAE1F,OAAO;CACT;;;;;;;CAQA,MAAMI,OAAO,EACX,OACA,QACA,OACA,iBACA,OAOgC;EAChC,MAAM,uBACJ,KAAKJ,IAAI,aACP,oBACA;GAAE,QAAQ;GAAO,aAAa,MAAM;EAAW,UACxC;GACL,MAAM,MAAM;GACZ,KAAK,MAAM;GACX,SAAS;GACT,QAAQ,MAAM;GACd,YAAY;EACd,EACF;EAEF,MAAM,UAAU,MAAM,eAAe;EACrC,IAAI,SAAS,OAAO,KAAKC,UAAU,OAAO;EAE1C,IAAI;GACF,MAAM,MAAM,MAAM,KAAKD,IAAI,UAA+B,oBAAoB;IAC5E,QAAQ;IACR,YAAY;IACZ,aAAa,MAAM;IACnB,MAAM,MAAM;IACZ,KAAK,MAAM;IACX,SAAS;IACT,QAAQ,MAAM;IACd,YAAY;IACZ,YAAY;GACd,CAAC;GACD,OAAO,KAAKC,UAAU,GAAG;EAC3B,SAAS,OAAO;GACd,IAAI,EAAE,iBAAiB,uBAAuB,MAAM;GAEpD,MAAM,MAAM,MAAM,eAAe;GACjC,IAAI,CAAC,KAAK,MAAM;GAChB,OAAO,KAAKA,UAAU,GAAG;EAC3B;CACF;CAEA,MAAM,KAAK,EAAE,SAA6D;EACxE,MAAM,OAAO,MAAM,KAAKD,IAAI,SAC1B,oBACA,EAAE,QAAQ,MAAM,GAChB,EAAE,SAAS,CAAC,CAAC,QAAQ,KAAK,CAAC,EAAE,CAC/B;EACA,OAAO,QAAQ,IAAI,KAAK,KAAI,QAAO,KAAKC,UAAU,GAAG,CAAC,CAAC;CACzD;CAEA,MAAM,OAAO,EAAE,OAAO,cAAuE;EAC3F,OAAQ,MAAM,KAAKD,IAAI,WAAW,oBAAoB;GAAE,QAAQ;GAAO,aAAa;EAAW,CAAC,IAAK;CACvG;AACF"}

package/dist/storage/domains/filesystem/base.d.ts

@@ -0,0 +1,26 @@
+import { FactoryStorageDomain } from '@mastra/core/storage';
+import type { CollectionSchema } from '@mastra/core/storage';
+export interface FilesystemFile {
+    path: string;
+}
+export interface ReplaceFilesystemFilesInput {
+    resourceId: string;
+    threadId: string;
+    files: FilesystemFile[];
+}
+export declare const FILESYSTEM_SCHEMAS: CollectionSchema[];
+export declare class FilesystemStorage extends FactoryStorageDomain {
+    constructor();
+    init(): Promise<void>;
+    dangerouslyClearAll(): Promise<void>;
+    replaceFiles(input: ReplaceFilesystemFilesInput): Promise<void>;
+    listFiles(args: {
+        resourceId: string;
+        threadId: string;
+    }): Promise<FilesystemFile[]>;
+    deleteFiles(args: {
+        resourceId: string;
+        threadId: string;
+    }): Promise<number>;
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/storage/domains/filesystem/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/storage/domains/filesystem/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAI7D,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,2BAA2B;IAC1C,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,cAAc,EAAE,CAAC;CACzB;AAED,eAAO,MAAM,kBAAkB,EAAE,gBAAgB,EAiBhD,CAAC;AAuCF,qBAAa,iBAAkB,SAAQ,oBAAoB;;IAKnD,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAIrB,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAIpC,YAAY,CAAC,KAAK,EAAE,2BAA2B,GAAG,OAAO,CAAC,IAAI,CAAC;IAY/D,SAAS,CAAC,IAAI,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IASpF,WAAW,CAAC,IAAI,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,CAAC;CAInF"}