testdriverai 7.8.0-test.7 → 7.8.0-test.71

package/agent/index.js

@@ -70,6 +70,7 @@ class TestDriverAgent extends EventEmitter2 {
     this.sandboxId = flags["sandbox-id"] || null;
     this.sandboxAmi = flags["sandbox-ami"] || null;
     this.sandboxInstance = flags["sandbox-instance"] || null;
+    this.e2bTemplateId = flags["e2b-template-id"] || null;
     this.sandboxOs = flags.os || "linux";
     this.ip = flags.ip || null;
     this.workingDir = flags.workingDir || process.cwd();
@@ -211,6 +212,14 @@ class TestDriverAgent extends EventEmitter2 {
       }
     }
 
+    // End the Sentry root session span so the trace is finalized
+    try {
+      const sentry = require("../lib/sentry");
+      sentry.clearSessionTraceContext(this.session && this.session.get());
+    } catch (e) {
+      // Sentry module may not be available, ignore
+    }
+
     shouldRunPostrun =
       !this.hasRunPostrun &&
       (shouldRunPostrun || this.cliArgs?.command == "run");
@@ -1941,18 +1950,19 @@ ${regression}
     // Allow explicit override via env (e.g. VITE_DOMAIN from .env)
     if (process.env.VITE_DOMAIN) return process.env.VITE_DOMAIN;
 
+    const environments = require("../lib/environments.json");
     const mapping = {
-      "https://api.testdriver.ai": "https://console.testdriver.ai",
-      "https://v6.testdriver.ai": "https://console.testdriver.ai",
-      "https://api-canary.testdriver.ai": "https://console-canary.testdriver.ai",
-      "https://api-test.testdriver.ai": "https://console-test.testdriver.ai",
+      "https://v6.testdriver.ai": environments.stable.consoleUrl,
     };
+    for (const env of Object.values(environments)) {
+      mapping[env.apiRoot] = env.consoleUrl;
+    }
     if (mapping[apiRoot]) return mapping[apiRoot];
     // Local dev: API on localhost:1337 -> Web on localhost:3001
     if (apiRoot.includes("localhost:1337") || apiRoot.includes("127.0.0.1:1337")) {
       return "http://localhost:3001";
     }
-    return "https://console.testdriver.ai";
+    return environments.stable.consoleUrl;
   }
 
   // Write session file for IDE preview (VSCode extension watches for these)
