testdriverai 5.0.0-beta.8 → 5.0.2

package/agent.js

@@ -130,8 +130,9 @@ const thisCommand = a.command;
 logger.info(chalk.green(`Howdy! I'm TestDriver v${package.version}`));
 logger.info(chalk.dim(`Working on ${thisFile}`));
 logger.info("");
-logger.info(chalk.yellow(`This is beta software!`));
-logger.info(`Join our Discord for help`);
+logger.info(`This is beta software!`);
+logger.info("");
+logger.info(chalk.yellow(`Join our Discord for help`));
 logger.info(`https://discord.com/invite/cWDFW8DzPm`);
 logger.info("");
 
@@ -1145,14 +1146,17 @@ const start = async () => {
   if (thisCommand !== "run") {
     speak("Howdy! I am TestDriver version " + package.version);
 
-    logger.info(
-      chalk.red("Warning!") +
-        chalk.dim(" TestDriver sends screenshots of the desktop to our API."),
-    );
-    logger.info(
-      chalk.dim("https://docs.testdriver.ai/security-and-privacy/agent"),
-    );
-    logger.info("");
+    if (!config.TD_VM) {
+      logger.info(
+        chalk.red("Warning!") +
+          chalk.dim("Local mode sends screenshots of the desktop to our API."),
+      );
+      logger.info(
+        chalk.dim("https://docs.testdriver.ai/security-and-privacy/agent"),
+      );
+      logger.info("");
+    }
+    
   }
 
   analytics.track("command", { command: thisCommand, file: thisFile });
@@ -1174,23 +1178,31 @@ const makeSandbox = async () => {
 
   if (config.TD_VM) {
           
-    logger.info(chalk.gray(`- creating linux sandbox...`));
-    await sandbox.boot();
-    logger.info(chalk.gray(`- authenticating...`));
-    await sandbox.send({type: 'authenticate', apiKey: config.TD_API_KEY });
-    logger.info(chalk.gray(`- setting up...`));
-    await sandbox.send({type: 'create', resolution: [1024, 768]});
-    logger.info(chalk.gray(`- starting stream...`));
-    await sandbox.send({type: 'stream.start'});
-    let {url} = await sandbox.send({type: 'stream.getUrl'});
-    logger.info(chalk.gray(`- rendering...`));
-    await sandbox.send({type: 'ready'});
-    emitter.emit(events.vm.show, {url});
-    logger.info(chalk.gray(`- booting...`));
-    await new Promise(resolve => setTimeout(resolve, 3000)); 
-    logger.info(chalk.green(``));
-    logger.info(chalk.green(`sandbox runner ready!`));
+    try {
 
+      logger.info(chalk.gray(`- creating linux sandbox...`));
+      await sandbox.boot();
+      logger.info(chalk.gray(`- authenticating...`));
+      await sandbox.send({type: 'authenticate', apiKey: config.TD_API_KEY });
+      logger.info(chalk.gray(`- setting up...`));
+      await sandbox.send({type: 'create', resolution: [1024, 768]});
+      logger.info(chalk.gray(`- starting stream...`));
+      await sandbox.send({type: 'stream.start'});
+      let {url} = await sandbox.send({type: 'stream.getUrl'});
+      logger.info(chalk.gray(`- rendering...`));
+      await sandbox.send({type: 'ready'});
+      emitter.emit(events.vm.show, {url});
+      logger.info(chalk.gray(`- booting...`));
+      await new Promise(resolve => setTimeout(resolve, 3000)); 
+      logger.info(chalk.green(``));
+      logger.info(chalk.green(`sandbox runner ready!`));
+
+    } catch  (e) {
+      logger.error(e)
+      logger.error(chalk.red(`sandbox runner failed to start`));
+      process.exit(1);
+    }
+ 
   }
 
   emitter.emit(events.interactive, false);

package/electron/overlay.js

@@ -48,10 +48,10 @@ app.whenReady().then(() => {
       autoHideMenuBar: true,
     };
 
-  }
+    if (process.platform !== 'darwin') {
+      windowOptions.fullscreen = true;
+    }
 
-  if (process.platform !== 'darwin') {
-    windowOptions.fullscreen = true;
   }
 
   const window = new BrowserWindow(windowOptions);

package/lib/commands.js

@@ -10,7 +10,6 @@ const {
 
 const { focusApplication } = require("./focus-application");
 const fs = require("fs").promises; // Using the promises version for async operations
-const robot = require("robotjs");
 const { findTemplateImage } = require("./subimage/index");
 const { cwd } = require("node:process");
 const path = require("path");
@@ -20,11 +19,13 @@ const cliProgress = require("cli-progress");
 const redraw = require("./redraw");
 const sandbox = require("./sandbox.js");
 const config = require("./config.js");
+let robot;
 
 let keymap;
 if (config.TD_VM) {
   keymap = require("./keymaps/sandbox.js");
 } else {
+  robot = require("robotjs");
   keymap = require("./keymaps/robot.js");
 }
 

package/lib/config.js

@@ -26,7 +26,7 @@ const config = {
   TD_ANALYTICS: true,
   TD_NOTIFY: false,
   TD_MINIMIZE: true,
-  TD_API_ROOT: "https://replayable-api-production.herokuapp.com",
+  TD_API_ROOT: "https://api.testdriver.ai",
   TD_API_KEY: null,
   TD_DEV: parseValue(process.env["DEV"]),
   TD_PROFILE: false,

package/lib/focus-application.js

@@ -3,9 +3,14 @@ const path = require("path");
 const { execSync } = require("child_process");
 const { platform } = require("./system");
 const scriptPath = path.join(__dirname, "focusWindow.ps1");
-const robot = require("robotjs");
 const { logger } = require("./logger");
 const sandbox = require("./sandbox");
+const config = require("./config.js");
+let robot;
+
+if (!config.TD_VM) {
+  robot = require("robotjs");
+}
 
 async function focusVSCode() {
 

package/lib/sandbox.js

@@ -28,12 +28,18 @@ class Sandbox {
       this.socket = new WebSocket(`${config.TD_API_ROOT.replace('https://', 'wss://')}`);
 
       // handle errors
+      this.socket.on('close', (err) => {
+        console.log('Socket Closed. Check your API KEY (TD_API_KEY)');
+        clearInterval(this.heartbeat);
+        reject();
+        process.exit(1);
+      }); 
+
       this.socket.on('error', (err) => {
-        console.log('Socket Closed');
-        console.log('If this happens during setup, check your API Key (`TD_API_KEY` in `.env`).');
+        console.log('Socket Error');
         err && console.log(err);
         clearInterval(this.heartbeat);
-        reject();
+        throw err;
       }); 
 
       this.socket.on('open', async () => {

package/lib/system.js

@@ -3,11 +3,15 @@ const fs = require("fs");
 const os = require("os");
 const path = require("path");
 const si = require("systeminformation");
-const robot = require("robotjs");
 const sharp = require("sharp");
 const { emitter, events } = require("./events.js");
 const sandbox = require("./sandbox.js");
 const config = require("./config.js");
+let robot;
+
+if (!config.TD_VM) {
+  robot = require("robotjs");
+}
 
 let scshotdesk;
 if (!config.TD_VM) {
@@ -179,7 +183,7 @@ const activeWin = async () => {
 
 const getMousePosition = async () => {
   // @todo vm does expose mouse position, add it back
-  return config.TD_VM ? await robot.getMousePos() : null;
+  return !config.TD_VM ? await robot.getMousePos() : null;
 };
 
 module.exports = {

package/package.json

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

package/postinstall.js

@@ -1,5 +1,7 @@
 let platform = require("os").platform();
 let exec = require("child_process").exec;
+const { execSync } = require("child_process");
+const readline = require("readline");
 
 if (platform !== "darwin") {
   console.log("TestDriver Setup: Skipping codesign becasue not on Mac");
@@ -18,3 +20,25 @@ exec(signScript, (error, stdout, stderr) => {
   console.log(`stdout: ${stdout}`);
   console.error(`stderr: ${stderr}`);
 });
+
+const rl = readline.createInterface({
+  input: process.stdin,
+  output: process.stdout
+});
+
+rl.question("Do you want to install Vue support? (y/n) ", (answer) => {
+  if (answer.toLowerCase() === "y") {
+    console.log("Installing Vue dependencies...");
+    execSync("npm install -g vue vue-router", { stdio: "inherit" });
+  }
+
+  rl.question("Do you want to install React support? (y/n) ", (answer2) => {
+    if (answer2.toLowerCase() === "y") {
+      console.log("Installing React dependencies...");
+      execSync("npm install -g react react-dom", { stdio: "inherit" });
+    }
+
+    console.log("Setup complete!");
+    rl.close();
+  });
+});