@mastra/factory 0.15.1-alpha.0 → 0.16.0-alpha.11

package/dist/factory.js

@@ -1,35 +1,39 @@
 import { timedPhase } from "./timing.js";
 import { buildAuthRoutes, createFactoryAuthGate, createFactoryRouteAuth, getFactoryAuthOrgId, getFactoryAuthUserFromContext, getFactoryAuthUserId } from "./auth.js";
 import { assertFactoryConfigVersion } from "./rules/validation.js";
-import { workItemPhaseSemantics } from "./boards/semantics.js";
+import { isTerminalWorkItem } from "./boards/semantics.js";
 import { createBoardRegistry } from "./boards/registry.js";
 import "./boards/index.js";
 import { touchFeed } from "./feed-events.js";
 import { reconcileGithubAcceptanceLabels } from "./integrations/github/acceptance-labels.js";
 import { recordFactoryPullRequestProvenance, resolveFactoryPullRequestParentWorkItemId } from "./integrations/github/provenance.js";
-import { isValidGitRef } from "./integrations/github/sandbox.js";
+import { PlatformApiClient, platformApiClientConfigFromEnv } from "./integrations/platform/api-client.js";
+import { buildPlatformConnectRoutes } from "./integrations/platform/connect/routes.js";
 import { WorkItemsStorage } from "./storage/domains/work-items/base.js";
 import { IntakeStorage } from "./storage/domains/intake/base.js";
+import { isValidGitRef } from "./sandbox/git-ref.js";
+import { createPlaintextFactorySecretEncryption } from "./secret-encryption.js";
+import { ModelCredentialsStorage } from "./storage/domains/credentials/base.js";
+import { createTenantCredentialPrimer, primeTenantCredentials, registerTenantCredentialResolver } from "./routes/tenant-credentials.js";
 import { PlatformGithubIntegration } from "./integrations/platform/github/integration.js";
+import { PlatformGitLabIntegration } from "./integrations/platform/gitlab/integration.js";
 import { PlatformIncidentioIntegration } from "./integrations/platform/incidentio/integration.js";
+import { PlatformJiraIntegration } from "./integrations/platform/jira/integration.js";
 import { PlatformLinearIntegration } from "./integrations/platform/linear/integration.js";
 import { createCustomProvidersPrimer, registerCustomProvidersSource } from "./routes/custom-provider-source.js";
 import { ProjectRoutes } from "./routes/projects.js";
 import { MemorySettingsStorage } from "./storage/domains/memory-settings/base.js";
+import { SourceControlStorage } from "./storage/domains/source-control/base.js";
 import { hydrateSessionMemorySettings } from "./session/memory-settings-hydration.js";
-import { refreshFactorySessionMemorySettings } from "./session/factory-session.js";
+import { createSourceControlSessionLookup, refreshFactorySessionMemorySettings } from "./session/factory-session.js";
 import { FactoryTransitionService } from "./rules/transition-service.js";
-import { SourceControlStorage } from "./storage/domains/source-control/base.js";
 import { AuditStorage } from "./storage/domains/audit/base.js";
 import { AuditDomain } from "./storage/domains/audit/domain.js";
 import { hydrateSessionModelPack } from "./session/model-pack-hydration.js";
 import { observeSessionFirstExec } from "./session/first-exec-capture.js";
 import { observeSessionFilesystem } from "./session/filesystem-capture.js";
-import { createPlaintextFactorySecretEncryption } from "./secret-encryption.js";
-import { ModelCredentialsStorage } from "./storage/domains/credentials/base.js";
 import { hydrateSupervisorSession, parseSupervisorResourceId, resolveSupervisorScope } from "./supervisor/session.js";
 import { FactoryWorkspaceRegistry, createWorkspaceFactory } from "./workspace.js";
-import { createTenantCredentialPrimer, primeTenantCredentials, registerTenantCredentialResolver } from "./routes/tenant-credentials.js";
 import { QueueHealthStorage } from "./storage/domains/queue-health/base.js";
 import { assembleFactoryApiRoutes, buildIntegrationContext } from "./routes/surface.js";
 import { TelemetryRoutes } from "./routes/telemetry.js";
@@ -43,6 +47,7 @@ import { SessionRetirementCoordinator } from "./sandbox/session-retirement.js";
 import { handleServerError } from "./server-error.js";
 import { observeSessionFirstMessage } from "./session/first-message-capture.js";
 import { LiveSessions } from "./session/live-sessions.js";
