@mastra/code-sdk 1.2.0-alpha.9 → 1.2.0

package/dist/index.js

@@ -29,6 +29,7 @@ import { hasExplicitOMConfiguration } from "./onboarding/om-settings.js";
 import { getAvailableModePacks, getAvailableOmPacks, selectPreferredOMPack } from "./onboarding/packs.js";
 import { getToolCategory } from "./permissions.js";
 import { PluginManager } from "./plugins/manager.js";
+import { PluginSignalLane } from "./plugins/signal-lane.js";
 import { PlanRejectionAbortProcessor } from "./processors/plan-rejection-abort.js";
 import { stateSchema } from "./schema.js";
 import { mastraBrand } from "./theme-palette.js";
@@ -37,6 +38,7 @@ import { createSignalsPubSub } from "./utils/signals-pubsub.js";
 import { DEFAULT_RETENTION, createStorageMaintenance, resolveLocalDbFiles } from "./utils/storage-maintenance.js";
 import { createStorage, createVectorStore } from "./utils/storage-factory.js";
 import { acquireThreadLock, releaseThreadLock } from "./utils/thread-lock.js";
+import { registerWorkflowBuilderPrimitives } from "./workflows/register-primitives.js";
 import { autoApprovePolicy, denyPolicy, permissionModeToPolicy } from "./headless/policy.js";
 import { runMC } from "./headless/run-mc.js";
 import { formatHuman, formatJsonl, renderJsonResult, renderTextResult } from "./headless/format.js";
@@ -50,6 +52,7 @@ import { AgentController } from "@mastra/core/agent-controller";
 import { createCodingAgent } from "@mastra/core/coding-agent";
 import { PROVIDER_REGISTRY } from "@mastra/core/llm";
 import { Mastra } from "@mastra/core/mastra";
+import { defaultNotificationDeliveryDecision } from "@mastra/core/notifications";
 import { AgentsMDInjector, PrefillErrorHandler, ProviderHistoryCompat, StreamErrorRetryProcessor, isBadRequestError } from "@mastra/core/processors";
 import { RequestContext } from "@mastra/core/request-context";
 import { InMemoryHarness, MastraCompositeStore } from "@mastra/core/storage";
