@mastra/factory 0.12.0-alpha.0 → 0.12.0-alpha.10

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

@@ -1,5 +1,5 @@
 import { isTerminalFactoryRuleStage } from "../../../rules/types.js";
-import { WORK_ITEM_COMMENTS_SCHEMA, WORK_ITEM_COMMENT_MENTIONS_SCHEMA } from "../comments/schema.js";
+import { WORK_ITEM_ACTIVITY_SCHEMA, WORK_ITEM_COMMENTS_SCHEMA, WORK_ITEM_COMMENT_MENTIONS_SCHEMA } from "../comments/schema.js";
 import { createHash } from "crypto";
 import { FactoryStorageDomain, UniqueViolationError } from "@mastra/core/storage";
 //#region src/storage/domains/work-items/base.ts
@@ -43,6 +43,8 @@ const FACTORY_DISPATCH_FAILURE_CODES = [
 	"source_repository_missing",
 	"unsupported_provider_item",
 	"notification_delivery_failed",
+	"plan_awaiting_approval",
+	"run_awaiting_input",
 	"repository_git_missing",
 	"repository_egress_blocked",
 	"repository_clone_failed",
@@ -70,6 +72,14 @@ function factoryMentionAttentionIdentity(commentId) {
 		occurrence: 0
 	};
 }
+/** Collapsed per work item, so the occurrence is what a new comment bumps. */
+function factoryActivityAttentionIdentity(workItemId, occurrence) {
+	return {
+		kind: "activity",
+		sourceId: workItemId,
+		occurrence
+	};
+}
 function factoryAttentionKey(factoryProjectId, identity) {
 	return `factory:${factoryProjectId}:attention:${identity.kind}:${identity.sourceId}:${identity.occurrence}`;
 }
@@ -119,6 +129,10 @@ const WORK_ITEMS_SCHEMA = {
 			type: "timestamp",
 			nullable: true
 		},
+		plans_preapproved_at: {
+			type: "timestamp",
+			nullable: true
+		},
 		comment_count: {
 			type: "integer",
 			default: 0
@@ -139,25 +153,39 @@ const WORK_ITEMS_SCHEMA = {
 		name: "work_items_project_source_key_unique",
 		columns: ["factory_project_id", "source_key"]
 	}],
-	indexes: [{
-		name: "work_items_org_project_created_at_id_idx",
-		columns: [
-			"org_id",
-			"factory_project_id",
-			"created_at",
-			"id"
-		]
-	}, {
-		name: "work_items_project_parent_idx",
-		columns: [
-			"org_id",
-			"factory_project_id",
-			"parent_work_item_id"
-		]
-	}]
+	indexes: [
+		{
+			name: "work_items_org_project_created_at_id_idx",
+			columns: [
+				"org_id",
+				"factory_project_id",
+				"created_at",
+				"id"
+			]
+		},
+		{
+			name: "work_items_project_parent_idx",
+			columns: [
+				"org_id",
+				"factory_project_id",
+				"parent_work_item_id"
+			]
+		},
+		{
+			name: "work_items_source_key_idx",
+			columns: ["source_key"]
+		}
+	]
 };
