@mastra/factory 0.15.0 → 0.16.0-alpha.5

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

@@ -42,6 +42,8 @@ const FACTORY_DISPATCH_FAILURE_CODES = [
 	"source_repository_missing",
 	"unsupported_provider_item",
 	"notification_delivery_failed",
+	"run_terminal_event_missing",
+	"skill_delivery_ambiguous",
 	"run_overdue",
 	"repository_git_missing",
 	"repository_egress_blocked",
@@ -135,6 +137,10 @@ const WORK_ITEMS_SCHEMA = {
 			type: "text",
 			nullable: true
 		},
+		claim_key: {
+			type: "text",
+			nullable: true
+		},
 		parent_work_item_id: {
 			type: "text",
 			nullable: true
@@ -182,6 +188,9 @@ const WORK_ITEMS_SCHEMA = {
 	uniqueIndexes: [{
 		name: "work_items_project_source_key_unique",
 		columns: ["factory_project_id", "source_key"]
+	}, {
+		name: "work_items_org_claim_key_unique",
+		columns: ["org_id", "claim_key"]
 	}],
 	indexes: [
 		{
@@ -224,6 +233,7 @@ function toWorkItem(row) {
 		factoryProjectId: String(row.factory_project_id),
 		board: row.board ?? null,
 		externalSource: row.external_source,
+		claimKey: row.claim_key ?? null,
 		parentWorkItemId: row.parent_work_item_id,
 		title: row.title,
 		stages: row.stages,
@@ -248,6 +258,7 @@ function patchColumns(changes) {
 		...changes.board !== void 0 ? { board: changes.board } : {},
 		...changes.parentWorkItemId !== void 0 ? { parent_work_item_id: changes.parentWorkItemId } : {},
 		...changes.title !== void 0 ? { title: changes.title } : {},
+		...changes.claimKey !== void 0 ? { claim_key: changes.claimKey } : {},
 		...changes.stages !== void 0 ? { stages: changes.stages } : {},
 		...changes.stageHistory !== void 0 ? { stage_history: changes.stageHistory } : {},
 		...changes.sessions !== void 0 ? { sessions: changes.sessions } : {},
@@ -274,6 +285,15 @@ function priorState(row) {
 var WorkItemRelationError = class extends Error {
 	code = "invalid_work_item_relation";
 };
+/** Another card in the org already holds the claim the write asked for. */
+var WorkItemClaimConflictError = class extends Error {
+	claimant;
+	code = "work_item_claim_conflict";
+	constructor(claimant) {
+		super(`Work item claim is held by ${claimant.id} in project ${claimant.factoryProjectId}`);
+		this.claimant = claimant;
+	}
+};
 var WorkItemUpdateConflictError = class extends Error {
 	reason;
 	code = "work_item_update_conflict";
@@ -842,6 +862,39 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	useTerminalPhasePredicate(isTerminal) {
 		this.#isTerminal = isTerminal;
 	}
+	/**
+	* The claim a new row may carry: the requested key, unless the row is born
+	* finished, in which case there is nothing to own and holding the key would
+	* only block the next project from filing the record.
+	*/
+	#claimForNewRow(input, stages) {
+		if (!input.claimKey) return null;
+		const shape = {
+			board: input.board ?? null,
+			externalSource: input.externalSource ?? null,
+			stages
+		};
+		return this.#isTerminal(shape) ? null : input.claimKey;
+	}
+	/** The claim an existing unclaimed row adopts from `input`, if it is not finished after `patch`. */
+	#claimToAdopt(current, patch, input) {
+		if (!input.claimKey || current.claim_key !== null) return null;
+		return this.#isTerminal(toWorkItem({
+			...current,
+			...patch
+		})) ? null : input.claimKey;
+	}
+	/** Drop the org-wide claim from a patch that moves the card into a terminal phase. */
+	#releaseClaimIfFinished(current, patch) {
+		if (patch.stages === void 0 || current.claim_key === null) return patch;
+		return this.#isTerminal(toWorkItem({
+			...current,
+			...patch
+		})) ? {
+			...patch,
+			claim_key: null
+		} : patch;
+	}
 	async init() {
 		await this.ensureCollections([
 			WORK_ITEMS_SCHEMA,
@@ -1116,6 +1169,56 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 		const rows = await this.#db.findMany("work_items", { source_key: externalSourceKey(source) });
 		return rows.length === 1 ? toWorkItem(rows[0]) : null;
 	}
+	/**
+	* Every card in the org that was linked from one external source, across
+	* Factory projects. Source keys are unique per project, not per org, so an
+	* issue whose routing moved between projects can own one card in each; intake
+	* consults this before creating another.
+	*/
+	async listBySource({ orgId, source }) {
+		return (await this.#db.findMany("work_items", {
+			org_id: orgId,
+			source_key: externalSourceKey(source)
+		})).map(toWorkItem);
+	}
+	/** The one card in the org holding `claimKey`, if any. */
+	async getByClaimKey({ orgId, claimKey }) {
+		const row = await this.#db.findOne("work_items", {
+			org_id: orgId,
+			claim_key: claimKey
+		});
+		return row ? toWorkItem(row) : null;
+	}
+	/**
+	* Give an unclaimed card the org-wide claim for its record. Returns the card
+	* once claimed, or null when the card already carries a different claim or
+	* another card in the org holds this one.
+	*/
+	async claimWorkItem({ orgId, id, claimKey }) {
+		const heldBy = async () => {
+			const holder = await this.#db.findOne("work_items", {
+				org_id: orgId,
+				claim_key: claimKey
+			});
+			return holder && holder.id !== id ? holder : null;
+		};
+		if (await heldBy()) return null;
+		try {
+			const row = await this.#db.updateAtomic("work_items", {
+				org_id: orgId,
+				id
+			}, async (current) => {
+				if (current.claim_key === claimKey) return {};
+				if (current.claim_key !== null) throw new WorkItemClaimConflictError(toWorkItem(current));
+				return { claim_key: claimKey };
+			});
+			return row ? toWorkItem(row) : null;
+		} catch (error) {
+			if (error instanceof WorkItemClaimConflictError) return null;
+			if (await heldBy()) return null;
+			throw error;
+		}
+	}
 	async getForProject(orgId, factoryProjectId, id) {
 		const row = await this.#db.findOne("work_items", {
 			id,
@@ -1185,6 +1288,10 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					...disarm ? { autonomyArmedAt: null } : {},
 					...accept ? { acceptedAt: now } : {},
 					...classified ? { triageType } : {},
+					...this.#isTerminal({
+						...existing,
+						stages: [input.destinationStage]
+					}) ? { claimKey: null } : {},
 					stages: [input.destinationStage],
 					stageHistory: applyStageTransition(existing.stageHistory, existing.stages, [input.destinationStage], input.actorId, now),
 					revision: existing.revision + 1,
