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

package/dist/factory.js

@@ -1,7 +1,7 @@
 import { timedPhase } from "./timing.js";
-import { buildAuthDisabledRoutes, buildAuthRoutes, createFactoryAuthGate, createFactoryRouteAuth, factoryUserOrgId, getFactoryAuthOrgId, getFactoryAuthUserFromContext, getFactoryAuthUserId } from "./auth.js";
+import { buildAuthDisabledRoutes, buildAuthRoutes, createFactoryAuthGate, createFactoryInteractiveMarker, createFactoryRouteAuth, factoryUserOrgId, getFactoryAuthOrgId, getFactoryAuthUserFromContext, getFactoryAuthUserId } from "./auth.js";
 import { assertFactoryConfigVersion } from "./rules/validation.js";
-import { isTerminalWorkItem } from "./boards/semantics.js";
+import { boardForWorkItem, isTerminalWorkItem } from "./boards/semantics.js";
 import { createBoardRegistry } from "./boards/registry.js";
 import "./boards/index.js";
 import { isValidGitRef } from "./sandbox/git-ref.js";
@@ -31,22 +31,29 @@ import { PlatformIncidentioIntegration } from "./integrations/platform/incidenti
 import { PlatformJiraIntegration } from "./integrations/platform/jira/integration.js";
 import { PlatformLinearIntegration } from "./integrations/platform/linear/integration.js";
 import { FactoryProjectsStorage } from "./storage/domains/projects/base.js";
+import { SourceControlStorage } from "./storage/domains/source-control/base.js";
+import { createSourceControlSessionLookup } from "./session/factory-session.js";
+import { AuditStorage } from "./storage/domains/audit/base.js";
+import { deliverFactorySessionNotification } from "./tools/deliver.js";
+import { createSessionApprovalNotifier, createSessionRunEndReporter, createStageChangeNotifier, createWorkItemRunEndReporter } from "./tools/reports.js";
+import { observeSessionRunEnd } from "./session/run-audit.js";
+import { observeSessionRunReports } from "./tools/run-watch.js";
+import { applyFactoryToolPermissions } from "./tools/permissions.js";
+import { resolveFactoryToolScope } from "./tools/scope.js";
+import { createFactoryTools } from "./tools/index.js";
+import { buildFactoryMcpRoutes, resolveMcpAuthorizationServers } from "./mcp/routes.js";
 import { resolveFactoryCredentialTenant } from "./session/credential-policy.js";
 import { hydrateSessionMemorySettings, layerPersonalMemorySettings } from "./session/memory-settings-hydration.js";
 import { MemorySettingsStorage, factoryMemorySettingsUserId } from "./storage/domains/memory-settings/base.js";
 import { resolveDeploymentModelProviders } from "./routes/config.js";
 import { createCustomProvidersPrimer, registerCustomProvidersSource } from "./routes/custom-provider-source.js";
 import { ProjectRoutes } from "./routes/projects.js";
-import { SourceControlStorage } from "./storage/domains/source-control/base.js";
-import { createSourceControlSessionLookup } from "./session/factory-session.js";
 import { FactoryTransitionService } from "./rules/transition-service.js";
-import { AuditStorage } from "./storage/domains/audit/base.js";
 import { AuditDomain } from "./storage/domains/audit/domain.js";
 import { QueueHealthStorage } from "./storage/domains/queue-health/base.js";
 import { assembleFactoryApiRoutes, buildIntegrationContext } from "./routes/surface.js";
 import { TelemetryRoutes } from "./routes/telemetry.js";
 import { readFactorySessionScope, resolveFactorySessionAddress } from "./rules/binding-context.js";
-import { observeSessionRunEnd } from "./session/run-audit.js";
 import { FactoryFeedReader } from "./storage/domains/comments/feed-context.js";
 import { FactoryDecisionDispatcher } from "./rules/dispatcher.js";
 import { FactoryPhaseStateProcessor, reportMemorySettingsUnavailable } from "./rules/processor.js";
