@testdriverai/mcp 7.10.0-canary.0 → 7.10.1-test

package/mcp-server/dist/env-utils.d.ts

@@ -0,0 +1,45 @@
+/**
+ * Utilities for resolving TestDriver environment variables.
+ *
+ * These helpers centralise the "param > env var > default" fallback logic
+ * so the server handler stays clean and the logic can be unit-tested
+ * independently.
+ */
+declare const VALID_OS_VALUES: readonly ["linux", "windows"];
+export type SandboxOs = (typeof VALID_OS_VALUES)[number];
+/**
+ * Resolve the sandbox OS with the following priority:
+ *   1. Explicit `os` parameter passed to session_start
+ *   2. `TD_OS` environment variable (validated against supported values)
+ *   3. `"linux"` default
+ *
+ * Mirrors the behaviour in `hooks.mjs` so the MCP server and the
+ * Vitest integration are consistent.
+ *
+ * @param explicitOs - The value passed explicitly by the caller (may be undefined)
+ * @param env        - The environment variables map (defaults to `process.env`)
+ * @returns          - The resolved OS and a warning message if TD_OS was invalid
+ */
+export declare function resolveOs(explicitOs: SandboxOs | undefined, env?: NodeJS.ProcessEnv): {
+    os: SandboxOs;
+    warning?: string;
+};
+/**
+ * Resolve the E2B template ID with the following priority:
+ *   1. Explicit `e2bTemplateId` parameter passed to session_start
+ *   2. `TD_E2B_TEMPLATE_ID` environment variable
+ *   3. `undefined` (let the SDK use its own default)
+ *
+ * Mirrors the behaviour in `hooks.mjs`:
+ *   ```js
+ *   if (!config.e2bTemplateId && process.env.TD_E2B_TEMPLATE_ID) {
+ *     config.e2bTemplateId = process.env.TD_E2B_TEMPLATE_ID;
+ *   }
+ *   ```
+ *
+ * @param explicitId - The value passed explicitly by the caller (may be undefined)
+ * @param env        - The environment variables map (defaults to `process.env`)
+ * @returns          - The resolved template ID, or undefined if none is set
+ */
+export declare function resolveE2bTemplateId(explicitId: string | undefined, env?: NodeJS.ProcessEnv): string | undefined;
+export {};

package/mcp-server/dist/env-utils.js

@@ -0,0 +1,63 @@
+/**
+ * Utilities for resolving TestDriver environment variables.
+ *
+ * These helpers centralise the "param > env var > default" fallback logic
+ * so the server handler stays clean and the logic can be unit-tested
+ * independently.
+ */
+// ============================================================
+// OS resolution
+// ============================================================
+const VALID_OS_VALUES = ["linux", "windows"];
+/**
+ * Resolve the sandbox OS with the following priority:
+ *   1. Explicit `os` parameter passed to session_start
+ *   2. `TD_OS` environment variable (validated against supported values)
+ *   3. `"linux"` default
+ *
+ * Mirrors the behaviour in `hooks.mjs` so the MCP server and the
+ * Vitest integration are consistent.
+ *
+ * @param explicitOs - The value passed explicitly by the caller (may be undefined)
+ * @param env        - The environment variables map (defaults to `process.env`)
+ * @returns          - The resolved OS and a warning message if TD_OS was invalid
+ */
+export function resolveOs(explicitOs, env = process.env) {
+    if (explicitOs) {
+        return { os: explicitOs };
+    }
+    const envOs = env.TD_OS;
+    if (envOs) {
+        if (VALID_OS_VALUES.includes(envOs)) {
+            return { os: envOs };
+        }
+        return {
+            os: "linux",
+            warning: `TD_OS has unsupported value "${envOs}" (supported: ${VALID_OS_VALUES.join(", ")}), ignoring`,
+        };
+    }
+    return { os: "linux" };
+}
+// ============================================================
+// E2B template ID resolution
+// ============================================================
+/**
+ * Resolve the E2B template ID with the following priority:
+ *   1. Explicit `e2bTemplateId` parameter passed to session_start
+ *   2. `TD_E2B_TEMPLATE_ID` environment variable
+ *   3. `undefined` (let the SDK use its own default)
+ *
+ * Mirrors the behaviour in `hooks.mjs`:
+ *   ```js
+ *   if (!config.e2bTemplateId && process.env.TD_E2B_TEMPLATE_ID) {
+ *     config.e2bTemplateId = process.env.TD_E2B_TEMPLATE_ID;
+ *   }
+ *   ```
+ *
+ * @param explicitId - The value passed explicitly by the caller (may be undefined)
+ * @param env        - The environment variables map (defaults to `process.env`)
+ * @returns          - The resolved template ID, or undefined if none is set
+ */
+export function resolveE2bTemplateId(explicitId, env = process.env) {
+    return explicitId || env.TD_E2B_TEMPLATE_ID;
+}