@@ -2056,13 +2163,17 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					org_id: input.orgId,
 					factory_project_id: input.factoryProjectId
 				});
-				if (!bindingRow || !itemRow) throw new Error("Factory start replay references missing state.");
-				return {
+				if (bindingRow && itemRow && bindingRow.status === "active") return {
 					item: toRow(itemRow),
 					binding: toBinding(bindingRow),
 					pendingStart: toPendingStart(prior),
 					replayed: true
 				};
+				await ops.deleteMany("factory_pending_starts", {
+					id: prior.id,
+					org_id: input.orgId,
+					factory_project_id: input.factoryProjectId
+				});
 			}
 			const now = /* @__PURE__ */ new Date();
 			const create = input.workItem.input;
@@ -2078,11 +2189,16 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			let item;
 			if (row) {
 				row = await ops.updateAtomic("work_items", { id: row.id }, (current) => {
-					return applyUpdate({
+					const next = applyUpdate({
 						current,
 						userId: input.userId,
 						input: { sessions: { [input.role]: input.session } }
 					});
+					const adopt = this.#claimToAdopt(current, next, create);
+					return adopt ? {
+						...next,
+						claim_key: adopt
+					} : next;
 				});
 				item = toRow(row);
 			} else {
@@ -2097,6 +2213,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					board: create.board ?? null,
 					external_source: create.externalSource ?? null,
 					source_key: externalSourceKey(create.externalSource),
+					claim_key: this.#claimForNewRow(create, create.stages ?? []),
 					parent_work_item_id: create.parentWorkItemId ?? null,
 					title: create.title,
 					stages: create.stages ?? [],
@@ -2172,6 +2289,12 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			return await prepare();
 		} catch (error) {
 			if (!(error instanceof UniqueViolationError)) throw error;
+			const claimKey = input.workItem.input.claimKey;
+			const claimant = claimKey ? await this.#db.findOne("work_items", {
+				org_id: input.orgId,
+				claim_key: claimKey
+			}) : null;
+			if (claimant && claimant.factory_project_id !== input.factoryProjectId) throw new WorkItemClaimConflictError(toWorkItem(claimant));
 			lastError = error;
 		}
 		throw lastError;