+import { createSourceControlTools } from "./session/source-control-tools.js";
 import { observeSessionThreadTitle } from "./session/thread-title-mirror.js";
 import { createSpaStaticMiddleware, resolveUiDistDir } from "./spa-static.js";
 import { createStateSigner } from "./state-signing.js";
@@ -58,6 +63,7 @@ import { FactoryProjectsStorage } from "./storage/domains/projects/base.js";
 import { FactorySupervisorHealthWorker } from "./supervisor/health-worker.js";
 import { SUPERVISOR_INSTRUCTIONS } from "./supervisor/instructions.js";
 import { createFactorySupervisorReadTools } from "./supervisor/read-tools.js";
+import { messageWorkerSession } from "./supervisor/session-messaging.js";
 import { createFactorySupervisorWriteTools } from "./supervisor/write-tools.js";
 import { hasAuthInit, isUserProvider } from "@mastra/core/server";
 import { MastraAuthStudio } from "@mastra/auth-studio";
@@ -70,9 +76,9 @@ import { EventEmitterPubSub } from "@mastra/core/events";
 *
 * The consumer's deploy entry constructs deployment-specific config instances
 * (auth adapter, pubsub) and passes them here explicitly. The only provider
-* defaults constructed here are Platform GitHub, incident.io, and Linear
-* integrations when Platform credentials exist and the caller did not provide
-* those integrations.
+* defaults constructed here are Platform GitHub, GitLab, Jira, incident.io, and
+* Linear integrations when Platform credentials exist and the caller did not
+* provide those integrations.
 *
 * `prepare()` resolves feature readiness, threads every dependency explicitly,
 * assembles the web routes/middleware, and returns the constructor args for
