Merge pull request #6 from pinanks/DOT-945

DOT-945

Author: sushobhit-lt

commands/utils/constants.js

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

commands/utils/dom.js

@@ -43,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');
@@ -52,11 +52,10 @@ 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('branch', commit.branch);
+    form.append('commitId', commit.shortHash);
+    form.append('commitAuthor', commit.author.name);
+    form.append('commitMessage', commit.subject);
 
     // Send DOM to render API
     await axios.post(constants[options.env].RENDER_API_URL, form, {

commands/utils/validate.js

@@ -1,6 +1,7 @@
 const axios = require('axios');
 var { constants } = require('./constants');
 const fs = require('fs');
+const { getLastCommit } = require('./git');
 
 function validateProjectToken(options) {
     if (process.env.PROJECT_TOKEN) { 
@@ -60,4 +61,28 @@ async function validateStorybookDir(dir) {
     }
 };
 
-module.exports = { validateProjectToken, validateStorybookUrl, validateStorybookDir };
+async function validateLatestBuild(options) {
+    let commit = await getLastCommit();
+    return axios.get(new URL(constants[options.env].SB_BUILD_VALIDATE_PATH, constants[options.env].BASE_URL).href, {
+        headers: {
+            projectToken: process.env.PROJECT_TOKEN
+        },
+        params: {
+            branch: commit.branch,
+            commitId: commit.shortHash
+        }})
+        .then(function (response) {
+            if (response.data.status === 'Failure') {
+                console.log(`[smartui] Build with commit '${commit.shortHash}' on branch '${commit.branch}' already exists.`);
+                console.log('[smartui] Use option --force-rebuild to forcefully push a new build.');
+                process.exit(0);
+            }
+        })
+        .catch(function (error) {
+            // TODO: Add retries
+            console.log('[smartui] Error: ', error.message);
+            process.exit(1);
+        });
+}
+
+module.exports = { validateProjectToken, validateStorybookUrl, validateStorybookDir, validateLatestBuild };

index.js

@@ -3,7 +3,7 @@
 const { Command, Option } = require('commander');
 const program = new Command();
 const { storybook } = require('./commands/storybook');
-const { validateProjectToken } = require('./commands/utils/validate');
+const { validateProjectToken, validateLatestBuild } = require('./commands/utils/validate');
 const { createConfig } = require('./commands/config');
 
 program
@@ -26,10 +26,12 @@ program.command('storybook')
     .description('Snapshot Storybook stories')
     .argument('<url|directory>', 'Storybook url or static build directory')
     .option('-c --config <file>', 'Config file path')
+    .option('--force-rebuild', 'Force a rebuild of an already existing build.', false)
     // .option('--env <prod|stage>', 'Runtime environment option', 'prod')
     .action(async function(serve, options) {
         options.env = program.opts().env || 'prod';
         await validateProjectToken(options);
+        if (!options.forceRebuild) await validateLatestBuild(options);
         storybook(serve, options);
     });