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

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
@@ -70,6 +70,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}`;
 }
@@ -644,7 +652,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) {
@@ -695,15 +703,25 @@ function toPendingStart(row) {
 	};
 }
 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();
 	}
@@ -1282,6 +1300,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 +1389,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 +1412,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 +1424,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 +1450,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 +1510,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 +1540,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 +1598,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 +1630,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) {
@@ -2101,6 +2141,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 +2155,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 +2184,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, factoryActivityAttentionIdentity, factoryAttentionKey, factoryDecisionAttentionIdentity, factoryDecisionHash, factoryMentionAttentionIdentity, isAgentActor, stampSessions, validateParentRelation };
 
 //# sourceMappingURL=base.js.map