@@ -191,7 +197,9 @@ var MastraFactory = class {
 		const integrations = [...this.#config.integrations ?? []];
 		if (hasPlatformCredentials()) {
 			if (!integrations.some((integration) => integration.id === "github")) integrations.push(new PlatformGithubIntegration({ slug: this.#config.platform?.githubAppSlug }));
-			if (process.env.MASTRA_INCIDENT_IO_CONNECTION_ID?.trim() && !integrations.some((integration) => integration.id === "incidentio")) integrations.push(new PlatformIncidentioIntegration());
+			if (!integrations.some((integration) => integration.id === "incidentio")) integrations.push(new PlatformIncidentioIntegration());
+			if (!integrations.some((integration) => integration.id === "jira")) integrations.push(new PlatformJiraIntegration());
+			if (!integrations.some((integration) => integration.id === "gitlab")) integrations.push(new PlatformGitLabIntegration());
 			if (!integrations.some((integration) => integration.id === "linear")) integrations.push(new PlatformLinearIntegration());
 		}
 		const integrationIds = /* @__PURE__ */ new Set();
@@ -204,7 +212,7 @@ var MastraFactory = class {
 		const auditStorage = storage.registerDomain(new AuditStorage());
 		const workItemsStorage = storage.registerDomain(new WorkItemsStorage());
 		workItemsStorage.onAttentionChanged((scope) => touchFeed(eventBus, scope));
-		workItemsStorage.useTerminalPhasePredicate((item) => workItemPhaseSemantics(this.#boards, item)?.kind === "terminal");
+		workItemsStorage.useTerminalPhasePredicate((item) => isTerminalWorkItem(this.#boards, item));
 		const modelCredentialsStorage = storage.registerDomain(new ModelCredentialsStorage(secretEncryption));
 		const modelPacksStorage = storage.registerDomain(new ModelPacksStorage());
 		const memorySettingsStorage = storage.registerDomain(new MemorySettingsStorage());
@@ -273,13 +281,17 @@ var MastraFactory = class {
 			authEnabled: Boolean(auth)
 		});
 		for (const integration of integrations) {
+			const integrationSourceControl = integration.versionControl ? sourceControlStorage.forIntegration(integration.id) : void 0;
 			integration.initialize?.({
 				storage: integrationStorage.forIntegration(integration.id),
 				projects: factoryProjectsStorage,
-				auth: routeAuth
+				auth: routeAuth,
+				intake: intakeStorage,
+				...integrationSourceControl ? { sourceControl: integrationSourceControl } : {}
 			});
-			if (integration.versionControl) integration.versionControl.initialize({ storage: sourceControlStorage.forIntegration(integration.id) });
+			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 integrationRegistrations = integrations.map((integration) => {
 			const requiredDomains = [
 				"integrations",
@@ -298,20 +310,46 @@ var MastraFactory = class {
 		const factoryReady = storage.isDomainReady("projects") && storage.isDomainReady("work-items");
 		const knowledgeEnabled = process.env.MASTRACODE_EXPERIMENTAL_SUBCONSCIOUS === "1";
 		const githubIntegration = integrations.find((integration) => integration.id === "github");
+		const gitlabIntegration = integrations.find((integration) => integration.id === "gitlab" && integration.intake && integration.versionControl);
 		const workItemsReady = storage.isDomainReady("work-items");
 		const sessionRetirement = sandboxConfig && storage.isDomainReady("source-control") ? new SessionRetirementCoordinator({ invalidateSession: (sessionId) => workspaceRegistry.invalidateSession(sessionId) }) : void 0;
-		const retireTerminalSessions = sessionRetirement && githubIntegration && workItemsReady ? async ({ orgId, workItemId }) => sessionRetirement.retireWorkItemSessions({
-			workItems: workItemsStorage,
-			sourceControl: sourceControlStorage.forIntegration(githubIntegration.id),
-			orgId,
-			workItemId
-		}) : void 0;
-		const onTerminalStage = workItemsReady ? createTerminalStageCleanup({
+		const retireTerminalSessions = sessionRetirement && workItemsReady ? async ({ orgId, workItemId }) => {
+			const item = await workItemsStorage.get({
+				orgId,
+				id: workItemId
+			});
+			if (!item) return;
+			const sessionIds = [...new Set(Object.values(item.sessions).map((session) => session.sessionId))];
+			await Promise.all(sessionIds.map(async (sessionId) => {
+				const sourceControl = await sourceControlSessions.getSourceControlBySessionId(sessionId);
+				if (!sourceControl) return;
+				await sessionRetirement.retireSession({
+					sourceControl,
+					orgId,
+					sessionId,
+					deleteSession: false
+				});
+			}));
+		} : void 0;
+		const terminalCleanup = workItemsReady ? createTerminalStageCleanup({
 			workItems: workItemsStorage,
 			reconcileBinding: async (binding) => {
 				await factoryProcessor?.reconcileBinding(binding);
 			},
+			abortSession: async (binding) => {
+				const controller = this.#prepared?.base.controller;
+				if (!controller) return;
+				const session = await controller.getSessionByResource(binding.resourceId);
+				if (session?.stream.isActive()) session.abort();
+			},
 			...retireTerminalSessions ? { releaseSandboxes: retireTerminalSessions } : {}
+		}) : void 0;
+		const onTerminalStage = terminalCleanup ? (args) => terminalCleanup({
+			orgId: args.orgId,
+			factoryProjectId: args.factoryProjectId,
+			workItemId: args.workItemId,
+			revision: args.revision,
+			...args.actor.type === "agent" ? { initiatingBindingId: args.actor.bindingId } : {}
 		}) : retireTerminalSessions;
 		const transitionService = workItemsReady ? new FactoryTransitionService({
 			configVersion,
@@ -333,28 +371,51 @@ var MastraFactory = class {
 			projects: factoryProjectsStorage,
 			sourceControl: sourceControlStorage,
 			versionControlIntegrationIds: integrations.filter((integration) => integration.versionControl).map((integration) => integration.id),
-			...githubIntegration ? { resolveRepository: async ({ integrationId, orgId, installationId, externalId, slug }) => {
-				if (integrationId !== githubIntegration.id) return null;
-				const installation = await githubIntegration.sourceControlStorage.installations.get({
+			...githubIntegration || gitlabIntegration ? { resolveRepository: async ({ integrationId, orgId, userId, installationId, externalId, slug }) => {
+				if (githubIntegration && integrationId === githubIntegration.id) {
+					const installation = await githubIntegration.sourceControlStorage.installations.get({
+						orgId,
+						id: installationId
+					});
+					if (!installation) return null;
+					const selected = (await githubIntegration.listInstallationRepos(Number(installation.externalId))).find((repo) => repo.id.toString() === externalId && repo.fullName === slug);
+					if (!selected) return null;
+					return githubIntegration.sourceControlStorage.repositories.upsert({
+						orgId,
+						input: {
+							installationId,
+							externalId,
+							slug: selected.fullName,
+							defaultBranch: isValidGitRef(selected.defaultBranch) ? selected.defaultBranch : "main",
+							providerMetadata: {
+								private: selected.private,
+								owner: selected.owner
+							}
+						}
+					});
+				}
+				if (!gitlabIntegration?.intake || !gitlabIntegration.versionControl || integrationId !== gitlabIntegration.id) return null;
+				const installation = await sourceControlStorage.forIntegration(gitlabIntegration.id).installations.get({
 					orgId,
 					id: installationId
 				});
 				if (!installation) return null;
-				const selected = (await githubIntegration.listInstallationRepos(Number(installation.externalId))).find((repo) => repo.id.toString() === externalId && repo.fullName === slug);
+				const selected = (await gitlabIntegration.intake.listSources({
+					orgId,
+					userId
+				})).find((source) => source.name === slug && typeof source.metadata?.projectId === "string" && source.metadata.projectId === externalId && source.metadata.connectionId === installation.externalId);
 				if (!selected) return null;
-				return githubIntegration.sourceControlStorage.repositories.upsert({
+				const [repository] = await gitlabIntegration.versionControl.registerRepositories({
 					orgId,
-					input: {
-						installationId,
+					installationId,
+					repositories: [{
 						externalId,
-						slug: selected.fullName,
-						defaultBranch: isValidGitRef(selected.defaultBranch) ? selected.defaultBranch : "main",
-						providerMetadata: {
-							private: selected.private,
-							owner: selected.owner
-						}
-					}
+						slug,
+						defaultBranch: typeof selected.metadata?.defaultBranch === "string" && isValidGitRef(selected.metadata.defaultBranch) ? selected.metadata.defaultBranch : "main",
+						metadata: selected.metadata
+					}]
 				});
+				return repository ?? null;
 			} } : {},
 			...sessionRetirement ? { sessionRetirement } : {},
 			...workItemsReady ? { workItems: workItemsStorage } : {}
@@ -381,13 +442,23 @@ var MastraFactory = class {
 			}
 		})();
 		const toolIntegrations = integrationRegistrations.filter(({ integration }) => integration.agentTools || integration.sessionTools);
+		const sourceControlToolProviders = integrations.flatMap((integration) => integration.versionControl ? [{
+			id: integration.id,
+			versionControl: integration.versionControl,
+			storage: sourceControlStorage.forIntegration(integration.id)
+		}] : []);
 		const prepared = await timedPhase("prepare.controllerMount", () => prepareAgentControllerMount({
 			controllerId: CONTROLLER_ID,
 			coAuthor: { name: "mastra-platform[bot]" },
 			workspace: createWorkspaceFactory({
 				...sandboxConfig ? { sandbox: sandboxConfig } : {},
 				...this.#config.sandboxStart ? { sandboxStart: this.#config.sandboxStart } : {},
-				...githubIntegration ? { github: githubIntegration } : {},
+				sourceControls: integrations.flatMap((integration) => integration.versionControl ? [{
+					id: integration.id,
+					versionControl: integration.versionControl,
+					storage: sourceControlStorage.forIntegration(integration.id),
+					...integration.id === "github" && githubIntegration ? { github: githubIntegration } : {}
+				}] : []),
 				...factoryProjectsStorage ? { projects: factoryProjectsStorage } : {},
 				...workItemsStorage ? { workItems: workItemsStorage } : {},
 				workspaceRegistry
@@ -408,7 +479,7 @@ var MastraFactory = class {
 			...mastraStorageBackend ? { storageBackend: mastraStorageBackend } : {},
 			...factoryProcessor ? { inputProcessors: [factoryProcessor] } : {},
 			...vector ? { vector } : {},
-			...toolIntegrations.length > 0 || workItemsStorage && transitionService ? { extraTools: async ({ requestContext }) => {
+			...toolIntegrations.length > 0 || sourceControlToolProviders.length > 0 || workItemsStorage && transitionService ? { extraTools: async ({ requestContext }) => {
 				const tools = {};
 				const toolOwners = /* @__PURE__ */ new Map();
 				const mergeTools = (ownerId, contributed) => {
@@ -419,12 +490,18 @@ var MastraFactory = class {
 						tools[name] = tool;
 					}
 				};
+				if (storage.isDomainReady("source-control")) mergeTools("source-control", createSourceControlTools({
+					requestContext,
+					providers: sourceControlToolProviders,
+					audit: auditDomain
+				}));
 				if (workItemsStorage && transitionService) {
 					mergeTools("factory", await createFactoryTransitionTools({
 						requestContext,
 						storage: workItemsStorage,
 						transitionService,
-						...storage.isDomainReady("source-control") ? { sessions: sourceControlStorage.forIntegration("github").sessions } : {}
+						boards: this.#boards,
+						...storage.isDomainReady("source-control") ? { sessions: sourceControlSessions } : {}
 					}));
 					const supervisorScope = await resolveSupervisorScope({
 						requestContext,
@@ -453,14 +530,13 @@ var MastraFactory = class {
 							audit: auditDomain,
 							transitionService,
 							...githubIntegration ? { reconcileAcceptanceLabels: (args) => reconcileGithubAcceptanceLabels(githubIntegration, sourceControlStorage.forIntegration(githubIntegration.id), args) } : {},
-							signalSession: async ({ sessionId, message }) => {
-								const session = await prepared.base.controller.getSessionByResource(sessionId);
-								if (!session) throw new Error("The worker session is not currently available.");
-								await session.sendMessage({
-									content: message,
-									...requestContext ? { requestContext } : {}
-								});
-							}
+							messageSession: ({ sessionId, message, delivery }) => messageWorkerSession({
+								controller: prepared.base.controller,
+								sessionId,
+								message,
+								delivery,
+								...requestContext ? { requestContext } : {}
+							})
 						}));
 					}
 				}
@@ -584,7 +660,11 @@ var MastraFactory = class {
 				}),
 				...projectRoutes.routes(),
 				...auditDomain.routes(),
-				...commentsDomain.routes()
+				...commentsDomain.routes(),
+				...hasPlatformCredentials() ? buildPlatformConnectRoutes({
+					auth: routeAuth,
+					client: new PlatformApiClient(platformApiClientConfigFromEnv())
+				}) : []
 			],
 			buildServerConfig: () => {
 				const cors = allowedOrigins.length ? { cors: {
@@ -626,11 +706,11 @@ var MastraFactory = class {
 		prepared.base.controller.onSessionCreated((session) => {
 			observeSessionFilesystem(session, {
 				filesystem: filesystemStorage,
-				sourceControl: sourceControlStorage.forIntegration("github")
+				sourceControl: { sessions: sourceControlSessions }
 			});
-			observeSessionFirstMessage(session, { sourceControl: sourceControlStorage.forIntegration("github") });
-			observeSessionFirstExec(session, { sourceControl: sourceControlStorage.forIntegration("github") });
-			observeSessionThreadTitle(session, { sourceControl: sourceControlStorage.forIntegration("github") });
+			observeSessionFirstMessage(session, { sourceControl: { sessions: sourceControlSessions } });
+			observeSessionFirstExec(session, { sourceControl: { sessions: sourceControlSessions } });
+			observeSessionThreadTitle(session, { sourceControl: { sessions: sourceControlSessions } });
 			observeSessionRunEnd(session, { audit: auditDomain });
 		});
 		prepared.base.controller.onSessionCreated((session) => hydrateSupervisorSession(session, {
@@ -640,12 +720,12 @@ var MastraFactory = class {
 			console.warn("[Factory Supervisor] Failed to hydrate supervisor session", { error: error instanceof Error ? error.message : String(error) });
 		}), { blocking: true });
 		prepared.base.controller.onSessionCreated((session) => hydrateSessionMemorySettings(session, {
-			sourceControl: sourceControlStorage.forIntegration("github"),
+			sourceControl: { sessions: sourceControlSessions },
 			projects: factoryProjectsStorage,
 			memorySettings: memorySettingsStorage
 		}), { blocking: true });
 		prepared.base.controller.onSessionCreated((session) => hydrateSessionModelPack(session, {
-			sourceControl: sourceControlStorage.forIntegration("github"),
+			sourceControl: { sessions: sourceControlSessions },
 			workItems: workItemsStorage,
 			modelPacks: modelPacksStorage
 		}), { blocking: true });
@@ -663,6 +743,7 @@ var MastraFactory = class {
 				factoryStorage: storage,
 				integrationStorage,
 				sourceControlStorage,
+				integrations: integrationRegistrations,
 				configVersion,
 				boardRegistry: this.#boards,
 				factoryReady,
@@ -682,6 +763,7 @@ var MastraFactory = class {
 				factoryStorage: storage,
 				integrationStorage,
 				sourceControlStorage,
+				integrations: integrationRegistrations,
 				configVersion,
 				boardRegistry: this.#boards,
 				factoryReady,
@@ -705,6 +787,7 @@ var MastraFactory = class {
 			factoryStorage: storage,
 			integrationStorage,
 			sourceControlStorage,
+			integrations: integrationRegistrations,
 			configVersion,
 			boardRegistry: this.#boards,
 			factoryReady,