@lambdatest/smartui-cli 4.1.2 → 4.1.3-beta.0

package/dist/index.cjs

@@ -625,6 +625,10 @@ var ConfigSchema = {
     useGlobalCache: {
       type: "boolean",
       errorMessage: "Invalid config; useGlobalCache must be true/false"
+    },
+    skipBuildCreation: {
+      type: "boolean",
+      errorMessage: "Invalid config; skipBuildCreation must be true/false"
     }
   },
   anyOf: [
@@ -845,6 +849,10 @@ var SnapshotSchema = {
         loadDomContent: {
           type: "boolean",
           errorMessage: "Invalid snapshot options; loadDomContent must be a boolean"
+        },
+        sessionId: {
+          type: "string",
+          errorMessage: "Invalid snapshot options; sessionId must be a string"
         }
       },
       additionalProperties: false
@@ -992,7 +1000,6 @@ var validateSnapshot = ajv.compile(SnapshotSchema);
 var validateFigmaDesignConfig = ajv.compile(FigmaDesignConfigSchema);
 var validateWebFigmaConfig = ajv.compile(FigmaWebConfigSchema);
 util.promisify(setTimeout);
-var isPollingActive = false;
 function delDir(dir) {
   if (fs5__default.default.existsSync(dir)) {
     fs5__default.default.rmSync(dir, { recursive: true });
@@ -1170,38 +1177,37 @@ function getRenderViewportsForOptions(options) {
     ...mobileRenderViewports[constants_default.MOBILE_OS_ANDROID]
   ];
 }
-process.on("SIGINT", () => __async(void 0, null, function* () {
-  if (isPollingActive) {
-    console.log("Fetching results interrupted. Exiting...");
-    isPollingActive = false;
-  } else {
-    console.log("\nExiting gracefully...");
-  }
-  process.exit(0);
-}));
-function startPolling(ctx) {
+function startPolling(ctx, build_id, baseline, projectToken) {
   return __async(this, null, function* () {
-    ctx.log.info("Fetching results in progress....");
-    ctx.log.debug(ctx.build);
-    isPollingActive = true;
+    if (build_id) {
+      ctx.log.info(`Fetching results for buildId ${build_id} in progress....`);
+    } else if (ctx.build && ctx.build.id) {
+      ctx.log.info(`Fetching results for buildId ${ctx.build.id} in progress....`);
+    }
     const intervalId = setInterval(() => __async(this, null, function* () {
-      if (!isPollingActive) {
-        clearInterval(intervalId);
-        return;
-      }
       try {
-        const resp = yield ctx.client.getScreenshotData(ctx.build.id, ctx.build.baseline || false, ctx.log);
+        let resp;
+        if (build_id) {
+          resp = yield ctx.client.getScreenshotData(build_id, baseline, ctx.log, projectToken);
+        } else if (ctx.build && ctx.build.id) {
+          resp = yield ctx.client.getScreenshotData(ctx.build.id, ctx.build.baseline, ctx.log, "");
+        } else {
+          return;
+        }
         if (!resp.build) {
           ctx.log.info("Error: Build data is null.");
           clearInterval(intervalId);
-          isPollingActive = false;
+          return;
+        }
+        let fileName = `${resp.build.build_id}.json`;
+        if (ctx.options.fetchResults && ctx.options.fetchResultsFileName && ctx.build && ctx.build.id && resp.build.build_id === ctx.build.id) {
+          fileName = `${ctx.options.fetchResultsFileName}`;
         }
-        fs5__default.default.writeFileSync(ctx.options.fetchResultsFileName, JSON.stringify(resp, null, 2));
-        ctx.log.debug(`Updated results in ${ctx.options.fetchResultsFileName}`);
+        fs5__default.default.writeFileSync(`${fileName}`, JSON.stringify(resp, null, 2));
+        ctx.log.debug(`Updated results in ${fileName}`);
         if (resp.build.build_status_ind === constants_default.BUILD_COMPLETE || resp.build.build_status_ind === constants_default.BUILD_ERROR) {
           clearInterval(intervalId);
-          ctx.log.info(`Fetching results completed. Final results written to ${ctx.options.fetchResultsFileName}`);
-          isPollingActive = false;
+          ctx.log.info(`Fetching results completed. Final results written to ${fileName}`);
           let totalScreenshotsWithMismatches = 0;
           let totalVariantsWithMismatches = 0;
           const totalScreenshots = Object.keys(resp.screenshots || {}).length;
@@ -1224,7 +1230,7 @@ function startPolling(ctx) {
           ctx.log.info(
             chalk__default.default.green.bold(
               `
-Summary of Mismatches:
+Summary of Mismatches for buildId: ${build_id}
 ${chalk__default.default.yellow("Total Variants with Mismatches:")} ${chalk__default.default.white(totalVariantsWithMismatches)} out of ${chalk__default.default.white(totalVariants)}
 ${chalk__default.default.yellow("Total Screenshots with Mismatches:")} ${chalk__default.default.white(totalScreenshotsWithMismatches)} out of ${chalk__default.default.white(totalScreenshots)}
 ${chalk__default.default.yellow("Branch Name:")} ${chalk__default.default.white(resp.build.branch)}
@@ -1242,7 +1248,6 @@ ${chalk__default.default.yellow("Build ID:")} ${chalk__default.default.white(res
           ctx.log.error(`Error fetching screenshot data: ${error.message}`);
         }
         clearInterval(intervalId);
-        isPollingActive = false;
       }
     }), 5e3);
   });
@@ -1289,15 +1294,37 @@ var server_default = (ctx) => __async(void 0, null, function* () {
     reply.code(200).send({ data: { dom: SMARTUI_DOM } });
   });
   server.post("/snapshot", opts, (request, reply) => __async(void 0, null, function* () {
-    var _a;
+    var _a, _b, _c;
     let replyCode;
     let replyBody;
     try {
       let { snapshot, testType } = request.body;
       if (!validateSnapshot(snapshot))
         throw new Error(validateSnapshot.errors[0].message);
+      const sessionId = (_a = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _a.sessionId;
+      let capsBuildId = "";
+      if (sessionId) {
+        if ((_b = ctx.sessionCapabilitiesMap) == null ? void 0 : _b.has(sessionId)) {
+          const cachedCapabilities = ctx.sessionCapabilitiesMap.get(sessionId);
+          capsBuildId = (cachedCapabilities == null ? void 0 : cachedCapabilities.buildId) || "";
+        } else {
+          try {
+            let fetchedCapabilitiesResp = yield ctx.client.getSmartUICapabilities(sessionId, ctx.config, ctx.git, ctx.log);
+            capsBuildId = (fetchedCapabilitiesResp == null ? void 0 : fetchedCapabilitiesResp.buildId) || "";
+            ctx.log.debug(`fetch caps for sessionId: ${sessionId} are ${JSON.stringify(fetchedCapabilitiesResp)}`);
+            if (capsBuildId) {
+              ctx.sessionCapabilitiesMap.set(sessionId, fetchedCapabilitiesResp);
+            } else if (fetchedCapabilitiesResp && (fetchedCapabilitiesResp == null ? void 0 : fetchedCapabilitiesResp.sessionId)) {
+              ctx.sessionCapabilitiesMap.set(sessionId, fetchedCapabilitiesResp);
+            }
+          } catch (error) {
+            ctx.log.debug(`Failed to fetch capabilities for sessionId ${sessionId}: ${error.message}`);
+            console.log(`Failed to fetch capabilities for sessionId ${sessionId}: ${error.message}`);
+          }
+        }
+      }
       ctx.testType = testType;
-      (_a = ctx.snapshotQueue) == null ? void 0 : _a.enqueue(snapshot);
+      (_c = ctx.snapshotQueue) == null ? void 0 : _c.enqueue(snapshot);
       ctx.isSnapshotCaptured = true;
       replyCode = 200;
       replyBody = { data: { message: "success", warnings: [] } };
@@ -1457,21 +1484,28 @@ var startServer_default = (ctx) => {
     rendererOptions: { persistentOutput: true }
   };
 };
-var auth_default = (ctx) => {
+var authExec_default = (ctx) => {
   return {
     title: `Authenticating with SmartUI`,
     task: (ctx2, task) => __async(void 0, null, function* () {
       updateLogContext({ task: "auth" });
       try {
-        const authResult = yield ctx2.client.auth(ctx2.log, ctx2.env);
-        if (authResult === 2) {
-          task.output = chalk__default.default.gray(`New project '${ctx2.env.PROJECT_NAME}' created successfully`);
-        } else if (authResult === 0) {
-          task.output = chalk__default.default.gray(`Using existing project token '******#${ctx2.env.PROJECT_TOKEN.split("#").pop()}'`);
-        } else if (authResult === 1) {
-          task.output = chalk__default.default.gray(`Using existing project '${ctx2.env.PROJECT_NAME}'`);
+        if (!ctx2.env.PROJECT_NAME && !ctx2.env.PROJECT_TOKEN) {
+          ctx2.authenticatedInitially = false;
+          task.output = chalk__default.default.gray(`Empty PROJECT_TOKEN and PROJECT_NAME. Skipping authentication. Expecting SmartUI Capabilities in driver!`);
+          task.title = "Skipped Authentication with SmartUI";
+        } else {
+          const authResult = yield ctx2.client.auth(ctx2.log, ctx2.env);
+          if (authResult === 2) {
+            task.output = chalk__default.default.gray(`New project '${ctx2.env.PROJECT_NAME}' created successfully`);
+          } else if (authResult === 0) {
+            task.output = chalk__default.default.gray(`Using existing project token '******#${ctx2.env.PROJECT_TOKEN.split("#").pop()}'`);
+          } else if (authResult === 1) {
+            task.output = chalk__default.default.gray(`Using existing project '${ctx2.env.PROJECT_NAME}'`);
+          }
+          ctx2.authenticatedInitially = true;
+          task.title = "Authenticated with SmartUI";
         }
-        task.title = "Authenticated with SmartUI";
       } catch (error) {
         ctx2.log.debug(error);
         task.output = chalk__default.default.gray(error.message);
@@ -1483,7 +1517,7 @@ var auth_default = (ctx) => {
 };
 
 // package.json
-var version = "4.1.2";
+var version = "4.1.3-beta.0";
 var package_default = {
   name: "@lambdatest/smartui-cli",
   version,
@@ -1563,7 +1597,9 @@ var httpClient = class {
     }
     this.axiosInstance = axios__default.default.create(axiosConfig);
     this.axiosInstance.interceptors.request.use((config) => {
-      config.headers["projectToken"] = this.projectToken;
+      if (!config.headers["projectToken"]) {
+        config.headers["projectToken"] = this.projectToken;
+      }
       config.headers["projectName"] = this.projectName;
       config.headers["username"] = this.username;
       config.headers["accessKey"] = this.accessKey;
@@ -1661,11 +1697,12 @@ var httpClient = class {
       }
     }, log2);
   }
-  getScreenshotData(buildId, baseline, log2) {
+  getScreenshotData(buildId, baseline, log2, projectToken) {
     return this.request({
       url: "/screenshot",
       method: "GET",
-      params: { buildId, baseline }
+      params: { buildId, baseline },
+      headers: { projectToken }
     }, log2);
   }
   ping(buildId, log2) {
@@ -1677,6 +1714,25 @@ var httpClient = class {
       }
     }, log2);
   }
+  getSmartUICapabilities(sessionId, config, git, log2) {
+    return this.request({
+      url: "/sessions/capabilities",
+      method: "GET",
+      params: {
+        sessionId
+      },
+      data: {
+        git,
+        config
+      },
+      headers: {
+        projectToken: "",
+        projectName: "",
+        username: "",
+        accessKey: ""
+      }
+    }, log2);
+  }
   finalizeBuild(buildId, totalSnapshots, log2) {
     let params = { buildId };
     if (totalSnapshots > -1)
@@ -1687,6 +1743,20 @@ var httpClient = class {
       params
     }, log2);
   }
+  finalizeBuildForCapsWithToken(buildId, totalSnapshots, projectToken, log2) {
+    let params = { buildId };
+    if (totalSnapshots > -1)
+      params.totalSnapshots = totalSnapshots;
+    return this.request({
+      url: "/build",
+      method: "DELETE",
+      params,
+      headers: {
+        projectToken
+        // Use projectToken dynamically
+      }
+    }, log2);
+  }
   uploadSnapshot(ctx, snapshot) {
     return this.request({
       url: `/builds/${ctx.build.id}/snapshot`,
@@ -1718,6 +1788,45 @@ var httpClient = class {
       }
     }, ctx.log);
   }
+  processSnapshotCaps(ctx, snapshot, snapshotUuid, capsBuildId, capsProjectToken) {
+    return this.request({
+      url: `/build/${capsBuildId}/snapshot`,
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        projectToken: capsProjectToken !== "" ? capsProjectToken : this.projectToken
+      },
+      data: {
+        name: snapshot.name,
+        url: snapshot.url,
+        snapshotUuid,
+        test: {
+          type: ctx.testType,
+          source: "cli"
+        },
+        async: false
+      }
+    }, ctx.log);
+  }
+  uploadSnapshotForCaps(ctx, snapshot, capsBuildId, capsProjectToken) {
+    const buildId = capsBuildId !== "" ? capsBuildId : ctx.build.id;
+    return this.request({
+      url: `/builds/${buildId}/snapshot`,
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        projectToken: capsProjectToken !== "" ? capsProjectToken : this.projectToken
+        // Use capsProjectToken dynamically
+      },
+      data: {
+        snapshot,
+        test: {
+          type: ctx.testType,
+          source: "cli"
+        }
+      }
+    }, ctx.log);
+  }
   uploadScreenshot({ id: buildId, name: buildName, baseline }, ssPath, ssName, browserName, viewport, log2) {
     browserName = browserName === constants_default.SAFARI ? constants_default.WEBKIT : browserName;
     const file = fs5__default.default.readFileSync(ssPath);
@@ -1799,6 +1908,21 @@ var httpClient = class {
       }
     }, ctx.log);
   }
+  getS3PresignedURLForSnapshotUploadCaps(ctx, snapshotName, snapshotUuid, capsBuildId, capsProjectToken) {
+    return this.request({
+      url: `/snapshotuploadurl`,
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        projectToken: capsProjectToken !== "" ? capsProjectToken : this.projectToken
+      },
+      data: {
+        buildId: capsBuildId,
+        snapshotName,
+        snapshotUuid
+      }
+    }, ctx.log);
+  }
   uploadLogs(ctx, uploadURL) {
     const fileStream = fs5__default.default.createReadStream(constants_default.LOG_FILE_PATH);
     const { size } = fs5__default.default.statSync(constants_default.LOG_FILE_PATH);
@@ -1830,6 +1954,21 @@ var httpClient = class {
       // prevent axios from limiting the content size
     }, ctx.log);
   }
+  uploadSnapshotToS3Caps(ctx, uploadURL, snapshot, capsProjectToken) {
+    return this.request({
+      url: uploadURL,
+      method: "PUT",
+      headers: {
+        "Content-Type": "application/json",
+        projectToken: capsProjectToken !== "" ? capsProjectToken : this.projectToken
+      },
+      data: snapshot,
+      maxBodyLength: Infinity,
+      // prevent axios from limiting the body size
+      maxContentLength: Infinity
+      // prevent axios from limiting the content size
+    }, ctx.log);
+  }
   processWebFigma(requestBody, log2) {
     return this.request({
       url: "figma-web/upload",
@@ -1852,7 +1991,7 @@ var httpClient = class {
   }
 };
 var ctx_default = (options) => {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
   let env = env_default();
   let webConfig;
   let mobileConfig;
@@ -1893,7 +2032,7 @@ var ctx_default = (options) => {
         process.exit(1);
       }
       fetchResultObj = true;
-      fetchResultsFileObj = options.fetchResults === true ? "results.json" : options.fetchResults;
+      fetchResultsFileObj = options.fetchResults === true ? "" : options.fetchResults;
     } else {
       fetchResultObj = false;
       fetchResultsFileObj = "";
@@ -1936,7 +2075,8 @@ var ctx_default = (options) => {
       smartIgnore: (_f = config.smartIgnore) != null ? _f : false,
       delayedUpload: (_g = config.delayedUpload) != null ? _g : false,
       useGlobalCache: (_h = config.useGlobalCache) != null ? _h : false,
-      ignoreHTTPSErrors: (_i = config.ignoreHTTPSErrors) != null ? _i : false
+      ignoreHTTPSErrors: (_i = config.ignoreHTTPSErrors) != null ? _i : false,
+      skipBuildCreation: (_j = config.skipBuildCreation) != null ? _j : false
     },
     uploadFilePath: "",
     webStaticConfig: [],
@@ -1970,7 +2110,10 @@ var ctx_default = (options) => {
     cliVersion: version,
     totalSnapshots: -1,
     isStartExec: false,
-    isSnapshotCaptured: false
+    isSnapshotCaptured: false,
+    sessionCapabilitiesMap: /* @__PURE__ */ new Map(),
+    buildToSnapshotCountMap: /* @__PURE__ */ new Map(),
+    fetchResultsForBuild: new Array()
   };
 };
 function executeCommand(command7) {
@@ -2045,22 +2188,27 @@ var getGitInfo_default = (ctx) => {
     rendererOptions: { persistentOutput: true }
   };
 };
-var createBuild_default = (ctx) => {
+var createBuildExec_default = (ctx) => {
   return {
     title: `Creating SmartUI build`,
     task: (ctx2, task) => __async(void 0, null, function* () {
       updateLogContext({ task: "createBuild" });
       try {
-        let resp = yield ctx2.client.createBuild(ctx2.git, ctx2.config, ctx2.log, ctx2.build.name, ctx2.isStartExec);
-        ctx2.build = {
-          id: resp.data.buildId,
-          name: resp.data.buildName,
-          url: resp.data.buildURL,
-          baseline: resp.data.baseline,
-          useKafkaFlow: resp.data.useKafkaFlow || false
-        };
-        task.output = chalk__default.default.gray(`build id: ${resp.data.buildId}`);
-        task.title = "SmartUI build created";
+        if (ctx2.authenticatedInitially && !ctx2.config.skipBuildCreation) {
+          let resp = yield ctx2.client.createBuild(ctx2.git, ctx2.config, ctx2.log, ctx2.build.name, ctx2.isStartExec);
+          ctx2.build = {
+            id: resp.data.buildId,
+            name: resp.data.buildName,
+            url: resp.data.buildURL,
+            baseline: resp.data.baseline,
+            useKafkaFlow: resp.data.useKafkaFlow || false
+          };
+          task.output = chalk__default.default.gray(`build id: ${resp.data.buildId}`);
+          task.title = "SmartUI build created";
+        } else {
+          task.output = chalk__default.default.gray(`Empty PROJECT_TOKEN and PROJECT_NAME. Skipping Creation of Build!`);
+          task.title = "Skipped SmartUI build creation";
+        }
       } catch (error) {
         ctx2.log.debug(error);
         task.output = chalk__default.default.gray(error.message);
@@ -2076,7 +2224,9 @@ var exec_default = (ctx) => {
     title: `Executing '${(_a = ctx.args.execCommand) == null ? void 0 : _a.join(" ")}'`,
     task: (ctx2, task) => __async(void 0, null, function* () {
       if (ctx2.options.fetchResults) {
-        startPolling(ctx2);
+        if (ctx2.build && ctx2.build.id) {
+          startPolling(ctx2, "", false, "");
+        }
       }
       updateLogContext({ task: "exec" });
       return new Promise((resolve, reject) => {
@@ -2160,14 +2310,50 @@ var finalizeBuild_default = (ctx) => {
       var _a, _b;
       updateLogContext({ task: "finalizeBuild" });
       try {
-        yield ctx2.client.finalizeBuild(ctx2.build.id, ctx2.totalSnapshots, ctx2.log);
-        task.output = chalk__default.default.gray(`build url: ${ctx2.build.url}`);
-        task.title = "Finalized build";
+        if (ctx2.build.id) {
+          yield ctx2.client.finalizeBuild(ctx2.build.id, ctx2.totalSnapshots, ctx2.log);
+        }
       } catch (error) {
         ctx2.log.debug(error);
         task.output = chalk__default.default.gray(error.message);
         throw new Error("Finalize build failed");
       }
+      let buildUrls = `build url: ${ctx2.build.url}
+`;
+      for (const [sessionId, capabilities] of ctx2.sessionCapabilitiesMap.entries()) {
+        try {
+          const buildId = (capabilities == null ? void 0 : capabilities.buildId) || "";
+          const projectToken = (capabilities == null ? void 0 : capabilities.projectToken) || "";
+          const totalSnapshots = (capabilities == null ? void 0 : capabilities.snapshotCount) || 0;
+          const sessionBuildUrl = (capabilities == null ? void 0 : capabilities.buildURL) || "";
+          const testId = (capabilities == null ? void 0 : capabilities.id) || "";
+          if (ctx2.options.fetchResults && ctx2.fetchResultsForBuild) {
+            if (!ctx2.fetchResultsForBuild.includes(buildId)) {
+              let is_baseline;
+              if (capabilities.baseline) {
+                is_baseline = true;
+              } else {
+                is_baseline = false;
+              }
+              console.log(`start polling was called at finalize build for buildId: ${buildId}`);
+              startPolling(ctx2, buildId, is_baseline, capabilities.projectToken);
+              yield new Promise((resolve) => setTimeout(resolve, 7e3));
+              ctx2.fetchResultsForBuild.push(buildId);
+            }
+          }
+          if (buildId && projectToken) {
+            yield ctx2.client.finalizeBuildForCapsWithToken(buildId, totalSnapshots, projectToken, ctx2.log);
+          }
+          if (testId && buildId) {
+            buildUrls += `TestId ${testId}: ${sessionBuildUrl}
+`;
+          }
+        } catch (error) {
+          ctx2.log.debug(`Error finalizing build for session ${sessionId}: ${error.message}`);
+        }
+      }
+      task.output = chalk__default.default.gray(buildUrls);
+      task.title = "Finalized build";
       try {
         yield (_a = ctx2.browser) == null ? void 0 : _a.close();
         ctx2.log.debug(`Closed browser`);
@@ -2360,6 +2546,16 @@ function processSnapshot(snapshot, ctx) {
       if (options.loadDomContent) {
         processedOptions.loadDomContent = true;
       }
+      if (options.sessionId) {
+        const sessionId = options.sessionId;
+        processedOptions.sessionId = sessionId;
+        if (ctx.sessionCapabilitiesMap && ctx.sessionCapabilitiesMap.has(sessionId)) {
+          const sessionCapabilities = ctx.sessionCapabilitiesMap.get(sessionId);
+          if (sessionCapabilities && sessionCapabilities.id) {
+            processedOptions.testId = sessionCapabilities.id;
+          }
+        }
+      }
       if (options.web && Object.keys(options.web).length) {
         processedOptions.web = {};
         if (options.web.viewports && options.web.viewports.length > 0) {
@@ -2750,6 +2946,7 @@ var Queue = class {
   }
   processNext() {
     return __async(this, null, function* () {
+      var _a, _b, _c;
       if (!this.isEmpty()) {
         let snapshot;
         if (this.ctx.config.delayedUpload) {
@@ -2777,22 +2974,71 @@ var Queue = class {
             this.processGenerateVariants(snapshot);
           }
           if (!drop) {
+            const sessionId = (_a = snapshot == null ? void 0 : snapshot.options) == null ? void 0 : _a.sessionId;
+            let capsBuildId = "";
+            let capsProjectToken = "";
+            let useCapsBuildId = false;
+            let useKafkaFlowCaps = false;
+            if (sessionId && ((_b = this.ctx.sessionCapabilitiesMap) == null ? void 0 : _b.has(sessionId))) {
+              const cachedCapabilities = this.ctx.sessionCapabilitiesMap.get(sessionId);
+              capsProjectToken = (cachedCapabilities == null ? void 0 : cachedCapabilities.projectToken) || "";
+              capsBuildId = (cachedCapabilities == null ? void 0 : cachedCapabilities.buildId) || "";
+              useKafkaFlowCaps = (cachedCapabilities == null ? void 0 : cachedCapabilities.useKafkaFlow) || false;
+              if (capsBuildId != "" && capsProjectToken != "") {
+                useCapsBuildId = true;
+              }
+            }
             let { processedSnapshot, warnings } = yield processSnapshot(snapshot, this.ctx);
-            if (this.ctx.build && this.ctx.build.useKafkaFlow) {
-              const snapshotUuid = uuid.v4();
-              const presignedResponse = yield this.ctx.client.getS3PresignedURLForSnapshotUpload(this.ctx, processedSnapshot.name, snapshotUuid);
-              const uploadUrl = presignedResponse.data.url;
-              yield this.ctx.client.uploadSnapshotToS3(this.ctx, uploadUrl, processedSnapshot);
-              yield this.ctx.client.processSnapshot(this.ctx, processedSnapshot, snapshotUuid);
+            if (useCapsBuildId) {
+              if (useKafkaFlowCaps) {
+                const snapshotUuid = uuid.v4();
+                const presignedResponse = yield this.ctx.client.getS3PresignedURLForSnapshotUploadCaps(this.ctx, processedSnapshot.name, snapshotUuid, capsBuildId, capsProjectToken);
+                const uploadUrl = presignedResponse.data.url;
+                yield this.ctx.client.uploadSnapshotToS3Caps(this.ctx, uploadUrl, processedSnapshot, capsProjectToken);
+                yield this.ctx.client.processSnapshotCaps(this.ctx, processedSnapshot, snapshotUuid, capsBuildId, capsProjectToken);
+              } else {
+                yield this.ctx.client.uploadSnapshotForCaps(this.ctx, processedSnapshot, capsBuildId, capsProjectToken);
+              }
+              const cachedCapabilities = this.ctx.sessionCapabilitiesMap.get(sessionId);
+              const currentCount = (cachedCapabilities == null ? void 0 : cachedCapabilities.snapshotCount) || 0;
+              cachedCapabilities.snapshotCount = currentCount + 1;
+              this.ctx.sessionCapabilitiesMap.set(sessionId, cachedCapabilities);
             } else {
-              yield this.ctx.client.uploadSnapshot(this.ctx, processedSnapshot);
+              if (!((_c = this.ctx.build) == null ? void 0 : _c.id)) {
+                if (this.ctx.authenticatedInitially) {
+                  let resp = yield this.ctx.client.createBuild(this.ctx.git, this.ctx.config, this.ctx.log, this.ctx.build.name);
+                  this.ctx.build = {
+                    id: resp.data.buildId,
+                    name: resp.data.buildName,
+                    url: resp.data.buildURL,
+                    baseline: resp.data.baseline,
+                    useKafkaFlow: resp.data.useKafkaFlow || false
+                  };
+                } else {
+                  throw new Error("SmartUI capabilities are missing in env variables or in driver capabilities");
+                }
+                if (this.ctx.options.fetchResults) {
+                  if (this.ctx.build && this.ctx.build.id) {
+                    startPolling(this.ctx, "", false, "");
+                  }
+                }
+              }
+              if (this.ctx.build && this.ctx.build.useKafkaFlow) {
+                const snapshotUuid = uuid.v4();
+                const presignedResponse = yield this.ctx.client.getS3PresignedURLForSnapshotUpload(this.ctx, processedSnapshot.name, snapshotUuid);
+                const uploadUrl = presignedResponse.data.url;
+                yield this.ctx.client.uploadSnapshotToS3(this.ctx, uploadUrl, processedSnapshot);
+                yield this.ctx.client.processSnapshot(this.ctx, processedSnapshot, snapshotUuid);
+              } else {
+                yield this.ctx.client.uploadSnapshot(this.ctx, processedSnapshot);
+              }
+              this.ctx.totalSnapshots++;
             }
-            this.ctx.totalSnapshots++;
-            this.processedSnapshots.push({ name: snapshot.name, warnings });
+            this.processedSnapshots.push({ name: snapshot == null ? void 0 : snapshot.name, warnings });
           }
         } catch (error) {
           this.ctx.log.debug(`snapshot failed; ${error}`);
-          this.processedSnapshots.push({ name: snapshot.name, error: error.message });
+          this.processedSnapshots.push({ name: snapshot == null ? void 0 : snapshot.name, error: error.message });
         }
         if (this.ctx.browser) {
           for (let context of this.ctx.browser.contexts()) {
@@ -2843,10 +3089,10 @@ command.name("exec").description("Run test commands around SmartUI").argument("<
     ctx.totalSnapshots = 0;
     let tasks = new listr2.Listr(
       [
-        auth_default(),
+        authExec_default(),
         startServer_default(),
         getGitInfo_default(),
-        createBuild_default(),
+        createBuildExec_default(),
         exec_default(ctx),
         processSnapshot_default(),
         finalizeBuild_default()
@@ -2991,6 +3237,55 @@ configWebFigma.name("config:create-figma-web").description("Create figma config
     createWebFigmaConfig(filepath);
   });
 });
+var auth_default = (ctx) => {
+  return {
+    title: `Authenticating with SmartUI`,
+    task: (ctx2, task) => __async(void 0, null, function* () {
+      updateLogContext({ task: "auth" });
+      try {
+        const authResult = yield ctx2.client.auth(ctx2.log, ctx2.env);
+        if (authResult === 2) {
+          task.output = chalk__default.default.gray(`New project '${ctx2.env.PROJECT_NAME}' created successfully`);
+        } else if (authResult === 0) {
+          task.output = chalk__default.default.gray(`Using existing project token '******#${ctx2.env.PROJECT_TOKEN.split("#").pop()}'`);
+        } else if (authResult === 1) {
+          task.output = chalk__default.default.gray(`Using existing project '${ctx2.env.PROJECT_NAME}'`);
+        }
+        task.title = "Authenticated with SmartUI";
+      } catch (error) {
+        ctx2.log.debug(error);
+        task.output = chalk__default.default.gray(error.message);
+        throw new Error("Authentication failed");
+      }
+    }),
+    rendererOptions: { persistentOutput: true }
+  };
+};
+var createBuild_default = (ctx) => {
+  return {
+    title: `Creating SmartUI build`,
+    task: (ctx2, task) => __async(void 0, null, function* () {
+      updateLogContext({ task: "createBuild" });
+      try {
+        let resp = yield ctx2.client.createBuild(ctx2.git, ctx2.config, ctx2.log, ctx2.build.name, ctx2.isStartExec);
+        ctx2.build = {
+          id: resp.data.buildId,
+          name: resp.data.buildName,
+          url: resp.data.buildURL,
+          baseline: resp.data.baseline,
+          useKafkaFlow: resp.data.useKafkaFlow || false
+        };
+        task.output = chalk__default.default.gray(`build id: ${resp.data.buildId}`);
+        task.title = "SmartUI build created";
+      } catch (error) {
+        ctx2.log.debug(error);
+        task.output = chalk__default.default.gray(error.message);
+        throw new Error("SmartUI build creation failed");
+      }
+    }),
+    rendererOptions: { persistentOutput: true }
+  };
+};
 function captureScreenshotsForConfig(ctx, browsers, urlConfig, browserName, renderViewports) {
   return __async(this, null, function* () {
     ctx.log.debug(`*** urlConfig  ${JSON.stringify(urlConfig)}`);
@@ -3318,7 +3613,7 @@ var captureScreenshots_default = (ctx) => {
       try {
         ctx2.task = task;
         if (ctx2.options.fetchResults) {
-          startPolling(ctx2);
+          startPolling(ctx2, "", false, "");
         }
         updateLogContext({ task: "capture" });
         if (ctx2.options.parallel) {
@@ -3417,7 +3712,7 @@ var uploadScreenshots_default = (ctx) => {
       try {
         ctx2.task = task;
         if (ctx2.options.fetchResults) {
-          startPolling(ctx2);
+          startPolling(ctx2, "", false, "");
         }
         updateLogContext({ task: "upload" });
         yield uploadScreenshots(ctx2);
@@ -3630,7 +3925,7 @@ var uploadWebFigma_default2 = (ctx) => {
           task.output = task.output + "\n" + chalk__default.default.green(`${output}`);
         }
         if (ctx2.options.fetchResults) {
-          startPolling(ctx2);
+          startPolling(ctx2, "", false, "");
         }
         task.title = "Web Figma images uploaded successfully to SmartUI";
         ctx2.log.debug(`Web Figma processed: ${results}`);
@@ -3774,7 +4069,7 @@ command4.name("exec:start").description("Start SmartUI server").option("-P, --po
       yield tasks.run(ctx);
       startPingPolling(ctx);
       if (ctx.options.fetchResults) {
-        startPolling(ctx);
+        startPolling(ctx, "", false, "");
       }
     } catch (error) {
       console.error("Error during server execution:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lambdatest/smartui-cli",
-  "version": "4.1.2",
+  "version": "4.1.3-beta.0",
   "description": "A command line interface (CLI) to run SmartUI tests on LambdaTest",
   "files": [
     "dist/**/*"