-function sourceKey(source) {
-	return source ? `${source.integrationId}:${source.type}:${source.externalId}` : null;
+/**
+* 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
+* cross-workspace collision the `workspaceId` is here to close.
+*/
+function externalSourceKey(source) {
+	if (!source) return null;
+	const workspace = source.workspaceId ? `${source.workspaceId}:` : "";
+	return `${source.integrationId}:${source.type}:${workspace}${source.externalId}`;
 }
 function toWorkItem(row) {
 	return {
@@ -173,6 +201,7 @@ function toWorkItem(row) {
 		metadata: row.metadata,
 		triageType: row.triage_type ?? null,
 		autonomyArmedAt: row.autonomy_armed_at ?? null,
+		plansPreapprovedAt: row.plans_preapproved_at ?? null,
 		commentCount: row.comment_count ?? 0,
 		feedActivityAt: row.feed_activity_at ?? null,
 		revision: row.revision,
@@ -191,7 +220,7 @@ function patchColumns(changes) {
 		...changes.sessions !== void 0 ? { sessions: changes.sessions } : {},
 		...changes.metadata !== void 0 ? { metadata: changes.metadata } : {},
 		...changes.triageType !== void 0 ? { triage_type: changes.triageType } : {},
-		...changes.autonomyArmedAt !== void 0 && changes.autonomyArmedAt !== null ? { autonomy_armed_at: changes.autonomyArmedAt } : {},
+		...changes.autonomyArmedAt !== void 0 ? { autonomy_armed_at: changes.autonomyArmedAt } : {},
 		...changes.revision !== void 0 ? { revision: changes.revision } : {},
 		...changes.updatedAt !== void 0 ? { updated_at: changes.updatedAt } : {}
 	};
@@ -577,6 +606,10 @@ const FACTORY_GOVERNANCE_SCHEMAS = [
 				type: "text",
 				nullable: true
 			},
+			failure_code: {
+				type: "text",
+				nullable: true
+			},
 			completed_at: {
 				type: "timestamp",
 				nullable: true
@@ -644,7 +677,7 @@ function toDeferredDecision(row) {
 	};
 }
 function attentionReceiptKind(value) {
-	if (value === "automation-failed" || value === "mention") return value;
+	if (value === "automation-failed" || value === "mention" || value === "activity") return value;
 	throw new Error(`Unsupported attention receipt kind '${String(value)}'.`);
 }
 function attentionReceiptState(value) {
@@ -689,21 +722,32 @@ function toPendingStart(row) {
 		leaseOwner: row.lease_owner ?? null,
 		leaseExpiresAt: row.lease_expires_at ?? null,
 		lastError: row.last_error ?? null,
+		failureCode: row.failure_code ?? null,
 		completedAt: row.completed_at ?? null,
 		createdAt: row.created_at,
 		updatedAt: row.updated_at
 	};
 }
 var WorkItemsStorage = class extends FactoryStorageDomain {
+	#attentionChanged = () => {};
 	constructor() {
 		super("work-items");
 	}
+	/**
+	* Wired once at boot. Every write below that changes what this project's
+	* attention list projects announces it here — the one place a new such write
+	* has to remember, since clients stop polling while their stream is up.
+	*/
+	onAttentionChanged(listener) {
+		this.#attentionChanged = listener;
+	}
 	async init() {
 		await this.ensureCollections([
 			WORK_ITEMS_SCHEMA,
 			...FACTORY_GOVERNANCE_SCHEMAS,
 			WORK_ITEM_COMMENTS_SCHEMA,
-			WORK_ITEM_COMMENT_MENTIONS_SCHEMA
+			WORK_ITEM_COMMENT_MENTIONS_SCHEMA,
+			WORK_ITEM_ACTIVITY_SCHEMA
 		]);
 		await this.repairLegacyAttentionState();
 	}
@@ -812,7 +856,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				lease_owner: null,
 				lease_expires_at: null,
 				last_error: input.lastError,
-				...table === "factory_deferred_decisions" ? { failure_code: input.failureCode } : {},
+				failure_code: input.failureCode,
 				completed_at: input.terminal ? input.now : null,
 				...table === "factory_deferred_decisions" && input.terminal ? { failure_occurrence: Number(current.failure_occurrence ?? 0) + 1 } : {},
 				updated_at: input.now
@@ -873,6 +917,16 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		});
 		return row ? toWorkItem(row) : null;
 	}
+	/**
+	* Resolve the card a platform thread created, given only its external source.
+	* Bare of org/project because an inbound platform message carries neither: an
+	* unlinked sender has no tenant. Tenant safety rides on the key's
+	* `workspaceId` instead. Ambiguous matches resolve to nothing, not a guess.
+	*/
+	async getBySource(source) {
+		const rows = await this.#db.findMany("work_items", { source_key: externalSourceKey(source) });
+		return rows.length === 1 ? toWorkItem(rows[0]) : null;
+	}
 	async getForProject(orgId, factoryProjectId, id) {
 		const row = await this.#db.findOne("work_items", {
 			id,
@@ -922,19 +976,22 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					reason = input.evaluation.reason;
 					return null;
 				}
-				const arm = input.armAutonomy === true && !existing.autonomyArmedAt;
+				const arm = input.autonomy === "arm" && !existing.autonomyArmedAt;
+				const disarm = input.autonomy === "disarm" && existing.autonomyArmedAt !== null;
 				const triageType = existing.triageType ?? input.triageType ?? null;
 				const classified = triageType !== existing.triageType;
-				if (existing.stages.length === 1 && existing.stages[0] === input.destinationStage) return arm || classified ? patchColumns({
+				if (existing.stages.length === 1 && existing.stages[0] === input.destinationStage) return arm || disarm || classified ? patchColumns({
 					...arm ? { autonomyArmedAt: now } : {},
-					...classified ? { triageType } : {},
+					...disarm ? { autonomyArmedAt: null } : {},
 					...classified ? {
+						triageType,
 						revision: existing.revision + 1,
 						updatedAt: now
 					} : {}
 				}) : null;
 				return patchColumns({
 					...arm ? { autonomyArmedAt: now } : {},
+					...disarm ? { autonomyArmedAt: null } : {},
 					...classified ? { triageType } : {},
 					stages: [input.destinationStage],
 					stageHistory: applyStageTransition(existing.stageHistory, existing.stages, [input.destinationStage], input.actorId, now),
@@ -983,29 +1040,34 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					causal_chain: input.causalChain,
 					created_at: now
 				});
-				if (outcome === "accepted" && input.evaluation.outcome === "accepted") for (const [index, decision] of input.evaluation.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,
-					source_key: decisionSourceKey(decision),
-					idempotency_key: String(decision.idempotencyKey),
-					effect_ordinal: index,
-					effect_hash: factoryDecisionHash(decision),
-					causal_chain: input.causalChain,
-					actor: null,
-					decision,
-					status: "pending",
-					attempts: 0,
-					delivery_generation: 0,
-					available_at: now,
-					lease_owner: null,
-					lease_expires_at: null,
-					last_error: null,
-					completed_at: null,
-					created_at: new Date(now.getTime() + index),
-					updated_at: now
-				});
+				if (outcome === "accepted" && input.evaluation.outcome === "accepted") for (const [index, decision] of input.evaluation.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,
+						source_key: decisionSourceKey(decision),
+						idempotency_key: String(decision.idempotencyKey),
+						effect_ordinal: index,
+						effect_hash: factoryDecisionHash(decision),
+						causal_chain: input.causalChain,
+						actor: null,
+						decision,
+						status: "pending",
+						attempts: 0,
+						delivery_generation: 0,
+						available_at: now,
+						lease_owner: null,
+						lease_expires_at: null,
+						last_error: null,
+						approved_at: approvedBy ? now : null,
+						approved_by: approvedBy,
+						completed_at: null,
+						created_at: new Date(now.getTime() + index),
+						updated_at: now
+					});
+				}
 			}
 			return {
 				status: "committed",
@@ -1282,6 +1344,13 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			});
 			return Boolean(decision) && currentOccurrence;
 		}