package/mcp-server/dist/provision-types.d.ts

@@ -147,7 +147,7 @@ export declare const SessionStartInputSchema: z.ZodObject<{
     /** Electron args (for electron) */
     electronArgs: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     /** Operating system for the sandbox */
-    os: z.ZodDefault<z.ZodEnum<["linux", "windows"]>>;
+    os: z.ZodOptional<z.ZodEnum<["linux", "windows"]>>;
     /** Keep sandbox alive duration in ms (default: 5 minutes) */
     keepAlive: z.ZodDefault<z.ZodNumber>;
     /** Path to test file - when provided, you MUST append generated code to this file after each action */
@@ -158,11 +158,12 @@ export declare const SessionStartInputSchema: z.ZodObject<{
     apiRoot: z.ZodOptional<z.ZodString>;
     /** Direct IP address of self-hosted instance (bypasses cloud provisioning) */
     ip: z.ZodOptional<z.ZodString>;
+    /** E2B sandbox template ID (overrides TD_E2B_TEMPLATE_ID env var) */
+    e2bTemplateId: z.ZodOptional<z.ZodString>;
     /** Sandbox ID to connect to (for debug-on-failure mode). When provided, connects to an existing sandbox instead of creating a new one. Skip provisioning. */
     sandboxId: z.ZodOptional<z.ZodString>;
 }, "strip", z.ZodTypeAny, {
     type: "chrome" | "chromeExtension" | "vscode" | "installer" | "electron";
-    os: "linux" | "windows";
     keepAlive: number;
     reconnect: boolean;
     url?: string | undefined;
@@ -178,9 +179,11 @@ export declare const SessionStartInputSchema: z.ZodObject<{
     installerUrl?: string | undefined;
     installerFilename?: string | undefined;
     electronArgs?: string[] | undefined;
+    os?: "linux" | "windows" | undefined;
     testFile?: string | undefined;
     apiRoot?: string | undefined;
     ip?: string | undefined;
+    e2bTemplateId?: string | undefined;
     sandboxId?: string | undefined;
 }, {
     url?: string | undefined;
@@ -203,6 +206,7 @@ export declare const SessionStartInputSchema: z.ZodObject<{
     reconnect?: boolean | undefined;
     apiRoot?: string | undefined;
     ip?: string | undefined;
+    e2bTemplateId?: string | undefined;
     sandboxId?: string | undefined;
 }>;
 export type SessionStartInput = z.infer<typeof SessionStartInputSchema>;

package/mcp-server/dist/provision-types.js

@@ -118,7 +118,7 @@ export const SessionStartInputSchema = z.object({
     electronArgs: z.array(z.string()).optional().describe("Additional Electron arguments"),
     // Common session options
     /** Operating system for the sandbox */
-    os: z.enum(["linux", "windows"]).default("linux").describe("Sandbox OS"),
+    os: z.enum(["linux", "windows"]).optional().describe("Sandbox OS (default: 'linux', or TD_OS environment variable if set)"),
     /** Keep sandbox alive duration in ms (default: 5 minutes) */
     keepAlive: z.number().default(300000).describe("Keep sandbox alive for this many ms"),
     /** Path to test file - when provided, you MUST append generated code to this file after each action */
@@ -130,6 +130,8 @@ export const SessionStartInputSchema = z.object({
     // Self-hosted connection options
     /** Direct IP address of self-hosted instance (bypasses cloud provisioning) */
     ip: z.string().optional().describe("Direct IP address of self-hosted Windows instance (e.g., from AWS). When provided, connects directly to this IP instead of using cloud provisioning."),
+    /** E2B sandbox template ID (overrides TD_E2B_TEMPLATE_ID env var) */
+    e2bTemplateId: z.string().optional().describe("E2B sandbox template ID. Overrides the TD_E2B_TEMPLATE_ID environment variable when provided."),
     // Debug mode - connect to existing sandbox
     /** Sandbox ID to connect to (for debug-on-failure mode). When provided, connects to an existing sandbox instead of creating a new one. Skip provisioning. */
     sandboxId: z.string().optional().describe("Existing sandbox ID to connect to (from debug-on-failure mode). Skips provisioning."),