@@ -59,6 +66,7 @@ import { handleServerError } from "./server-error.js";
 import { hydrateSessionDefaultModel } from "./session/default-model-hydration.js";
 import { observeSessionFirstMessage } from "./session/first-message-capture.js";
 import { LiveSessions } from "./session/live-sessions.js";
+import { FactoryReplyCredentialProcessor } from "./session/reply-credential-processor.js";
 import { createSourceControlTools } from "./session/source-control-tools.js";
 import { readFactoryThinkingLevel, resolveFactoryThinkingLevel } from "./session/thinking-default.js";
 import { observeSessionThreadTitle } from "./session/thread-title-mirror.js";
@@ -72,6 +80,7 @@ import { CustomProvidersStorage } from "./storage/domains/custom-providers/base.
 import { FilesystemStorage } from "./storage/domains/filesystem/base.js";
 import { IntegrationStorage } from "./storage/domains/integrations/base.js";
 import { ModelDefaultsStorage } from "./storage/domains/model-defaults/base.js";
+import { FactorySessionsStateProcessor } from "./tools/sessions-state-processor.js";
 import { hasAuthInit, isUserProvider } from "@mastra/core/server";
 import { MastraAuthStudio } from "@mastra/auth-studio";
 import { prepareAgentControllerMount } from "@mastra/code-sdk";
@@ -199,6 +208,7 @@ var MastraFactory = class {
 	#configVersion;
 	#prepared;
 	#dispatcher;
+	#startCoordinator;
 	#factoryProcessor;
 	/** Environment builds through the sandbox's `builds`; undefined when the sandbox has none. */
 	#environmentBuilds;
@@ -243,6 +253,8 @@ var MastraFactory = class {
 		const secretEncryption = this.#config.secretEncryption ?? createPlaintextFactorySecretEncryption();
 		const deploymentModelProviders = resolveDeploymentModelProviders(this.#config.deploymentModelProviders);
 		const routeAuth = createFactoryRouteAuth(auth);
+		const mcpConfig = this.#config.mcp;
+		const mcpAuthorizationServers = mcpConfig === false || mcpConfig?.enabled === false ? [] : resolveMcpAuthorizationServers(auth?.name, mcpConfig);
 		const integrations = [...this.#config.integrations ?? []];
 		const explicitGithub = integrations.some((integration) => integration.id === "github");
 		let installedPlatformGithub = false;
@@ -349,7 +361,8 @@ var MastraFactory = class {
 			});
 			if (integration.versionControl && integrationSourceControl) integration.versionControl.initialize({ storage: integrationSourceControl });
 		}
-		const sourceControlSessions = createSourceControlSessionLookup([.../* @__PURE__ */ new Set(["github", ...integrations.filter((integration) => integration.versionControl).map(({ id }) => id)])].map((id) => sourceControlStorage.forIntegration(id)));
+		const sourceControlHandles = [.../* @__PURE__ */ new Set(["github", ...integrations.filter((integration) => integration.versionControl).map(({ id }) => id)])].map((id) => sourceControlStorage.forIntegration(id));
+		const sourceControlSessions = createSourceControlSessionLookup(sourceControlHandles);
 		if (auth) registerTenantCredentialResolver(modelCredentialsStorage, deploymentModelProviders, (requestContext) => resolveFactoryCredentialTenant(requestContext, {
 			projects: factoryProjectsStorage,
 			sourceControl: sourceControlSessions,
@@ -407,6 +420,28 @@ var MastraFactory = class {
 			},
 			...retireTerminalSessions ? { releaseSandboxes: retireTerminalSessions } : {}
 		}) : void 0;
+		const factoryMessaging = async () => {
+			const controller = this.#prepared?.base.controller;
+			const notifications = await controller?.getMastra()?.getStorage()?.getStore("notifications");
+			return controller && notifications ? {
+				controller,
+				notifications
+			} : null;
+		};
+		const factoryReporting = async () => {
+			const messaging = await factoryMessaging();
+			if (!messaging) return null;
+			return {
+				notifications: messaging.notifications,
+				deliver: (target, notification, options) => deliverFactorySessionNotification({
+					controller: messaging.controller,
+					sessions: sourceControlSessions,
+					target,
+					notification,
+					...options
+				})
+			};
+		};
 		const onTerminalStage = terminalCleanup ? (args) => terminalCleanup({
 			orgId: args.orgId,
 			factoryProjectId: args.factoryProjectId,
@@ -426,9 +461,37 @@ var MastraFactory = class {
 					id: factoryProjectId
 				}))?.autoApprovePlans ?? false;
 			},
+			selfReviewEnabled: async ({ orgId, factoryProjectId }) => {
+				await factoryProjectsStorage.ensureReady();
+				return (await factoryProjectsStorage.get({
+					orgId,
+					id: factoryProjectId
+				}))?.selfReviewEnabled ?? false;
+			},
 			...onTerminalStage ? { onTerminalStage } : {},
+			onTransitionCommitted: createStageChangeNotifier({
+				workItems: workItemsStorage,
+				reporting: factoryReporting
+			}),
 			...githubIntegration ? { onAccepted: (args) => reconcileGithubAcceptanceLabels(githubIntegration, sourceControlStorage.forIntegration(githubIntegration.id), args) } : {}
 		}) : void 0;