@@ -201,6 +204,7 @@ async function createMastraCodeAgentController(config) {
 	const homeDir = config?.homeDir ?? config?.initialState?.homeDir;
 	const configDir = config?.configDir ?? ".mastracode";
 	let activeSession;
+	let pluginRuntimeController;
 	if (configDir !== ".mastracode") validateConfigDirName(configDir);
 	try {
 		process.loadEnvFile(path.join(cwd, ".env"));
@@ -314,66 +318,136 @@ async function createMastraCodeAgentController(config) {
 	});
 	const memory = config?.memory === false ? void 0 : config?.memory ?? getDynamicMemory(storage, vector);
 	const mcpManager = config?.disableMcp ? void 0 : createMcpManager(project.rootPath, configDir, config?.mcpServers);
-	const hookManager = config?.disableHooks ? void 0 : new HookManager(project.rootPath, "session-init", configDir, homeDir);
+	const hookManager = config?.disableHooks ? void 0 : new HookManager(project.rootPath, "session-init", configDir, homeDir, project.isWorktree ? {
+		path: project.rootPath,
+		branch: project.gitBranch,
+		mainRepoPath: project.mainRepoPath
+	} : void 0);
 	const pluginManager = config?.disablePlugins ? void 0 : config?.pluginManager ?? new PluginManager({
 		projectRoot: project.rootPath,
 		configDir,
 		homeDir
 	});
+	pluginManager?.setRuntime({
+		getController: () => pluginRuntimeController,
+		getActiveSession: () => activeSession
+	});
 	const loadedPlugins = pluginManager ? await pluginManager.reload() : [];
 	const pluginTools = pluginManager?.getPluginTools() ?? {};
 	const outcomeScorer = createOutcomeScorer();
 	const efficiencyScorer = createEfficiencyScorer();
+	const getNotificationStreamOptions = async ({ resourceId, threadId }) => {
+		const session = await controller.getSessionByResource(resourceId) ?? activeSession;
+		if (!session) return void 0;
+		const modeId = session.mode.get();
+		const defaultModeModelId = controller.listModes().find((mode) => mode.id === modeId)?.defaultModelId;
+		const modelId = session.model.get() || activeSession?.model.get() || defaultModeModelId || "";
+		const requestContext = new RequestContext();
+		const agentControllerContext = {
+			controllerId: controller.id,
+			state: session.state.get(),
+			getState: () => session.state.get(),
+			setState: (updates) => session.state.set(updates),
+			threadId,
+			resourceId,
+			session: {
+				id: session.identity.getId(),
+				ownerId: session.identity.getOwnerId(),
+				modeId,
+				modelId,
+				state: {
+					get: () => session.state.get(),
+					set: (updates) => session.state.set(updates),
+					update: (updater) => session.state.update(updater)
+				}
+			},
+			workspace: session.getWorkspace(),
+			getSubagentModelId: (params) => session.subagents.model.get(params ?? {})
+		};
+		requestContext.set("controller", agentControllerContext);
+		return {
+			memory: {
+				thread: threadId,
+				resource: resourceId
+			},
+			requestContext,
+			maxSteps: 1e3,
+			savePerStep: false,
+			requireToolApproval: session.state.get().yolo !== true,
+			modelSettings: { temperature: 1 }
+		};
+	};
 	const githubSignals = globalSettings.signals?.experimentalGithubSignals && !config?.disableGithubSignals ? new GithubSignals({
 		cwd: project.rootPath,
 		gitcrawlCommand: process.env.MASTRACODE_GITCRAWL_BIN ?? process.env.GITCRAWL_BIN ?? process.env.MASTRACODE_GITCRAWL_COMMAND ?? process.env.GITCRAWL_COMMAND,
-		getNotificationStreamOptions: async ({ resourceId, threadId }) => {
-			const session = await controller.getSessionByResource(resourceId) ?? activeSession;
-			const modeId = session.mode.get();
-			const defaultModeModelId = controller.listModes().find((mode) => mode.id === modeId)?.defaultModelId;
-			const modelId = session.model.get() || activeSession?.model.get() || defaultModeModelId || "";
-			const requestContext = new RequestContext();
-			const agentControllerContext = {
-				controllerId: controller.id,
-				state: session.state.get(),
-				getState: () => session.state.get(),
-				setState: (updates) => session.state.set(updates),
-				threadId,
-				resourceId,
-				session: {
-					id: session.identity.getId(),
-					ownerId: session.identity.getOwnerId(),
-					modeId,
-					modelId,
-					state: {
-						get: () => session.state.get(),
-						set: (updates) => session.state.set(updates),
-						update: (updater) => session.state.update(updater)
-					}
-				},
-				workspace: controller.getWorkspace(),
-				getSubagentModelId: (params) => session.subagents.model.get(params ?? {})
-			};
-			requestContext.set("controller", agentControllerContext);
-			return {
-				memory: {
-					thread: threadId,
-					resource: resourceId
-				},
-				requestContext,
-				maxSteps: 1e3,
-				savePerStep: false,
-				requireToolApproval: session.state.get().yolo !== true,
-				modelSettings: { temperature: 1 }
-			};
-		}
+		getNotificationStreamOptions
 	}) : void 0;
+	const mastraCodeInputProcessors = [
+		...config?.inputProcessors ?? [],
+		new PlanRejectionAbortProcessor(),
+		new AgentsMDInjector({
+			isEnabled: ({ requestContext }) => {
+				const state = getInjectorSessionState(requestContext);
+				return state?.untrustedCheckout !== true || typeof state?.baseRef === "string";
+			},
+			getReader: ({ requestContext }) => {
+				const state = getInjectorSessionState(requestContext);
+				if (state?.untrustedCheckout !== true || typeof state?.baseRef !== "string") return void 0;
+				return createGitRefReminderReader(state?.projectPath ?? project.rootPath, state.baseRef);
+			},
+			getIgnoredInstructionPaths: ({ requestContext }) => {
+				const state = getInjectorSessionState(requestContext);
+				const projectPath = state?.projectPath ?? project.rootPath;
+				return getStaticallyLoadedInstructionPaths(projectPath, void 0, state?.untrustedCheckout === true && typeof state?.baseRef === "string" ? createGitRefInstructionReader(projectPath, state.baseRef) : void 0);
+			}
+		}),
+		new ProviderHistoryCompat()
+	];
+	const taskSignalProvider = new TaskSignalProvider();
+	const NO_PLUGIN_PROCESSORS = {
+		input: [],
+		output: []
+	};
+	let pluginProcessorReadWarned = false;
+	const pluginSignalLane = pluginManager ? new PluginSignalLane({ reservedProviderIds: [taskSignalProvider.id, ...githubSignals ? [githubSignals.id] : []] }) : void 0;
+	let unsubscribePluginReload;
+	/**
+	* Plugin processors are read through a function so that enabling, disabling or
+	* updating a plugin takes effect on the next request rather than requiring a
+	* new agent. This runs before every LLM call, and also outside the request
+	* path when the Agent catalogues its configured processors — where a throw is
+	* swallowed into a debug log. So it only reads already-resolved state: no
+	* filesystem, no network, no construction, and it never throws.
+	*/
+	const readPluginProcessors = () => {
+		try {
+			return pluginManager?.getPluginProcessors() ?? NO_PLUGIN_PROCESSORS;
+		} catch (error) {
+			if (!pluginProcessorReadWarned) {
+				pluginProcessorReadWarned = true;
+				console.warn("Failed to read plugin processors:", error);
+			}
+			return NO_PLUGIN_PROCESSORS;
+		}
+	};
 	const codeAgent = createCodingAgent({
 		id: CODE_AGENT_ID,
 		name: "Code Agent",
 		workspace: void 0,
 		instructions: getDynamicInstructions,
 		model: (ctx) => getDynamicModel(ctx, config?.settingsPath),
+		notifications: { deliveryPolicy: { decide: async (input) => {
+			const decision = defaultNotificationDeliveryDecision(input);
+			if (!input.record.resourceId) return decision;
+			const streamOptions = await getNotificationStreamOptions({
+				resourceId: input.record.resourceId,
+				threadId: input.record.threadId
+			});
+			return streamOptions ? {
+				...decision,
+				streamOptions
+			} : decision;
+		} } },
 		tools: createDynamicTools(mcpManager, config?.extraTools, config?.disabledTools, storage, pluginTools),
 		hooks: createToolHooks(hookManager, config?.postToolObserver),
 		scorers: {
@@ -389,7 +463,7 @@ async function createMastraCodeAgentController(config) {
 				}
 			}
 		},
-		signals: [new TaskSignalProvider(), ...githubSignals ? [githubSignals] : []],
+		signals: [taskSignalProvider, ...githubSignals ? [githubSignals] : []],
 		goal: {
 			judge: (ctx) => getGoalJudgeModel(ctx, config?.settingsPath),
 			maxRuns: globalSettings.models.goalMaxTurns ?? 50,
@@ -397,27 +471,12 @@ async function createMastraCodeAgentController(config) {
 			prompt: DEFAULT_GOAL_JUDGE_PROMPT,
 			tools: getGoalJudgeTools
 		},
-		inputProcessors: [
-			...config?.inputProcessors ?? [],
-			new PlanRejectionAbortProcessor(),
-			new AgentsMDInjector({
-				isEnabled: ({ requestContext }) => {
-					const state = getInjectorSessionState(requestContext);
-					return state?.untrustedCheckout !== true || typeof state?.baseRef === "string";
-				},
-				getReader: ({ requestContext }) => {
-					const state = getInjectorSessionState(requestContext);
-					if (state?.untrustedCheckout !== true || typeof state?.baseRef !== "string") return void 0;
-					return createGitRefReminderReader(state?.projectPath ?? project.rootPath, state.baseRef);
-				},
-				getIgnoredInstructionPaths: ({ requestContext }) => {
-					const state = getInjectorSessionState(requestContext);
-					const projectPath = state?.projectPath ?? project.rootPath;
-					return getStaticallyLoadedInstructionPaths(projectPath, void 0, state?.untrustedCheckout === true && typeof state?.baseRef === "string" ? createGitRefInstructionReader(projectPath, state.baseRef) : void 0);
-				}
-			}),
-			new ProviderHistoryCompat()
+		inputProcessors: () => [
+			...mastraCodeInputProcessors,
+			...readPluginProcessors().input.map((entry) => entry.value),
+			...pluginSignalLane?.getInputProcessors() ?? []
 		],
+		outputProcessors: () => [...readPluginProcessors().output.map((entry) => entry.value), ...pluginSignalLane?.getOutputProcessors() ?? []],
 		errorProcessors: [
 			new ProviderHistoryCompat(),
 			new StreamErrorRetryProcessor({ matchers: [
@@ -566,6 +625,11 @@ async function createMastraCodeAgentController(config) {
 			release: releaseThreadLock
 		}
 	});
+	pluginRuntimeController = controller;
+	if (pluginSignalLane && pluginManager) {
+		pluginSignalLane.sync(pluginManager.getPluginSignalProviders());
+		unsubscribePluginReload = pluginManager.onReload(() => pluginSignalLane.sync(pluginManager.getPluginSignalProviders()));
+	}
 	return {
 		controller,
 		storage,
@@ -588,8 +652,59 @@ async function createMastraCodeAgentController(config) {
 		githubSignals,
 		sessionId,
 		ownerId,
+		projectPath: project.rootPath,
+		codeAgent,
 		setActiveSession: (session) => {
 			activeSession = session;
+		},
+		/**
+		* Starts the signal providers contributed by plugins. Called by the
+		* composition layer once the controller is inited, because that is when a
+		* Mastra instance exists — a provider without one has no storage, and
+		* nothing else will hand it one: the Agent propagates Mastra only to the
+		* providers in its own `signals` array, which these deliberately are not in.
+		*/
+		startPluginSignalProviders: () => {
+			const mastra = controller.getMastra();
+			if (!pluginSignalLane || !mastra) return;
+			pluginSignalLane.setMastra(mastra, codeAgent);
+		},
+		/**
+		* Stops every plugin-contributed signal provider and stops listening for
+		* plugin reloads. The inverse of `startPluginSignalProviders`, for an
+		* embedder that is done with this controller: a `pluginManager` shared
+		* across controllers (`MastraCodeConfig.pluginManager`) outlives any one of
+		* them, so without this its providers keep polling and its reload listener
+		* keeps firing for a controller that is gone.
+		*/
+		stopPluginSignalProviders: () => {
+			unsubscribePluginReload?.();
+			unsubscribePluginReload = void 0;
+			pluginSignalLane?.stopAll();
+		},
+		/**
+		* Hands Mastra to the statically configured input processors.
+		*
+		* The Agent does this itself, but only for processors configured as a
+		* plain array (`Array.isArray` in `__registerMastra`). This lane is a
+		* function so plugins can contribute to it, which takes those processors
+		* out of that branch — including any an embedder passed as
+		* `config.inputProcessors`, some of which need Mastra to work at all
+		* (`CostGuardProcessor` reads observability storage there). Doing it here
+		* keeps that unchanged.
+		*
+		* Plugin processors are deliberately not included: they come and go with
+		* their plugin, and the registry keeps the first instance registered under
+		* an id forever, which would leave a retired instance behind. Plugins
+		* reach Mastra through `getController()` on the plugin context instead.
+		*/
+		registerConfiguredProcessorsWithMastra: () => {
+			const mastra = controller.getMastra();
+			if (!mastra) return;
+			for (const processor of mastraCodeInputProcessors) {
+				mastra.addProcessor(processor);
+				mastra.addProcessorConfiguration(processor, CODE_AGENT_ID, "input");
+			}
 		}
 	};
 }
@@ -639,9 +754,17 @@ async function wireSessionConcerns(base, session) {
 */
 async function bootLocalAgentController(config) {
 	const base = await createMastraCodeAgentController(config);
-	const { controller, sessionId, ownerId } = base;
+	const { controller, sessionId, ownerId, projectPath, codeAgent, mcpManager } = base;
 	await controller.init();
-	await controller.getMastra()?.startWorkers();
+	const mastra = controller.getMastra();
+	if (mastra) await registerWorkflowBuilderPrimitives(mastra, {
+		projectPath,
+		codeAgent,
+		mcpManager
+	});
+	await mastra?.startWorkers();
+	base.registerConfiguredProcessorsWithMastra();
+	base.startPluginSignalProviders();
 	const session = await controller.createSession({
 		id: sessionId,
 		ownerId
@@ -697,17 +820,19 @@ async function mountAgentControllerOnMastra(config) {
 */
 async function prepareAgentControllerMount(config) {
 	const base = await createMastraCodeAgentController(config);
-	const { controller, storage, authStorage } = base;
+	const { controller, storage, authStorage, projectPath, codeAgent, mcpManager } = base;
 	const controllerId = config?.controllerId ?? controller.id;
 	const apiRoutes = config?.buildApiRoutes?.({
 		controller,
 		authStorage
 	});
+	const extraServerConfig = config?.buildServerConfig?.({
+		controller,
+		authStorage
+	});
+	const weOwnTheMastra = !config?.mastra;
 	const serverConfig = {
-		...config?.buildServerConfig?.({
-			controller,
-			authStorage
-		}),
+		...extraServerConfig,
 		...apiRoutes?.length ? { apiRoutes } : {}
 	};
 	const mastraArgs = {
@@ -718,7 +843,17 @@ async function prepareAgentControllerMount(config) {
 	};
 	const finalize = async () => {
 		await controller.init();
+		if (weOwnTheMastra) {
+			const mastra = controller.getMastra();
+			if (mastra) await registerWorkflowBuilderPrimitives(mastra, {
+				projectPath,
+				codeAgent,
+				mcpManager
+			});
+		}
 		await controller.getMastra()?.startWorkers();
+		base.registerConfiguredProcessorsWithMastra();
+		base.startPluginSignalProviders();
 	};
 	return {
 		base,