+		if (identity.kind === "activity") return await ops.findOne("work_item_activity", {
+			work_item_id: identity.sourceId,
+			participant_id: userId,
+			org_id: orgId,
+			factory_project_id: factoryProjectId,
+			occurrence: identity.occurrence
+		}) !== null;
 		if (identity.occurrence !== 0) return false;
 		if (!await ops.findOne("work_item_comment_mentions", {
 			comment_id: identity.sourceId,
@@ -1364,7 +1433,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	}
 	async failDeferredDecision(input) {
 		const row = await this.#failLease("factory_deferred_decisions", input);
-		return row ? toDeferredDecision(row) : null;
+		if (!row) return null;
+		const record = toDeferredDecision(row);
+		if (input.terminal) this.#attentionChanged(record);
+		return record;
 	}
 	/** Park a claimed effect for human approval; the dispatcher never claims `proposed` rows. */
 	async proposeDeferredDecision(identity, now) {
@@ -1384,7 +1456,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				updated_at: now
 			};
 		});
-		return proposed && row ? toDeferredDecision(row) : null;
+		if (!proposed || !row) return null;
+		const record = toDeferredDecision(row);
+		this.#attentionChanged(record);
+		return record;
 	}
 	/**
 	* Release an approved effect back to the dispatcher; only `proposed` rows are
@@ -1393,7 +1468,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	* leaving its follow-up work parked for re-approval.
 	*/
 	async approveDeferredDecision(orgId, factoryProjectId, decisionId, now, approvedBy) {
-		return this.storage.withTransaction(async (ops) => {
+		const approved = await this.storage.withTransaction(async (ops) => {
 			let settled = false;
 			const row = await ops.updateAtomic("factory_deferred_decisions", {
 				id: decisionId,
@@ -1419,6 +1494,8 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			}, (current) => current.autonomy_armed_at ? null : { autonomy_armed_at: now });
 			return record;
 		});
+		if (approved) this.#attentionChanged(approved);
+		return approved;
 	}
 	/** Retire a proposal nobody wants: `dismissed` is terminal, so the run never happens. */
 	async dismissDeferredDecision(orgId, factoryProjectId, decisionId, now) {
@@ -1477,10 +1554,13 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			settled = true;
 			return patch;
 		});
-		return settled && row ? toDeferredDecision(row) : null;
+		if (!settled || !row) return null;
+		const record = toDeferredDecision(row);
+		this.#attentionChanged(record);
+		return record;
 	}
 	async #resolveFailedDecision({ orgId, factoryProjectId, decisionId, status, now }) {