@@ -2187,6 +2197,9 @@ Please check your network connection, TD_API_KEY, or the service status.`,
     if (this.sandboxInstance) {
       sandboxConfig.instanceType = this.sandboxInstance;
     }
+    if (this.e2bTemplateId) {
+      sandboxConfig.e2bTemplateId = this.e2bTemplateId;
+    }
     // Add keepAlive TTL if specified
     if (this.keepAlive !== undefined && this.keepAlive !== null) {
       sandboxConfig.keepAlive = this.keepAlive;

package/agent/lib/commands.js

@@ -580,11 +580,12 @@ const createCommands = (
         } else if (action === "mouseDown") {
           await sandbox.send({ type: "mousePress", button: "left", x, y, ...elementData });
         } else if (action === "mouseUp") {
+          // Move first to create drag motion, then release
+          // (pyautogui.mouseUp with x/y teleports instead of dragging)
+          await sandbox.send({ type: "moveMouse", x, y, ...elementData });
           await sandbox.send({
             type: "mouseRelease",
             button: "left",
-            x,
-            y,
             ...elementData
           });
         }

package/agent/lib/http.js

@@ -0,0 +1,162 @@
+/**
+ * Shared HTTP client for the TestDriver SDK.
+ *
+ * All SDK HTTP traffic should go through these helpers so that
+ * User-Agent, timeouts, Sentry tracing headers, and response
+ * parsing are handled in one place.
+ *
+ * Uses axios under the hood — the same library the rest of the SDK
+ * already depends on.
+ */
+
+const axios = require("axios");
+const crypto = require("crypto");
+const { version } = require("../../package.json");
+
+const USER_AGENT = `TestDriverSDK/${version} (Node.js ${process.version})`;
+
+/**
+ * Generate Sentry distributed-tracing headers.
+ *
+ * When Sentry is initialized and a span is active, uses Sentry.getTraceData()
+ * so the headers reference the real active span (proper parent-child linkage).
+ * Falls back to MD5(sessionId)-based headers when Sentry is not available or
+ * has no active span (e.g. TD_TELEMETRY=false).
+ *
+ * @param {string} sessionId
+ * @returns {object} Headers object (empty if no sessionId and no active span)
+ */
+function getSentryTraceHeaders(sessionId) {
+  // Prefer Sentry's own trace propagation when available
+  try {
+    const Sentry = require("@sentry/node");
+    if (typeof Sentry.getTraceData === "function") {
+      const traceData = Sentry.getTraceData();
+      if (traceData && traceData["sentry-trace"]) {
+        return traceData;
+      }
+    }
+  } catch (e) {
+    // Sentry not available — fall through to manual derivation
+  }
+
+  // Fallback: derive deterministic trace from session ID
+  if (!sessionId) return {};
+  const traceId = crypto.createHash("md5").update(sessionId).digest("hex");
+  const spanId = crypto.randomBytes(8).toString("hex");
+  return {
+    "sentry-trace": traceId + "-" + spanId + "-1",
+    baggage:
+      "sentry-trace_id=" +
+      traceId +
+      ",sentry-sample_rate=1.0,sentry-sampled=true",
+  };
+}
+
+/**
+ * Build common request headers.
+ * @param {object} [extra] - Additional headers to merge
+ * @returns {object}
+ */
+function baseHeaders(extra) {
+  return {
+    "User-Agent": USER_AGENT,
+    ...extra,
+  };
+}
+
+/**
+ * POST JSON to `url` and return the parsed response body.
+ *
+ * @param {string} url    - Absolute URL
+ * @param {object} [data] - JSON body
+ * @param {object} [opts] - Extra axios config (headers, timeout, …)
+ * @returns {Promise<object>} Parsed response data
+ */
+async function httpPost(url, data, opts = {}) {
+  const { headers: extraHeaders, ...rest } = opts;
+  const res = await axios({
+    method: "post",
+    url,
+    headers: baseHeaders({
+      "Content-Type": "application/json",
+      ...extraHeaders,
+    }),
+    data,
+    timeout: opts.timeout || 30000,
+    ...rest,
+  });
+  return res.data;
+}
+
+/**
+ * GET `url` and return the parsed response body.
+ *
+ * @param {string} url    - Absolute URL
+ * @param {object} [opts] - Extra axios config
+ * @returns {Promise<object>} Parsed response data
+ */
+async function httpGet(url, opts = {}) {
+  const { headers: extraHeaders, ...rest } = opts;
+  const res = await axios({
+    method: "get",
+    url,
+    headers: baseHeaders(extraHeaders),
+    timeout: opts.timeout || 30000,
+    ...rest,
+  });
+  return res.data;
+}
+
+/**
+ * PUT data to `url` (e.g. S3 presigned upload).
+ *
+ * @param {string} url    - Absolute URL
+ * @param {Buffer|string} data - Request body
+ * @param {object} [opts] - Extra axios config (headers, timeout, …)
+ * @returns {Promise<object>} Parsed response data (or empty object for 2xx with no body)
+ */
+async function httpPut(url, data, opts = {}) {
+  const { headers: extraHeaders, ...rest } = opts;
+  const res = await axios({
+    method: "put",
+    url,
+    headers: baseHeaders(extraHeaders),
+    data,
+    timeout: opts.timeout || 30000,
+    maxBodyLength: Infinity,
+    maxContentLength: Infinity,
+    ...rest,
+  });
+  return res.data;
+}
+
+/**
+ * Download a URL as a Buffer (e.g. screenshot from S3).
+ *
+ * @param {string} url    - Absolute URL
+ * @param {object} [opts] - Extra axios config
+ * @returns {Promise<Buffer>}
+ */
+async function downloadBuffer(url, opts = {}) {
+  const { headers: extraHeaders, ...rest } = opts;
+  const res = await axios({
+    method: "get",
+    url,
+    headers: baseHeaders(extraHeaders),
+    responseType: "arraybuffer",
+    timeout: opts.timeout || 60000,
+    ...rest,
+  });
+  return Buffer.from(res.data);
+}
+
+module.exports = {
+  httpPost,
+  httpGet,
+  httpPut,
+  downloadBuffer,
+  getSentryTraceHeaders,
+  USER_AGENT,
+  baseHeaders,
+};

package/agent/lib/logger.js

@@ -7,6 +7,7 @@
  */
 
 const useStderr = process.env.TD_STDIO === 'stderr';
+const isDebug = process.env.TD_DEBUG === 'true' || process.env.VERBOSE === 'true';
 
 /**
  * Log a message - uses stdout by default, stderr if TD_STDIO=stderr
@@ -40,6 +41,19 @@ function warn(...args) {
   }
 }
 
+/**
+ * Log a debug message - only outputs when DEBUG=true
+ * @param {...any} args - Arguments to log
+ */
+function debug(...args) {
+  if (!isDebug) return;
+  if (useStderr) {
+    console.error(...args);
+  } else {
+    console.log(...args);
+  }
+}
+
 /**
  * Check if logger is configured to use stderr
  * @returns {boolean}
@@ -50,6 +64,7 @@ function isStderrMode() {
 
 module.exports = {
   log,
+  debug,
   error,
   warn,
   isStderrMode,