mastracode 0.48.1 → 0.48.2-alpha.0

package/dist/{tui-CEuwbh2h.js → tui-Db51Stlv.js}

@@ -64,6 +64,70 @@ import { RequestContext } from "@mastra/core/request-context";
 import { MaintenanceLockError, acquireMaintenanceLock } from "@mastra/code-sdk/utils/maintenance-lock";
 import { runStorageMaintenance } from "@mastra/code-sdk/utils/storage-maintenance";
 import { loadCustomCommands } from "@mastra/code-sdk/utils/slash-command-loader";
+// src/browser-lifecycle.ts
+/**
+* The browser Mastra Code's agents currently use. Mastra Code owns it: a local
+* browser keeps running (and holds its profile's lock) until it is closed, so
+* it must be closed when replaced and when Mastra Code exits.
+*/
+let activeBrowser;
+const CLOSE_TIMEOUT_MS = 5e3;
+/**
+* Retire a browser: it closes and never relaunches, even for a tool call that
+* still holds it. If close fails or takes longer than CLOSE_TIMEOUT_MS, its
+* processes are killed so the old Chrome cannot keep holding the profile.
+*/
+async function closeBrowser(browser) {
+	let timer;
+	const closed = await Promise.race([browser.retire().then(() => true, () => false), new Promise((resolve) => {
+		timer = setTimeout(() => resolve(false), CLOSE_TIMEOUT_MS);
+		timer.unref?.();
+	})]);
+	clearTimeout(timer);
+	if (!closed) browser.killProcesses();
+}
+/**
+* Counts replacements, so a caller that started creating a browser can tell
+* whether another caller replaced the active browser in the meantime.
+*/
+let generation = 0;
+function activeBrowserGeneration() {
+	return generation;
+}
+/** Close a browser that never became active. Bounded like any other close. */
+function discardBrowser(browser) {
+	return closeBrowser(browser);
+}
+/**
+* Every close still in flight, chained so a later replacement (or exit) waits
+* for all of them, not just the browser it replaced. Each browser is retired once.
+*/
+let retiring = Promise.resolve();
+let retiringBrowser;
+/**
+* Make `next` the active browser, closing the previous one first so it
+* releases its profile before `next` launches on it. Also waits for any
+* browser an earlier replacement is still closing.
+*
+* Returns whether `next` is still the active browser once those closes finish.
+* False means a later call replaced (and closed) it meanwhile, so the caller
+* must not hand `next` to agents.
+*/
+async function replaceActiveBrowser(next) {
+	const previous = activeBrowser;
+	activeBrowser = next;
+	const mine = ++generation;
+	if (previous && previous !== next && previous !== retiringBrowser) {
+		retiringBrowser = previous;
+		retiring = Promise.all([retiring, closeBrowser(previous)]).then(() => {});
+	}
+	await retiring;
+	return generation === mine;
+}
+/** Close the active browser. Bounded so a hung browser cannot block exit. */
+async function closeActiveBrowser() {
+	await replaceActiveBrowser(void 0);
+}
 // src/commit-attribution.ts
 const TUI_CO_AUTHOR = { name: "mastracode" };
 // src/tui/bundled-themes.ts
