Merge pull request #4 from pinanks/DOT-884

DOT-884

Author: sushobhit-lt

commands/utils/constants.js

@@ -2,7 +2,8 @@ var constants = {}
 
 constants.stage = {
     AUTH_URL: "https://stage-api.lambdatestinternal.com/storybook/auth",
-    RENDER_API_URL: "https://stage-api.lambdatestinternal.com/storybook/render"
+    RENDER_API_URL: "https://stage-api.lambdatestinternal.com/storybook/render",
+    BUILD_STATUS_URL: "https://stage-api.lambdatestinternal.com/storybook/status"
 };
 constants.prod = {
     AUTH_URL: "https://api.lambdatest.com/storybook/auth",

commands/utils/dom.js

@@ -6,6 +6,9 @@ const { JSDOM } = require("jsdom");
 var { constants } = require('./constants');
 const { getLastCommit } = require('./git')
 
+var INTERVAL = 2000
+const MAX_INTERVAL = 512000
+
 async function sendDoM(storybookUrl, stories, storybookConfig, options) {
     const createBrowser = require('browserless')
     const browser = createBrowser()
@@ -40,7 +43,7 @@ async function sendDoM(storybookUrl, stories, storybookConfig, options) {
     await browser.close()
 
     // Create form
-    let commit = await getLastCommit();
+    // let commit = await getLastCommit();
     const form = new formData();
     for (const [storyId, storyInfo] of Object.entries(stories)) {
         const file = fs.readFileSync('doms/' + storyId + '.html');
@@ -49,38 +52,88 @@ async function sendDoM(storybookUrl, stories, storybookConfig, options) {
     form.append('resolution', storybookConfig.resolutions);
     form.append('browser', storybookConfig.browsers);
     form.append('projectToken', process.env.PROJECT_TOKEN);
-    form.append('buildName', options.buildname);
-    form.append('branch', commit.branch);
-    form.append('commitId', commit.shortHash);
-    form.append('commitAuthor', commit.author.name);
-    form.append('commitMessage', commit.subject);
+    // form.append('buildName', options.buildname);
+    // form.append('branch', commit.branch);
+    // form.append('commitId', commit.shortHash);
+    // form.append('commitAuthor', commit.author.name);
+    // form.append('commitMessage', commit.subject);
 
     // Send DOM to render API
-    axios.post(constants[options.env].RENDER_API_URL, form, {
+    await axios.post(constants[options.env].RENDER_API_URL, form, {
         headers: {
             ...form.getHeaders()
         }
         })
-        .then(function (response) {
-            console.log('[smartui] Build successful')
+        .then(async function (response) {
+            console.log('[smartui] Build in progress...');
+            await shortPolling(response.data.buildId, 0, options);
         })
         .catch(function (error) {
-            fs.rm('doms', {recursive: true}, (err) => {
-                if (err) {
-                    return console.error(err);
-                }
-            });
             console.log('[smartui] Build failed: Error: ', error.message);
-            process.exit(0);
         });
-
+    
     fs.rm('doms', {recursive: true}, (err) => {
         if (err) {
             return console.error(err);
         }
     });
 };
 
+async function shortPolling(buildId, retries = 0, options) {
+    await axios.get(new URL('?buildId=' + buildId, constants[options.env].BUILD_STATUS_URL).href, {
+        headers: {
+            projectToken: process.env.PROJECT_TOKEN
+        }})
+        .then(function (response) {
+            if (response.data) {
+                if (response.data.buildStatus === 'completed') {
+                    console.log('[smartui] Build successful\n');
+                    console.log('[smartui] Build details:\n',
+                        // 'Build URL: ', response.data.buildId, '\n',
+                        'Build Name: ', response.data.buildName, '\n',
+                        'Total Screenshots: ', response.data.screenshots.length, '\n',
+                        'Approved: ', response.data.buildResults.approved, '\n',
+                        'Changes found: ', response.data.buildResults.changesFound, '\n'
+                    );
+    
+                    response.data.screenshots.forEach(screenshot => {
+                        console.log(screenshot.storyName, ' | Mis-match: ', screenshot.mismatchPercentage);
+                    });
+    
+                    return;
+                } else {
+                    if (response.data.screenshots && response.data.screenshots.length > 0) {
+                        // TODO: show Screenshots processed 8/10 
+                        console.log('[smartui] Screenshots processed: ', response.data.screenshots.length)
+                    }
+                }
+            }
+            
+            // Double the INTERVAL, up to the maximum INTERVAL of 512 secs (so ~15 mins in total)
+            INTERVAL = Math.min(INTERVAL * 2, MAX_INTERVAL);
+            if (INTERVAL == MAX_INTERVAL) {
+                console.log('[smartui] Please check the build status on LambdaTest SmartUI.');
+                return;
+            }
+    
+            setTimeout(function () {
+                shortPolling(buildId, 0, options)
+            }, INTERVAL);
+        })
+        .catch(function (error) {
+            if (retries >= 3) {
+                console.log('[smartui] Error: Failed getting build status.', error.message);
+                console.log('[smartui] Please check the build status on LambdaTest SmartUI.');
+                return;
+            }
+    
+            console.log('here2');
+            setTimeout(function () {
+                shortPolling(buildId, retries+1, options);
+            }, 2000);
+        });
+};
+
 function getBase64(url) {
     return axios.get(url, {
             responseType: "text",