testdriverai 5.1.0 → 5.1.1

package/agent.js

@@ -691,6 +691,26 @@ const actOnMarkdown = async (content, depth, pushToHistory = false, dry = false)
   }
 };
 
+const ensureMacScreenPerms = async () => {
+
+  // if os is mac, check for screen capture permissions
+  if (
+    !config.TD_VM &&
+    process.platform === "darwin" &&
+    !macScreenPerms.hasScreenCapturePermission()
+  ) {
+    logger.info(chalk.red("Screen capture permissions not enabled."));
+    logger.info(
+      "You must enable screen capture permissions for the application calling `testdriverai`.",
+    );
+    logger.info(
+      "Read More: https://docs.testdriver.ai/faq/screen-recording-permissions-mac-only",
+    );
+    analytics.track("noMacPermissions");
+    return exit();
+  }
+}
+
 const newSession = async () => {
   // should be start of new session
   const sessionRes = await sdk.req("session/start", {
@@ -1122,6 +1142,13 @@ const embed = async (file, depth) => {
   logger.info(`${file} (end)`);
 };
 
+const buildEnv = async () => {
+  let win = await system.activeWin();
+  setTerminalApp(win);
+  await ensureMacScreenPerms();
+  await makeSandbox();
+}
+
 const start = async () => {
   // logger.info(await  system.getPrimaryDisplay());
 
@@ -1135,28 +1162,12 @@ const start = async () => {
 
   // await sdk.auth();
 
-  // if os is mac, check for screen capture permissions
-  if (
-    process.platform === "darwin" &&
-    !macScreenPerms.hasScreenCapturePermission()
-  ) {
-    logger.info(chalk.red("Screen capture permissions not enabled."));
-    logger.info(
-      "You must enable screen capture permissions for the application calling `testdriverai`.",
-    );
-    logger.info(
-      "Read More: https://docs.testdriver.ai/faq/screen-recording-permissions-mac-only",
-    );
-    analytics.track("noMacPermissions");
-    return exit();
-  }
-
   if (thisCommand !== "run") {
     speak("Howdy! I am TestDriver version " + package.version);
 
     if (!config.TD_VM) {
       logger.info(
-        chalk.red("Warning!") +
+        chalk.red("Warning!" ) +
           chalk.dim("Local mode sends screenshots of the desktop to our API."),
       );
       logger.info(
@@ -1170,11 +1181,11 @@ const start = async () => {
   analytics.track("command", { command: thisCommand, file: thisFile });
 
   if (thisCommand == "edit") {
-    await makeSandbox();
+    await buildEnv();
     firstPrompt();
   } else if (thisCommand == "run") {
+    await buildEnv();
     errorLimit = 100;
-    await makeSandbox();
     run(thisFile);
   } else if (thisCommand == "init") {
     await init();

package/index.js

@@ -10,11 +10,9 @@ if (process.argv[2] === "--help" || process.argv[2] === "-h") {
 }
 
 (async () => {
-  let win = await system.activeWin();
 
   if (!config.TD_OVERLAY) {
     let agent = require("./agent.js");
-    agent.setTerminalApp(win);
     agent.start();
   } else {
     // Intercept all stdout and stderr calls (works with console as well)

package/lib/logger.js

@@ -19,6 +19,13 @@ const logFormat = printf(({ message }) => {
 
 let interpolationVars = JSON.parse(process.env.TD_INTERPOLATION_VARS || '{}');
 
+// this handles local `TD_*` variables
+for (const [key, value] of Object.entries(process.env)) {
+  if (key.startsWith("TD_") && key !== "TD_INTERPOLATION_VARS") {
+    interpolationVars[key] = value;
+  }
+}
+
 const censorSensitiveData = (message) => {
   for (let value of Object.values(interpolationVars)) {
 

package/lib/system.js

@@ -171,19 +171,10 @@ const initializeActiveWindow = async () => {
 const activeWin = async () => {
   
   if (config.TD_VM) { 
-    
-  return "error getting active window, proceed normally";
-
+    return "error getting active window, proceed normally";
   } else {
-
-    try {
-      const activeWindow = await initializeActiveWindow();
-      return await activeWindow();
-    } catch (error) {
-      logger.error('Error getting active window: %s', error);
-      return null;
-    }
-
+    const activeWindow = await initializeActiveWindow();
+    return await activeWindow();
   }
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "5.1.0",
+  "version": "5.1.1",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "index.js",
   "bin": {