+		const factoryToolDeps = {
+			workItems: workItemsStorage,
+			comments: workItemCommentsStorage,
+			feed: commentsDomain,
+			audit: auditStorage,
+			boards: this.#boards,
+			transitionService,
+			auditRecorder: auditDomain,
+			sessions: sourceControlSessions,
+			sessionModel: {
+				projects: factoryProjectsStorage,
+				modelDefaults: modelDefaultsStorage
+			},
+			intake: intakeStorage,
+			startCoordinator: () => this.#startCoordinator,
+			messaging: factoryMessaging
+		};
 		this.#environmentBuilds = this.#sandbox?.builds ? new EnvironmentBuildRunner({
 			sandbox: this.#sandbox,
 			projects: factoryProjectsStorage,
@@ -622,9 +685,24 @@ var MastraFactory = class {
 			...this.#config.knowledge ? { knowledge: this.#config.knowledge } : {},
 			inputProcessors: async ({ requestContext }) => {
 				const environmentProcessor = new FactoryEnvironmentStateProcessor();
+				const sessionsProcessor = workItemsStorage && storage.isDomainReady("source-control") ? [new FactorySessionsStateProcessor({
+					authEnabled: routeAuth.enabled(),
+					sessions: sourceControlSessions,
+					projects: factoryProjectsStorage,
+					workItems: workItemsStorage,
+					boards: this.#boards,
+					isRunning: async (sessionId) => {
+						return (await this.#prepared?.base.controller.getSessionByResource(sessionId))?.stream.isActive() ?? false;
+					},
+					notifications: async () => (await factoryMessaging())?.notifications ?? null
+				})] : [];
 				if (factoryProcessor) {
 					await factoryProcessor.prepareMemorySettings(requestContext);
-					return [environmentProcessor, factoryProcessor];
+					return [
+						environmentProcessor,
+						...sessionsProcessor,
+						factoryProcessor
+					];
 				}
 				try {
 					await loadMemorySettings({
@@ -634,8 +712,9 @@ var MastraFactory = class {
 				} catch (error) {
 					reportMemorySettingsUnavailable(requestContext, error instanceof Error ? error.message : String(error));
 				}
-				return [environmentProcessor];
+				return [environmentProcessor, ...sessionsProcessor];
 			},
+			outputProcessors: [new FactoryReplyCredentialProcessor()],
 			...vector ? { vector } : {},
 			...toolIntegrations.length > 0 || sourceControlToolProviders.length > 0 || workItemsStorage && transitionService ? { extraTools: async ({ requestContext }) => {
 				const tools = {};
@@ -651,7 +730,55 @@ var MastraFactory = class {
 				if (storage.isDomainReady("source-control")) mergeTools("source-control", await createSourceControlTools({
 					requestContext,
 					providers: sourceControlToolProviders,
-					audit: auditDomain
+					audit: auditDomain,
+					testChecksEnabled: async ({ orgId, factoryProjectId }) => {
+						await factoryProjectsStorage.ensureReady();
+						return (await factoryProjectsStorage.get({
+							orgId,
+							id: factoryProjectId
+						}))?.reviewTestChecksEnabled ?? false;
+					},
+					...workItemsStorage ? {
+						selfReviewApprovedHead: async () => {
+							const resolution = await resolveFactorySessionAddress({
+								requestContext,
+								storage: workItemsStorage
+							});
+							if (!resolution) return void 0;
+							const binding = resolution.binding ?? await workItemsStorage.findActiveRunBinding(resolution.address);
+							if (!binding) return void 0;
+							const item = await workItemsStorage.get({
+								orgId: binding.orgId,
+								id: binding.workItemId
+							});
+							if (!item || boardForWorkItem(item) !== "work") return void 0;
+							if (typeof item.metadata?.openPullRequestNumber === "number") return void 0;
+							if (item.metadata?.selfReviewVerdict === void 0) {
+								await factoryProjectsStorage.ensureReady();
+								if (!(await factoryProjectsStorage.get({
+									orgId: binding.orgId,
+									id: binding.factoryProjectId
+								}))?.selfReviewEnabled) return void 0;
+							}
+							const sha = item.metadata?.selfReviewedHeadSha;
+							return item.metadata?.selfReviewVerdict === "approve" && typeof sha === "string" ? sha : null;
+						},
+						boundChangeRequestId: async () => {
+							const resolution = await resolveFactorySessionAddress({
+								requestContext,
+								storage: workItemsStorage
+							});
+							if (!resolution) return void 0;
+							const binding = resolution.binding ?? await workItemsStorage.findActiveRunBinding(resolution.address);
+							if (binding?.role !== "review") return void 0;
+							const source = (await workItemsStorage.get({
+								orgId: binding.orgId,
+								id: binding.workItemId
+							}))?.externalSource;
+							if (!source || !/pull-request|merge-request/.test(source.type)) return void 0;
+							return /(\d+)$/.exec(source.externalId)?.[1];
+						}
+					} : {}
 				}));
 				if (workItemsStorage && transitionService) {
 					mergeTools("factory", await createFactoryTransitionTools({
@@ -668,6 +795,19 @@ var MastraFactory = class {
 						uiOrigin: reviewSourceUiOrigin,
 						...storage.isDomainReady("source-control") ? { sessions: sourceControlSessions } : {}
 					}));
+					if (storage.isDomainReady("source-control")) {
+						const scope = await resolveFactoryToolScope({
+							requestContext,
+							authEnabled: routeAuth.enabled(),
+							sessions: sourceControlSessions,
+							projects: factoryProjectsStorage,
+							workItems: workItemsStorage
+						});
+						if (scope) mergeTools("factory-tools", createFactoryTools({
+							scope,
+							deps: factoryToolDeps
+						}));
+					}
 				}
 				for (const { integration, ready, ensureReady } of toolIntegrations) {
 					if (!ready && ensureReady) try {
@@ -718,6 +858,14 @@ var MastraFactory = class {
 			} : {},
 			buildApiRoutes: ({ controller, authStorage }) => [
 				...auth ? buildAuthRoutes(auth, { publicUrl: publicOrigin }) : buildAuthDisabledRoutes(),
+				...auth && mcpAuthorizationServers.length > 0 ? buildFactoryMcpRoutes({
+					auth: routeAuth,
+					publicOrigin,
+					authorizationServers: mcpAuthorizationServers,
+					projects: factoryProjectsStorage,
+					sourceControls: sourceControlHandles,
+					tools: factoryToolDeps
+				}) : [],
 				...new TelemetryRoutes({
 					auth: routeAuth,
 					providerName: auth?.name,
@@ -751,7 +899,8 @@ var MastraFactory = class {
 					configVersion,
 					boardRegistry: this.#boards,
 					factoryTransitionService: transitionService,
-					onFactoryRuntime: ({ transitionService: runtimeTransitionService, prepareBinding }) => {
+					onFactoryRuntime: ({ transitionService: runtimeTransitionService, startCoordinator, prepareBinding }) => {
+						this.#startCoordinator = startCoordinator;
 						this.#dispatcher ??= new FactoryDecisionDispatcher({
 							controller,
 							transitionService: runtimeTransitionService,
@@ -787,6 +936,10 @@ var MastraFactory = class {
 								tenant,
 								credentials: modelCredentialsStorage
 							}),
+							reportRunEnd: createWorkItemRunEndReporter({
+								workItems: storage.getDomain("work-items"),
+								reporting: factoryReporting
+							}),
 							resolveLinkedWorkItemParentId: async ({ orgId, factoryProjectId, decision }) => {
 								if (decision.source !== "github-pr") return null;
 								const repositoryId = decision.metadata?.githubRepositoryId;
@@ -819,11 +972,15 @@ var MastraFactory = class {
 				const uiDist = resolveUiDistDir();
 				const spa = uiDist ? [createSpaStaticMiddleware(uiDist)] : [];
 				if (!auth) return {
-					middleware: [createCustomProvidersPrimer({
-						auth: routeAuth,
-						storage: customProvidersStorage,
-						authEnabled: false
-					}), ...spa],
+					middleware: [
+						createFactoryInteractiveMarker(),
+						createCustomProvidersPrimer({
+							auth: routeAuth,
+							storage: customProvidersStorage,
+							authEnabled: false
+						}),
+						...spa
+					],
 					...cors,
 					...onError
 				};
@@ -847,6 +1004,13 @@ var MastraFactory = class {
 				};
 			}
 		}));
+		const sessionReports = {
+			sessions: sourceControlSessions,
+			workItems: storage.getDomain("work-items"),
+			reporting: factoryReporting
+		};
+		const reportSessionRunEnd = createSessionRunEndReporter(sessionReports);
+		const notifyApprovalRequested = createSessionApprovalNotifier(sessionReports);
 		prepared.base.controller.onSessionCreated((session) => {
 			observeSessionFilesystem(session, {
 				filesystem: filesystemStorage,
@@ -856,6 +1020,16 @@ var MastraFactory = class {
 			observeSessionFirstExec(session, { sourceControl: { sessions: sourceControlSessions } });
 			observeSessionThreadTitle(session, { sourceControl: { sessions: sourceControlSessions } });
 			observeSessionRunEnd(session, { audit: auditDomain });
+			observeSessionRunReports(session, (outcome, since) => reportSessionRunEnd({
+				sessionId: session.identity.getResourceId(),
+				threadId: session.thread.getId(),
+				outcome,
+				since
+			}), { onApprovalRequested: (request) => notifyApprovalRequested({
+				sessionId: session.identity.getResourceId(),
+				threadId: session.thread.getId(),
+				...request
+			}) });
 		});
 		prepared.base.controller.onSessionCreated((session) => hydrateSessionMemorySettings(session, { sourceControl: { sessions: sourceControlSessions } }), { blocking: true });
 		prepared.base.controller.onSessionCreated((session) => hydrateSessionDefaultModel(session, {
@@ -864,6 +1038,7 @@ var MastraFactory = class {
 			workItems: workItemsStorage,
 			modelDefaults: modelDefaultsStorage
 		}), { blocking: true });
+		prepared.base.controller.onSessionCreated((session) => applyFactoryToolPermissions(session, { sessions: sourceControlSessions }), { blocking: true });
 		this.#prepared = prepared;
 		this.#factoryProcessor = factoryProcessor;
 		const channelRegistrations = integrationRegistrations.filter(({ integration, ready }) => ready && integration.channels);