testdriverai 5.5.4 → 5.5.5

package/lib/sdk.js

@@ -9,7 +9,7 @@ const axios = require('axios');
 
 const { logger } = require('./logger');
 
-// let token = null;
+let token = null;
 
 const outputError = async (error) => {
   logger.info(chalk.red(error.status), chalk.red(error.statusText));
@@ -62,29 +62,32 @@ const parseBody = async (response, body) => {
 };
 
 let auth = async () => {
-  // data.apiKey = process.env.DASHCAM_API_KEY; @todo add-auth
 
-  // if (!data.apiKey) {
-  //   logger.info(chalk.red('API key not found. Set DASHCAM_API_KEY in your environment.'));
-  //   process.exit(1);
-  // }
-
-  const url = [root, "auth/exchange-api-key"].join("/");
-  const config = {
-    method: "post",
-    headers: {
-      "Content-Type": "application/json",
-    },
-    // data
-  };
-
-  try {
-    await axios(url, config);
-    // token = res.data.token;
-  } catch (error) {
-    await outputError(error);
-    process.exit(1);
+  if (config["TD_API_KEY"]) {
+
+    const url = [root, "auth/exchange-api-key"].join("/");
+    const c = {
+      method: "post",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      data: {
+        apiKey: config["TD_API_KEY"],
+        version
+      }
+    };
+  
+    try {
+      let res = await axios(url, c);
+  
+      token = res.data.token;
+      return token;
+    } catch (error) {
+      await outputError(error);
+      process.exit(1);
+    }
   }
+
 };
 
 const req = async (path, data, onChunk) => {
@@ -98,17 +101,19 @@ const req = async (path, data, onChunk) => {
   const url = path.startsWith("/api")
     ? [root, path].join("")
     : [root, "api", version, "testdriver", path].join("/");
-
-  const config = {
-    method: "post",
-    headers: { "Content-Type": "application/json" }, 
-    responseType: typeof onChunk === "function" ? "stream" : "json",
-    data: {
+    const config = {
+      method: "post",
+      headers: { 
+      "Content-Type": "application/json",
+      ...(token && { "Authorization": `Bearer ${token}` }) // Add the authorization bearer token only if token is set
+      }, 
+      responseType: typeof onChunk === "function" ? "stream" : "json",
+      data: {
       ...data,
       session: session.get(),
       stream: typeof onChunk === "function",
-    },
-  };
+      },
+    };
 
   try {
     let response;

package/package.json

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