-		return this.storage.withTransaction(async (ops) => {
+		const resolved = await this.storage.withTransaction(async (ops) => {
 			let resolved = false;
 			const row = await ops.updateAtomic("factory_deferred_decisions", {
 				id: decisionId,
@@ -1504,6 +1584,8 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			});
 			return toDeferredDecision(row);
 		});
+		if (resolved) this.#attentionChanged(resolved);
+		return resolved;
 	}
 	async supersedeTerminalDecisionsForWorkItem(input) {
 		await this.supersedeDecisionsForWorkItem(input);
@@ -1560,7 +1642,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	}
 	/** Requeue the same idempotent terminal effect; non-failed decisions are never rerun. */
 	async retryDeferredDecision(orgId, factoryProjectId, decisionId, now) {
-		return this.storage.withTransaction(async (ops) => {
+		const retriedRecord = await this.storage.withTransaction(async (ops) => {
 			let retried = false;
 			const row = await ops.updateAtomic("factory_deferred_decisions", {
 				id: decisionId,
@@ -1592,6 +1674,8 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			});
 			return toDeferredDecision(row);
 		});
+		if (retriedRecord) this.#attentionChanged(retriedRecord);
+		return retriedRecord;
 	}
 	/** Resolve exact active agent authority; partial session matches never authorize. */
 	async findActiveRunBinding(address) {
@@ -1769,10 +1853,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				id: input.workItem.id,
 				org_id: input.orgId,
 				factory_project_id: input.factoryProjectId
-			}) : sourceKey(create.externalSource) ? await ops.findOne("work_items", {
+			}) : externalSourceKey(create.externalSource) ? await ops.findOne("work_items", {
 				org_id: input.orgId,
 				factory_project_id: input.factoryProjectId,
-				source_key: sourceKey(create.externalSource)
+				source_key: externalSourceKey(create.externalSource)
 			}) : null;
 			let item;
 			if (row) {
@@ -1794,7 +1878,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					created_by: input.userId,
 					factory_project_id: input.factoryProjectId,
 					external_source: create.externalSource ?? null,
-					source_key: sourceKey(create.externalSource),
+					source_key: externalSourceKey(create.externalSource),
 					parent_work_item_id: create.parentWorkItemId ?? null,
 					title: create.title,
 					stages: create.stages ?? [],
@@ -1811,6 +1895,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				const armedRow = await ops.updateAtomic("work_items", { id: item.id }, (current) => current.autonomy_armed_at ? null : { autonomy_armed_at: now });
 				if (armedRow) item = toRow(armedRow);
 			}
+			if (input.preapprovePlans && !item.plansPreapprovedAt) {
+				const grantedRow = await ops.updateAtomic("work_items", { id: item.id }, (current) => current.plans_preapproved_at ? null : { plans_preapproved_at: now });
+				if (grantedRow) item = toRow(grantedRow);
+			}
 			await ops.updateMany("factory_run_bindings", {
 				org_id: input.orgId,
 				factory_project_id: input.factoryProjectId,
@@ -1857,6 +1945,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				lease_owner: null,
 				lease_expires_at: null,
 				last_error: null,
+				failure_code: null,
 				completed_at: null,
 				created_at: now,
 				updated_at: now
@@ -1904,7 +1993,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		}
 	}
 	async #upsert({ orgId, userId, factoryProjectId, input, reuseMode = "update" }, ops) {
-		const key = sourceKey(input.externalSource);
+		const key = externalSourceKey(input.externalSource);
 		const reuse = async () => {
 			if (!key) return null;
 			const existing = await ops.findOne("work_items", {
@@ -2101,6 +2190,13 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			if (comments.length < ATTENTION_RECEIPT_QUERY_BATCH_SIZE) break;
 		}
 		await ops.deleteMany("work_item_comment_mentions", where);
+		await ops.deleteMany("factory_attention_receipts", {
+			org_id: orgId,
+			factory_project_id: factoryProjectId,
+			kind: "activity",
+			source_id: workItemId
+		});
+		await ops.deleteMany("work_item_activity", where);
 	}
 	async delete({ orgId, id }) {
 		const candidate = await this.#db.findOne("work_items", {
@@ -2108,7 +2204,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			id
 		});
 		if (!candidate) return null;
-		return this.#withProjectRelationTransaction(orgId, candidate.factory_project_id, async (ops) => {
+		const removed = await this.#withProjectRelationTransaction(orgId, candidate.factory_project_id, async (ops) => {
 			const existing = await ops.findOne("work_items", {
 				org_id: orgId,
 				id
@@ -2137,9 +2233,11 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			});
 			return toWorkItem(existing);
 		});
+		if (removed) this.#attentionChanged(removed);
+		return removed;
 	}
 };
 //#endregion
-export { FACTORY_PULL_REQUEST_RECONCILIATION_KEY, FACTORY_RULE_MATERIALIZATION_KEY, WORK_ITEMS_SCHEMA, WorkItemRelationError, WorkItemsStorage, applyStageTransition, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
+export { FACTORY_PULL_REQUEST_RECONCILIATION_KEY, FACTORY_RULE_MATERIALIZATION_KEY, WORK_ITEMS_SCHEMA, WorkItemRelationError, WorkItemsStorage, applyStageTransition, externalSourceKey, factoryActivityAttentionIdentity, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
 
 //# sourceMappingURL=base.js.map