@@ -19792,16 +19856,28 @@ function resolveModeAgent(mode, agentControllerState) {
 /**
 * @param chatModelId the chat model passed to `createBrowserFromSettings` for
 *   this browser, so the recorded model matches what was really launched.
+* @param startedAt `activeBrowserGeneration()` read before this call started
+*   creating `browser`. If another /browser call replaced the active browser
+*   since then, that call owns the agents: `browser` is discarded and false returned.
 */
-function applyBrowserToAgents(ctx, browser, browserSettings, chatModelId) {
-	const modes = ctx.controller.listModes();
-	let agentControllerState;
-	for (const mode of modes) resolveModeAgent(mode, agentControllerState ??= ctx.state.session.state.get())?.setBrowser?.(browser);
-	ctx.state.local?.controller.setBrowser(browser);
+async function applyBrowserToAgents(ctx, browser, browserSettings, chatModelId, startedAt) {
+	if (startedAt !== void 0 && activeBrowserGeneration() !== startedAt) {
+		if (browser) await discardBrowser(browser);
+		return false;
+	}
+	setAgentsBrowser(ctx, void 0);
+	if (!await replaceActiveBrowser(browser)) return false;
+	setAgentsBrowser(ctx, browser);
 	ctx.state.session.state.set({
 		[ACTIVE_BROWSER_KEY]: browserSettings ? toActiveBrowserSettings(browserSettings) : void 0,
 		[ACTIVE_BROWSER_MODEL_KEY]: browserSettings?.enabled ? resolveStagehandModel(browserSettings, { chatModelId }) : void 0
 	});
+	return true;
+}
+function setAgentsBrowser(ctx, browser) {
+	let agentControllerState;
+	for (const mode of ctx.controller.listModes()) resolveModeAgent(mode, agentControllerState ??= ctx.state.session.state.get())?.setBrowser?.(browser);
+	ctx.state.local?.controller.setBrowser(browser);
 }
 /**
 * Get a summary key for browser settings to detect config drift.
@@ -20009,9 +20085,12 @@ async function handleBrowserCommand(ctx, args = []) {
 			...settings.browser,
 			enabled: false
 		};
+		if (!await applyBrowserToAgents(ctx, void 0, disabledSettings)) {
+			ctx.showInfo("Another /browser command replaced this browser.");
+			return;
+		}
 		settings.browser = disabledSettings;
 		saveSettings(settings);
-		applyBrowserToAgents(ctx, void 0, disabledSettings);
 		ctx.showInfo("Browser disabled.");
 		return;
 	}
@@ -20026,7 +20105,11 @@ async function handleBrowserCommand(ctx, args = []) {
 		}
 		try {
 			const chatModelId = launchChatModelId(ctx);
-			applyBrowserToAgents(ctx, await createBrowserFromSettings(nextBrowser, { chatModelId }), nextBrowser, chatModelId);
+			const startedAt = activeBrowserGeneration();
+			if (!await applyBrowserToAgents(ctx, await createBrowserFromSettings(nextBrowser, { chatModelId }), nextBrowser, chatModelId, startedAt)) {
+				ctx.showInfo("Another /browser command replaced this browser.");
+				return;
+			}
 			if (nextBrowser.profile && nextBrowser.provider) setProfileProvider(nextBrowser.profile, nextBrowser.provider);
 			settings.browser = nextBrowser;
 			saveSettings(settings);
@@ -20050,18 +20133,29 @@ async function handleBrowserCommand(ctx, args = []) {
 					height: 720
 				}
 			};
-			saveSettings(settings);
-			if (wasEnabled) try {
-				const chatModelId = launchChatModelId(ctx);
-				applyBrowserToAgents(ctx, await createBrowserFromSettings(settings.browser, { chatModelId }), settings.browser, chatModelId);
-			} catch (err) {
-				settings.browser.enabled = false;
-				saveSettings(settings);
-				applyBrowserToAgents(ctx, void 0);
-				ctx.showError(`Browser settings reset, but failed to restart: ${err instanceof Error ? err.message : String(err)}`);
+			if (wasEnabled) {
+				const startedAt = activeBrowserGeneration();
+				try {
+					const chatModelId = launchChatModelId(ctx);
+					if (!await applyBrowserToAgents(ctx, await createBrowserFromSettings(settings.browser, { chatModelId }), settings.browser, chatModelId, startedAt)) {
+						ctx.showInfo("Another /browser command replaced this browser.");
+						return;
+					}
+				} catch (err) {
+					if (!await applyBrowserToAgents(ctx, void 0, void 0, void 0, startedAt)) {
+						ctx.showInfo("Another /browser command replaced this browser.");
+						return;
+					}
+					settings.browser.enabled = false;
+					saveSettings(settings);
+					ctx.showError(`Browser settings reset, but failed to restart: ${err instanceof Error ? err.message : String(err)}`);
+					return;
+				}
+			} else if (!await applyBrowserToAgents(ctx, void 0)) {
+				ctx.showInfo("Another /browser command replaced this browser.");
 				return;
 			}
-			else applyBrowserToAgents(ctx, void 0);
+			saveSettings(settings);
 			ctx.showInfo("Browser settings reset to defaults.");
 			return;
 		}
@@ -20111,8 +20205,12 @@ async function handleBrowserCommand(ctx, args = []) {
 		let browserInstance = resolveModeAgent(ctx.state.session.mode.resolve(), ctx.state.session.state.get())?.browser;
 		if (!browserInstance && browser.enabled) {
 			const chatModelId = launchChatModelId(ctx);
+			const startedAt = activeBrowserGeneration();
 			browserInstance = await createBrowserFromSettings(browser, { chatModelId });
-			applyBrowserToAgents(ctx, browserInstance, browser, chatModelId);
+			if (!await applyBrowserToAgents(ctx, browserInstance, browser, chatModelId, startedAt)) {
+				ctx.showError("Another /browser command replaced the browser. Run the export again.");
+				return;
+			}
 		}
 		if (!browserInstance) {
 			ctx.showError("Browser not enabled. Run /browser on first.");
@@ -20170,9 +20268,12 @@ async function handleBrowserCommand(ctx, args = []) {
 	}
 	if (enableChoice === "No") {
 		if (browser.enabled) {
+			if (!await applyBrowserToAgents(ctx, void 0)) {
+				ctx.showInfo("Another /browser command replaced this browser.");
+				return;
+			}
 			settings.browser.enabled = false;
 			saveSettings(settings);
-			applyBrowserToAgents(ctx, void 0);
 			ctx.showInfo("Browser automation disabled.");
 		} else ctx.showInfo("Browser automation remains disabled.");
 		return;
@@ -20318,7 +20419,11 @@ async function handleBrowserCommand(ctx, args = []) {
 	}
 	const chatModelId = launchChatModelId(ctx);
 	try {
-		applyBrowserToAgents(ctx, await createBrowserFromSettings(nextBrowser, { chatModelId }), nextBrowser, chatModelId);
+		const startedAt = activeBrowserGeneration();
+		if (!await applyBrowserToAgents(ctx, await createBrowserFromSettings(nextBrowser, { chatModelId }), nextBrowser, chatModelId, startedAt)) {
+			ctx.showInfo("Another /browser command replaced this browser.");
+			return;
+		}
 		if (nextBrowser.profile && nextBrowser.provider) setProfileProvider(nextBrowser.profile, nextBrowser.provider);
 		settings.browser = nextBrowser;
 		saveSettings(settings);
@@ -28706,6 +28811,6 @@ var MastraTUI = class MastraTUI {
 		}, { startup: true });
 	}
 };
-export { mastraBrand as C, mastra as S, TUI_CO_AUTHOR as T, normalizeThemePreference as _, formatUpdateHeader as a, getMarkdownTheme as b, LoginDialogComponent as c, createTUIState as d, UserMessageComponent as f, applyThemePreference as g, AssistantMessageComponent as h, formatUpToDate as i, OMProgressComponent as l, ModelSelectorComponent as m, formatInstallingLabel as n, formatUpdateOutcome as o, ToolExecutionComponentEnhanced as p, formatRegistryError as r, LoginSelectorComponent as s, MastraTUI as t, formatOMStatus as u, applyThemeMode as v, theme as w, getThemeMode as x, getEditorTheme as y };
+export { mastraBrand as C, closeActiveBrowser as D, activeBrowserGeneration as E, discardBrowser as O, mastra as S, TUI_CO_AUTHOR as T, normalizeThemePreference as _, formatUpdateHeader as a, getMarkdownTheme as b, LoginDialogComponent as c, createTUIState as d, UserMessageComponent as f, applyThemePreference as g, AssistantMessageComponent as h, formatUpToDate as i, replaceActiveBrowser as k, OMProgressComponent as l, ModelSelectorComponent as m, formatInstallingLabel as n, formatUpdateOutcome as o, ToolExecutionComponentEnhanced as p, formatRegistryError as r, LoginSelectorComponent as s, MastraTUI as t, formatOMStatus as u, applyThemeMode as v, theme as w, getThemeMode as x, getEditorTheme as y };
 
-//# sourceMappingURL=tui-CEuwbh2h.js.map
+//# sourceMappingURL=tui-Db51Stlv.js.map