@mastra/factory 0.2.2-alpha.1 → 0.2.2-alpha.2

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

@@ -1,252 +1,317 @@
-// src/storage/domains/integrations/base.ts
 import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
-var INTEGRATION_CONNECTIONS_SCHEMA = {
-  name: "integration_connections",
-  columns: {
-    id: { type: "uuid-pk" },
-    integration_id: { type: "text" },
-    org_id: { type: "text" },
-    /** Who connected it (audit only — never scopes reads). */
-    user_id: { type: "text", nullable: true },
-    data: { type: "json" },
-    metadata: { type: "json" },
-    created_at: { type: "timestamp" },
-    updated_at: { type: "timestamp" }
-  },
-  uniqueIndexes: [{ name: "integration_connections_integration_org_unique", columns: ["integration_id", "org_id"] }]
+//#region src/storage/domains/integrations/base.ts
+/**
+* Generic integration storage — the default persistence surface for
+* `FactoryIntegration`s.
+*
+* Most integrations persist the same three shapes: an org-owned OAuth
+* **connection**, **subscriptions** binding external objects to
+* sessions/threads, and per-`(org, user)` **settings**. This built-in domain
+* stores all three in shared collections keyed by `integration_id`, so an
+* integration author writes zero schema, zero DDL, and zero queries — and
+* org scoping is enforced by construction through the pre-scoped handle
+* returned by {@link IntegrationStorage.forIntegration}.
+*
+* Source-control integrations additionally use the shared, provider-scoped
+* source-control domain for installations, projects, worktrees, and sandboxes.
+*
+* Payloads (`data` / `config`) are JSON documents typed per-integration via
+* the handle's generics. JSON round-trips exactly what JSON can represent:
+* store timestamps inside payloads as epoch millis or ISO strings, not
+* `Date`s.
+*/
+const INTEGRATION_CONNECTIONS_SCHEMA = {
+	name: "integration_connections",
+	columns: {
+		id: { type: "uuid-pk" },
+		integration_id: { type: "text" },
+		org_id: { type: "text" },
+		/** Who connected it (audit only — never scopes reads). */
+		user_id: {
+			type: "text",
+			nullable: true
+		},
+		data: { type: "json" },
+		metadata: { type: "json" },
+		created_at: { type: "timestamp" },
+		updated_at: { type: "timestamp" }
+	},
+	uniqueIndexes: [{
+		name: "integration_connections_integration_org_unique",
+		columns: ["integration_id", "org_id"]
+	}]
 };
-var INTEGRATION_SUBSCRIPTIONS_SCHEMA = {
-  name: "integration_subscriptions",
-  columns: {
-    id: { type: "uuid-pk" },
-    integration_id: { type: "text" },
-    org_id: { type: "text" },
-    /** Canonical external-object key the integration derives (e.g. `pr:{installation}:{repo}:{number}`). */
-    target_key: { type: "text" },
-    session_id: { type: "text", nullable: true },
-    resource_id: { type: "text", nullable: true },
-    thread_id: { type: "text", nullable: true },
-    session_scope: { type: "text", nullable: true },
-    status: { type: "text" },
-    data: { type: "json" },
-    created_at: { type: "timestamp" },
-    updated_at: { type: "timestamp" }
-  },
-  indexes: [
-    { name: "integration_subscriptions_target_idx", columns: ["integration_id", "target_key"] },
-    { name: "integration_subscriptions_org_idx", columns: ["integration_id", "org_id"] }
-  ]
+const INTEGRATION_SUBSCRIPTIONS_SCHEMA = {
+	name: "integration_subscriptions",
+	columns: {
+		id: { type: "uuid-pk" },
+		integration_id: { type: "text" },
+		org_id: { type: "text" },
+		/** Canonical external-object key the integration derives (e.g. `pr:{installation}:{repo}:{number}`). */
+		target_key: { type: "text" },
+		session_id: {
+			type: "text",
+			nullable: true
+		},
+		resource_id: {
+			type: "text",
+			nullable: true
+		},
+		thread_id: {
+			type: "text",
+			nullable: true
+		},
+		session_scope: {
+			type: "text",
+			nullable: true
+		},
+		status: { type: "text" },
+		data: { type: "json" },
+		created_at: { type: "timestamp" },
+		updated_at: { type: "timestamp" }
+	},
+	indexes: [{
+		name: "integration_subscriptions_target_idx",
+		columns: ["integration_id", "target_key"]
+	}, {
+		name: "integration_subscriptions_org_idx",
+		columns: ["integration_id", "org_id"]
+	}]
 };
-var INTEGRATION_SETTINGS_SCHEMA = {
-  name: "integration_settings",
-  columns: {
-    id: { type: "uuid-pk" },
-    integration_id: { type: "text" },
-    org_id: { type: "text" },
-    user_id: { type: "text" },
-    config: { type: "json" },
-    created_at: { type: "timestamp" },
-    updated_at: { type: "timestamp" }
-  },
-  uniqueIndexes: [
-    { name: "integration_settings_integration_org_user_unique", columns: ["integration_id", "org_id", "user_id"] }
-  ]
+const INTEGRATION_SETTINGS_SCHEMA = {
+	name: "integration_settings",
+	columns: {
+		id: { type: "uuid-pk" },
+		integration_id: { type: "text" },
+		org_id: { type: "text" },
+		user_id: { type: "text" },
+		config: { type: "json" },
+		created_at: { type: "timestamp" },
+		updated_at: { type: "timestamp" }
+	},
+	uniqueIndexes: [{
+		name: "integration_settings_integration_org_user_unique",
+		columns: [
+			"integration_id",
+			"org_id",
+			"user_id"
+		]
+	}]
 };
+/**
+* The built-in generic integration domain, written once against the
+* `FactoryStorageOps` surface — works on any `FactoryStorage` backend.
+*/
 var IntegrationStorage = class extends FactoryStorageDomain {
-  constructor() {
-    super("integrations");
-  }
-  async init() {
-    await this.ensureCollections([
-      INTEGRATION_CONNECTIONS_SCHEMA,
-      INTEGRATION_SUBSCRIPTIONS_SCHEMA,
-      INTEGRATION_SETTINGS_SCHEMA
-    ]);
-  }
-  async dangerouslyClearAll() {
-    await this.ops.deleteMany("integration_subscriptions", {});
-    await this.ops.deleteMany("integration_settings", {});
-    await this.ops.deleteMany("integration_connections", {});
-  }
-  get #db() {
-    return this.ops;
-  }
-  /** A typed handle pre-scoped to `integrationId`. */
-  forIntegration(integrationId) {
-    if (!integrationId) throw new Error("[IntegrationStorage] integrationId is required.");
-    const db = () => this.#db;
-    const scoped = { integration_id: integrationId };
-    const mapConnection = (row) => ({
-      id: row.id,
-      orgId: row.org_id,
-      userId: row.user_id,
-      data: row.data,
-      metadata: row.metadata ?? {},
-      createdAt: row.created_at,
-      updatedAt: row.updated_at
-    });
-    const mapSubscription = (row) => ({
-      id: row.id,
-      orgId: row.org_id,
-      targetKey: row.target_key,
-      sessionId: row.session_id,
-      resourceId: row.resource_id,
-      threadId: row.thread_id,
-      sessionScope: row.session_scope,
-      status: row.status,
-      data: row.data,
-      createdAt: row.created_at,
-      updatedAt: row.updated_at
-    });
-    return {
-      integrationId,
-      connections: {
-        get: async (orgId) => {
-          const row = await db().findOne("integration_connections", { ...scoped, org_id: orgId });
-          return row ? mapConnection(row) : null;
-        },
-        upsert: async (orgId, input) => {
-          const where = { ...scoped, org_id: orgId };
-          const update = async () => db().updateMany("integration_connections", where, {
-            user_id: input.userId ?? null,
-            data: input.data,
-            metadata: input.metadata ?? {},
-            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 db().insertOne("integration_connections", {
-                ...where,
-                user_id: input.userId ?? null,
-                data: input.data,
-                metadata: input.metadata ?? {},
-                created_at: now,
-                updated_at: now
-              });
-              return;
-            } catch (error) {
-              if (!(error instanceof UniqueViolationError)) throw error;
-              lastError = error;
-            }
-          }
-          throw lastError;
-        },
-        update: async (orgId, fn) => {
-          const row = await db().updateAtomic(
-            "integration_connections",
-            { ...scoped, org_id: orgId },
-            (current) => ({ data: fn(current.data), updated_at: /* @__PURE__ */ new Date() })
-          );
-          return row ? mapConnection(row) : null;
-        },
-        delete: async (orgId) => {
-          const deleted = await db().deleteMany("integration_connections", { ...scoped, org_id: orgId });
-          return deleted > 0;
-        }
-      },
-      subscriptions: {
-        create: async (input) => {
-          const now = /* @__PURE__ */ new Date();
-          const row = await db().insertOne("integration_subscriptions", {
-            ...scoped,
-            org_id: input.orgId,
-            target_key: input.targetKey,
-            session_id: input.sessionId ?? null,
-            resource_id: input.resourceId ?? null,
-            thread_id: input.threadId ?? null,
-            session_scope: input.sessionScope ?? null,
-            status: input.status ?? "active",
-            data: input.data ?? {},
-            created_at: now,
-            updated_at: now
-          });
-          return mapSubscription(row);
-        },
-        listByTarget: async (targetKey, opts) => {
-          const rows = await db().findMany(
-            "integration_subscriptions",
-            { ...scoped, target_key: targetKey, ...opts?.status !== void 0 ? { status: opts.status } : {} },
-            { orderBy: [["created_at", "asc"]] }
-          );
-          return rows.map(mapSubscription);
-        },
-        listBySession: async (sessionId) => {
-          const rows = await db().findMany(
-            "integration_subscriptions",
-            { ...scoped, session_id: sessionId },
-            { orderBy: [["created_at", "asc"]] }
-          );
-          return rows.map(mapSubscription);
-        },
-        listByThread: async (resourceId, threadId) => {
-          const rows = await db().findMany(
-            "integration_subscriptions",
-            { ...scoped, resource_id: resourceId, thread_id: threadId },
-            { orderBy: [["created_at", "asc"]] }
-          );
-          return rows.map(mapSubscription);
-        },
-        updateStatus: async (id, status) => {
-          await db().updateMany("integration_subscriptions", { ...scoped, id }, { status, updated_at: /* @__PURE__ */ new Date() });
-        },
-        delete: async (id) => {
-          const deleted = await db().deleteMany("integration_subscriptions", { ...scoped, id });
-          return deleted > 0;
-        },
-        deleteWhere: async (where) => {
-          return db().deleteMany("integration_subscriptions", {
-            ...scoped,
-            org_id: where.orgId,
-            ...where.targetKey !== void 0 ? { target_key: where.targetKey } : {},
-            ...where.sessionId !== void 0 ? { session_id: where.sessionId } : {}
-          });
-        }
-      },
-      settings: {
-        get: async (orgId, userId) => {
-          const row = await db().findOne("integration_settings", {
-            ...scoped,
-            org_id: orgId,
-            user_id: userId
-          });
-          return row ? structuredClone(row.config) : null;
-        },
-        save: async (orgId, userId, config) => {
-          const where = { ...scoped, org_id: orgId, user_id: userId };
-          const update = async () => db().updateMany("integration_settings", where, {
-            config,
-            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 db().insertOne("integration_settings", {
-                ...where,
-                config,
-                created_at: now,
-                updated_at: now
-              });
-              return;
-            } catch (error) {
-              if (!(error instanceof UniqueViolationError)) throw error;
-              lastError = error;
-            }
-          }
-          throw lastError;
-        }
-      }
-    };
-  }
-};
-export {
-  INTEGRATION_CONNECTIONS_SCHEMA,
-  INTEGRATION_SETTINGS_SCHEMA,
-  INTEGRATION_SUBSCRIPTIONS_SCHEMA,
-  IntegrationStorage
+	constructor() {
+		super("integrations");
+	}
+	async init() {
+		await this.ensureCollections([
+			INTEGRATION_CONNECTIONS_SCHEMA,
+			INTEGRATION_SUBSCRIPTIONS_SCHEMA,
+			INTEGRATION_SETTINGS_SCHEMA
+		]);
+	}
+	async dangerouslyClearAll() {
+		await this.ops.deleteMany("integration_subscriptions", {});
+		await this.ops.deleteMany("integration_settings", {});
+		await this.ops.deleteMany("integration_connections", {});
+	}
+	get #db() {
+		return this.ops;
+	}
+	/** A typed handle pre-scoped to `integrationId`. */
+	forIntegration(integrationId) {
+		if (!integrationId) throw new Error("[IntegrationStorage] integrationId is required.");
+		const db = () => this.#db;
+		const scoped = { integration_id: integrationId };
+		const mapConnection = (row) => ({
+			id: row.id,
+			orgId: row.org_id,
+			userId: row.user_id,
+			data: row.data,
+			metadata: row.metadata ?? {},
+			createdAt: row.created_at,
+			updatedAt: row.updated_at
+		});
+		const mapSubscription = (row) => ({
+			id: row.id,
+			orgId: row.org_id,
+			targetKey: row.target_key,
+			sessionId: row.session_id,
+			resourceId: row.resource_id,
+			threadId: row.thread_id,
+			sessionScope: row.session_scope,
+			status: row.status,
+			data: row.data,
+			createdAt: row.created_at,
+			updatedAt: row.updated_at
+		});
+		return {
+			integrationId,
+			connections: {
+				get: async (orgId) => {
+					const row = await db().findOne("integration_connections", {
+						...scoped,
+						org_id: orgId
+					});
+					return row ? mapConnection(row) : null;
+				},
+				upsert: async (orgId, input) => {
+					const where = {
+						...scoped,
+						org_id: orgId
+					};
+					const update = async () => db().updateMany("integration_connections", where, {
+						user_id: input.userId ?? null,
+						data: input.data,
+						metadata: input.metadata ?? {},
+						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 db().insertOne("integration_connections", {
+								...where,
+								user_id: input.userId ?? null,
+								data: input.data,
+								metadata: input.metadata ?? {},
+								created_at: now,
+								updated_at: now
+							});
+							return;
+						} catch (error) {
+							if (!(error instanceof UniqueViolationError)) throw error;
+							lastError = error;
+						}
+					}
+					throw lastError;
+				},
+				update: async (orgId, fn) => {
+					const row = await db().updateAtomic("integration_connections", {
+						...scoped,
+						org_id: orgId
+					}, (current) => ({
+						data: fn(current.data),
+						updated_at: /* @__PURE__ */ new Date()
+					}));
+					return row ? mapConnection(row) : null;
+				},
+				delete: async (orgId) => {
+					return await db().deleteMany("integration_connections", {
+						...scoped,
+						org_id: orgId
+					}) > 0;
+				}
+			},
+			subscriptions: {
+				create: async (input) => {
+					const now = /* @__PURE__ */ new Date();
+					const row = await db().insertOne("integration_subscriptions", {
+						...scoped,
+						org_id: input.orgId,
+						target_key: input.targetKey,
+						session_id: input.sessionId ?? null,
+						resource_id: input.resourceId ?? null,
+						thread_id: input.threadId ?? null,
+						session_scope: input.sessionScope ?? null,
+						status: input.status ?? "active",
+						data: input.data ?? {},
+						created_at: now,
+						updated_at: now
+					});
+					return mapSubscription(row);
+				},
+				listByTarget: async (targetKey, opts) => {
+					return (await db().findMany("integration_subscriptions", {
+						...scoped,
+						target_key: targetKey,
+						...opts?.status !== void 0 ? { status: opts.status } : {}
+					}, { orderBy: [["created_at", "asc"]] })).map(mapSubscription);
+				},
+				listBySession: async (sessionId) => {
+					return (await db().findMany("integration_subscriptions", {
+						...scoped,
+						session_id: sessionId
+					}, { orderBy: [["created_at", "asc"]] })).map(mapSubscription);
+				},
+				listByThread: async (resourceId, threadId) => {
+					return (await db().findMany("integration_subscriptions", {
+						...scoped,
+						resource_id: resourceId,
+						thread_id: threadId
+					}, { orderBy: [["created_at", "asc"]] })).map(mapSubscription);
+				},
+				updateStatus: async (id, status) => {
+					await db().updateMany("integration_subscriptions", {
+						...scoped,
+						id
+					}, {
+						status,
+						updated_at: /* @__PURE__ */ new Date()
+					});
+				},
+				delete: async (id) => {
+					return await db().deleteMany("integration_subscriptions", {
+						...scoped,
+						id
+					}) > 0;
+				},
+				deleteWhere: async (where) => {
+					return db().deleteMany("integration_subscriptions", {
+						...scoped,
+						org_id: where.orgId,
+						...where.targetKey !== void 0 ? { target_key: where.targetKey } : {},
+						...where.sessionId !== void 0 ? { session_id: where.sessionId } : {}
+					});
+				}
+			},
+			settings: {
+				get: async (orgId, userId) => {
+					const row = await db().findOne("integration_settings", {
+						...scoped,
+						org_id: orgId,
+						user_id: userId
+					});
+					return row ? structuredClone(row.config) : null;
+				},
+				save: async (orgId, userId, config) => {
+					const where = {
+						...scoped,
+						org_id: orgId,
+						user_id: userId
+					};
+					const update = async () => db().updateMany("integration_settings", where, {
+						config,
+						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 db().insertOne("integration_settings", {
+								...where,
+								config,
+								created_at: now,
+								updated_at: now
+							});
+							return;
+						} catch (error) {
+							if (!(error instanceof UniqueViolationError)) throw error;
+							lastError = error;
+						}
+					}
+					throw lastError;
+				}
+			}
+		};
+	}
 };
+//#endregion
+export { INTEGRATION_CONNECTIONS_SCHEMA, INTEGRATION_SETTINGS_SCHEMA, INTEGRATION_SUBSCRIPTIONS_SCHEMA, IntegrationStorage };
+
 //# sourceMappingURL=base.js.map