@@ -2199,6 +2322,17 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 			return await execute();
 		} catch (error) {
 			if (!(error instanceof UniqueViolationError)) throw error;
+			const claimKey = params.input.claimKey;
+			const claimant = claimKey ? await this.#db.findOne("work_items", {
+				org_id: params.orgId,
+				claim_key: claimKey
+			}) : null;
+			if (claimant && claimant.factory_project_id !== params.factoryProjectId) throw new WorkItemClaimConflictError(toWorkItem(claimant));
+			if (claimant) return {
+				item: toWorkItem(claimant),
+				created: false,
+				previous: priorState(claimant)
+			};
 			return execute();
 		}
 	}
@@ -2212,11 +2346,18 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				source_key: key
 			});
 			if (!existing) return null;
-			if (reuseMode === "preserve") return {
-				created: false,
-				item: toWorkItem(existing),
-				previous: priorState(existing)
-			};
+			if (reuseMode === "preserve") {
+				const adopt = this.#claimToAdopt(existing, {}, input);
+				return {
+					created: false,
+					item: toWorkItem((adopt ? await ops.updateAtomic("work_items", {
+						org_id: orgId,
+						factory_project_id: factoryProjectId,
+						source_key: key
+					}, (current) => current.claim_key === null ? { claim_key: adopt } : null) : null) ?? existing),
+					previous: priorState(existing)
+				};
+			}
 			let previous = emptyPrior();
 			const updated = await ops.updateAtomic("work_items", {
 				org_id: orgId,
@@ -2234,7 +2375,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 					metadata: input.metadata
 				} : fullPatch;
 				if (patch.parentWorkItemId !== void 0) validateParentRelation(await this.#listWithOps(ops, orgId, factoryProjectId), current.id, patch.parentWorkItemId);
-				return {
+				const next = {
 					board: reuseMode === "non-stage" ? current.board : input.board ?? current.board ?? null,
 					external_source: input.externalSource ?? null,
 					...applyUpdate({
@@ -2243,6 +2384,11 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 						input: patch
 					})
 				};
+				const adopt = this.#claimToAdopt(current, next, input);
+				return adopt ? {
+					...next,
+					claim_key: adopt
+				} : next;
 			});
 			return updated ? {
 				item: toWorkItem(updated),
@@ -2262,6 +2408,7 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				board: input.board ?? null,
 				external_source: input.externalSource ?? null,
 				source_key: key,
+				claim_key: this.#claimForNewRow(input, stages),
 				parent_work_item_id: input.parentWorkItemId ?? null,
 				title: input.title,
 				stages,
@@ -2314,11 +2461,11 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 				if (expectedBoard !== void 0 && current.board !== expectedBoard) throw new WorkItemUpdateConflictError("board");
 				if (expectedRevision !== void 0 && current.revision !== expectedRevision) throw new WorkItemUpdateConflictError("revision");
 				if (patch.parentWorkItemId !== void 0) validateParentRelation(await this.#listWithOps(ops, orgId, current.factory_project_id), current.id, patch.parentWorkItemId);
-				return applyUpdate({
+				return this.#releaseClaimIfFinished(current, applyUpdate({
 					current,
 					userId,
 					input: patch
-				});
+				}));
 			});
 			return row ? {
 				item: toWorkItem(row),
@@ -2440,6 +2587,6 @@ var WorkItemsStorage = class extends FactoryStorageDomain {
 	}
 };
 //#endregion
-export { FACTORY_PULL_REQUEST_RECONCILIATION_KEY, FACTORY_RULE_MATERIALIZATION_KEY, WORK_ITEMS_SCHEMA, WorkItemRelationError, WorkItemUpdateConflictError, WorkItemsStorage, applyStageTransition, externalSourceKey, factoryActivityAttentionIdentity, factoryAgentWaitingAttentionIdentity, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, factoryProposalAttentionIdentity, factorySupervisorFindingAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
+export { 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, factorySupervisorFindingAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
 
 //# sourceMappingURL=base.js.map