@mastra/e2b 0.13.0 → 0.14.0-alpha.0

package/dist/index.d.ts

@@ -4,5 +4,6 @@ export { createDefaultMountableTemplate, isNamedTemplateSpec, isDeferredNamedTem
 export { createRepoTemplate, refreshRepoTemplate, repoTemplateRef, type RepoTemplateOptions, type RepoTemplateRepository, type RepoTemplateIdentity, type RepositoryAccess, type RefreshRepoTemplateResult, } from './utils/repo-template.js';
 export { type E2BS3MountConfig, type E2BGCSMountConfig, type E2BAzureBlobMountConfig, type E2BMountConfig, } from './sandbox/mounts/index.js';
 export { e2bSandboxProvider } from './provider.js';
+export { E2BFactorySandbox, type E2BFactorySandboxOptions, type E2BFactorySandboxSettings } from './factory-sandbox.js';
 export { E2BCodeModeTransport } from './code-mode/transport.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EACL,8BAA8B,EAC9B,mBAAmB,EACnB,2BAA2B,EAC3B,oBAAoB,EACpB,KAAK,YAAY,EACjB,KAAK,iBAAiB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,uBAAuB,EAC5B,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,eAAe,EACf,KAAK,mBAAmB,EACxB,KAAK,sBAAsB,EAC3B,KAAK,oBAAoB,EACzB,KAAK,gBAAgB,EACrB,KAAK,yBAAyB,GAC/B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,iBAAiB,EACtB,KAAK,uBAAuB,EAC5B,KAAK,cAAc,GACpB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC9D,OAAO,EACL,8BAA8B,EAC9B,mBAAmB,EACnB,2BAA2B,EAC3B,oBAAoB,EACpB,KAAK,YAAY,EACjB,KAAK,iBAAiB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,uBAAuB,EAC5B,KAAK,wBAAwB,GAC9B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,kBAAkB,EAClB,mBAAmB,EACnB,eAAe,EACf,KAAK,mBAAmB,EACxB,KAAK,sBAAsB,EAC3B,KAAK,oBAAoB,EACzB,KAAK,gBAAgB,EACrB,KAAK,yBAAyB,GAC/B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,KAAK,gBAAgB,EACrB,KAAK,iBAAiB,EACtB,KAAK,uBAAuB,EAC5B,KAAK,cAAc,GACpB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,KAAK,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AACrH,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/index.js

@@ -1,5 +1,5 @@
-import { MastraSandbox, ProcessHandle, SandboxError, SandboxNotReadyError, SandboxProcessManager, UnsupportedStdinCloseError, assertModesUnsupported } from "@mastra/core/workspace";
-import { Sandbox, Template } from "e2b";
+import { FactorySandbox, MastraSandbox, ProcessHandle, SandboxError, SandboxNotReadyError, SandboxProcessManager, UnsupportedStdinCloseError, assertModesUnsupported } from "@mastra/core/workspace";
+import { ApiClient, ConnectionConfig, Sandbox, Template } from "e2b";
 import { createHash } from "crypto";
 import { execFile } from "child_process";
 import { promisify } from "util";
@@ -2145,6 +2145,222 @@ const e2bSandboxProvider = {
 	createSandbox: (config) => new E2BSandbox(config)
 };
 //#endregion
+//#region src/factory-sandbox.ts
+/** Applied when neither the environment setting nor a host default is set. */
+const DEFAULT_SETTINGS = {
+	cpuCount: 2,
+	memoryMb: 1024,
+	idleTimeoutMinutes: 5
+};
+const SETTINGS_SCHEMA = {
+	type: "object",
+	properties: {
+		cpuCount: {
+			type: "integer",
+			title: "CPU",
+			description: "vCPUs for the sandbox template",
+			default: DEFAULT_SETTINGS.cpuCount,
+			minimum: 1,
+			maximum: 8
+		},
+		memoryMb: {
+			type: "integer",
+			title: "Memory (MB)",
+			description: "Memory in MB for the sandbox template",
+			default: DEFAULT_SETTINGS.memoryMb,
+			minimum: 512,
+			maximum: 65536
+		},
+		idleTimeoutMinutes: {
+			type: "integer",
+			title: "Idle timeout (minutes)",
+			description: "Minutes a sandbox stays alive before E2B stops it",
+			default: DEFAULT_SETTINGS.idleTimeoutMinutes,
+			minimum: 1,
+			maximum: 1440
+		}
+	},
+	additionalProperties: false
+};
+/**
+* The settings schema with the provider's effective defaults as `default`:
+* the built-in values, overridden by the host's `defaults` option. The schema
+* is what clients show as placeholders, so it must match what the provider
+* applies when a setting is unset.
+*/
+function settingsSchema(options) {
+	const { defaults, timeout } = options;
+	const effective = {
+		...DEFAULT_SETTINGS,
+		...timeout !== void 0 ? { idleTimeoutMinutes: Math.ceil(timeout / 6e4) } : {},
+		...defaults
+	};
+	const properties = Object.fromEntries(Object.entries(SETTINGS_SCHEMA.properties).map(([key, property]) => [key, {
+		...property,
+		default: effective[key]
+	}]));
+	return {
+		...SETTINGS_SCHEMA,
+		properties
+	};
+}
+/** Log entries E2B returns per build-status call. */
+const BUILD_LOG_PAGE = 100;
+/** The id of a build that needed no build: the sha-tagged template already existed. */
+const EXISTING_BUILD = "existing";
+/**
+* The E2B host contract for factory: every session sandbox is an
+* {@link E2BSandbox} whose template is {@link createRepoTemplate} over the
+* session's repositories, sized by the factory's environment settings. The
+* `builds` capability starts the same sha-tagged template ahead of any
+* session and reads its status, logs and history back from E2B.
+*/
+var E2BFactorySandbox = class extends FactorySandbox {
+	provider = "e2b";
+	settings;
+	builds;
+	#options;
+	#connection;
+	constructor(options = {}) {
+		super();
+		this.#options = options;
+		this.settings = settingsSchema(options);
+		this.#connection = {
+			...options.domain && { domain: options.domain },
+			...options.apiUrl && { apiUrl: options.apiUrl },
+			...options.apiKey && { apiKey: options.apiKey },
+			...options.accessToken && { accessToken: options.accessToken }
+		};
+		this.builds = {
+			start: (ctx, settings) => this.#startBuild(ctx, settings),
+			get: (_ctx, _settings, buildId) => this.#getBuild(buildId),
+			list: (ctx, settings) => this.#listBuilds(ctx, settings)
+		};
+	}
+	template(ctx, settings) {
+		return createRepoTemplate(this.#templateOptions(ctx, settings));
+	}
+	create(ctx, settings) {
+		const { defaults, ...options } = this.#options;
+		return new E2BSandbox({
+			...options,
+			id: ctx.sessionId,
+			sandboxId: ctx.sandboxId,
+			template: this.template(ctx, settings),
+			...ctx.workingDirectory ? { workingDirectory: ctx.workingDirectory } : {},
+			timeout: this.#timeout(settings)
+		});
+	}
+	/** The setting, then the host default, then the host's own `timeout` option, then 5 minutes. */
+	#timeout(settings) {
+		const minutes = settings.idleTimeoutMinutes ?? this.#options.defaults?.idleTimeoutMinutes;
+		if (minutes !== void 0) return minutes * 6e4;
+		return this.#options.timeout ?? DEFAULT_SETTINGS.idleTimeoutMinutes * 6e4;
+	}
+	#templateOptions(ctx, settings) {
+		const { defaults } = this.#options;
+		const { resolveHead: _resolveHead, ...rest } = ctx;
+		return {
+			...rest,
+			cpuCount: settings.cpuCount ?? defaults?.cpuCount ?? DEFAULT_SETTINGS.cpuCount,
+			memoryMB: settings.memoryMb ?? defaults?.memoryMb ?? DEFAULT_SETTINGS.memoryMb
+		};
+	}
+	async #startBuild(ctx, settings) {
+		const { spec } = await resolveSpecAtHead(this.#templateOptions(ctx, settings));
+		if (await Template.exists(spec.ref, this.#connection)) return {
+			buildId: `${spec.ref}:${EXISTING_BUILD}`,
+			templateId: spec.ref,
+			status: "ready"
+		};
+		const info = await Template.buildInBackground(spec.template, spec.ref, {
+			...this.#connection,
+			...spec.buildTags?.length ? { tags: spec.buildTags } : {},
+			...spec.buildResources
+		});
+		return {
+			buildId: `${info.templateId}:${info.buildId}`,
+			templateId: info.templateId,
+			status: "pending"
+		};
+	}
+	async #getBuild(compositeId) {
+		const { templateId, buildId } = splitBuildId(compositeId);
+		if (buildId === EXISTING_BUILD) return {
+			buildId: compositeId,
+			templateId,
+			status: "ready"
+		};
+		let response = await Template.getBuildStatus({
+			templateId,
+			buildId
+		}, this.#connection);
+		const entries = [...response.logEntries ?? []];
+		while (response.logEntries && response.logEntries.length >= BUILD_LOG_PAGE) {
+			response = await Template.getBuildStatus({
+				templateId,
+				buildId
+			}, {
+				...this.#connection,
+				logsOffset: entries.length
+			});
+			entries.push(...response.logEntries ?? []);
+		}
+		const logs = entries.length ? entries.map((entry) => entry.toString()) : response.logs?.length ? response.logs : void 0;
+		return {
+			buildId: compositeId,
+			templateId: response.templateID,
+			status: mapStatus(response.status),
+			...logs ? { logs } : {},
+			...response.reason?.message ? { error: response.reason.message } : {}
+		};
+	}
+	async #listBuilds(ctx, settings) {
+		const { spec } = await resolveSpecAtHead(this.#templateOptions(ctx, settings));
+		const name = templateNameOf(spec.ref);
+		const client = new ApiClient(new ConnectionConfig(this.#connection));
+		const templates = await client.api.GET("/templates");
+		if (templates.error || !templates.data) throw new Error(`E2B template listing failed: ${describeApiError(templates.error)}`);
+		const template = templates.data.find((t) => t.names.includes(name) || t.aliases.includes(name));
+		if (!template) return [];
+		const detail = await client.api.GET("/templates/{templateID}", { params: { path: { templateID: template.templateID } } });
+		if (detail.error || !detail.data) throw new Error(`E2B template build listing failed: ${describeApiError(detail.error)}`);
+		return detail.data.builds.slice().sort((a, b) => b.createdAt.localeCompare(a.createdAt)).map((build) => ({
+			buildId: `${template.templateID}:${build.buildID}`,
+			templateId: template.templateID,
+			status: mapStatus(build.status),
+			startedAt: build.createdAt,
+			...build.finishedAt ? { finishedAt: build.finishedAt } : {}
+		}));
+	}
+};
+function describeApiError(error) {
+	return error ? JSON.stringify(error) : "empty response";
+}
+/** `<templateId>:<buildId>`, split on the last colon because a template ref itself carries one (`name:sha-<head>`). */
+function splitBuildId(compositeId) {
+	const at = compositeId.lastIndexOf(":");
+	if (at <= 0 || at === compositeId.length - 1) throw new Error(`Invalid E2B build id "${compositeId}": expected "<templateId>:<buildId>"`);
+	return {
+		templateId: compositeId.slice(0, at),
+		buildId: compositeId.slice(at + 1)
+	};
+}
+/** The template name of a `name:tag` ref. */
+function templateNameOf(ref) {
+	const at = ref.lastIndexOf(":");
+	return at > 0 ? ref.slice(0, at) : ref;
+}
+function mapStatus(status) {
+	switch (status) {
+		case "waiting": return "pending";
+		case "building": return "building";
+		case "ready": return "ready";
+		case "error": return "failed";
+		default: return "unknown";
+	}
+}
+//#endregion
 //#region src/code-mode/transport.ts
 /**
 * Code Mode — E2B transport
@@ -2350,6 +2566,6 @@ var E2BCodeModeTransport = class {
 	}
 };
 //#endregion
-export { DEFAULT_NODE_VERSION, E2BCodeModeTransport, E2BProcessManager, E2BSandbox, createDefaultMountableTemplate, createRepoTemplate, e2bSandboxProvider, isDeferredNamedTemplateSpec, isNamedTemplateSpec, refreshRepoTemplate, repoTemplateRef };
+export { DEFAULT_NODE_VERSION, E2BCodeModeTransport, E2BFactorySandbox, E2BProcessManager, E2BSandbox, createDefaultMountableTemplate, createRepoTemplate, e2bSandboxProvider, isDeferredNamedTemplateSpec, isNamedTemplateSpec, refreshRepoTemplate, repoTemplateRef };
 
 //# sourceMappingURL=index.js.map