@mastra/factory 0.24.0-alpha.1 → 0.24.0-alpha.11

package/dist/storage/domains/work-items/base.js

@@ -61,6 +61,9 @@ const FACTORY_DISPATCH_FAILURE_CODES = [
 	"repository_cli_missing",
 	"repository_pr_failed",
 	"run_configuration_invalid",
+	"run_credentials_unavailable",
+	"run_user_unauthenticated",
+	"run_failed",
 	"unknown"
 ];
 /** Written until a pause stopped counting as a failure; stored rows still read through them. */
@@ -224,6 +227,47 @@ const WORK_ITEMS_SCHEMA = {
 		}
 	]
 };
+const FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA = {
+	name: "factory_work_item_subscriptions",
+	columns: {
+		id: { type: "uuid-pk" },
+		org_id: { type: "text" },
+		factory_project_id: { type: "text" },
+		work_item_id: { type: "text" },
+		session_id: { type: "text" },
+		thread_id: { type: "text" },
+		resource_id: { type: "text" },
+		user_id: { type: "text" },
+		reason: { type: "text" },
+		created_at: { type: "timestamp" }
+	},
+	uniqueIndexes: [{
+		name: "factory_work_item_subscriptions_item_session_unique",
+		columns: ["work_item_id", "session_id"]
+	}],
+	indexes: [{
+		name: "factory_work_item_subscriptions_session_idx",
+		columns: [
+			"org_id",
+			"factory_project_id",
+			"session_id"
+		]
+	}]
+};
+function toWorkItemSubscription(row) {
+	return {
+		id: row.id,
+		orgId: row.org_id,
+		factoryProjectId: String(row.factory_project_id),
+		workItemId: row.work_item_id,
+		sessionId: row.session_id,
+		threadId: row.thread_id,
+		resourceId: row.resource_id,
+		userId: row.user_id,
+		reason: row.reason,
+		createdAt: row.created_at
+	};
+}
 /**
 * The one shape a platform id takes in a lookup key, for cards and for the
 * comments mirrored off them alike: whoever writes a second builder reopens the
@@ -890,12 +934,14 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			...FACTORY_GOVERNANCE_SCHEMAS,
 			WORK_ITEM_COMMENTS_SCHEMA,
 			WORK_ITEM_COMMENT_MENTIONS_SCHEMA,
-			WORK_ITEM_ACTIVITY_SCHEMA
+			WORK_ITEM_ACTIVITY_SCHEMA,
+			FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA
 		]);
 		await this.repairLegacyAttentionState();
 	}
 	async dangerouslyClearAll() {
 		await this.ops.deleteMany("work_items", {});
+		await super.ops.deleteMany(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {});
 		await super.ops.deleteMany(FACTORY_BOARD_VERSIONS_SCHEMA.name, {});
 		this.#listCache.clear();
 	}
@@ -962,6 +1008,8 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 							attempts: Number(current.attempts) + 1,
 							lease_owner: input.ownerId,
 							lease_expires_at: input.leaseExpiresAt,
+							last_error: null,
+							failure_code: null,
 							updated_at: input.now
 						};
 					});
@@ -1420,29 +1468,34 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				causal_chain: input.causalChain,
 				created_at: input.now
 			});
-			for (const [effectOrdinal, decision] of decisions.entries()) await ops.insertOne("factory_deferred_decisions", {
-				org_id: input.orgId,
-				factory_project_id: input.factoryProjectId,
-				evaluation_id: evaluation.id,
-				work_item_id: item?.id ?? null,
-				source_key: decisionSourceKey(decision),
-				idempotency_key: String(decision.idempotencyKey),
-				effect_ordinal: effectOrdinal,
-				effect_hash: factoryDecisionHash(decision),
-				causal_chain: input.causalChain,
-				actor: input.actor,
-				decision,
-				status: "pending",
-				attempts: 0,
-				delivery_generation: 0,
-				available_at: input.now,
-				lease_owner: null,
-				lease_expires_at: null,
-				last_error: null,
-				completed_at: null,
-				created_at: new Date(input.now.getTime() + effectOrdinal),
-				updated_at: input.now
-			});
+			for (const [effectOrdinal, decision] of decisions.entries()) {
+				const approvedBy = decision.type === "invokeSkill" ? input.consentedBy ?? null : null;
+				await ops.insertOne("factory_deferred_decisions", {
+					org_id: input.orgId,
+					factory_project_id: input.factoryProjectId,
+					evaluation_id: evaluation.id,
+					work_item_id: item?.id ?? null,
+					source_key: decisionSourceKey(decision),
+					idempotency_key: String(decision.idempotencyKey),
+					effect_ordinal: effectOrdinal,
+					effect_hash: factoryDecisionHash(decision),
+					causal_chain: input.causalChain,
+					actor: input.actor,
+					decision,
+					status: "pending",
+					attempts: 0,
+					delivery_generation: 0,
+					available_at: input.now,
+					lease_owner: null,
+					lease_expires_at: null,
+					last_error: null,
+					completed_at: null,
+					approved_at: approvedBy ? input.now : null,
+					approved_by: approvedBy,
+					created_at: new Date(input.now.getTime() + effectOrdinal),
+					updated_at: input.now
+				});
+			}
 			return {
 				status: "committed",
 				result
@@ -2032,6 +2085,14 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		if (rows.length !== 1) return null;
 		return toBinding(rows[0]);
 	}
+	/** Whether this session was ever bound to a run, on any thread and whether or not the binding was revoked. */
+	async hasRunBindingForSession(input) {
+		return await this.#db.findOne("factory_run_bindings", {
+			org_id: input.orgId,
+			factory_project_id: input.factoryProjectId,
+			session_id: input.sessionId
+		}) !== null;
+	}
 	/** Resolve exact bound-session state for processor awareness; ambiguous cross-tenant matches return null. */
 	async findRunBindingBySession(address) {
 		const rows = await this.#db.findMany("factory_run_bindings", {
@@ -2082,17 +2143,17 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		});
 	}
 	/**
-	* Revoke leaked/legacy active bindings: older than `olderThan`, or whose
-	* work item is gone, malformed, or already terminal. Terminal-stage cleanup
-	* handles bindings go-forward; this sweep drains anything that slipped past
-	* it. Returns the number of bindings revoked.
+	* Revoke leaked/legacy active bindings: older than `olderThan` (unless
+	* `keep` vouches for them), or whose work item is gone, malformed, or
+	* already terminal. Terminal-stage cleanup handles bindings go-forward; this
+	* sweep drains anything that slipped past it. Returns the number of bindings revoked.
 	*/
 	async revokeStaleRunBindings(input) {
 		const bindings = await this.listActiveRunBindings();
 		const itemCache = /* @__PURE__ */ new Map();
 		let revoked = 0;
 		for (const binding of bindings) {
-			let stale = binding.createdAt.getTime() < input.olderThan.getTime();
+			let stale = binding.createdAt.getTime() < input.olderThan.getTime() && !await input.keep?.(binding);
 			if (!stale) {
 				const key = `${binding.orgId}:${binding.workItemId}`;
 				let item = itemCache.get(key);
@@ -2119,6 +2180,29 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	async listActiveRunBindings() {
 		return (await this.#db.findMany("factory_run_bindings", { status: "active" })).map(toBinding);
 	}
+	/**
+	* The work items a session was ever bound to run, revoked bindings included.
+	* `null` stands for a bound item that no longer loads. Empty when the session
+	* never ran a work item.
+	*/
+	async listRunBoundItems(sessionId) {
+		const rows = await this.#db.findMany("factory_run_bindings", { session_id: sessionId });
+		const keys = new Map(rows.map((row) => [`${row.org_id}:${row.work_item_id}`, row]));
+		return Promise.all([...keys.values()].map((row) => this.get({
+			orgId: String(row.org_id),
+			id: String(row.work_item_id)
+		})));
+	}
+	/**
+	* Move a session's bindings from the legacy `resourceId = sessionId` onto
+	* `resourceId`. Only rows still on the session id change, so a rerun is a no-op.
+	*/
+	async moveRunBindingsToResource(input) {
+		return this.#db.updateMany("factory_run_bindings", {
+			session_id: input.sessionId,
+			resource_id: input.sessionId
+		}, { resource_id: input.resourceId });
+	}
 	/** List binding history, optionally narrowed to one work item. */
 	async listRunBindings(orgId, factoryProjectId, workItemId) {
 		return (await this.#db.findMany("factory_run_bindings", {
@@ -2560,6 +2644,60 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		});
 		await ops.deleteMany("work_item_activity", where);
 	}
+	/**
+	* Make a session follow a work item. Idempotent: following again keeps the
+	* first subscription, including its reason (filing beats a later subscribe).
+	*/
+	async subscribeToWorkItem(input) {
+		const ops = super.ops;
+		const where = {
+			work_item_id: input.workItemId,
+			session_id: input.sessionId
+		};
+		const existing = await ops.findOne(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, where);
+		if (existing) return toWorkItemSubscription(existing);
+		try {
+			return toWorkItemSubscription(await ops.insertOne(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {
+				org_id: input.orgId,
+				factory_project_id: input.factoryProjectId,
+				work_item_id: input.workItemId,
+				session_id: input.sessionId,
+				thread_id: input.threadId,
+				resource_id: input.resourceId,
+				user_id: input.userId,
+				reason: input.reason,
+				created_at: /* @__PURE__ */ new Date()
+			}));
+		} catch (error) {
+			if (!(error instanceof UniqueViolationError)) throw error;
+			const raced = await ops.findOne(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, where);
+			if (!raced) throw error;
+			return toWorkItemSubscription(raced);
+		}
+	}
+	/** Stop a session following a work item. Returns whether it was following. */
+	async unsubscribeFromWorkItem({ orgId, workItemId, sessionId }) {
+		return await super.ops.deleteMany(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {
+			org_id: orgId,
+			work_item_id: workItemId,
+			session_id: sessionId
+		}) > 0;
+	}
+	/** The sessions following a work item, earliest first. */
+	async listWorkItemSubscribers({ orgId, workItemId }) {
+		return (await super.ops.findMany(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {
+			org_id: orgId,
+			work_item_id: workItemId
+		}, { orderBy: [["created_at", "asc"], ["id", "asc"]] })).map(toWorkItemSubscription);
+	}
+	/** The work items one session follows in a Factory, earliest first. */
+	async listSessionSubscriptions({ orgId, factoryProjectId, sessionId }) {
+		return (await super.ops.findMany(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {
+			org_id: orgId,
+			factory_project_id: factoryProjectId,
+			session_id: sessionId
+		}, { orderBy: [["created_at", "asc"], ["id", "asc"]] })).map(toWorkItemSubscription);
+	}
 	async delete({ orgId, id, purgeRuleState = true }) {
 		const candidate = await this.#db.findOne("work_items", {
 			org_id: orgId,
@@ -2581,6 +2719,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				factoryProjectId: existing.factory_project_id,
 				workItemId: id
 			});
+			await ops.deleteMany(FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA.name, {
+				org_id: orgId,
+				work_item_id: id
+			});
 			if (purgeRuleState && existing.external_source?.externalId) await this.#purgeRuleState(ops, {
 				orgId,
 				factoryProjectId: existing.factory_project_id,
@@ -2600,6 +2742,6 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	}
 };
 //#endregion
-export { ACTIVE_RUN_BINDING_STAGES, FACTORY_PULL_REQUEST_RECONCILIATION_KEY, FACTORY_RULE_MATERIALIZATION_KEY, WORK_ITEMS_SCHEMA, WorkItemClaimConflictError, WorkItemRelationError, WorkItemUpdateConflictError, WorkItemsStorage, applyStageTransition, externalSourceKey, factoryActivityAttentionIdentity, factoryAgentWaitingAttentionIdentity, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, factoryProposalAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
+export { ACTIVE_RUN_BINDING_STAGES, FACTORY_PULL_REQUEST_RECONCILIATION_KEY, FACTORY_RULE_MATERIALIZATION_KEY, FACTORY_WORK_ITEM_SUBSCRIPTIONS_SCHEMA, WORK_ITEMS_SCHEMA, WorkItemClaimConflictError, WorkItemRelationError, WorkItemUpdateConflictError, WorkItemsStorage, applyStageTransition, externalSourceKey, factoryActivityAttentionIdentity, factoryAgentWaitingAttentionIdentity, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, factoryProposalAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
 
 //# sourceMappingURL=base.js.map